From b631703aa6bfd50fe6454abb4f7128bdf4f67a88 Mon Sep 17 00:00:00 2001
From: Gabe <yugang2002@gmail.com>
Date: Wed, 3 Sep 2025 14:02:55 +0800
Subject: [PATCH] fix: update ollama api

---
 src/apis/trans.js | 18 +++++++++++++++---
 src/config/api.js |  2 +-
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/src/apis/trans.js b/src/apis/trans.js
index e7273d4..232a8a2 100644
--- a/src/apis/trans.js
+++ b/src/apis/trans.js
@@ -554,6 +554,8 @@ const genOllama = ({
   systemPrompt,
   userPrompt,
   model,
+  temperature,
+  maxTokens,
   customHeader,
   customBody,
   docInfo,
@@ -565,9 +567,19 @@ const genOllama = ({
 
   const data = {
     model,
-    system: systemPrompt,
-    prompt: userPrompt,
-    think: think,
+    messages: [
+      {
+        role: "system",
+        content: systemPrompt,
+      },
+      {
+        role: "user",
+        content: userPrompt,
+      },
+    ],
+    temperature,
+    max_tokens: maxTokens,
+    think,
     stream: false,
     ...customBody,
   };
diff --git a/src/config/api.js b/src/config/api.js
index ab37e5c..3483342 100644
--- a/src/config/api.js
+++ b/src/config/api.js
@@ -379,7 +379,7 @@ const defaultOpenaiApi = {
 };
 const defaultOllamaApi = {
   ...defaultApi,
-  url: "http://localhost:11434/api/generate",
+  url: "http://localhost:11434/v1/chat/completions",
   model: "llama3.1",
 };
 export const DEFAULT_TRANS_APIS = {