agent.py aktualisiert
This commit is contained in:
@@ -5,9 +5,9 @@ from util.logger import get_logger
|
|||||||
import asyncio
|
import asyncio
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
# Import des neuen Web-Search Tools
|
# Import des neuen Web-Search Tools und der IR Sender Api
|
||||||
from tools.web_search import web_search_tool, web_search
|
from tools.web_search import web_search_tool, web_search
|
||||||
|
from script_api_tool import script_api_tool
|
||||||
|
|
||||||
class Agent:
|
class Agent:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
@@ -61,6 +61,8 @@ class Agent:
|
|||||||
else:
|
else:
|
||||||
bedrock_schema = schema
|
bedrock_schema = schema
|
||||||
|
|
||||||
|
self.tools.append(script_api_tool)
|
||||||
|
|
||||||
self.tools.append({
|
self.tools.append({
|
||||||
"toolSpec": {
|
"toolSpec": {
|
||||||
"name": t.name,
|
"name": t.name,
|
||||||
@@ -69,6 +71,7 @@ class Agent:
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|
||||||
self.logger.info(f"{len(self.tools)} OpenHAB-Tools erfolgreich geladen")
|
self.logger.info(f"{len(self.tools)} OpenHAB-Tools erfolgreich geladen")
|
||||||
|
|
||||||
# 2. Web-Search Tool manuell hinzufügen
|
# 2. Web-Search Tool manuell hinzufügen
|
||||||
@@ -96,6 +99,12 @@ class Agent:
|
|||||||
# Web-Search Tool
|
# Web-Search Tool
|
||||||
if name == "web_search":
|
if name == "web_search":
|
||||||
return await web_search(**args)
|
return await web_search(**args)
|
||||||
|
if tool_name == "run_script":
|
||||||
|
|
||||||
|
result = await run_script(
|
||||||
|
arguments["command"],
|
||||||
|
arguments.get("args")
|
||||||
|
)
|
||||||
|
|
||||||
# Normale MCP Tools (OpenHAB)
|
# Normale MCP Tools (OpenHAB)
|
||||||
result = await self.mcp.call_tool(name, args)
|
result = await self.mcp.call_tool(name, args)
|
||||||
|
|||||||
Reference in New Issue
Block a user