From 92e4ec2a57ef6f52d3316906131e756e7399ded9 Mon Sep 17 00:00:00 2001 From: Zhonglin Han Date: Thu, 19 Dec 2024 09:18:09 -0800 Subject: [PATCH 1/2] add response com agent_instance.py --- .../providers/inline/agents/meta_reference/agent_instance.py | 1 + 1 file changed, 1 insertion(+) diff --git a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py index d7930550d7..2f2ca9580c 100644 --- a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py +++ b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py @@ -430,6 +430,7 @@ async def _run( input_messages, tools=self._get_tools(), tool_prompt_format=self.agent_config.tool_prompt_format, + response_format=self.agent_config.response_format, stream=True, sampling_params=sampling_params, ): From 506d9289f62173fdd1fa78da722b36fc0c172369 Mon Sep 17 00:00:00 2001 From: Zhonglin Han Date: Thu, 19 Dec 2024 09:19:07 -0800 Subject: [PATCH 2/2] add response_format in AgentConfig --- llama_stack/apis/agents/agents.py | 1 + 1 file changed, 1 insertion(+) diff --git a/llama_stack/apis/agents/agents.py b/llama_stack/apis/agents/agents.py index 5fd90ae7ad..b0fcbf56ee 100644 --- a/llama_stack/apis/agents/agents.py +++ b/llama_stack/apis/agents/agents.py @@ -297,6 +297,7 @@ class AgentConfig(AgentConfigCommon): model: str instructions: str enable_session_persistence: bool + response_format: Optional[ResponseFormat] = None class AgentConfigOverridablePerTurn(AgentConfigCommon):