apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-05-31 19:15:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add RAG tool logging and update to use chunks toggle, ragK value

Browse source
This commit is contained in:
akhisud3195 2025-05-09 10:26:16 +05:30
parent 770a080232
commit 889b7ee166
2 changed files with 16 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -153,7 +153,7 @@ def get_rag_tool(config: dict, complete_request: dict) -> FunctionTool:
"""
project_id = complete_request.get("projectId", "")
if config.get("ragDataSources", None):
print("rag_search")
print(f"Creating rag_search tool with params:\n-Data Sources: {config.get('ragDataSources', [])}\n-Return Type: {config.get('ragReturnType', 'chunks')}\n-K: {config.get('ragK', 3)}")
params = {
"type": "object",
"properties": {
@ -168,10 +168,10 @@ def get_rag_tool(config: dict, complete_request: dict) -> FunctionTool:
]
}
tool = FunctionTool(
name="rag_search",
description="Get information about an article",
params_json_schema=params,
on_invoke_tool=lambda ctx, args: call_rag_tool(project_id, json.loads(args)['query'], config.get("ragDataSources", []), "chunks", 3)
name="rag_search",
description="Get information about an article",
params_json_schema=params,
on_invoke_tool=lambda ctx, args: call_rag_tool(project_id, json.loads(args)['query'], config.get("ragDataSources", []), config.get("ragReturnType", "chunks"), config.get("ragK", 3))
)
return tool
else: