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

add acreate_assistants

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-09 09:33:41 -07:00
parent 06926920d5
commit f4f07e13f3
2 changed files with 74 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

30
litellm/llms/openai.py
View file

@ -2384,6 +2384,29 @@ class OpenAIAssistantsAPI(BaseLLM):
return response
# Create Assistant
async def async_create_assistants(
self,
api_key: Optional[str],
api_base: Optional[str],
timeout: Union[float, httpx.Timeout],
max_retries: Optional[int],
organization: Optional[str],
client: Optional[AsyncOpenAI],
create_assistant_data: dict,
) -> Assistant:
openai_client = self.async_get_openai_client(
api_key=api_key,
api_base=api_base,
timeout=timeout,
max_retries=max_retries,
organization=organization,
client=client,
)
response = await openai_client.beta.assistants.create(**create_assistant_data)
return response
def create_assistants(
self,
api_key: Optional[str],
@ -2393,16 +2416,17 @@ class OpenAIAssistantsAPI(BaseLLM):
organization: Optional[str],
create_assistant_data: dict,
client=None,
acreate_assistants=None,
async_create_assistants=None,
):
if acreate_assistants is not None and acreate_assistants == True:
return self.async_get_assistants(
if async_create_assistants is not None and async_create_assistants == True:
return self.async_create_assistants(
api_key=api_key,
api_base=api_base,
timeout=timeout,
max_retries=max_retries,
organization=organization,
client=client,
create_assistant_data=create_assistant_data,
)
openai_client = self.get_openai_client(
api_key=api_key,