This commit is contained in:
pablodanswer 2024-10-11 17:39:09 -07:00 committed by GitHub
parent 35d32ea3b0
commit d892203821
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -37,7 +37,7 @@ def compute_all_tool_tokens(tools: list[Tool], llm_tokenizer: BaseTokenizer) ->
def is_image_generation_available(db_session: Session) -> bool:
providers = db_session.query(LLMProvider).all()
for provider in providers:
if provider.name == "OpenAI":
if provider.provider == "openai":
return True
return bool(AZURE_DALLE_API_KEY)