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

fix: fix imports

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-26 22:23:29 -07:00
parent 81e62aec16
commit 3ec8a0d25d
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/__init__.py
View file

@ -862,7 +862,7 @@ from .llms.vertex_ai_and_google_ai_studio.vertex_ai_anthropic import (
from .llms.vertex_ai_and_google_ai_studio.vertex_ai_partner_models import (
VertexAILlama3Config,
)
from .llms.sagemaker import SagemakerConfig
from .llms.sagemaker.sagemaker import SagemakerConfig
from .llms.ollama import OllamaConfig
from .llms.ollama_chat import OllamaChatConfig
from .llms.maritalk import MaritTalkConfig

2
litellm/tests/test_streaming.py
View file

@ -1664,7 +1664,7 @@ def test_sagemaker_weird_response():
try:
import json
from litellm.llms.sagemaker import TokenIterator
from litellm.llms.sagemaker.sagemaker import TokenIterator
chunk = """<s>[INST] Hey, how's it going? [/INST],
I'm doing well, thanks for asking! How about you? Is there anything you'd like to chat about or ask? I'm here to help with any questions you might have."""