feat: update agentstack-sdk (#1405)

Author: xjacka

python/beeai_framework/adapters/agentstack/agents/agent.py

@@ -152,9 +152,18 @@ async def _get_metadata(self, client: PlatformClient) -> dict[str, Any]:
 
         context_token = await self._agent_stack_context.generate_token(
             client=client,
-            grant_global_permissions=Permissions(llm={"*"}, embeddings={"*"}, a2a_proxy={"*"}, contexts={"read"}),
+            grant_global_permissions=Permissions(
+                llm={"*"},
+                embeddings={"*"},
+                a2a_proxy={"*"},
+                contexts={"read"},
+                model_providers={"read"},
+                providers={"read"},
+            ),
             grant_context_permissions=ContextPermissions(files={"*"}, vector_stores={"*"}, context_data={"*"}),
         )
+        client.headers["Authorization"] = f"Bearer {context_token.token.get_secret_value()}"
+
         llm_spec = LLMServiceExtensionSpec.from_agent_card(self._agent.agent_card)
         embedding_spec = EmbeddingServiceExtensionSpec.from_agent_card(self._agent.agent_card)
         agent_stack_extension_spec = PlatformApiExtensionSpec.from_agent_card(self._agent.agent_card)

python/beeai_framework/adapters/agentstack/serve/_dummy_context_store.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from collections.abc import AsyncIterator
+from uuid import UUID
 
 try:
     import a2a.types as a2a_types
@@ -23,6 +24,9 @@ async def load_history(self) -> AsyncIterator[a2a_types.Message | a2a_types.Arti
     async def store(self, data: a2a_types.Message | a2a_types.Artifact) -> None:
         pass
 
+    async def delete_history_from_id(self, from_id: UUID) -> None:
+        pass
+
 
 class DummyContextStore(agentstack_context_store.ContextStore):
     _cs = DummyContextStoreInstance()

python/beeai_framework/adapters/agentstack/serve/factories.py

@@ -1,6 +1,5 @@
 # Copyright 2025 © BeeAI a Series of LF Projects, LLC
 # SPDX-License-Identifier: Apache-2.0
-import uuid
 from collections.abc import AsyncGenerator
 from typing import Annotated, Any, Unpack, cast
 
@@ -87,17 +86,16 @@ async def run(
             embedding=extra_extensions.get("embedding"),
             extra_extensions=extra_extensions,  # type: ignore[arg-type]
         ) as stack_context:
-            artifact_id = uuid.uuid4()
-            append = False
+            stream = False
 
             @cloned_agent.emitter.on("partial_update")
             async def on_partial_update(data: ReActAgentUpdateEvent, _: EventMeta) -> None:
-                nonlocal append
+                nonlocal stream
                 if data.update.key == "final_answer":
                     update = data.update.value
                     update = update.get_text_content() if hasattr(update, "get_text_content") else str(update)
-                    await _send_final_answer_update(context, artifact_id, update, append=append)
-                    append = True
+                    await context.yield_async(update)
+                    stream = True
 
             @cloned_agent.emitter.on("update")
             async def on_update(data: ReActAgentUpdateEvent, _: EventMeta) -> None:
@@ -117,9 +115,7 @@ async def on_update(data: ReActAgentUpdateEvent, _: EventMeta) -> None:
                 await context.store(message)
                 await context.store(agent_response)
 
-            if append:
-                await _send_final_answer_update(context, artifact_id, last_chunk=True)
-            else:
+            if not stream:
                 yield agent_response
 
     return agentstack_agent.agent(**agent_metadata)(run)
@@ -207,14 +203,13 @@ async def run(
             embedding=extra_extensions.get("embedding"),
             extra_extensions=extra_extensions,  # type: ignore[arg-type]
         ) as stack_context:
-            artifact_id = uuid.uuid4()
-            append = False
+            stream = False
 
             @cloned_agent.emitter.on("final_answer")
             async def on_final_answer(data: RequirementAgentFinalAnswerEvent, _: EventMeta) -> None:
-                nonlocal append
-                await _send_final_answer_update(context, artifact_id, data.delta, append=append)
-                append = True
+                nonlocal stream
+                await context.yield_async(data.delta)
+                stream = True
 
             result = await cloned_agent.run([convert_a2a_to_framework_message(message)]).middleware(
                 create_tool_trajectory_middleware(stack_context)
@@ -225,37 +220,12 @@ async def on_final_answer(data: RequirementAgentFinalAnswerEvent, _: EventMeta)
                 await context.store(message)
                 await context.store(agent_response)
 
-            if append:
-                await _send_final_answer_update(context, artifact_id, last_chunk=True)
-            else:
+            if not stream:
                 yield agent_response
 
     return agentstack_agent.agent(**agent_metadata)(run)
 
 
-async def _send_final_answer_update(
-    context: agentstack_context.RunContext,
-    artifact_id: uuid.UUID,
-    update: str = "",
-    *,
-    append: bool = True,
-    last_chunk: bool = False,
-) -> None:
-    await context.yield_async(
-        a2a_types.TaskArtifactUpdateEvent(
-            append=append,
-            context_id=context.context_id,
-            task_id=context.task_id,
-            last_chunk=last_chunk,
-            artifact=a2a_types.Artifact(
-                name="final_answer",
-                artifact_id=str(artifact_id),
-                parts=[a2a_types.Part(root=a2a_types.TextPart(text=update))],
-            ),
-        )
-    )
-
-
 def _runnable_factory(
     runnable: Runnable[Any], *, metadata: AgentStackServerMetadata | None = None, memory_manager: MemoryManager
 ) -> agentstack_agent.AgentFactory: