highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-06-21 13:30:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add GPT-4o-Mini & fix a missing gpt-4o

Browse Source
This commit is contained in:
versecafe 2024-07-19 12:12:09 -05:00 committed by Chris Weaver
parent 1ebae50d0c
commit 86d1804eb0
4 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backend/danswer/llm/llm_provider_options.py
View File

@ -26,6 +26,7 @@ OPENAI_PROVIDER_NAME = "openai"
OPEN_AI_MODEL_NAMES = [ OPEN_AI_MODEL_NAMES = [
"gpt-4", "gpt-4",
"gpt-4o", "gpt-4o",
"gpt-4o-mini",
"gpt-4-turbo", "gpt-4-turbo",
"gpt-4-turbo-preview", "gpt-4-turbo-preview",
"gpt-4-1106-preview", "gpt-4-1106-preview",

2
backend/danswer/tools/utils.py
View File

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

2
backend/scripts/test-openapi-key.py
View File

@ -2,6 +2,8 @@ from openai import OpenAI
VALID_MODEL_LIST = [ VALID_MODEL_LIST = [
"gpt-4o-mini",
"gpt-4o",
"gpt-4-1106-preview", "gpt-4-1106-preview",
"gpt-4-vision-preview", "gpt-4-vision-preview",
"gpt-4", "gpt-4",

2
web/src/lib/llm/utils.ts
View File

@ -36,10 +36,12 @@ export function getFinalLLM(
const MODELS_SUPPORTING_IMAGES = [ const MODELS_SUPPORTING_IMAGES = [
["openai", "gpt-4o"], ["openai", "gpt-4o"],
["openai", "gpt-4o-mini"],
["openai", "gpt-4-vision-preview"], ["openai", "gpt-4-vision-preview"],
["openai", "gpt-4-turbo"], ["openai", "gpt-4-turbo"],
["openai", "gpt-4-1106-vision-preview"], ["openai", "gpt-4-1106-vision-preview"],
["azure", "gpt-4o"], ["azure", "gpt-4o"],
["azure", "gpt-4o-mini"],
["azure", "gpt-4-vision-preview"], ["azure", "gpt-4-vision-preview"],
["azure", "gpt-4-turbo"], ["azure", "gpt-4-turbo"],
["azure", "gpt-4-1106-vision-preview"], ["azure", "gpt-4-1106-vision-preview"],