phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-13 21:29:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix duplicate tool msg

Browse source
This commit is contained in:
Xi Yan 2025-02-20 22:16:37 -08:00
parent 0de38a2b48
commit 99bc54b033
1 changed files with 3 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

13
llama_stack/providers/inline/agents/meta_reference/agent_instance.py
View file

@ -269,6 +269,9 @@ class ChatAgent(ShieldRunnerMixin):
messages.extend(self.turn_to_messages(turn)) messages.extend(self.turn_to_messages(turn))
messages.extend(request.tool_responses) messages.extend(request.tool_responses)
last_turn_messages = [
x for x in messages if isinstance(x, UserMessage) or isinstance(x, ToolResponseMessage)
]
# get the steps from the turn id # get the steps from the turn id
steps = [] steps = []
@ -326,19 +329,9 @@ class ChatAgent(ShieldRunnerMixin):
assert output_message is not None assert output_message is not None
last_turn_messages = []
last_turn_start_time = datetime.now() last_turn_start_time = datetime.now()
if len(turns) > 0: if len(turns) > 0:
last_turn_start_time = turns[-1].started_at last_turn_start_time = turns[-1].started_at
last_turn_messages = self.turn_to_messages(turns[-1])
# add tool responses to the last turn messages
last_turn_messages.extend(request.tool_responses)
# filter out non User / Tool messages
# TODO: should we just keep all message types in Turn.input_messages?
last_turn_messages = [
m for m in last_turn_messages if isinstance(m, UserMessage) or isinstance(m, ToolResponseMessage)
]
turn = Turn( turn = Turn(
turn_id=request.turn_id, turn_id=request.turn_id,