main.py

import threading
from agents import AICompanionAgent, EntitiesExtractionAgent
from prompts import COMPANION_PROMPT_TEMPLATE
from tools import entities_extraction_tools

companion_agent = AICompanionAgent(COMPANION_PROMPT_TEMPLATE, verbose=False)

user_profile_updater = EntitiesExtractionAgent(
    tools=entities_extraction_tools)

print("\nYou've just bumped into your AI friend...")

while True:
    user_input = input("\nHuman: ")

    # Create a new thread to execute update_user_profile concurrently to avoid waiting for AI agent's response back to user
    profile_update_thread = threading.Thread(
        target=user_profile_updater.update_user_profile, args=(user_input,))
    profile_update_thread.start()

    # Perform companion agent talk in the main thread to avoid waiting for entity extraction
    print("\nAI:" + " " + companion_agent.talk(user_input))