Update agents.py

agents.py

@@ -19,21 +19,19 @@ from dotenv import load_dotenv
 # supervisor_prompt = os.getenv("SUPERVISOR_PROMPT")
 
 def build_graph():
-   a_debug = False
-   model  = init_chat_model("llama3-70b-8192", model_provider="groq")
-   summarization_node = SummarizationNode(token_counter=count_tokens_approximately,
+    a_debug = False
+    model  = init_chat_model("llama3-70b-8192", model_provider="groq")
+    summarization_node = SummarizationNode(token_counter=count_tokens_approximately,
                                            model=model,
                                            max_tokens=2048,
                                            max_summary_tokens=1024,
                                            output_messages_key="llm_input_messages",)
     
-   class State(AgentState):
-      context: dict[str, Any]
-   
-   checkpointer = InMemorySaver()
-       
-
-   agent0= create_react_agent(model=model,
+    class State(AgentState):
+        context: dict[str, Any]
+        checkpointer = InMemorySaver()
+        
+    agent0= create_react_agent(model=model,
                               tools= [custom_tools.stock_data_tool],
                               prompt=("Give the company name to the stock tool produce the stock fundamentals. Summarize all the information in one paragraph in 200 words."),
                               name="Stock Analyst",
@@ -41,8 +39,8 @@ def build_graph():
                               state_schema=State, 
                               checkpointer=checkpointer,
                               debug=a_debug)
-   
-   agent1= create_react_agent(model=model,
+        
+    agent1= create_react_agent(model=model,
                               tools= [custom_tools.web_search],
                               prompt='''You are a geek that can find anything from the internet. Do the research and provide the summery in 250 words and in only one paragraph.''',
                               name="Web Surfer",
@@ -50,8 +48,8 @@ def build_graph():
                               state_schema=State, 
                               checkpointer=checkpointer,
                               debug=a_debug)
-    
-   agent2= create_react_agent(model=model,
+
+    agent2= create_react_agent(model=model,
                               tools= [custom_tools.reddit_search_tool],
                               prompt='''You are a social media analyst. your job is to search social media and find out the sentiments of public on any topic. Do the research and provide the summery in 250 words and in only one paragraph.''', 
                               name="Social Media Analyst",
@@ -60,7 +58,7 @@ def build_graph():
                               checkpointer=checkpointer,
                               debug=a_debug)
     
-   agent3= create_react_agent(model=model,
+    agent3= create_react_agent(model=model,
                               tools= [custom_tools.tech_news_tool],
                               prompt='''You are a Technology news analyst. You find the latest news related to a topic and summerize it to findout the overall picture. Do the research and provide the summery in 250 words and in only one paragraph.''', 
                               name="Tech Journalist",
@@ -69,7 +67,7 @@ def build_graph():
                               checkpointer=checkpointer,
                               debug=a_debug)
    
-   agent4= create_react_agent(model=model,
+    agent4= create_react_agent(model=model,
                               tools= [custom_tools.politics_news_tool],
                               prompt='''You are a Politics news analyst. You find the latest news related to a topic and summerize it to findout the overall picture. Do the research and provide the summery in 250 words and in only one paragraph.''', 
                               name="Political Journalist",
@@ -78,7 +76,7 @@ def build_graph():
                               checkpointer=checkpointer,
                               debug=a_debug)
    
-   agent5= create_react_agent(model=model,
+    agent5= create_react_agent(model=model,
                               tools= [custom_tools.business_news_tool],
                               prompt='''You are a Business news analyst. You find the latest news related to a topic and summerize it to findout the overall picture. Do the research and provide the summery in 250 words and in only one paragraph.''', 
                               name="Business Journalist",
@@ -86,17 +84,15 @@ def build_graph():
                               state_schema=State, 
                               checkpointer=checkpointer,
                               debug=a_debug)
-   
-   agent6= create_react_agent(model=model,
+    
+    agent6= create_react_agent(model=model,
                               tools= [custom_tools.world_news_tool],
                               prompt='''You are a Geopolitical news analyst. You find the latest news related to a topic and summerize it to findout the overall picture. Do the research and provide the summery in 250 words and in only one paragraph.''', 
                               name="Geopolitical Journalist",
                               pre_model_hook= summarization_node,
                               state_schema=State, 
                               checkpointer=checkpointer,
-                              debug=a_debug)
-
-
+                              debug=a_debug)     
     supervisor_prompt = '''You are a stock market research supervisor based in the United Kingdom. You lead a team of expert agents, each specializing in a specific domain. Your job is to orchestrate these agents to compile a comprehensive report based on the user's stock-related question.
 IMPORTANT: You DO NOT analyze anything yourself — your role is to build a strategy, decide when and whom to call, and summarize their findings.
 ---
@@ -155,13 +151,9 @@ Optimize calls:
 Your output should contain intermediate decisions, the agent/tool called, the query issued, and the response. After all tools are used, generate the final report.
 
 '''
-   
-     
-
-
 
-   graph = create_supervisor(agents=[agent0,agent1,agent2,agent3,agent4,agent5,agent6],
+    graph = create_supervisor(agents=[agent0,agent1,agent2,agent3,agent4,agent5,agent6],
                              model=model,
                              prompt= supervisor_prompt)
-   return graph.compile()
+    return graph.compile()