This commit is contained in:
pablodanswer
2024-10-11 17:39:09 -07:00
committed by GitHub
parent 35d32ea3b0
commit d892203821

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)