Report generator

orchestrator_agent.py

@@ -57,36 +57,15 @@ def create_agent(csv_url: str, api_key: str) -> Agent:
     csv_metadata = get_csv_basic_info(csv_url)
     
     system_prompt = (
-        "You are a data analyst assistant with limited tool capabilities. "
-        "Available tools can only handle simple data queries: "
-        "- Count rows/columns\n- Calculate basic stats (avg, sum, min/max)\n"
-        "- Create simple visualizations (pie charts, bar graphs)\n"
-        "- Show column names/types\n\n"
-        
-        "Query Handling Rules:\n"
-        "1. If query is complex, ambiguous, or exceeds tool capabilities:\n"
-        "   - Break into simpler sub-questions\n"
-        "   - Ask for clarification\n"
-        "   - Rephrase to nearest simple query\n"
-        "2. For 'full report' requests:\n"
-        "   - Outline possible analysis steps\n"
-        "   - Ask user to select one component at a time\n\n"
-        
-        "Examples:\n"
-        "- Bad query: 'Show me everything'\n"
-        "  Response: 'I can show row count (10), columns (5: Name, Age...), "
-        "or a pie chart of categories. Which would you like?'\n"
-        "- Bad query: 'Analyze trends'\n"
-        "  Response: 'For trend analysis, I can show monthly averages or "
-        "year-over-year comparisons. Please specify time period and metric.'\n\n"
-        
-        "Current CSV Context:\n"
-        f"- URL: {csv_url}\n"
-        f"- Metadata: {csv_metadata}\n\n"
-        
-        "Always format images as: ![Chart Description](direct-image-url)"
+        "You are a data analyst."
+        "You have all the tools you need to answer any question."
+        "If user asking for multiple answers or charts then break the question into multiple proper questions."
+        "Pass csv_url/path with the questions to the tools to generate the answer."
+        "Explain the answer in a friendly way."
+        "**Format images** in Markdown: `![alt_text](direct-image-url)`"
+        f"Your csv url is {csv_url}"
+        f"Your csv metadata is {csv_metadata}"
     )
-    
     return Agent(
         model=initialize_model(api_key),
         deps_type=str,
@@ -94,7 +73,6 @@ def create_agent(csv_url: str, api_key: str) -> Agent:
         system_prompt=system_prompt
     )
 
-
 def csv_orchestrator_chat(csv_url: str, user_question: str) -> str:
     print("CSV URL:", csv_url)
     print("User questions:", user_question)