Skip to content

Commit 6816379

Browse files
badraxasbmovi
andauthored
feat(engine/ollama.ts): extract message from response data and clean content by removing unnecessary tags (#445)
Co-authored-by: Bastien Mottier <[email protected]>
1 parent 9d80991 commit 6816379

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/engine/ollama.ts

+7-2
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,14 @@ export class OllamaEngine implements AiEngine {
3333
params
3434
);
3535

36-
const message = response.data.message;
36+
const { message } = response.data;
37+
let content = message?.content;
3738

38-
return message?.content;
39+
if (content && content.includes('<think>')) {
40+
return content.replace(/<think>[\s\S]*?<\/think>/g, '').trim();
41+
}
42+
43+
return content;
3944
} catch (err: any) {
4045
const message = err.response?.data?.error ?? err.message;
4146
throw new Error(`Ollama provider error: ${message}`);

0 commit comments

Comments
 (0)