🐛 fix: fix ollama host issue (lobehub#2210)

Co-authored-by: eyebrowkang <[email protected]>
zwl2828596 · Apr 26, 2024 · c49d4c7 · c49d4c7
1 parent 1ca7cad
commit c49d4c7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/libs/agent-runtime/ollama/index.ts b/src/libs/agent-runtime/ollama/index.ts
@@ -24,7 +24,7 @@ export class LobeOllamaAI implements LobeRuntimeAI {
       throw AgentRuntimeError.createError(AgentRuntimeErrorType.InvalidOllamaArgs);
     }
 
-    this.client = new Ollama(!baseURL ? undefined : { host: new URL(baseURL).host });
+    this.client = new Ollama(!baseURL ? undefined : { host: baseURL });
 
     if (baseURL) this.baseURL = baseURL;
   }

diff --git a/src/services/ollama.ts b/src/services/ollama.ts
@@ -27,8 +27,7 @@ export class OllamaService {
   getHost = (): string => {
     const config = modelConfigSelectors.ollamaConfig(useGlobalStore.getState());
 
-    const url = new URL(config.endpoint || DEFAULT_BASE_URL);
-    return url.host;
+    return config.endpoint || DEFAULT_BASE_URL;
   };
 
   getOllamaClient = () => {