added together ai agent

controller.py

@@ -306,7 +306,7 @@ def langchain_csv_chat(csv_url: str, question: str, chart_required: bool):
                 return_intermediate_steps=True
             )
 
-            prompt = _prompt_generator(question, chart_required)
+            prompt = _prompt_generator(question, chart_required, csv_url)
             result = agent.invoke({"input": prompt})
             return result.get("output")
 
@@ -537,7 +537,7 @@ def langchain_csv_chart(csv_url: str, question: str, chart_required: bool):
                 return_intermediate_steps=True
             )
 
-            result = agent.invoke({"input": _prompt_generator(question, True)})
+            result = agent.invoke({"input": _prompt_generator(question, True, csv_url)})
             output = result.get("output", "")
 
             # Close figures to avoid interference

lc_groq_chart.py

@@ -64,7 +64,7 @@ def langchain_csv_chart(csv_url: str, question: str, chart_required: bool):
                 return_intermediate_steps=True
             )
 
-            result = agent.invoke({"input": _prompt_generator(f"{question} and use this csv_url: {csv_url} to read the csv file", True)})
+            result = agent.invoke({"input": _prompt_generator(question, True, csv_url)})
             output = result.get("output", "")
 
             # Verify chart file creation

lc_groq_chat.py

@@ -63,7 +63,7 @@ def langchain_csv_chat(csv_url: str, question: str, chart_required: bool):
                 return_intermediate_steps=True
             )
 
-            prompt = _prompt_generator(question, chart_required)
+            prompt = _prompt_generator(question, chart_required, csv_url)
             result = agent.invoke({"input": prompt})
             return result.get("output")
 

orchestrator_functions.py

@@ -184,7 +184,7 @@ def langchain_csv_chat(csv_url: str, question: str, chart_required: bool):
                 return_intermediate_steps=True
             )
 
-            prompt = _prompt_generator(question, chart_required)
+            prompt = _prompt_generator(question, chart_required, csv_url)
             result = agent.invoke({"input": prompt})
             return result.get("output")
 
@@ -324,7 +324,7 @@ def langchain_csv_chart(csv_url: str, question: str, chart_required: bool):
                 return_intermediate_steps=True
             )
 
-            result = agent.invoke({"input": _prompt_generator(f"{question} and use this csv_url: {csv_url} to read the csv file", True)})
+            result = agent.invoke({"input": _prompt_generator(question, True, csv_url)})
             output = result.get("output", "")
 
             # Verify chart file creation

util_service.py

@@ -22,7 +22,7 @@ def process_answer(answer):
     return False
 
 
-def _prompt_generator(question: str, chart_required: bool):
+def _prompt_generator(question: str, chart_required: bool, csv_url: str):
 
     chat_prompt = f"""You are a senior data analyst working with CSV data. Adhere strictly to the following guidelines:  
 
@@ -30,6 +30,7 @@ def _prompt_generator(question: str, chart_required: bool):
                          2. **Data Integrity:** Ensure proper handling of null values to maintain accuracy and reliability.  
                          3. **Communication:** Provide concise, professional, and well-structured responses.
                          4. Avoid including any internal processing details or references to the methods used to generate your response (ex: based on the tool call, using the function -> These types of phrases.)
+                         5. Always use pd.read_csv({csv_url}) to read the CSV file.
 
                          **Query:** {question}  
          
@@ -65,6 +66,7 @@ def _prompt_generator(question: str, chart_required: bool):
                        - Use THE SAME unique_id throughout entire process
                        - NEVER generate new UUIDs after initial creation
                        - Return EXACT filepath string of the final saved chart
+                       - Always use pd.read_csv({csv_url}) to read the CSV file
                        """