prompting, simpler models

app.py

@@ -29,7 +29,7 @@ system_prompt = """You are working with a pandas dataframe in Python. The name o
 
                 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. 
-                Do not ask if the user wants additional information or analysis."""
+                Do not mention next steps. Do not ask if the user has any follow up questions. 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
@@ -72,7 +72,7 @@ def chain_node(state: AgentState, chain, name):
 # cell 3
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 llm = ChatOpenAI(model="gpt-4o-mini-2024-07-18", temperature=0, api_key=OPENAI_API_KEY)
-llm_big = ChatOpenAI(model="gpt-4o", temperature=0, api_key=OPENAI_API_KEY)
+llm_big = ChatOpenAI(model="gpt-4o-mini-2024-07-18", temperature=0, api_key=OPENAI_API_KEY)
 
 eda_task = """Using the data in the dataframe `df` and the package statsmodels, first run an augmented dickey fuller test on the data.
             Using matplotlib plot the time series, display it and save it to 'plot.png'.