fix import

This commit is contained in:
Ishaan Jaff 2024-09-05 14:42:56 -07:00
parent 8ca7c2c9af
commit a30917311e
3 changed files with 3 additions and 3 deletions

View file

@ -2291,7 +2291,7 @@ class AzureAssistantsAPI(BaseLLM):
""" """
Here's an example: Here's an example:
``` ```
from litellm.llms.openai import OpenAIAssistantsAPI, MessageData from litellm.llms.OpenAI.openai import OpenAIAssistantsAPI, MessageData
# create thread # create thread
message: MessageData = {"role": "user", "content": "Hey, how's it going?"} message: MessageData = {"role": "user", "content": "Hey, how's it going?"}

View file

@ -3056,7 +3056,7 @@ class OpenAIAssistantsAPI(BaseLLM):
""" """
Here's an example: Here's an example:
``` ```
from litellm.llms.openai import OpenAIAssistantsAPI, MessageData from litellm.llms.OpenAI.openai import OpenAIAssistantsAPI, MessageData
# create thread # create thread
message: MessageData = {"role": "user", "content": "Hey, how's it going?"} message: MessageData = {"role": "user", "content": "Hey, how's it going?"}

View file

@ -81,7 +81,7 @@ class VertexAIPartnerModels(BaseLLM):
from google.cloud import aiplatform from google.cloud import aiplatform
from litellm.llms.databricks import DatabricksChatCompletion from litellm.llms.databricks import DatabricksChatCompletion
from litellm.llms.openai import OpenAIChatCompletion from litellm.llms.OpenAI.openai import OpenAIChatCompletion
from litellm.llms.text_completion_codestral import CodestralTextCompletion from litellm.llms.text_completion_codestral import CodestralTextCompletion
from litellm.llms.vertex_ai_and_google_ai_studio.gemini.vertex_and_google_ai_studio_gemini import ( from litellm.llms.vertex_ai_and_google_ai_studio.gemini.vertex_and_google_ai_studio_gemini import (
VertexLLM, VertexLLM,