diff --git a/src/engine/ollama.ts b/src/engine/ollama.ts index 225a24b..af3b927 100644 --- a/src/engine/ollama.ts +++ b/src/engine/ollama.ts @@ -33,9 +33,14 @@ export class OllamaEngine implements AiEngine { params ); - const message = response.data.message; + const { message } = response.data; + let content = message?.content; - return message?.content; + if (content && content.includes('')) { + return content.replace(/[\s\S]*?<\/think>/g, '').trim(); + } + + return content; } catch (err: any) { const message = err.response?.data?.error ?? err.message; throw new Error(`Ollama provider error: ${message}`);