From ea8c944dae15e0ec81cd4a81f7b9767e12aa6f14 Mon Sep 17 00:00:00 2001
From: bleedblack1 <nd4157958@gmail.com>
Date: Sun, 1 Mar 2026 02:24:50 +0530
Subject: [PATCH] Fix: Parallelize preprocessing to reduce latency

---
 src/agent/profiles/base.py | 43 +++++++++++++++++++++-----------------
 1 file changed, 24 insertions(+), 19 deletions(-)

diff --git a/src/agent/profiles/base.py b/src/agent/profiles/base.py
index b11d653..4170b36 100644
--- a/src/agent/profiles/base.py
+++ b/src/agent/profiles/base.py
@@ -1,5 +1,6 @@
 from typing import Annotated, Literal, TypedDict
 
+import asyncio
 from langchain_core.embeddings import Embeddings
 from langchain_core.language_models.chat_models import BaseChatModel
 from langchain_core.messages import BaseMessage
@@ -50,25 +51,29 @@ def __init__(
         self.search_workflow: Runnable = create_search_workflow(llm)
 
     async def preprocess(self, state: BaseState, config: RunnableConfig) -> BaseState:
-        rephrased_input: str = await self.rephrase_chain.ainvoke(
-            {
-                "user_input": state["user_input"],
-                "chat_history": state.get("chat_history", []),
-            },
-            config,
-        )
-        safety_check: SafetyCheck = await self.safety_checker.ainvoke(
-            {"rephrased_input": rephrased_input}, config
-        )
-        detected_language: str = await self.language_detector.ainvoke(
-            {"user_input": state["user_input"]}, config
-        )
-        return BaseState(
-            rephrased_input=rephrased_input,
-            safety=safety_check.safety,
-            reason_unsafe=safety_check.reason_unsafe,
-            detected_language=detected_language,
-        )
+            rephrased_input: str = await self.rephrase_chain.ainvoke(
+                {
+                    "user_input": state["user_input"],
+                    "chat_history": state.get("chat_history", []),
+                },
+                config,
+            )
+
+            safety_check, detected_language = await asyncio.gather(
+                self.safety_checker.ainvoke(
+                    {"rephrased_input": rephrased_input}, config
+                ),
+                self.language_detector.ainvoke(
+                    {"user_input": state["user_input"]}, config
+                ),
+            )
+
+            return BaseState(
+                rephrased_input=rephrased_input,
+                safety=safety_check.safety,
+                reason_unsafe=safety_check.reason_unsafe,
+                detected_language=detected_language,
+            )
 
     def proceed_with_research(self, state: BaseState) -> Literal["Continue", "Finish"]:
         return "Continue" if state["safety"] == "true" else "Finish"