phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

feat(assistants/main.py): support litellm.get_assistants() and litellm.get_messages()

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-04 21:30:28 -07:00
parent cad01fb586
commit 8fe6c9b401
3 changed files with 265 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/llms/openai.py
View file

@ -1282,10 +1282,10 @@ class OpenAIAssistantsAPI(BaseLLM):
def get_assistants(
self,
api_key: str,
api_key: Optional[str],
api_base: Optional[str],
timeout: Union[float, httpx.Timeout],
max_retries: int,
max_retries: Optional[int],
organization: Optional[str],
client: Optional[OpenAI],
) -> SyncCursorPage[Assistant]:
@ -1340,10 +1340,10 @@ class OpenAIAssistantsAPI(BaseLLM):
def get_messages(
self,
thread_id: str,
api_key: str,
api_key: Optional[str],
api_base: Optional[str],
timeout: Union[float, httpx.Timeout],
max_retries: int,
max_retries: Optional[int],
organization: Optional[str],
client: Optional[OpenAI] = None,
) -> SyncCursorPage[OpenAIMessage]:
@ -1440,10 +1440,10 @@ class OpenAIAssistantsAPI(BaseLLM):
model: Optional[str],
stream: Optional[Literal[False]] | Literal[True],
tools: Optional[Iterable[AssistantToolParam]],
api_key: str,
api_key: Optional[str],
api_base: Optional[str],
timeout: Union[float, httpx.Timeout],
max_retries: int,
max_retries: Optional[int],
organization: Optional[str],
client: Optional[OpenAI],
) -> Run: