Update tools.py

tools.py

@@ -6,6 +6,7 @@ from langchain_openai import OpenAIEmbeddings
 from langchain_community.tools import WikipediaQueryRun, ArxivQueryRun
 from langchain_community.utilities import WikipediaAPIWrapper, ArxivAPIWrapper
 from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain_openai import ChatOpenAI
 from dotenv import load_dotenv
 import os
 import requests
@@ -16,6 +17,7 @@ import requests
 
 load_dotenv()
 
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 API_KEY = os.getenv("ALPHAVANTAGE_API_KEY")
 NEWS_API_KEY = os.getenv("NEWS_API_KEY")
 WEATHER_API_KEY = os.getenv("WEATHER_API_KEY")
@@ -102,26 +104,25 @@ def create_rag_tool():
 
         context = "\n\n".join(d.page_content for d in docs)
 
-        return f"""
-        You are given extracted content from a document.
+        summary_prompt = f"""
+        Summarize the following content clearly and concisely.
         
-        Your task:
-        - Produce ONE concise summary only.
-        - Use bullet points where appropriate.
-        - Do NOT repeat information.
-        - Do NOT restate the same idea in different words.
-        - Do NOT explain what the document is about.
-        - Do NOT add conclusions or meta commentary.
-        - Stop once the summary is complete.
+        Rules:
+        - Use bullet points when helpful
+        - Do NOT repeat ideas
+        - Do NOT mention the document
+        - Stop after the summary
         
-        IMPORTANT:
-        Return ONLY the final summarized content.
-        Do NOT add introductions, explanations, or commentary.
-        
-        DOCUMENT CONTENT:
+        CONTENT:
         {context}
         """
 
+        llm = ChatOpenAI(model="gpt-4.1-nano", temperature=0.2)
+
+        summary = llm.invoke(summary_prompt)
+
+        return summary.content.strip()
+
 
     return rag_search