From ffa793056d69116c1cdaca825ac917e8a5113e52 Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <95534749+ItzCrazyKns@users.noreply.github.com>
Date: Thu, 20 Mar 2025 11:47:54 +0530
Subject: [PATCH] feat(chains): remove think tags

---
 src/lib/chains/imageSearchAgent.ts | 2 ++
 src/lib/chains/videoSearchAgent.ts | 2 ++
 src/lib/search/metaSearchAgent.ts  | 2 ++
 3 files changed, 6 insertions(+)

diff --git a/src/lib/chains/imageSearchAgent.ts b/src/lib/chains/imageSearchAgent.ts
index d194612..4fd684f 100644
--- a/src/lib/chains/imageSearchAgent.ts
+++ b/src/lib/chains/imageSearchAgent.ts
@@ -58,6 +58,8 @@ const createImageSearchChain = (llm: BaseChatModel) => {
     llm,
     strParser,
     RunnableLambda.from(async (input: string) => {
+      input = input.replace(/<think>.*?<\/think>/g, '');
+
       const res = await searchSearxng(input, {
         engines: ['bing images', 'google images'],
       });
diff --git a/src/lib/chains/videoSearchAgent.ts b/src/lib/chains/videoSearchAgent.ts
index e1051f6..f7cb156 100644
--- a/src/lib/chains/videoSearchAgent.ts
+++ b/src/lib/chains/videoSearchAgent.ts
@@ -59,6 +59,8 @@ const createVideoSearchChain = (llm: BaseChatModel) => {
     llm,
     strParser,
     RunnableLambda.from(async (input: string) => {
+      input = input.replace(/<think>.*?<\/think>/g, '');
+
       const res = await searchSearxng(input, {
         engines: ['youtube'],
       });
diff --git a/src/lib/search/metaSearchAgent.ts b/src/lib/search/metaSearchAgent.ts
index 38ea2a0..4a3d817 100644
--- a/src/lib/search/metaSearchAgent.ts
+++ b/src/lib/search/metaSearchAgent.ts
@@ -202,6 +202,8 @@ class MetaSearchAgent implements MetaSearchAgentType {
 
           return { query: question, docs: docs };
         } else {
+          question = question.replace(/<think>.*?<\/think>/g, '');
+
           const res = await searchSearxng(question, {
             language: 'en',
             engines: this.config.activeEngines,