diff --git a/src/common/types.ts b/src/common/types.ts index 2c81e3c3..1ca9000b 100644 --- a/src/common/types.ts +++ b/src/common/types.ts @@ -154,12 +154,12 @@ export interface UiTabs { } export const ApiProviders = { - Ollama: 'ollama', - OllamaWebUi: 'ollamawebui', + LiteLLM: 'litellm', LlamaCpp: 'llamacpp', LMStudio: 'lmstudio', + Ollama: 'ollama', Oobabooga: 'oobabooga', - LiteLLM: 'litellm' + OpenWebUI: 'openwebui', } as const export interface ApiModel { diff --git a/src/extension/provider-options.ts b/src/extension/provider-options.ts index 731576c2..4426ba62 100644 --- a/src/extension/provider-options.ts +++ b/src/extension/provider-options.ts @@ -20,7 +20,7 @@ export function createStreamRequestBody( ): StreamBodyBase | StreamOptionsOllama | StreamBodyOpenAI { switch (provider) { case ApiProviders.Ollama: - case ApiProviders.OllamaWebUi: + case ApiProviders.OpenWebUI: return { model: options.model, prompt, @@ -63,7 +63,7 @@ export function createStreamRequestBodyFim( ): StreamBodyBase | StreamOptionsOllama | StreamBodyOpenAI { switch (provider) { case ApiProviders.Ollama: - case ApiProviders.OllamaWebUi: + case ApiProviders.OpenWebUI: return { model: options.model, prompt, diff --git a/src/extension/utils.ts b/src/extension/utils.ts index 2741bba4..9c0079fc 100644 --- a/src/extension/utils.ts +++ b/src/extension/utils.ts @@ -305,7 +305,7 @@ export const getChatDataFromProvider = ( ) => { switch (provider) { case ApiProviders.Ollama: - case ApiProviders.OllamaWebUi: + case ApiProviders.OpenWebUI: return data?.choices[0].delta?.content ? data?.choices[0].delta.content : '' @@ -326,6 +326,7 @@ export const getFimDataFromProvider = ( ) => { switch (provider) { case ApiProviders.Ollama: + case ApiProviders.OpenWebUI: return data?.response case ApiProviders.LlamaCpp: return data?.content