updated openwebui api provider options and fix fim

twinnydotdev · Apr 26, 2024 · 60ee0fd · 60ee0fd
1 parent 4682248
commit 60ee0fd
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git 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
@@ -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
@@ -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