feat(server,sdk): add context token as authentication method to agents

Signed-off-by: Radek Ježek <[email protected]>

Author: jezekra1

agents/chat/.vscode/launch.json

@@ -7,6 +7,7 @@
         {
             "name": "agent-chat",
             "type": "debugpy",
+            "justMyCode": false,
             "request": "launch",
             "program": "${workspaceFolder}/src/chat/agent.py",
             "console": "integratedTerminal"

agents/chat/Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.13-alpine3.22
+FROM python:3.13-slim
 ARG RELEASE_VERSION="main"
 COPY ./agents/chat/ /app/agents/chat
 COPY ./apps/agentstack-sdk-py/ /app/apps/agentstack-sdk-py/

agents/chat/pyproject.toml

@@ -7,7 +7,7 @@ authors = [
 ]
 requires-python = ">=3.13,<3.14"
 dependencies = [
-    "beeai-framework[duckduckgo,wikipedia,a2a]>=0.1.70",
+    "beeai-framework[duckduckgo,wikipedia,a2a]>=0.1.74",
     "tiktoken>=0.12.0",  # constraint for beeai-framework dependency (first version with musl arm64 wheel)
     "fastuuid>=0.14.0",  # constraint for beeai-framework dependency (first version with musl arm64 wheel)
     "openai>=1.107.1",

agents/chat/src/chat/agent.py

@@ -2,71 +2,58 @@
 # SPDX-License-Identifier: Apache-2.0
 import logging
 import os
-from typing import Annotated
 from textwrap import dedent
+from typing import Annotated
 
 from a2a.types import (
     AgentSkill,
     Message,
 )
-from beeai_framework.agents.requirement.utils._tool import FinalAnswerTool
-from beeai_framework.errors import FrameworkError
-from pydantic import BaseModel
-
 from agentstack_sdk.a2a.extensions import (
     AgentDetail,
     AgentDetailContributor,
     AgentDetailTool,
-    BaseExtensionServer,
-    BaseExtensionSpec,
     CitationExtensionServer,
     CitationExtensionSpec,
     ErrorExtensionParams,
     ErrorExtensionServer,
     ErrorExtensionSpec,
-    TrajectoryExtensionServer,
-    TrajectoryExtensionSpec,
     LLMServiceExtensionServer,
     LLMServiceExtensionSpec,
+    TrajectoryExtensionServer,
+    TrajectoryExtensionSpec,
 )
-
-# Monkey-patch to remove FormExtensionSpec which no longer exists
-# TODO: remove after next release
-import agentstack_sdk.a2a.extensions as agentstack_extensions
-from chat.tools.files.file_reader import FileReaderTool
-
-agentstack_extensions.FormExtensionSpec = BaseExtensionSpec
-agentstack_extensions.FormExtensionServer = BaseExtensionServer
-agentstack_extensions.TextField = BaseModel
-
+from agentstack_sdk.a2a.extensions.services.platform import (
+    PlatformApiExtensionServer,
+    PlatformApiExtensionSpec,
+)
+from agentstack_sdk.a2a.types import AgentArtifact, AgentMessage
+from agentstack_sdk.server import Server
+from agentstack_sdk.server.context import RunContext
+from agentstack_sdk.server.middleware.platform_auth_backend import PlatformAuthBackend
+from agentstack_sdk.server.store.platform_context_store import PlatformContextStore
 from beeai_framework.adapters.agentstack.backend.chat import AgentStackChatModel
 from beeai_framework.agents.requirement import RequirementAgent
 from beeai_framework.agents.requirement.events import (
-    RequirementAgentSuccessEvent,
     RequirementAgentFinalAnswerEvent,
+    RequirementAgentSuccessEvent,
 )
+from beeai_framework.agents.requirement.utils._tool import FinalAnswerTool
+from beeai_framework.backend import AssistantMessage, ChatModelParameters
+from beeai_framework.errors import FrameworkError
 from beeai_framework.middleware.trajectory import GlobalTrajectoryMiddleware
-from beeai_framework.tools import Tool, AnyTool
+from beeai_framework.tools import AnyTool, Tool
 from beeai_framework.tools.search.duckduckgo import DuckDuckGoSearchTool
 from beeai_framework.tools.search.wikipedia import WikipediaTool
 from beeai_framework.tools.weather import OpenMeteoTool
-from beeai_framework.backend import ChatModelParameters, AssistantMessage
-from agentstack_sdk.a2a.extensions.services.platform import (
-    PlatformApiExtensionServer,
-    PlatformApiExtensionSpec,
-)
-from agentstack_sdk.a2a.types import AgentMessage, AgentArtifact
-from agentstack_sdk.server import Server
-from agentstack_sdk.server.context import RunContext
 from openinference.instrumentation.beeai import BeeAIInstrumentor
 
 from chat.helpers.citations import extract_citations
 from chat.helpers.trajectory import TrajectoryContent
 from chat.tools.files.file_creator import FileCreatorTool, FileCreatorToolOutput
+from chat.tools.files.file_reader import FileReaderTool
 from chat.tools.files.utils import extract_files, to_framework_message
 
-from agentstack_sdk.server.store.platform_context_store import PlatformContextStore
-
 BeeAIInstrumentor().instrument()
 
 logger = logging.getLogger(__name__)
@@ -300,6 +287,7 @@ def serve():
             port=int(os.getenv("PORT", 8000)),
             configure_telemetry=True,
             context_store=PlatformContextStore(),
+            auth_backend=PlatformAuthBackend(),
         )
     except KeyboardInterrupt:
         pass