Add memory to APIChain

[DYSpider13]

Hello,

I'm using Langchain's APIChain to build a conversationnal agent that calls my web app API. I'm trying to add memory to the conversation, but does not seem to work. Any suggestions ? :

import os

from langchain.chains import APIChain
from langchain.chat_models import ChatOpenAI
from langchain.memory import ConversationBufferMemory

os.environ["OPENAI_API_KEY"] = "mykey"

llm = ChatOpenAI(temperature=0, model_name="gpt-3.5-turbo", max_tokens=256, verbose=True)
memory = ConversationBufferMemory(memory_key="chat_history")

apiSpec = """API documentation:...." 

chain_new = APIChain.from_llm_and_api_docs(llm, apiSpec, verbose=True, headers=headers, memory=memory)

user_input = input("\nAgent:  \u001B[3mHow can I help you ?\u001B[0m \n\n")

response = chain_new.run(user_input)
print(response)

cont = True
while cont:

    user_input = input("Agent:  \u001B[3mAnything else ?\u001B[0m \n\n")
    if user_input == "exit":
        break

    response = chain_new.run(user_input)

    print(response)

print(" --> End of Chat <-- ")