Tutorial: Building a Conversational Chat App


  • Level: Intermediate
  • Time to complete: 10 minutes
  • Nodes Used: PromptNode, ConversationalAgent and ConversationSummaryMemory
  • Goal: After completing this tutorial, you will have learned how to use ConversationalAgent to build a conversational chat application
  • Prerequisites: A Hugging Face API Key

Overview

A ConversationalAgent is a type of agent that is specifically implemented to create chat applications. With its memory integration, the new ConversationalAgent enables human-like conversation with large language models (LLMs).

This tutorial introduces you to the ConversationalAgent, ConversationSummaryMemory and explains how you can create your conversational chat application.

Preparing the Colab Environment

Installing Haystack

To start, let’s install the latest release of Haystack with pip:

%%bash

pip install --upgrade pip
pip install farm-haystack[colab]

Enabling Telemetry

Knowing you’re using this tutorial helps us decide where to invest our efforts to build a better product but you can always opt out by commenting the following line. See Telemetry for more details.

from haystack.telemetry import tutorial_running

tutorial_running(24)

Initializing the ConversationalAgent

To initialize a ConversationalAgent, we’ll first need to create a PromptNode to define the LLM that our chat application will use. Then, we’ll add a memory to enable the application to store previous conversation and use this memory to make the interaction more human-like.

Let’s create necessary components for a ConversationalAgent:

1) Provide a Hugging Face API Key

Hugging Face offers hosted Inference API which we can use to access machine learning models using simple HTTP requests. This way, we don’t need to download models from the hub. To use service, we need to provide an API key from Hugging Face:

from getpass import getpass

model_api_key = getpass("Enter model provider API key:")

2) Create a PromptNode

We’ll initialize a PromptNode with the model_name, api_key, and max_length to control the output length of the model. In this tutorial, we’ll use OpenAssistant/oasst-sft-1-pythia-12b, an open source Transformer-based text generation model.

from haystack.nodes import PromptNode

model_name = "OpenAssistant/oasst-sft-1-pythia-12b"
prompt_node = PromptNode(model_name, api_key=model_api_key, max_length=256)

3) Create a ConversationSummaryMemory

To have a chat application closer to a human interaction, we need to provide memory to our ConversationalAgent. There are two types of memory options in Haystack:

  1. ConversationMemory: stores the conversation history (default).
  2. ConversationSummaryMemory: stores the conversation history and periodically generates summaries.

These memory nodes inject the conversation history into the prompt for the large language model with every run. Instead of using the full conversation history, we’ll use ConversationSummaryMemory that sums up the conversation without losing important information, thus saving the token limit.

We can use the same PromptNode in ConversationSummaryMemory, so the same OpenAssistant/oasst-sft-1-pythia-12b model generates chat summaries. By default, ConversationSummaryMemory summarizes the chat with every 3 runs using the predefined conversational-summary PromptTemplate.

from haystack.agents.memory import ConversationSummaryMemory

summary_memory = ConversationSummaryMemory(prompt_node)

Optionally, you can define a separate PromptNode with another LLM and PromptTemplate for generating conversation summary and use it in the ConversationSummaryMemory.

4) Create a ConversationalAgent

Now that we have all the necessary components, we can initialize the ConversationalAgent. As PromptTemplate, ConversationalAgent uses conversational-agent by default.

from haystack.agents.conversational import ConversationalAgent

conversational_agent = ConversationalAgent(prompt_node=prompt_node, memory=summary_memory)

Now, our conversational agent is ready to chat!

Trying Out a Prompt

conversational_agent.run("Tell me three most interesting things about Istanbul, Turkey")
conversational_agent.run("Can you elaborate on the second item?")
conversational_agent.run("Can you turn this info into a twitter thread?")
  • At any point during the chat, you can use load() function to check the chat summary:
print(conversational_agent.memory.load())
  • To delete the whole chat history, call clear() method:
conversational_agent.memory.clear()

Congratulations! 🎉 You’ve learned how to use ConversationalAgent to create a chat application with a summarized memory.

💬 Example Application

To take the chat experience to another level, check out this example application. Run the code cell below and use the textarea to interact with the conversational agent. Use the buttons on the right to load or delete the chat summary.

import ipywidgets as widgets
from IPython.display import clear_output

## Text Input
user_input = widgets.Textarea(
    value="",
    placeholder="Type your prompt here",
    disabled=False,
    style={"description_width": "initial"},
    layout=widgets.Layout(width="100%", height="100%"),
)

## Submit Button
submit_button = widgets.Button(
    description="Submit", button_style="success", layout=widgets.Layout(width="100%", height="80%")
)


def on_button_clicked(b):
    user_prompt = user_input.value
    user_input.value = ""
    print("\nUser:\n", user_prompt)
    conversational_agent.run(user_prompt)


submit_button.on_click(on_button_clicked)

## Show Memory Button
memory_button = widgets.Button(
    description="Show Memory", button_style="info", layout=widgets.Layout(width="100%", height="100%")
)


def on_memory_button_clicked(b):
    memory = conversational_agent.memory.load()
    if len(memory):
        print("\nMemory:\n", memory)
    else:
        print("Memory is empty")


memory_button.on_click(on_memory_button_clicked)

## Clear Memory Button
clear_button = widgets.Button(
    description="Clear Memory", button_style="warning", layout=widgets.Layout(width="100%", height="100%")
)


def on_clear_button_button_clicked(b):
    conversational_agent.memory.clear()
    print("\nMemory is cleared\n")


clear_button.on_click(on_clear_button_button_clicked)

## Layout
grid = widgets.GridspecLayout(3, 3, height="200px", width="800px", grid_gap="10px")
grid[0, 2] = clear_button
grid[0:2, 0:2] = user_input
grid[2, 0:] = submit_button
grid[1, 2] = memory_button
display(grid)