phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source 
* 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 05731f698b
commit 26918487d6
64 changed files with 135 additions and 81 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/sagemaker/chat/handler.py
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