Phase 3 — WS frame + REST fallbacka
This commit is contained in:
@@ -489,6 +489,13 @@ def _memory_tools(user_id: str, trace_id: str | None) -> list[Any]:
|
||||
]
|
||||
|
||||
|
||||
def _read_only_memory_tools(user_id: str, trace_id: str | None) -> list[Any]:
|
||||
"""Return memory tools that only read — safe for the read-only brief-agent subset."""
|
||||
all_mem = _memory_tools(user_id, trace_id)
|
||||
_read_names = {"memory_list_blocks", "memory_get", "archival_memory_search", "conversation_search"}
|
||||
return [t for t in all_mem if t.name in _read_names]
|
||||
|
||||
|
||||
def _all_tools_for_user(user_id: str, trace_id: str | None) -> list[Any]:
|
||||
return [*_all_tools(), *_memory_tools(user_id, trace_id)]
|
||||
|
||||
@@ -792,12 +799,14 @@ async def _run_single_agent_stream(
|
||||
max_steps: int = 6,
|
||||
langfuse_prompt: Any = None,
|
||||
agent_name: str = "agent",
|
||||
tools: list[Any] | None = None,
|
||||
) -> AsyncGenerator[tuple[str, Any], None]:
|
||||
trace_id = _trace_id_from_context(context)
|
||||
session_id = _session_id_from_context(context)
|
||||
lf = get_langfuse()
|
||||
llm = get_agent_llm(agent_name)
|
||||
tools = _all_tools_for_user(user_id, trace_id)
|
||||
if tools is None:
|
||||
tools = _all_tools_for_user(user_id, trace_id)
|
||||
model_context = _context_for_model(context)
|
||||
logger.info("deep_agent: run_single_agent_stream_start trace=%s user=%s", trace_id or "-", user_id)
|
||||
llm_with_tools = llm.bind_tools(tools)
|
||||
|
||||
Reference in New Issue
Block a user