test(test_assistants.py): cleanup tests

This commit is contained in:
Krrish Dholakia 2024-05-04 21:31:07 -07:00
parent 8fe6c9b401
commit 5406205e4b

View file

@ -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")