From e882831afbfcd961457774a70b70e7913c50f7c4 Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Fri, 6 Feb 2026 10:47:07 +0100 Subject: [PATCH] Fix execute_code prompt --- haiku_rag_slim/haiku/rag/agents/rlm/agent.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/haiku_rag_slim/haiku/rag/agents/rlm/agent.py b/haiku_rag_slim/haiku/rag/agents/rlm/agent.py index 1758e1da..37fb065e 100644 --- a/haiku_rag_slim/haiku/rag/agents/rlm/agent.py +++ b/haiku_rag_slim/haiku/rag/agents/rlm/agent.py @@ -32,12 +32,11 @@ def create_rlm_agent(config: AppConfig) -> Agent[RLMDeps, RLMResult]: @agent.tool async def execute_code(ctx: RunContext[RLMDeps], code: str) -> CodeExecution: - """Execute Python code in the sandboxed environment. + """Execute Python code in a Docker-sandboxed environment. The code has access to haiku.rag functions (search, list_documents, - get_document, get_docling_document, ask) and safe standard library - modules (json, re, collections, math, statistics, itertools, - functools, datetime, typing). + get_document, get_docling_document, llm) and any Python standard + library module. Use print() to output results.