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

test assistants fixes

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-19 08:08:12 -07:00
parent dee5182fc8
commit 0a35c208d7
1 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/local_testing/test_assistants.py
View file

@ -236,11 +236,18 @@ async def test_aarun_thread_litellm(sync_mode, provider, is_streaming):
"""
import openai
try:
get_assistants_data = {
"custom_llm_provider": provider,
}
if provider == "azure":
get_assistants_data = _add_azure_related_dynamic_params(get_assistants_data)
if sync_mode:
assistants = litellm.get_assistants(custom_llm_provider=provider)
assistants = litellm.get_assistants(**get_assistants_data)
else:
assistants = await litellm.aget_assistants(custom_llm_provider=provider)
assistants = await litellm.aget_assistants(**get_assistants_data)
## get the first assistant ###
try: