feat: updates for platform

Signed-off-by: Tomas Weiss <tomas.weiss2@gmail.com>

Author: tomkis

.env.example

@@ -0,0 +1,51 @@
+name: Build and Push Agent Image
+
+on:
+  push:
+    tags:
+      - 'release-*'
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v3
+
+    - name: Set up Python environment
+      uses: actions/setup-python@v4
+      with:
+        python-version: '3.x'
+
+    - name: Install BeeAI CLI
+      run: pip install beeai-cli
+
+    - name: Login to GitHub Container Registry
+      uses: docker/login-action@v2
+      with:
+        registry: ghcr.io
+        username: ${{ github.actor }}
+        password: ${{ secrets.GITHUB_TOKEN }}
+
+    - name: Set up Docker
+      uses: docker/setup-docker-action@v4
+      with:
+        daemon-config: '{"features": {"containerd-snapshotter": true}}'
+
+    - name: Set up Docker Buildx
+      uses: docker/setup-buildx-action@v3
+
+    - name: Extract version from tag
+      id: extract_version
+      run: |
+        # Extract version from tag (e.g., release-1.2.3 -> 1.2.3)
+        VERSION=${GITHUB_REF#refs/tags/release-}
+        echo "version=$VERSION" >> $GITHUB_OUTPUT
+
+    - name: Build and Push
+      run: |
+        beeai build ./ --tag ghcr.io/${{ github.repository }}/my-agent:${{ steps.extract_version.outputs.version }} --no-import --multi-platform --push

.github/workflows/build-agent.yaml

@@ -0,0 +1,16 @@
+FROM python:3.13-bookworm
+COPY --from=ghcr.io/astral-sh/uv:0.7.15 /uv /bin/
+
+ENV UV_LINK_MODE=copy \
+    PRODUCTION_MODE=true
+
+ADD . /app
+WORKDIR /app
+
+RUN uv sync --no-cache --locked --link-mode copy
+
+ENV PRODUCTION_MODE=True \
+    PATH="/app/.venv/bin:$PATH" \
+    HOME=/tmp
+
+CMD ["uv", "run", "--no-sync", "server"]

Dockerfile

@@ -8,12 +8,13 @@ authors = [
 license = {text = "Apache-2.0"}
 requires-python = ">=3.11,<4.0"
 dependencies = [
-    "beeai-framework[beeai-platform]>=0.1.44",
+    "beeai-framework[beeai-platform]>=0.1.52",
     "aiohttp>=3.8.0",
     "mcp>=1.0.0",
     "python-dotenv>=1.0.0",
     "openinference-instrumentation-beeai>=0.1.10",
-    "beeai-sdk>=0.3.1"
+    "beeai-sdk>=0.3.5-rc4",
+    "pydantic>=2.11.9",
 ]
 
 [project.optional-dependencies]

pyproject.toml

@@ -4,21 +4,21 @@
 from beeai_framework.agents.experimental.requirements.conditional import ConditionalRequirement
 from beeai_framework.middleware.trajectory import GlobalTrajectoryMiddleware
 from beeai_framework.tools import Tool
+from agents.session_context import SessionContext
+from agents.utils import ToolNotFoundError, get_tools_by_names, create_repo_scoped_tool
 
-from agents.utils import ToolNotFoundError, get_tools_by_names, llm, session_manager, create_repo_scoped_tool
 
-
-async def get_agent_analyst():
+async def get_agent_analyst(session_context: SessionContext):
     """Create and configure the duplicate issue analyzer agent."""
-    tools = await session_manager.get_tools()
+    tools = await session_context.get_tools()
     tool_names = ["get_issue", "list_issues", "search_issues"]
 
     try:
         original_tools = await get_tools_by_names(tools, tool_names)
         # Create repo-scoped versions of all tools
         available_tools = []
         for tool in original_tools:
-            scoped_tool = await create_repo_scoped_tool(tool)
+            scoped_tool = await create_repo_scoped_tool(tool, session_context.get_repository())
             available_tools.append(scoped_tool)
     except ToolNotFoundError as e:
         raise RuntimeError(f"Failed to configure duplicate finder agent: {e}") from e
@@ -54,7 +54,7 @@ async def get_agent_analyst():
 
     return RequirementAgent(
         name="Analyst",
-        llm=llm,
+        llm=session_context.get_llm(),
         role=role,
         instructions=instruction,
         tools=available_tools,

src/agents/agent_analyst.py

@@ -1,28 +1,27 @@
 import json
-import os
 from textwrap import dedent, indent
 
 from beeai_framework.agents.experimental import RequirementAgent
 from beeai_framework.agents.experimental.prompts import (
     RequirementAgentSystemPromptInput,
-    RequirementAgentTaskPromptInput,
 )
-from beeai_framework.agents.experimental.requirements.ask_permission import AskPermissionRequirement
 from beeai_framework.agents.experimental.requirements.conditional import ConditionalRequirement
+from beeai_framework.memory import BaseMemory
 from beeai_framework.middleware.trajectory import GlobalTrajectoryMiddleware
 from beeai_framework.template import PromptTemplate, PromptTemplateInput
 from beeai_framework.tools import Tool
 from beeai_framework.tools.handoff import HandoffTool
 
 from agents.agent_analyst import get_agent_analyst
 from agents.agent_writer import get_agent_writer
+from agents.session_context import SessionContext
 from agents.simple_think import SimpleThinkTool
-from agents.utils import ToolNotFoundError, create_repo_scoped_tool, get_tools_by_names, llm, session_manager
+from agents.utils import ToolNotFoundError, create_repo_scoped_tool, get_tools_by_names
 
 
-async def get_agent_manager():
+async def get_agent_manager(session_context: SessionContext, memory: BaseMemory):
     """Create and configure the issue workflow management agent."""
-    tools = await session_manager.get_tools()
+    tools = await session_context.get_tools()
 
     try:
         tools = await get_tools_by_names(tools, ["create_issue", "list_issue_types"])
@@ -32,9 +31,9 @@ async def get_agent_manager():
 
         for tool in tools:
             if tool.name == "create_issue":
-                create_issue = await create_repo_scoped_tool(tool)
+                create_issue = await create_repo_scoped_tool(tool, session_context.get_repository())
             elif tool.name == "list_issue_types":
-                list_issue_types = await create_repo_scoped_tool(tool)
+                list_issue_types = await create_repo_scoped_tool(tool, session_context.get_repository())
 
     except ToolNotFoundError as e:
         raise RuntimeError(f"Failed to configure the agent: {e}") from e
@@ -55,13 +54,11 @@ async def get_agent_manager():
     issue_types_lines = [f"- {issue_type['name']}: {issue_type['description']}" for issue_type in issue_types_data]
     issue_types_text = indent("\n".join(issue_types_lines), "    ")
 
-    repository = os.getenv("GITHUB_REPOSITORY")
-
     role = "helpful coordinator"
     instruction = f"""\
 As the Coordinator, your responsibilities include routing tasks to experts, managing processes sequentially, and handling all user-facing communication. You do not perform technical writing or reasoning yourself.
 
-You work in the following repository: {repository}
+You work in the following repository: {session_context.get_repository()}
 
 ## Operating Principles
 - Manage the full lifecycle of a GitHub issue from user request to creation.
@@ -113,8 +110,8 @@ async def get_agent_manager():
 """
 
     # Get the specialized agents
-    writer = await get_agent_writer()
-    analyst = await get_agent_analyst()
+    writer = await get_agent_writer(session_context)
+    analyst = await get_agent_analyst(session_context)
 
     handoff_writer = HandoffTool(
         target=writer,
@@ -171,8 +168,9 @@ async def get_agent_manager():
 
     return RequirementAgent(
         name="Project Manager",
-        llm=llm,
+        llm=session_context.get_llm(),
         role=role,
+        memory=memory,
         instructions=instruction,
         tools=[
             SimpleThinkTool(),
@@ -182,7 +180,7 @@ async def get_agent_manager():
         ],
         requirements=[
             ConditionalRequirement(SimpleThinkTool, force_at_step=1, force_after=[Tool], consecutive_allowed=False),
-            AskPermissionRequirement(create_issue),
+            # AskPermissionRequirement(create_issue),
         ],
         templates={
             "system": PromptTemplate(PromptTemplateInput(schema=RequirementAgentSystemPromptInput, template=template)),

src/agents/agent_manager.py

@@ -3,7 +3,8 @@
 from beeai_framework.agents.tool_calling import ToolCallingAgent
 from beeai_framework.agents.tool_calling.prompts import ToolCallingAgentSystemPrompt, ToolCallingAgentTaskPrompt
 
-from agents.utils import fetch_content, llm
+from agents.session_context import SessionContext
+from agents.utils import fetch_content
 
 
 async def get_template(template_type: str) -> str:
@@ -41,7 +42,7 @@ def _strip_yaml_frontmatter(content: str) -> str:
     return content
 
 
-async def get_agent_writer():
+async def get_agent_writer(session_context: SessionContext):
     """Create and configure the technical issue writing agent."""
     # Get documentation content from environment variable
     docs_url = os.getenv("DOCS_URL")
@@ -140,7 +141,7 @@ async def get_agent_writer():
 
 """
 
-    clonedLlm = await llm.clone()
+    clonedLlm = await session_context.get_llm().clone()
     clonedLlm.emitter.on("start", lambda data, event: data.input.tools.pop(0))  # removes the final answer tool
 
     return ToolCallingAgent(