Feat/tool based coordinator routing (#5)

Co-authored-by: Michael <michael@example.com>
Reviewed-on: #5
This commit is contained in:
2026-06-29 21:17:42 +02:00
parent e38a80532b
commit 7f66d09d9e
10 changed files with 301 additions and 87 deletions
+11
View File
@@ -1,8 +1,11 @@
import asyncio
import logging
import lmstudio as lms
from typing import Any, Callable
from .prompt import CAVEMAN_PROMPT
logger: logging.Logger = logging.getLogger("agent-caveman")
class _ActResponseCapture:
"""Captures the AI response from LMStudio act() callback."""
@@ -60,6 +63,7 @@ class CavemanAgent:
async def initialize(self) -> None:
"""Initialize the LM Studio model."""
logger.info(f"Initializing CavemanAgent with model: {self.model_name}")
self.model = lms.llm(self.model_name)
async def run(self, user_input: str) -> str:
@@ -74,9 +78,12 @@ class CavemanAgent:
]
try:
logger.info(f"Running CavemanAgent interactively (input length: {len(user_input)})")
response = await self.model.respond(user_input, messages=messages)
logger.info(f"CavemanAgent responded successfully (response length: {len(response)})")
return response
except Exception as e:
logger.error(f"CavemanAgent execution error: {e}")
return f"Error in agent execution: {str(e)}"
async def run_with_tools(self, user_input: str, tools: list[Any]) -> str:
@@ -87,10 +94,14 @@ class CavemanAgent:
try:
capture = _ActResponseCapture()
logger.info(f"Calling LMStudio act() on CavemanAgent with {len(tools)} tools...")
result: lms.ActResult = self.model.act(user_input, tools=tools, on_message=capture)
logger.info(f"act() on CavemanAgent returned: {result}")
response: str = capture.full_response
if not response or response == "No response captured.":
logger.warning(f"Act completed with {result.rounds} rounds but no response was captured.")
return f"Act completed with {result.rounds} rounds but no response captured."
return response
except Exception as e:
logger.error(f"CavemanAgent tool execution error: {e}")
return f"Error in agent tool execution: {str(e)}"