diff options
Diffstat (limited to 'client.py')
| -rw-r--r-- | client.py | 14 |
1 files changed, 12 insertions, 2 deletions
@@ -1,7 +1,14 @@ from agent import Agent from identity import load_agent_identity, load_user_profile from llm import LLM -from tools import CalculatorTool, HelloTool, ToolRegistry +from tools import ( + CalculatorTool, + FileSystemListTool, + FileSystemReadTool, + HelloTool, + ToolRegistry, + WeatherTool, +) llm = LLM( "http://localhost:8080/v1/chat/completions", @@ -13,6 +20,9 @@ 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, @@ -26,6 +36,6 @@ while True: if user_input.lower() in {"exit", "quit"}: break - + answer = agent.chat(user_input) print(f"Agent: {answer}") |
