Using groq for summary

agent.py

@@ -13,7 +13,8 @@ from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_community.document_loaders import WikipediaLoader
 from langchain_community.document_loaders import ArxivLoader
 
-from langchain_google_genai import ChatGoogleGenerativeAI
+# from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_groq import ChatGroq
 
 from langchain_huggingface import HuggingFaceEmbeddings
 
@@ -23,16 +24,17 @@ from supabase import create_client, Client
 
 load_dotenv()
 
-# os.environ["TAVILY_API_KEY"] = os.environ.get("TAVILY_API_KEY")
-# os.environ["GOOGLE_API_KEY"] = os.environ.get("GOOGLE_API_KEY")
+#os.environ["TAVILY_API_KEY"] = os.environ.get("TAVILY_API_KEY")
+#os.environ["GOOGLE_API_KEY"] = os.environ.get("GOOGLE_API_KEY")
+#os.environ["GROQ_API_KEY"] = os.environ.get("GROQ_API_KEY")
 
 __embeddings = HuggingFaceEmbeddings(
     model_name="sentence-transformers/all-mpnet-base-v2",
     model_kwargs= { 'device': 'cpu' })
 
 # connect to supabase
-url: str = os.environ.get("SUPABASE_URL")
-key: str = os.environ.get("SUPABASE_SECRET_KEY")
+url: str = os.getenv("SUPABASE_URL")
+key: str = os.getenv("SUPABASE_SECRET_KEY")
 __supabase: Client = create_client(url, key)
 
 # build retriever
@@ -173,7 +175,8 @@ class AgentState(TypedDict):
 
 
 def build_graph():
-    llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+    # llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+    llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
     llm_with_tools = llm.bind_tools(tools)
 
     # Node 
@@ -207,8 +210,8 @@ def build_graph():
 
 # Test
 if __name__ == "__main__":
-    question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"
-    # question = "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia."
+    # question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"
+    question = "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia."
     graph = build_graph()
     messages = [HumanMessage(content=question)]
     messages = graph.invoke({ "messages": messages })

app.py

@@ -10,6 +10,9 @@ from agent import build_graph
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+os.environ["TAVILY_API_KEY"] = os.getenv("TAVILY_API_KEY")
+os.environ["GROQ_API_KEY"] = os.getenv("GROQ_API_KEY")
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
@@ -24,7 +27,7 @@ class BasicAgent:
         # print(f"Agent received question (first 50 chars): {question[:50]}...")
         # fixed_answer = "This is a default answer."
         # print(f"Agent returning fixed answer: {fixed_answer}")
-        return answer
+        return answer[:14]
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """

prompt.txt

@@ -1,49 +1,5 @@
-You are a helpful agent responsible for answering questions using a set of tools provided.
-If the tool(s) not available, you can try to search and find the solution or information online.
-You can also use your own knowledge to answer the question.
-==========================
-Here is a few examples showing you how to answer the question step by step.
-
-Question 1: In terms of geographical distance between capital cities, which 2 countries are the furthest from each other within the ASEAN bloc according to wikipedia? Answer using a comma separated list, ordering the countries by alphabetical order.
-Steps:
-1. Search the web for "ASEAN bloc".
-2. Click the Wikipedia result for the ASEAN Free Trade Area.
-3. Scroll down to find the list of member states.
-4. Click into the Wikipedia pages for each member state, and note its capital.
-5. Search the web for the distance between the first two capitals. The results give travel distance, not geographic distance, which might affect the answer.
-6. Thinking it might be faster to judge the distance by looking at a map, search the web for "ASEAN bloc" and click into the images tab.
-7. View a map of the member countries. Since they're clustered together in an arrangement that's not very linear, it's difficult to judge distances by eye.
-8. Return to the Wikipedia page for each country. Click the GPS coordinates for each capital to get the coordinates in decimal notation.
-9. Place all these coordinates into a spreadsheet.
-10. Write formulas to calculate the distance between each capital.
-11. Write formula to get the largest distance value in the spreadsheet.
-12. Note which two capitals that value corresponds to: Jakarta and Naypyidaw.
-13. Return to the Wikipedia pages to see which countries those respective capitals belong to: Indonesia, Myanmar.
-Tools:
-1. Search engine
-2. Web browser
-3. Microsoft Excel / Google Sheets
-Final Answer: Indonesia, Myanmar
-
-Question 2: Review the chess position provided in the image. It is black's turn. Provide the correct next move for black which guarantees a win. Please provide your response in algebraic notation.
-Steps:
-Step 1: Evaluate the position of the pieces in the chess position
-Step 2: Report the best move available for black: "Rd5"
-Tools:
-1. Image recognition tools
-Final Answer: Rd5
-
-Question 3: Solve the equation x^2 + 5x = -6
-Steps:
-Step 1: Moving all terms to left-hand side until the right-hand side become zero.
-Step 2: Identify the highest power of polynomial in left-hand side. In this case the highest power is 2, this equation is a quadratic equation.
-Step 3: Identify the coefficients of each term in this quadratic equation.
-Step 3: Write quadratic formula and calculate the possible solutions.
-Tools:
-1. Search engine
-2. Web browser
-3. Calculator 
-Final Answer: x=-2, x=-3
-
-==========================
-Now, please answer the following question step by step.
+You are a helpful assistant tasked with answering questions using a set of tools. 
+Now, I will ask you a question. Report your thoughts, and finish your answer with the following template: 
+FINAL ANSWER: [YOUR FINAL ANSWER]. 
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+Your answer should only start with "FINAL ANSWER: ", then follows with the answer.

requirements.txt

@@ -7,6 +7,7 @@ langchain-google-genai
 langchain-huggingface
 langchain-tavily
 langchain-chroma
+langchain-groq
 langgraph
 huggingface_hub
 supabase