updating prompts

app.py

@@ -28,7 +28,8 @@ system_prompt = """You are working with a pandas dataframe in Python. The name o
                 </df>
 
                 You are not meant to use only these rows to answer questions - they are meant as a way of telling you about the shape and schema of the dataframe. 
-                You also do not have use only the information here to answer questions - you can run intermediate queries to do exporatory data analysis to give you more information as needed. """
+                You also do not have use only the information here to answer questions - you can run intermediate queries to do exporatory data analysis to give you more information as needed. 
+                Do not ask if the user wants additional information or analysis."""
 system_prompt = system_prompt.format(dhead=df.head().to_markdown())
 
 # The agent state is the input to each node in the graph
@@ -59,7 +60,7 @@ def create_agent(llm: ChatOpenAI, tools: list, task: str):
 # AIMessage will have all kinds of metadata, so treat it all as HumanMessage I suppose?
 def agent_node(state: AgentState, agent, name):
     result = agent.invoke(state)
-    return {"messages": [HumanMessage(content=result["output"], name=name)]}
+    return {"messages": [HumanMessage(content=name + ": " + result["output"], name=name)]}
     # return {"messages": [result]}
 
 # I need to write the message to state here? or is that handled automatically?