From ed627bc5d278dc6f8d90820e7e35cf2a5e63ecd9 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 5 Sep 2024 09:56:33 -0700 Subject: [PATCH] fix linting error --- litellm/assistants/main.py | 19 +++++++++++++++++-- 1 file changed, 17 insertions(+), 2 deletions(-) diff --git a/litellm/assistants/main.py b/litellm/assistants/main.py index ba169f5e20..0ea5860ae7 100644 --- a/litellm/assistants/main.py +++ b/litellm/assistants/main.py @@ -21,8 +21,8 @@ from litellm.utils import ( supports_httpx_timeout, ) -from ..llms.azure import AzureAssistantsAPI -from ..llms.openai import OpenAIAssistantsAPI +from ..llms.AzureOpenAI.azure import AzureAssistantsAPI +from ..llms.OpenAI.openai import OpenAIAssistantsAPI from ..types.llms.openai import * from ..types.router import * from .utils import get_optional_params_add_message @@ -184,6 +184,21 @@ def get_assistants( request=httpx.Request(method="create_thread", url="https://github.com/BerriAI/litellm"), # type: ignore ), ) + + if response is None: + raise litellm.exceptions.BadRequestError( + message="LiteLLM doesn't support {} for 'get_assistants'. Only 'openai' is supported.".format( + custom_llm_provider + ), + model="n/a", + llm_provider=custom_llm_provider, + response=httpx.Response( + status_code=400, + content="Unsupported provider", + request=httpx.Request(method="create_thread", url="https://github.com/BerriAI/litellm"), # type: ignore + ), + ) + return response