(Refactor) Code Quality improvement - remove /prompt_templates/ , base_aws_llm.py from /llms folder (#7164)

* fix move base_aws_llm

* fix import

* update enforce llms folder style

* move prompt_templates

* update prompt_templates location

* fix imports

* fix imports

* fix imports

* fix imports

* fix checks
This commit is contained in:
Ishaan Jaff 2024-12-11 00:02:46 -08:00 committed by GitHub
parent b328d42ebc
commit b5d55688e5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
64 changed files with 135 additions and 81 deletions

View file

@ -6,8 +6,8 @@ import httpx
from litellm.utils import ModelResponse, get_secret
from ...base_aws_llm import BaseAWSLLM
from ...prompt_templates.factory import custom_prompt, prompt_factory
from litellm.llms.bedrock.base_aws_llm import BaseAWSLLM
from litellm.litellm_core_utils.prompt_templates.factory import custom_prompt, prompt_factory
from ..common_utils import AWSEventStreamDecoder
from .transformation import SagemakerChatConfig