from agent import Agent
from identity import load_agent_identity, load_user_profile
from llm import LLM
from tools import (
    CalculatorTool,
    FileSystemListTool,
    FileSystemReadTool,
    HelloTool,
    ToolRegistry,
    WeatherTool,
)

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

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

tools = ToolRegistry()
tools.register(HelloTool())
tools.register(CalculatorTool())
tools.register(WeatherTool())
tools.register(FileSystemListTool())
tools.register(FileSystemReadTool())

agent = Agent(
    llm,
    agent_identity,
    user_profile,
    tools,
)

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

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

    answer = agent.chat(user_input)
    print(f"Agent: {answer}")