budget agent

Author: amritav

backend/agents/agents_framework.py

@@ -15,12 +15,14 @@
 from .outreach_social_tool import OutreachSocialTool
 from .agent_permit import PermitSearchTool
 from .schedule_tool import ScheduleTool
+from .budget_tool import BudgetTool
 
 resourse_search_tool = SearchTool()
 outreach_social_tool = OutreachSocialTool()
 schedule_tool = ScheduleTool()
+budget_tool = BudgetTool()
 
-members = ["Resources", "Social_Outreach", "Volunteer_Outreach", "Schedule", "Permits"]
+members = ["Budget", "Resources", "Social_Outreach", "Volunteer_Outreach", "Schedule", "Permits"]
 options = members + ["FINISH"]
 
 # Our team supervisor is an LLM node. It just picks the next agent to process
@@ -154,6 +156,30 @@
     "- Temporary Event Permit: Authorizes use of public space for the event.\n"
 )
 
+budget_prompt = """You are an expert budget planning assistant specializing in all types of events. When a user asks about budget allocation:
+
+1. Use the calculate_budget_breakdown tool to get the budget amount and event type
+2. Based on the event type, first determine ALL relevant budget categories that would be needed. Consider:
+   - The specific nature and purpose of the event
+   - Standard requirements for this type of event
+   - Special equipment or resources typically needed
+   - Both essential and optional components
+   
+3. For each identified category:
+   - Assign an appropriate percentage based on importance and typical costs
+   - Calculate the actual dollar amount
+
+4. Your response should:
+   - List all relevant budget categories specific to the event type
+   - Show percentage and dollar amount for each category
+   - Ensure percentages sum to exactly 100%
+   - Group related items logically
+
+Remember that categories should be specifically tailored to the exact event type provided, not using generic templates.
+Only output the answer, no other text.
+If you need any clarification about the event type or special requirements, ask the user.
+"""
+
 prompt = ChatPromptTemplate.from_messages(
     [
         ("system", system_prompt),
@@ -270,6 +296,23 @@ def permit_node(state: dict) -> Command[str]:
         goto="Supervisor",
     )
 
+budget_agent = create_react_agent(
+    llm,
+    tools=[budget_tool],
+    prompt=budget_prompt
+)
+
+def budget_node(state: State) -> Command[Literal["Supervisor"]]:
+    result = budget_agent.invoke(state)
+    return Command(
+        update={
+            "messages": [
+                HumanMessage(content=result["messages"][-1].content, name="Budget")
+            ]
+        },
+        goto="Supervisor",
+    )
+
 def supervisor_node(state: State) -> Command[Literal[*members, "__end__"]]:
     messages = [
         {"role": "system", "content": system_prompt},
@@ -291,6 +334,7 @@ def initialize_graph():
     builder.add_node("Volunteer_Outreach", outreach_volunteer_node)
     builder.add_node("Schedule", schedule_node)
     builder.add_node("Permits", permit_node)
+    builder.add_node("Budget", budget_node)
     builder.set_entry_point("Supervisor")
     return builder.compile()
 

backend/agents/budget_tool.py

@@ -0,0 +1,49 @@
+from pydantic import BaseModel, Field
+from langchain.tools import BaseTool
+from typing import Type
+from langchain_core.messages import HumanMessage
+from langchain_core.tools import tool
+
+class BudgetInput(BaseModel):
+    budget_str: str = Field(description="The budget amount and event type in the format 'amount:event_type'.")
+
+class BudgetTool(BaseTool):
+    name: str = "budget_tool"
+    description: str = "Calculates budget allocation for a given event."
+    args_schema: Type[BaseModel] = BudgetInput
+
+    def _run(self, budget_str: str) -> str:
+        """Calculate budget allocation based on the provided amount and event type."""
+        try:
+            amount, event_type = budget_str.split(':')
+            amount = float(amount.replace('$', '').replace(',', ''))
+
+            # General budget breakdown prompt
+            response = f"""
+You are an expert budget planning assistant specializing in all types of events. When a user asks about budget allocation:
+
+1.⁠ ⁠Use the provided budget amount and event type to determine relevant budget categories.
+2.⁠ ⁠Assign an appropriate percentage based on importance and typical costs.
+3.⁠ ⁠Calculate the actual dollar amount for each category.
+
+For example, if the user provides a budget of ${amount:,.2f} for a '{event_type}', you might allocate it as follows:
+
+•⁠  ⁠Venue and setup: 10%, ${amount * 0.10:,.2f}
+•⁠  ⁠Equipment and supplies: 25%, ${amount * 0.25:,.2f}
+•⁠  ⁠Staff and coordination: 15%, ${amount * 0.15:,.2f}
+•⁠  ⁠Marketing and outreach: 15%, ${amount * 0.15:,.2f}
+•⁠  ⁠Administrative costs: 10%, ${amount * 0.10:,.2f}
+•⁠  ⁠Contingency fund: 25%, ${amount * 0.25:,.2f}
+
+Remember that categories should be specifically tailored to the exact event type provided, not using generic templates.
+Only output the answer, no other text.
+If you need any clarification about the event type or special requirements, ask the user.
+"""
+            return response.strip()  # Return the response without leading/trailing whitespace
+            
+        except ValueError:
+            return "Invalid input format. Please use 'amount:event_type' format (e.g. '15000:community event')"
+
+    async def _arun(self, budget_str: str) -> str:
+        """Use this tool asynchronously."""
+        raise NotImplementedError("Async not implemented")