Update tools.py

tools.py

@@ -83,41 +83,41 @@ def get_retriever():
 
 def create_rag_tool():
 
-    @tool()
+    @tool
     def rag_search(query: str) -> str:
         """
-        Retrieve and summarize information from uploaded documents.
-        Returns the final summarized answer directly.
+        Retrieve relevant information from the uploaded document.
+
+        Use this tool when the user asks questions related to the uploaded PDF
+        or any document-based knowledge. If no document is available or no
+        relevant information is found, return an appropriate message.
         """
+        global retriever
 
-        retriever = get_retriever()
         if retriever is None:
-            return "No document has been uploaded yet."
+            return "No document uploaded yet."
 
         docs = retriever.invoke(query)
 
         if not docs:
-            return "No relevant information found in the document."
+            return "No relevant information found in the uploaded document."
 
-        context = "\n".join(d.page_content for d in docs)
+        context = "\n\n".join(d.page_content for d in docs)
 
-        from langchain_openai import ChatOpenAI
-        llm = ChatOpenAI(model="gpt-4.1-nano", temperature=0.2)
+        return f"""
+            You are given extracted content from a document.
 
-        summary_prompt = f"""
-Summarize the following content clearly and concisely.
+            Your task:
+            - Summarize the content clearly
+            - Use bullet points where appropriate
+            - Keep formatting clean and readable
+            - Do NOT repeat unnecessary text
+            - Do NOT mention that this came from a document
 
-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
+            DOCUMENT CONTENT:
+            {context}
+            """
 
-CONTENT:
-{context}
-"""
-        return response.content.strip()
 
     return rag_search