highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-07-08 05:30:24 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add image generation for gpt-4o

Browse Source
This commit is contained in:
Weves
2024-06-03 16:36:15 -07:00
committed by Chris Weaver
parent 3788041115
commit 0c642f25dd
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
web/src/app/admin/assistants/AssistantEditor.tsx
View File

@ -36,6 +36,7 @@ import { Option } from "@/components/Dropdown";
import { ToolSnapshot } from "@/lib/tools/interfaces"; import { ToolSnapshot } from "@/lib/tools/interfaces";
import { checkUserIsNoAuthUser } from "@/lib/user"; import { checkUserIsNoAuthUser } from "@/lib/user";
import { addAssistantToList } from "@/lib/assistants/updateAssistantPreferences"; import { addAssistantToList } from "@/lib/assistants/updateAssistantPreferences";
import { checkLLMSupportsImageInput } from "@/lib/llm/utils";
function findSearchTool(tools: ToolSnapshot[]) { function findSearchTool(tools: ToolSnapshot[]) {
return tools.find((tool) => tool.in_code_tool_id === "SearchTool"); return tools.find((tool) => tool.in_code_tool_id === "SearchTool");
@ -45,10 +46,6 @@ function findImageGenerationTool(tools: ToolSnapshot[]) {
return tools.find((tool) => tool.in_code_tool_id === "ImageGenerationTool"); return tools.find((tool) => tool.in_code_tool_id === "ImageGenerationTool");
} }
function checkLLMSupportsImageGeneration(provider: string, model: string) {
return provider === "openai" && model === "gpt-4-turbo";
}
function Label({ children }: { children: string | JSX.Element }) { function Label({ children }: { children: string | JSX.Element }) {
return ( return (
<div className="block font-medium text-base text-emphasis">{children}</div> <div className="block font-medium text-base text-emphasis">{children}</div>
@ -261,7 +258,7 @@ export function AssistantEditor({
if ( if (
values.image_generation_tool_enabled && values.image_generation_tool_enabled &&
imageGenerationTool && imageGenerationTool &&
checkLLMSupportsImageGeneration( checkLLMSupportsImageInput(
providerDisplayNameToProviderName.get( providerDisplayNameToProviderName.get(
values.llm_model_provider_override || "" values.llm_model_provider_override || ""
) || ) ||
@ -566,7 +563,7 @@ export function AssistantEditor({
)} )}
{imageGenerationTool && {imageGenerationTool &&
checkLLMSupportsImageGeneration( checkLLMSupportsImageInput(
providerDisplayNameToProviderName.get( providerDisplayNameToProviderName.get(
values.llm_model_provider_override || "" values.llm_model_provider_override || ""
) || ) ||