from agent import Agent
from identity import load_agent_identity, load_user_profile
from llm import LLM

llm = LLM("http://localhost:8080/v1/chat/completions",)

agent_identity = load_agent_identity("agent.toml")
user_profile = load_user_profile("user.toml")

agent = Agent(
    llm,
    agent_identity,
    user_profile,
)

while True:
    user_input = input("You: ")

    if user_input.lower() in {"exit", "quit"}:
        break

    print("Agent: ", end="", flush=True)

    for chunk in agent.chat(user_input):
        print(chunk, end="", flush=True)

    print()