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

Merge branch 'agents-unify-tools' into agents-unify-tools-2

Browse source
This commit is contained in:
Xi Yan 2025-02-20 15:29:11 -08:00
commit 4dbe3fd9e6
4 changed files with 8 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

11
docs/_static/llama-stack-spec.html vendored
View file

@ -3954,23 +3954,16 @@
"ContinueAgentTurnRequest": { "ContinueAgentTurnRequest": {
"type": "object", "type": "object",
"properties": { "properties": {
"new_messages": { "tool_responses": {
"type": "array", "type": "array",
"items": { "items": {
"oneOf": [
{
"$ref": "#/components/schemas/UserMessage"
},
{
"$ref": "#/components/schemas/ToolResponseMessage" "$ref": "#/components/schemas/ToolResponseMessage"
} }
]
}
} }
}, },
"additionalProperties": false, "additionalProperties": false,
"required": [ "required": [
"new_messages" "tool_responses"
], ],
"title": "ContinueAgentTurnRequest" "title": "ContinueAgentTurnRequest"
}, },

8
docs/_static/llama-stack-spec.yaml vendored
View file

@ -2573,15 +2573,13 @@ components:
ContinueAgentTurnRequest: ContinueAgentTurnRequest:
type: object type: object
properties: properties:
new_messages: tool_responses:
type: array type: array
items: items:
oneOf: $ref: '#/components/schemas/ToolResponseMessage'
- $ref: '#/components/schemas/UserMessage'
- $ref: '#/components/schemas/ToolResponseMessage'
additionalProperties: false additionalProperties: false
required: required:
- new_messages - tool_responses
title: ContinueAgentTurnRequest title: ContinueAgentTurnRequest
InferenceStep: InferenceStep:
type: object type: object

7
llama_stack/apis/agents/agents.py
View file

@ -354,12 +354,7 @@ class Agents(Protocol):
agent_id: str, agent_id: str,
session_id: str, session_id: str,
turn_id: str, turn_id: str,
new_messages: List[ tool_responses: List[ToolResponseMessage],
Union[
UserMessage,
ToolResponseMessage,
]
],
) -> Union[Turn, AsyncIterator[AgentTurnResponseStreamChunk]]: ... ) -> Union[Turn, AsyncIterator[AgentTurnResponseStreamChunk]]: ...
@webmethod( @webmethod(

7
llama_stack/providers/inline/agents/meta_reference/agents.py
View file

@ -174,12 +174,7 @@ class MetaReferenceAgentsImpl(Agents):
agent_id: str, agent_id: str,
session_id: str, session_id: str,
turn_id: str, turn_id: str,
new_messages: List[ tool_responses: List[ToolResponseMessage],
Union[
UserMessage,
ToolResponseMessage,
]
],
) -> AsyncGenerator: ) -> AsyncGenerator:
pass pass