minimal_chatbot.py

from rich.console import Console
import instructor
import openai

# `BaseAgent` is the core class for creating chat agents in the Atomic Agents framework.
from atomic_agents.agents.base_agent import BaseAgent, BaseAgentConfig

# Create an instance of `BaseAgent` with default settings.
# For all supported clients such as Anthropic & Gemini, have a look at the `instructor` library documentation.
agent = BaseAgent(
    config=BaseAgentConfig(
        client=instructor.from_openai(openai.OpenAI()),
        model='gpt-3.5-turbo'
    )
)

# Initialize a `Console` object for rich text output in the terminal.
console = Console()

# Main chat loop for testing the chat agent.
while True:
    user_input = input('You: ')
    if user_input.lower() in ['/exit', '/quit']:
        print('Exiting chat...')
        break

    response = agent.run(agent.input_schema(chat_message=user_input))
    console.print(f'Agent: {response.chat_message}')