introduce two agent type , react and normal. Change the graph

Agents/agent.py

@@ -11,6 +11,7 @@ from langchain.tools.retriever import create_retriever_tool
 from langchain_community.vectorstores import SupabaseVectorStore
 from langfuse.langchain import CallbackHandler
 from langgraph.graph import END
+from langgraph.prebuilt import create_react_agent
 
 load_dotenv()
 
@@ -65,6 +66,16 @@ tools = [
     remainder
 ]
 
+def use_react_agent():
+    graph = create_react_agent(
+        "anthropic:claude-3-7-sonnet-latest",
+        tools=[check_weather],
+        prompt="You are a helpful assistant",
+    )
+    inputs = {"messages": [{"role": "user", "content": "what is the weather in sf"}]}
+    for chunk in graph.stream(inputs, stream_mode="updates"):
+        print(chunk)
+
 def build_graph(provider: str="groq"):
     if provider== "groq":
         print(f"Running llm groq")
@@ -117,10 +128,7 @@ def build_graph(provider: str="groq"):
         builder.add_conditional_edges(
             "assistant",
             tools_condition,
-            path_map={
-                True: "tools",
-                False: END
-            }
+            {"tools": "tools", "__end__": "__end__"}
         )
         builder.add_edge("tools","assistant")
 

app.py

@@ -4,8 +4,9 @@ import requests
 import inspect
 import pandas as pd
 from langchain_core.messages import HumanMessage
-from Agents.agent import build_graph
+from Agents.agent import build_graph, tools,systemPrompt
 from langfuse.langchain import CallbackHandler
+from langgraph.prebuilt import create_react_agent
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -13,9 +14,25 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 langfuse_handler = CallbackHandler()
 testMode = bool(os.getenv("TestMode"))
 langFuseOn = bool(os.getenv("LangFuseOn"))
+agentType = os.getenv("AgentType")
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+class ReactAgent:
+    def __init__(self):
+        print(f"tools: {tools}")
+        print(f"prompt : {systemPrompt}")
+        graph = create_react_agent(
+            "anthropic:claude-3-7-sonnet-latest",
+            tools=tools,
+            prompt=systemPrompt,
+        )
+
+    def __call__(self,question:str) -> str:
+        inputs = {"messages": [{"role": "user", "content": "{question}"}]}
+        for chunk in self.graph.stream(inputs, stream_mode="updates"):
+            print(chunk)
+
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
@@ -63,7 +80,10 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        if(agentType == "React"):
+            agent = ReactAgent()
+        else:
+            agent = BasicAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None