92 lines
2.5 KiB
Python
92 lines
2.5 KiB
Python
import json
|
|
|
|
from config import OPENROUTER_MODEL
|
|
from openrouter_client import client
|
|
from mcp_client import MCPClient
|
|
|
|
|
|
class Agent:
|
|
def __init__(self):
|
|
self.mcp = MCPClient()
|
|
self.tools = []
|
|
self.messages = [
|
|
{
|
|
"role": "system",
|
|
"content": (
|
|
"Du bist JARVIS, ein Smart-Home Assistent.\n"
|
|
"Du steuerst ein Haus über MCP Tools.\n"
|
|
"Antworte normal oder nutze Tools wenn nötig."
|
|
)
|
|
}
|
|
]
|
|
async def load_tools(self):
|
|
res = await self.mcp.list_tools()
|
|
|
|
self.tools = []
|
|
|
|
for t in res.tools:
|
|
self.tools.append({
|
|
"type": "function",
|
|
"function": {
|
|
"name": t.name,
|
|
"description": t.description,
|
|
"parameters": t.inputSchema
|
|
}
|
|
})
|
|
|
|
print(f"[JARVIS] {len(self.tools)} Tools geladen")
|
|
async def _run_tool(self, tool_call):
|
|
name = tool_call.function.name
|
|
args = json.loads(tool_call.function.arguments or "{}")
|
|
|
|
result = await self.mcp.call_tool(name, args)
|
|
|
|
return result
|
|
async def run(self, user_input: str):
|
|
|
|
self.messages.append({
|
|
"role": "user",
|
|
"content": user_input
|
|
})
|
|
|
|
for _ in range(8):
|
|
|
|
response = await client.chat.completions.create(
|
|
model=OPENROUTER_MODEL,
|
|
messages=self.messages,
|
|
tools=self.tools
|
|
)
|
|
|
|
message = response.choices[0].message
|
|
|
|
# 🧠 TOOL CALL PATH
|
|
if getattr(message, "tool_calls", None):
|
|
|
|
self.messages.append(message)
|
|
|
|
for tool_call in message.tool_calls:
|
|
|
|
result = await self._run_tool(tool_call)
|
|
|
|
self.messages.append({
|
|
"role": "tool",
|
|
"tool_call_id": tool_call.id,
|
|
"content": str(result)
|
|
})
|
|
|
|
continue
|
|
|
|
# 🧠 NORMAL RESPONSE PATH
|
|
if message.content:
|
|
|
|
self.messages.append({
|
|
"role": "assistant",
|
|
"content": message.content
|
|
})
|
|
|
|
return message.content
|
|
|
|
# 🧠 SAFETY FALLBACK
|
|
return "Ich konnte keine gültige Antwort generieren."
|
|
|
|
return "Tool Loop Limit erreicht." |