mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-12 23:01:41 +00:00
test(test_assistants.py): cleanup tests
This commit is contained in:
parent
8fe6c9b401
commit
5406205e4b
1 changed files with 0 additions and 72 deletions
|
|
@ -100,75 +100,3 @@ def test_run_thread_litellm():
|
|||
assert isinstance(messages.data[0], Message)
|
||||
else:
|
||||
pytest.fail("An unexpected error occurred when running the thread")
|
||||
|
||||
|
||||
def test_run_thread_openai_direct():
|
||||
"""
|
||||
- Get Assistants
|
||||
- Create thread
|
||||
- Create run w/ Assistants + Thread
|
||||
"""
|
||||
openai_api = OpenAIAssistantsAPI()
|
||||
|
||||
assistants = openai_api.get_assistants(
|
||||
api_key=os.getenv("OPENAI_API_KEY"),
|
||||
api_base=None,
|
||||
timeout=600,
|
||||
max_retries=2,
|
||||
organization=None,
|
||||
client=None,
|
||||
)
|
||||
|
||||
## get the first assistant ###
|
||||
assistant_id = assistants.data[0].id
|
||||
|
||||
## create a thread w/ message ###
|
||||
new_thread = test_create_thread()
|
||||
|
||||
thread_id = new_thread.id
|
||||
|
||||
# add message to thread
|
||||
message: MessageData = {"role": "user", "content": "Hey, how's it going?"} # type: ignore
|
||||
added_message = openai_api.add_message(
|
||||
thread_id=new_thread.id,
|
||||
message_data=message,
|
||||
api_key=os.getenv("OPENAI_API_KEY"),
|
||||
api_base=None,
|
||||
timeout=600,
|
||||
max_retries=2,
|
||||
organization=None,
|
||||
client=None,
|
||||
)
|
||||
|
||||
run = openai_api.run_thread(
|
||||
thread_id=thread_id,
|
||||
assistant_id=assistant_id,
|
||||
additional_instructions=None,
|
||||
instructions=None,
|
||||
metadata=None,
|
||||
model=None,
|
||||
stream=None,
|
||||
tools=None,
|
||||
api_key=os.getenv("OPENAI_API_KEY"),
|
||||
api_base=None,
|
||||
timeout=600,
|
||||
max_retries=2,
|
||||
organization=None,
|
||||
client=None,
|
||||
)
|
||||
|
||||
print(f"run: {run}")
|
||||
|
||||
if run.status == "completed":
|
||||
messages = openai_api.get_messages(
|
||||
thread_id=new_thread.id,
|
||||
api_key=os.getenv("OPENAI_API_KEY"),
|
||||
api_base=None,
|
||||
timeout=600,
|
||||
max_retries=2,
|
||||
organization=None,
|
||||
client=None,
|
||||
)
|
||||
assert isinstance(messages.data[0], Message)
|
||||
else:
|
||||
pytest.fail("An unexpected error occurred when running the thread")
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue