gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

changed settings and helpers to use watsonx instead of ibm for model provider

Browse source
This commit is contained in:
Lucas Oliveira 2025-09-29 17:40:12 -03:00
parent 530a223518
commit 9f83811ab4
2 changed files with 1124 additions and 1109 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

156
frontend/src/app/settings/helpers/model-helpers.tsx
View file

@ -1,87 +1,97 @@
import OpenAILogo from "@/components/logo/openai-logo";
import OllamaLogo from "@/components/logo/ollama-logo";
import IBMLogo from "@/components/logo/ibm-logo"; import IBMLogo from "@/components/logo/ibm-logo";
import OllamaLogo from "@/components/logo/ollama-logo";
import OpenAILogo from "@/components/logo/openai-logo";
export type ModelProvider = 'openai' | 'ollama' | 'ibm'; export type ModelProvider = "openai" | "ollama" | "watsonx";
export interface ModelOption { export interface ModelOption {
value: string; value: string;
label: string; label: string;
} }
// Helper function to get model logo based on provider or model name // Helper function to get model logo based on provider or model name
export function getModelLogo(modelValue: string, provider?: ModelProvider) { export function getModelLogo(modelValue: string, provider?: ModelProvider) {
// First check by provider // First check by provider
if (provider === 'openai') { if (provider === "openai") {
return <OpenAILogo className="w-4 h-4" />; return <OpenAILogo className="w-4 h-4" />;
} else if (provider === 'ollama') { } else if (provider === "ollama") {
return <OllamaLogo className="w-4 h-4" />; return <OllamaLogo className="w-4 h-4" />;
} else if (provider === 'ibm') { } else if (provider === "watsonx") {
return <IBMLogo className="w-4 h-4" />; return <IBMLogo className="w-4 h-4" />;
} }
// Fallback to model name analysis // Fallback to model name analysis
if (modelValue.includes('gpt') || modelValue.includes('text-embedding')) { if (modelValue.includes("gpt") || modelValue.includes("text-embedding")) {
return <OpenAILogo className="w-4 h-4" />; return <OpenAILogo className="w-4 h-4" />;
} else if (modelValue.includes('llama') || modelValue.includes('ollama')) { } else if (modelValue.includes("llama") || modelValue.includes("ollama")) {
return <OllamaLogo className="w-4 h-4" />; return <OllamaLogo className="w-4 h-4" />;
} else if (modelValue.includes('granite') || modelValue.includes('slate') || modelValue.includes('ibm')) { } else if (
return <IBMLogo className="w-4 h-4" />; modelValue.includes("granite") ||
} modelValue.includes("slate") ||
modelValue.includes("ibm")
) {
return <IBMLogo className="w-4 h-4" />;
}
return <OpenAILogo className="w-4 h-4" />; // Default to OpenAI logo return <OpenAILogo className="w-4 h-4" />; // Default to OpenAI logo
} }
// Helper function to get fallback models by provider // Helper function to get fallback models by provider
export function getFallbackModels(provider: ModelProvider) { export function getFallbackModels(provider: ModelProvider) {
switch (provider) { switch (provider) {
case 'openai': case "openai":
return { return {
language: [ language: [
{ value: 'gpt-4', label: 'GPT-4' }, { value: "gpt-4", label: "GPT-4" },
{ value: 'gpt-4-turbo', label: 'GPT-4 Turbo' }, { value: "gpt-4-turbo", label: "GPT-4 Turbo" },
{ value: 'gpt-3.5-turbo', label: 'GPT-3.5 Turbo' }, { value: "gpt-3.5-turbo", label: "GPT-3.5 Turbo" },
], ],
embedding: [ embedding: [
{ value: 'text-embedding-ada-002', label: 'text-embedding-ada-002' }, { value: "text-embedding-ada-002", label: "text-embedding-ada-002" },
{ value: 'text-embedding-3-small', label: 'text-embedding-3-small' }, { value: "text-embedding-3-small", label: "text-embedding-3-small" },
{ value: 'text-embedding-3-large', label: 'text-embedding-3-large' }, { value: "text-embedding-3-large", label: "text-embedding-3-large" },
], ],
}; };
case 'ollama': case "ollama":
return { return {
language: [ language: [
{ value: 'llama2', label: 'Llama 2' }, { value: "llama2", label: "Llama 2" },
{ value: 'llama2:13b', label: 'Llama 2 13B' }, { value: "llama2:13b", label: "Llama 2 13B" },
{ value: 'codellama', label: 'Code Llama' }, { value: "codellama", label: "Code Llama" },
], ],
embedding: [ embedding: [
{ value: 'mxbai-embed-large', label: 'MxBai Embed Large' }, { value: "mxbai-embed-large", label: "MxBai Embed Large" },
{ value: 'nomic-embed-text', label: 'Nomic Embed Text' }, { value: "nomic-embed-text", label: "Nomic Embed Text" },
], ],
}; };
case 'ibm': case "watsonx":
return { return {
language: [ language: [
{ value: 'meta-llama/llama-3-1-70b-instruct', label: 'Llama 3.1 70B Instruct' }, {
{ value: 'ibm/granite-13b-chat-v2', label: 'Granite 13B Chat v2' }, value: "meta-llama/llama-3-1-70b-instruct",
], label: "Llama 3.1 70B Instruct",
embedding: [ },
{ value: 'ibm/slate-125m-english-rtrvr', label: 'Slate 125M English Retriever' }, { value: "ibm/granite-13b-chat-v2", label: "Granite 13B Chat v2" },
], ],
}; embedding: [
default: {
return { value: "ibm/slate-125m-english-rtrvr",
language: [ label: "Slate 125M English Retriever",
{ value: 'gpt-4', label: 'GPT-4' }, },
{ value: 'gpt-4-turbo', label: 'GPT-4 Turbo' }, ],
{ value: 'gpt-3.5-turbo', label: 'GPT-3.5 Turbo' }, };
], default:
embedding: [ return {
{ value: 'text-embedding-ada-002', label: 'text-embedding-ada-002' }, language: [
{ value: 'text-embedding-3-small', label: 'text-embedding-3-small' }, { value: "gpt-4", label: "GPT-4" },
{ value: 'text-embedding-3-large', label: 'text-embedding-3-large' }, { value: "gpt-4-turbo", label: "GPT-4 Turbo" },
], { value: "gpt-3.5-turbo", label: "GPT-3.5 Turbo" },
}; ],
} embedding: [
} { value: "text-embedding-ada-002", label: "text-embedding-ada-002" },
{ value: "text-embedding-3-small", label: "text-embedding-3-small" },
{ value: "text-embedding-3-large", label: "text-embedding-3-large" },
],
};
}
}

2077
frontend/src/app/settings/page.tsx
View file

File diff suppressed because it is too large Load diff