Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: propagate use_assistant_message through send paths (#615)

Browse Source
This commit is contained in:
cthomas
2025-01-13 10:26:53 -08:00
committed by GitHub
parent 576d5a77e1
commit 10b3dd670f
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
letta/server/rest_api/routers/v1/agents.py
View File

@@ -535,6 +535,7 @@ async def send_message_streaming(
stream_steps=True,
stream_tokens=request.stream_tokens,
# Support for AssistantMessage
use_assistant_message=request.use_assistant_message,
assistant_message_tool_name=request.assistant_message_tool_name,
assistant_message_tool_kwarg=request.assistant_message_tool_kwarg,
)
@@ -547,6 +548,7 @@ async def process_message_background(
actor: User,
agent_id: str,
messages: list,
use_assistant_message: bool,
assistant_message_tool_name: str,
assistant_message_tool_kwarg: str,
) -> None:
@@ -560,6 +562,7 @@ async def process_message_background(
messages=messages,
stream_steps=False, # NOTE(matt)
stream_tokens=False,
use_assistant_message=use_assistant_message,
assistant_message_tool_name=assistant_message_tool_name,
assistant_message_tool_kwarg=assistant_message_tool_kwarg,
metadata={"job_id": job_id}, # Pass job_id through metadata
@@ -624,6 +627,7 @@ async def send_message_async(
actor=actor,
agent_id=agent_id,
messages=request.messages,
use_assistant_message=request.use_assistant_message,
assistant_message_tool_name=request.assistant_message_tool_name,
assistant_message_tool_kwarg=request.assistant_message_tool_kwarg,
)