phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge d3b75abc86 into b82af5b826

Browse source
This commit is contained in:
Jean Carlo de Souza 2025-04-24 01:02:01 -07:00 • committed by GitHub
commit 29eea8fe9b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 432 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

63
litellm/llms/openai/openai.py
View file

@ -2222,6 +2222,69 @@ class OpenAIAssistantsAPI(BaseLLM):
response = openai_client.beta.assistants.delete(assistant_id=assistant_id)
return response
# Modify Assistant
async def async_modify_assistants(
self,
assistant_id: str,
api_key: Optional[str],
api_base: Optional[str],
timeout: Union[float, httpx.Timeout],
max_retries: Optional[int],
organization: Optional[str],
client: Optional[AsyncOpenAI],
modify_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.update(assistant_id, **modify_assistant_data)
return response
def modify_assistants(
self,
assistant_id: str,
api_key: Optional[str],
api_base: Optional[str],
timeout: Union[float, httpx.Timeout],
max_retries: Optional[int],
organization: Optional[str],
modify_assistant_data: dict,
client=None,
async_modify_assistants=None,
):
if async_modify_assistants is not None and async_modify_assistants is True:
return self.async_modify_assistants(
assistant_id=assistant_id,
api_key=api_key,
api_base=api_base,
timeout=timeout,
max_retries=max_retries,
organization=organization,
client=client,
modify_assistant_data=modify_assistant_data,
)
openai_client = self.get_openai_client(
api_key=api_key,
api_base=api_base,
timeout=timeout,
max_retries=max_retries,
organization=organization,
client=client,
)
response = openai_client.beta.assistants.update(assistant_id, **modify_assistant_data)
return response
### MESSAGES ###
async def a_add_message(