tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-29 02:26:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix imports

Browse Source
This commit is contained in:
Krrish Dholakia
2024-08-26 22:24:30 -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
litellm/__init__.py
+1 -1
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
litellm/tests/test_streaming.py
+1 -1
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."""