diff --git a/litellm/tests/test_assistants.py b/litellm/tests/test_assistants.py index 940b874ff61..7f20a6df067 100644 --- a/litellm/tests/test_assistants.py +++ b/litellm/tests/test_assistants.py @@ -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")