apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-07-06 20:42:15 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Clean up tools.py and interactive.py

Browse source
This commit is contained in:
akhisud3195 2025-05-09 10:17:36 +05:30
parent 968dfacd65
commit 770a080232
3 changed files with 135 additions and 170 deletions
Download patch file Download diff file Expand all files Collapse all files

1
apps/rowboat_agents/src/graph/execute_turn.py
View file

@ -235,7 +235,6 @@ def get_agents(agent_configs, tool_configs, complete_request):
)
if tool:
new_tools.append(tool)
logger.debug(f"Added tool {tool_name} to agent {agent_config['name']}")
print(f"Added tool {tool_name} to agent {agent_config['name']}")
else:
print(f"WARNING: Tool {tool_name} not found in tool_configs")

39
apps/rowboat_agents/src/graph/tools.py
View file

@ -1,43 +1,8 @@
import json
import random
from src.utils.common import common_logger
logger = common_logger
RAG_TOOL = {
"name": "getArticleInfo",
"type": "rag",
"description": "Fetch articles with knowledge relevant to the query",
"parameters": {
"type": "object",
"properties": {
"question": {
"type": "string",
"description": "The query to retrieve articles for"
}
},
"required": [
"query"
]
}
}
CLOSE_CHAT_TOOL = {
"name": "close_chat",
"type": "close_chat",
"description": "Close the chat",
"parameters": {
"type": "object",
"properties": {
"error_message": {
"type": "string", "description": "The error message to close the chat with"
}
}
}
}
def tool_raise_error(error_message):
logger.error(f"Raising error: {error_message}")
print(f"Raising error: {error_message}")
raise ValueError(f"Raising error: {error_message}")
def respond_to_tool_raise_error(tool_calls, mock=False):
@ -45,7 +10,7 @@ def respond_to_tool_raise_error(tool_calls, mock=False):
return _create_tool_response(tool_calls, tool_raise_error(error_message))
def tool_close_chat(error_message):
logger.error(f"Closing chat: {error_message}")
print(f"Closing chat: {error_message}")
raise ValueError(f"Closing chat: {error_message}")
def respond_to_tool_close_chat(tool_calls, mock=False):