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 ( from .llms.vertex_ai_and_google_ai_studio.vertex_ai_partner_models import (
VertexAILlama3Config, VertexAILlama3Config,
) )
from .llms.sagemaker import SagemakerConfig from .llms.sagemaker.sagemaker import SagemakerConfig
from .llms.ollama import OllamaConfig from .llms.ollama import OllamaConfig
from .llms.ollama_chat import OllamaChatConfig from .llms.ollama_chat import OllamaChatConfig
from .llms.maritalk import MaritTalkConfig from .llms.maritalk import MaritTalkConfig

2
litellm/tests/test_streaming.py
View file

@ -1664,7 +1664,7 @@ def test_sagemaker_weird_response():
try: try:
import json import json
from litellm.llms.sagemaker import TokenIterator from litellm.llms.sagemaker.sagemaker import TokenIterator
chunk = """<s>[INST] Hey, how's it going? [/INST], 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.""" 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."""