Refactor error handling in Api class to ensure consistent response for invalid providers

This commit is contained in:
hlohaus 2025-09-05 03:12:13 +02:00
parent 1ddf666da4
commit 2832a98454

View file

@ -435,7 +435,8 @@ class Api:
if provider in model_map:
config.model = provider
provider = None
return ErrorResponse.from_message("Invalid provider.", HTTP_404_NOT_FOUND)
else:
return ErrorResponse.from_message("Invalid provider.", HTTP_404_NOT_FOUND)
try:
if config.provider is None:
config.provider = AppConfig.provider if provider is None else provider
@ -541,7 +542,7 @@ class Api:
if provider in model_map:
config.model = provider
provider = None
return ErrorResponse.from_message("Invalid provider.", HTTP_404_NOT_FOUND)
return ErrorResponse.from_message("", HTTP_404_NOT_FOUND)
if config.provider is None:
config.provider = provider
if config.provider is None:
@ -624,7 +625,8 @@ class Api:
if provider in model_map:
model = provider
provider = None
return ErrorResponse.from_message("Invalid provider.", HTTP_404_NOT_FOUND)
else:
return ErrorResponse.from_message("Invalid provider.", HTTP_404_NOT_FOUND)
kwargs = {"modalities": ["text"]}
if provider == "MarkItDown":
kwargs = {
@ -669,7 +671,8 @@ class Api:
if provider in model_map:
config.model = provider
provider = None
return ErrorResponse.from_message("Invalid provider.", HTTP_404_NOT_FOUND)
else:
return ErrorResponse.from_message("Invalid provider.", HTTP_404_NOT_FOUND)
try:
audio = filter_none(voice=config.voice, format=config.response_format, language=config.language)
response = await self.client.chat.completions.create(