Fix black

This commit is contained in:
Weves 2024-07-19 15:39:42 -07:00 committed by Chris Weaver
parent 864c552a17
commit 7aa3602fca

View File

@ -6,7 +6,13 @@ from danswer.llm.utils import get_default_llm_tokenizer
from danswer.tools.tool import Tool
OPEN_AI_TOOL_CALLING_MODELS = {"gpt-3.5-turbo", "gpt-4-turbo", "gpt-4", "gpt-4o", "gpt-4o-mini"}
OPEN_AI_TOOL_CALLING_MODELS = {
"gpt-3.5-turbo",
"gpt-4-turbo",
"gpt-4",
"gpt-4o",
"gpt-4o-mini",
}
def explicit_tool_calling_supported(model_provider: str, model_name: str) -> bool: