scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

async test assistants

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-09 09:44:18 -07:00
parent f4f07e13f3
commit e4ad792d5d
1 changed files with 15 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
litellm/tests/test_assistants.py
View file

@ -62,7 +62,7 @@ async def test_get_assistants(provider, sync_mode):
@pytest.mark.parametrize("provider", ["openai"])
@pytest.mark.parametrize(
"sync_mode",
[True, False],
[False],
)
@pytest.mark.asyncio
async def test_create_assistants(provider, sync_mode):
@ -86,8 +86,20 @@ async def test_create_assistants(provider, sync_mode):
)
assert assistant.id is not None
else:
assistants = await litellm.aget_assistants(**data)
assert isinstance(assistants, AsyncCursorPage)
assistant = await litellm.acreate_assistants(
custom_llm_provider="openai",
model="gpt-4-turbo",
instructions="You are a personal math tutor. When asked a question, write and run Python code to answer the question.",
name="Math Tutor",
tools=[{"type": "code_interpreter"}],
)
print("New assistants", assistant)
assert isinstance(assistant, Assistant)
assert (
assistant.instructions
== "You are a personal math tutor. When asked a question, write and run Python code to answer the question."
)
assert assistant.id is not None
@pytest.mark.parametrize("provider", ["openai", "azure"])