Update tools.py

tools.py

@@ -83,17 +83,14 @@ def get_retriever():
 
 def create_rag_tool():
 
-    @tool
+    @tool(return_direct=True)
     def rag_search(query: str) -> str:
         """
         Retrieve and summarize information from uploaded documents.
-
-        Use this tool whenever the user asks about uploaded documents.
-        Summarize clearly and avoid repeating raw content.
+        Returns the final summarized answer directly.
         """
 
         retriever = get_retriever()
-
         if retriever is None:
             return "No document has been uploaded yet."
 
@@ -102,27 +99,27 @@ def create_rag_tool():
         if not docs:
             return "No relevant information found in the document."
 
-        context = "\n\n".join(d.page_content for d in docs)
-
-        summary_prompt = f"""
-        Summarize the following content clearly and concisely.
-        
-        Rules:
-        - Use bullet points when helpful
-        - Do NOT repeat ideas
-        - Do NOT mention the document
-        - Stop after the summary
-        
-        CONTENT:
-        {context}
-        """
+        context = "\n".join(d.page_content for d in docs)
 
+        from langchain_openai import ChatOpenAI
         llm = ChatOpenAI(model="gpt-4.1-nano", temperature=0.2)
 
-        summary = llm.invoke(summary_prompt)
+        summary_prompt = f"""
+Summarize the following content clearly and concisely.
+
+Rules:
+- Use bullet points where appropriate
+- Do NOT repeat ideas
+- Do NOT explain what the document is
+- Do NOT add commentary
+- Stop after the summary
 
-        return summary.content.strip()
+CONTENT:
+{context}
+"""
 
+        response = llm.invoke(summary_prompt)
+        return response.content.strip()
 
     return rag_search