Using Agent

[pgvetrivel]

Hi Team

I am trying to code using agent to create an chatbot which can remember earlier questions (Conversational chatbot). But the agent is not remembering my earlier questions. Can you help me out?

import os
from dotenv import load_dotenv
import pandas as pd
from pandasai import Agent
from pandasai import SmartDatalake
import streamlit as st

st.header("🤖 MIS ChatBot", divider = 'rainbow')

sales_by_country = pd.DataFrame({
"country": ["United States", "United Kingdom", "France", "Germany", "Italy", "Spain", "Canada", "Australia", "Japan", "China"],
"sales": [5000, 3200, 2900, 4100, 2300, 2100, 2500, 2600, 4500, 7000],
"deals_opened": [142, 80, 70, 90, 60, 50, 40, 30, 110, 120],
"deals_closed": [120, 70, 60, 80, 50, 40, 30, 20, 100, 110]
})

load_dotenv()
pandaskey = os.getenv('PANDASAI_API_KEY')

if "chat_history" not in st.session_state:
st.session_state.chat_history = []

for message in st.session_state.chat_history:
with st.chat_message(message["role"]):
st.markdown(message["content"])

user_prompt = st.chat_input("Ask MIS Agent...")

if user_prompt:
st.chat_message("user").markdown(user_prompt)
st.session_state.chat_history.append({"role":"user","content": user_prompt})
agent = Agent(sales_by_country)
assistant_response = agent.chat(user_prompt)
st.session_state.chat_history.append({"role":"assistant", "content": assistant_response})
with st.chat_message("assistant"):
st.markdown(assistant_response)

[dosubot[bot]]

To make the Agent remember earlier questions in your chatbot, you need to set the memory_size parameter when initializing the Agent. This allows the agent to retain conversation history. Here's how you can modify your code:

1. Set the memory_size Parameter:

   When creating the Agent instance, specify the memory_size to a value greater than 1. This will enable the agent to remember previous interactions.

   agent = Agent(sales_by_country, memory_size=10)

2. Ensure Memory is Utilized:

   The memory_size parameter controls how many past interactions the agent can remember. Adjust this value based on how much history you want the agent to retain.

By making these changes, your chatbot should be able to remember earlier questions and provide more contextually aware responses ^[1].

To continue talking to Dosu, mention @dosu.