agentkernel 0.4.0__tar.gz → 0.5.1__tar.gz

{agentkernel-0.4.0 → agentkernel-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentkernel
-Version: 0.4.0
+Version: 0.5.1
 Summary: Agent Kernel - Unified AI Agents Runtime
 Author: Yaala Labs
 Author-email: Yaala Labs <agentkernel@yaalalabs.com>
@@ -26,6 +26,7 @@ Requires-Dist: chromadb>=0.4.0 ; extra == 'chromadb'
 Requires-Dist: crewai>=0.150.0 ; extra == 'crewai'
 Requires-Dist: openinference-instrumentation-crewai>=0.1.16 ; extra == 'crewai'
 Requires-Dist: openinference-instrumentation-litellm>=0.1.28 ; extra == 'crewai'
+Requires-Dist: google-cloud-firestore>=2.19.0 ; extra == 'gcp'
 Requires-Dist: google-auth>=2.0.0 ; extra == 'gmail'
 Requires-Dist: google-auth-oauthlib>=1.0.0 ; extra == 'gmail'
 Requires-Dist: google-auth-httplib2>=0.2.0 ; extra == 'gmail'
@@ -37,9 +38,8 @@ Requires-Dist: langgraph~=1.0.5 ; extra == 'langgraph'
 Requires-Dist: langchain~=1.2.3 ; extra == 'langgraph'
 Requires-Dist: langchain-community~=0.4.1 ; extra == 'langgraph'
 Requires-Dist: litellm~=1.74.9 ; extra == 'langgraph'
-Requires-Dist: fastmcp>=2.12.4 ; extra == 'mcp'
+Requires-Dist: fastmcp>=2.14.2,<3.0.0 ; extra == 'mcp'
 Requires-Dist: httpx>=0.27.0 ; extra == 'messenger'
-Requires-Dist: litellm>=1.74.9 ; extra == 'multimodal'
 Requires-Dist: neo4j>=5.0.0 ; extra == 'neo4j'
 Requires-Dist: openai-agents>=0.6.5 ; extra == 'openai'
 Requires-Dist: openinference-instrumentation-openai-agents>=1.4.0 ; extra == 'openai'
@@ -48,6 +48,7 @@ Requires-Dist: traceloop-sdk>=0.48.0 ; extra == 'openllmetry'
 Requires-Dist: redis>=7.1.0 ; extra == 'redis'
 Requires-Dist: slack-bolt==1.22.0 ; extra == 'slack'
 Requires-Dist: httpx>=0.27.0 ; extra == 'slack'
+Requires-Dist: smolagents>=1.0.0 ; extra == 'smolagents'
 Requires-Dist: botbuilder-core>=4.17.0 ; extra == 'teams'
 Requires-Dist: botbuilder-schema>=4.17.0 ; extra == 'teams'
 Requires-Dist: msal>=1.31.1 ; extra == 'teams'
@@ -76,18 +77,19 @@ Provides-Extra: azure
 Provides-Extra: chromadb
 Provides-Extra: cli
 Provides-Extra: crewai
+Provides-Extra: gcp
 Provides-Extra: gmail
 Provides-Extra: instagram
 Provides-Extra: langfuse
 Provides-Extra: langgraph
 Provides-Extra: mcp
 Provides-Extra: messenger
-Provides-Extra: multimodal
 Provides-Extra: neo4j
 Provides-Extra: openai
 Provides-Extra: openllmetry
 Provides-Extra: redis
 Provides-Extra: slack
+Provides-Extra: smolagents
 Provides-Extra: teams
 Provides-Extra: telegram
 Provides-Extra: test
@@ -101,18 +103,18 @@ Description-Content-Type: text/markdown
 [![PyPI version](https://badge.fury.io/py/agentkernel.svg)](https://badge.fury.io/py/agentkernel)
 [![Python 3.12+](https://img.shields.io/badge/python-3.12+-blue.svg)](https://www.python.org/downloads/)
 
-Agent Kernel is a lightweight **multi-cloud AI agent runtime** and adapter layer for building and running AI agents across multiple frameworks and cloud providers. Deploy the same agent code to **AWS or Azure** without modification. Migrate your existing agents to Agent Kernel and instantly utilize pre-built execution and testing capabilities.
+Agent Kernel is a lightweight **multi-cloud AI agent runtime** and adapter layer for building and running AI agents across multiple frameworks and cloud providers. Deploy the same agent code to **AWS, Azure, or GCP** without modification. Migrate your existing agents to Agent Kernel and instantly utilize pre-built execution and testing capabilities.
 
-**Supported Cloud Platforms:** AWS, Azure
+**Supported Cloud Platforms:** AWS, Azure, GCP
 
 ## Features
 
 - **Unified API**: Common abstractions (Agent, Runner, Session, Module, Runtime) across frameworks
-- **Multi-Framework Support**: OpenAI Agents SDK, CrewAI, LangGraph, Google ADK
-- **Multi-Cloud Deployment**: Deploy to AWS (Lambda, ECS/Fargate) or Azure (Functions, Container Apps) with the same code
-- **Session Management**: Built-in session abstraction with multi-cloud storage (Redis, DynamoDB, Cosmos DB)
+- **Multi-Framework Support**: OpenAI Agents SDK, CrewAI, LangGraph, Google ADK, and Smolagents
+- **Multi-Cloud Deployment**: Deploy to AWS (Lambda, ECS/Fargate), Azure (Functions, Container Apps), or GCP (Cloud Run serverless/containerized) with the same code
+- **Session Management**: Built-in session abstraction with multi-cloud storage (Redis, DynamoDB, Cosmos DB, Firestore)
 - **Knowledge Bases**: Unified `KnowledgeBase` interface with ChromaDB, Neo4j, and Starburst/Trino backends via `KnowledgeBuilder`
-- **Flexible Deployment**: Interactive CLI, REST API, serverless (AWS Lambda, Azure Functions), containerized (AWS ECS, Azure Container Apps)
+- **Flexible Deployment**: Interactive CLI, REST API, serverless (AWS Lambda, Azure Functions, GCP Cloud Run), containerized (AWS ECS, Azure Container Apps, GCP Cloud Run)
 - **Pluggable Architecture**: Easy to extend with custom framework adapters and cloud providers
 - **MCP Server**: Built-in Model Context Protocol server for exposing agents as MCP tools and exposing any custom tool
 - **A2A Server**: Built-in Agent-to-Agent communication server for exposing agents with a simple configuration change
@@ -357,6 +359,31 @@ Azure Functions also accepts the normalized envelope, and flat run payloads are
 - `400` — No agent available
 - `500` — Unexpected error
 
+### GCP Cloud Run Deployment
+
+Deploy your agents to GCP Cloud Run using the built-in `CloudRun` handler.
+
+```python
+from agentkernel.gcp import CloudRun
+from agentkernel.openai import OpenAIModule
+
+OpenAIModule([...])
+
+@CloudRun.register("/app", method="GET")
+def app_handler() -> dict:
+    return {"status": "ok"}
+
+def main() -> None:
+    CloudRun.run()
+
+if __name__ == "__main__":
+    main()
+```
+
+`CloudRun` is the GCP equivalent of `Lambda` (AWS) and `AzureFunctions` (Azure). It wraps `RESTAPI` and starts a FastAPI/uvicorn server. Custom routes are registered with `@CloudRun.register(path, method)`. Use `CloudRun.run()` instead of `RESTAPI.run()` when deploying to GCP.
+
+For full Terraform deployment configuration, see [`ak-deployment/ak-gcp/`](https://github.com/yaalalabs/agent-kernel/tree/develop/ak-deployment/ak-gcp) or the [GCP deployment docs](https://github.com/yaalalabs/agent-kernel/tree/develop/docs/docs/deployment/gcp-serverless.md).
+
 ## Configuration
 
 Agent Kernel can be configured via environment variables, `.env` files, or YAML/JSON configuration files.
@@ -404,7 +431,7 @@ Configure where agent sessions are stored (supports multi-cloud storage backends
 
 - **Field**: `session.type`
 - **Type**: string
-- **Options**: `in_memory`, `redis`, `dynamodb` (AWS), `cosmosdb` (Azure)
+- **Options**: `in_memory`, `redis`, `dynamodb` (AWS), `cosmosdb` (Azure), `firestore` (GCP)
 - **Default**: `in_memory`
 - **Environment Variable**: `AK_SESSION__TYPE`
 
@@ -564,10 +591,15 @@ Configure the REST API server (if using the API module).
   - **Description**: List of agent names to expose as MCP tools
   - **Environment Variable**: `AK_MCP__AGENTS` (comma-separated)
 
-- **URL**
-  - **Field**: `mcp.url`
-  - **Default**: `http://localhost:8000/mcp`
-  - **Environment Variable**: `AK_MCP__URL`
+- **Stateless HTTP**
+  - **Field**: `mcp.stateless_http`
+  - **Default**: `false`
+  - **Description**: Run MCP in stateless HTTP mode (no `Mcp-Session-Id`)
+  - **Environment Variable**: `AK_MCP__STATELESS_HTTP`
+
+- **Endpoint** (not configurable)
+  - The MCP server is always mounted at `/mcp` on the main API server.
+  - Full URL: `http://{api.host}:{api.port}/mcp` — use `api.port` / `AK_API__PORT` to change the port.
 
 #### Trace (Observability) Configuration
 
@@ -1023,7 +1055,6 @@ mcp:
   enabled: false
   expose_agents: false
   agents: ["*"]
-  url: http://localhost:8000/mcp
 trace:
   enabled: true
   type: langfuse
@@ -1095,8 +1126,7 @@ gmail:
   "mcp": {
     "enabled": false,
     "expose_agents": false,
-    "agents": ["*"],
-    "url": "http://localhost:8000/mcp"
+    "agents": ["*"]
   },
   "trace": {
     "enabled": true,

{agentkernel-0.4.0 → agentkernel-0.5.1}/README.md

@@ -3,18 +3,18 @@
 [![PyPI version](https://badge.fury.io/py/agentkernel.svg)](https://badge.fury.io/py/agentkernel)
 [![Python 3.12+](https://img.shields.io/badge/python-3.12+-blue.svg)](https://www.python.org/downloads/)
 
-Agent Kernel is a lightweight **multi-cloud AI agent runtime** and adapter layer for building and running AI agents across multiple frameworks and cloud providers. Deploy the same agent code to **AWS or Azure** without modification. Migrate your existing agents to Agent Kernel and instantly utilize pre-built execution and testing capabilities.
+Agent Kernel is a lightweight **multi-cloud AI agent runtime** and adapter layer for building and running AI agents across multiple frameworks and cloud providers. Deploy the same agent code to **AWS, Azure, or GCP** without modification. Migrate your existing agents to Agent Kernel and instantly utilize pre-built execution and testing capabilities.
 
-**Supported Cloud Platforms:** AWS, Azure
+**Supported Cloud Platforms:** AWS, Azure, GCP
 
 ## Features
 
 - **Unified API**: Common abstractions (Agent, Runner, Session, Module, Runtime) across frameworks
-- **Multi-Framework Support**: OpenAI Agents SDK, CrewAI, LangGraph, Google ADK
-- **Multi-Cloud Deployment**: Deploy to AWS (Lambda, ECS/Fargate) or Azure (Functions, Container Apps) with the same code
-- **Session Management**: Built-in session abstraction with multi-cloud storage (Redis, DynamoDB, Cosmos DB)
+- **Multi-Framework Support**: OpenAI Agents SDK, CrewAI, LangGraph, Google ADK, and Smolagents
+- **Multi-Cloud Deployment**: Deploy to AWS (Lambda, ECS/Fargate), Azure (Functions, Container Apps), or GCP (Cloud Run serverless/containerized) with the same code
+- **Session Management**: Built-in session abstraction with multi-cloud storage (Redis, DynamoDB, Cosmos DB, Firestore)
 - **Knowledge Bases**: Unified `KnowledgeBase` interface with ChromaDB, Neo4j, and Starburst/Trino backends via `KnowledgeBuilder`
-- **Flexible Deployment**: Interactive CLI, REST API, serverless (AWS Lambda, Azure Functions), containerized (AWS ECS, Azure Container Apps)
+- **Flexible Deployment**: Interactive CLI, REST API, serverless (AWS Lambda, Azure Functions, GCP Cloud Run), containerized (AWS ECS, Azure Container Apps, GCP Cloud Run)
 - **Pluggable Architecture**: Easy to extend with custom framework adapters and cloud providers
 - **MCP Server**: Built-in Model Context Protocol server for exposing agents as MCP tools and exposing any custom tool
 - **A2A Server**: Built-in Agent-to-Agent communication server for exposing agents with a simple configuration change
@@ -259,6 +259,31 @@ Azure Functions also accepts the normalized envelope, and flat run payloads are
 - `400` — No agent available
 - `500` — Unexpected error
 
+### GCP Cloud Run Deployment
+
+Deploy your agents to GCP Cloud Run using the built-in `CloudRun` handler.
+
+```python
+from agentkernel.gcp import CloudRun
+from agentkernel.openai import OpenAIModule
+
+OpenAIModule([...])
+
+@CloudRun.register("/app", method="GET")
+def app_handler() -> dict:
+    return {"status": "ok"}
+
+def main() -> None:
+    CloudRun.run()
+
+if __name__ == "__main__":
+    main()
+```
+
+`CloudRun` is the GCP equivalent of `Lambda` (AWS) and `AzureFunctions` (Azure). It wraps `RESTAPI` and starts a FastAPI/uvicorn server. Custom routes are registered with `@CloudRun.register(path, method)`. Use `CloudRun.run()` instead of `RESTAPI.run()` when deploying to GCP.
+
+For full Terraform deployment configuration, see [`ak-deployment/ak-gcp/`](https://github.com/yaalalabs/agent-kernel/tree/develop/ak-deployment/ak-gcp) or the [GCP deployment docs](https://github.com/yaalalabs/agent-kernel/tree/develop/docs/docs/deployment/gcp-serverless.md).
+
 ## Configuration
 
 Agent Kernel can be configured via environment variables, `.env` files, or YAML/JSON configuration files.
@@ -306,7 +331,7 @@ Configure where agent sessions are stored (supports multi-cloud storage backends
 
 - **Field**: `session.type`
 - **Type**: string
-- **Options**: `in_memory`, `redis`, `dynamodb` (AWS), `cosmosdb` (Azure)
+- **Options**: `in_memory`, `redis`, `dynamodb` (AWS), `cosmosdb` (Azure), `firestore` (GCP)
 - **Default**: `in_memory`
 - **Environment Variable**: `AK_SESSION__TYPE`
 
@@ -466,10 +491,15 @@ Configure the REST API server (if using the API module).
   - **Description**: List of agent names to expose as MCP tools
   - **Environment Variable**: `AK_MCP__AGENTS` (comma-separated)
 
-- **URL**
-  - **Field**: `mcp.url`
-  - **Default**: `http://localhost:8000/mcp`
-  - **Environment Variable**: `AK_MCP__URL`
+- **Stateless HTTP**
+  - **Field**: `mcp.stateless_http`
+  - **Default**: `false`
+  - **Description**: Run MCP in stateless HTTP mode (no `Mcp-Session-Id`)
+  - **Environment Variable**: `AK_MCP__STATELESS_HTTP`
+
+- **Endpoint** (not configurable)
+  - The MCP server is always mounted at `/mcp` on the main API server.
+  - Full URL: `http://{api.host}:{api.port}/mcp` — use `api.port` / `AK_API__PORT` to change the port.
 
 #### Trace (Observability) Configuration
 
@@ -925,7 +955,6 @@ mcp:
   enabled: false
   expose_agents: false
   agents: ["*"]
-  url: http://localhost:8000/mcp
 trace:
   enabled: true
   type: langfuse
@@ -997,8 +1026,7 @@ gmail:
   "mcp": {
     "enabled": false,
     "expose_agents": false,
-    "agents": ["*"],
-    "url": "http://localhost:8000/mcp"
+    "agents": ["*"]
   },
   "trace": {
     "enabled": true,

{agentkernel-0.4.0 → agentkernel-0.5.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "agentkernel"
-version = "0.4.0"
+version = "0.5.1"
 description = "Agent Kernel - Unified AI Agents Runtime"
 readme = "README.md"
 requires-python = ">=3.12,<3.14"
@@ -42,6 +42,9 @@ langgraph = [
     "langchain_community~=0.4.1",
     "litellm~=1.74.9"
 ]
+smolagents = [
+    "smolagents>=1.0.0",
+]
 cli = [
 
 ]
@@ -114,6 +117,9 @@ gmail = [
     "google-auth-httplib2>=0.2.0",
     "google-api-python-client>=2.0.0",
 ]
+gcp = [
+    "google-cloud-firestore>=2.19.0",
+]
 test = [
     "pytest>=8.4.1",
     "pytest-asyncio>=1.2.0",
@@ -130,10 +136,7 @@ a2a = [
     "a2a-sdk[http-server]>=0.3.6",
 ]
 mcp = [
-    "fastmcp>=2.12.4",
-]
-multimodal = [
-    "litellm>=1.74.9",
+    "fastmcp>=2.14.2,<3.0.0"
 ]
 
 [project.scripts]

{agentkernel-0.4.0 → agentkernel-0.5.1}/src/agentkernel/__init__.py

@@ -5,7 +5,7 @@ This package provides a unified framework for working with different agent imple
 It includes core functionality for agent execution, modular integration, and runtime management.
 
 Features:
-- Support for multiple agent frameworks (OpenAI, CrewAI, LangGraph)
+- Support for multiple agent frameworks (OpenAI, CrewAI, LangGraph, Smolagents)
 - Modular architecture for easy extension
 - Unified runtime environment
 - CLI interface for agent interaction

agentkernel-0.5.1/src/agentkernel/api/handler.py

@@ -0,0 +1,87 @@
+import logging
+from abc import ABC, abstractmethod
+from typing import List, Optional
+
+from fastapi import APIRouter, File, Form, UploadFile
+from pydantic import ConfigDict
+
+from ..core import Config
+from ..core.chat_service import ChatService
+from ..core.model import BaseChatRequest, BaseRunRequest
+from ..core.runtime import Runtime
+
+
+class RESTRequestHandler(ABC):
+    @abstractmethod
+    def get_router(self) -> APIRouter:
+        """
+        Returns the APIRouter instance which has configured routes
+        E.g.:
+        - GET /api/v1/agents: List available agents
+
+        router = APIRouter()
+
+        @router.get("/api/v1/agents")
+        def list_agents():
+            from ..core.runtime import Runtime
+            return {"agents": list(Runtime.current().agents().keys())}
+
+        """
+        pass
+
+
+class AgentRESTRequestHandler(RESTRequestHandler):
+    """
+    API routers that expose endpoints to interact with Agent Kernel.
+    Endpoints:
+    - GET /api/v1/agents: List available agents
+    - POST /api/v1/chat: Run an agent with a prompt
+      Payload JSON: { "prompt": str, "agent": str | null, "session_id": str | null }
+    """
+
+    class BaseMultimodalRunRequest(BaseChatRequest):
+        """Chat request with multipart file and image uploads (UploadFile format)."""
+
+        files: Optional[List[UploadFile]] = None
+        images: Optional[List[UploadFile]] = None
+        model_config = ConfigDict(extra="allow")
+
+    def __init__(self):
+        self._log = logging.getLogger("ak.api.agent")
+        self._max_file_size = Config.get().api.max_file_size
+        self.chat_service = ChatService(rest_api_mode=True)
+
+    def get_router(self) -> APIRouter:
+        """
+        Returns the APIRouter instance.
+        """
+
+        router = APIRouter()
+
+        @router.get("/api/v1/agents")
+        def list_agents():
+            return {"agents": list(Runtime.current().agents().keys())}
+
+        @router.post("/api/v1/chat")
+        async def run(body: BaseRunRequest):
+            return await self.chat_service.process_async_chat_request(req=body)
+
+        @router.post("/api/v1/chat-multipart")
+        async def run_multipart(
+            prompt: str = Form(...),
+            agent: Optional[str] = Form(None),
+            session_id: Optional[str] = Form(None),
+            files: Optional[List[UploadFile]] = File(None),
+            images: Optional[List[UploadFile]] = File(None),
+        ):
+            # Construct BaseMultimodalRunRequest from form parameters
+            req = AgentRESTRequestHandler.BaseMultimodalRunRequest(
+                prompt=prompt,
+                agent=agent,
+                session_id=session_id,
+                files=files,
+                images=images,
+            )
+            return await self.chat_service.process_async_chat_request(req=req)
+
+        return router

{agentkernel-0.4.0 → agentkernel-0.5.1}/src/agentkernel/api/mcp/akmcp.py

@@ -59,7 +59,10 @@ class MCP:
         if cls._built or not AKConfig.get().mcp.enabled:
             return
         if cls._fastmcp is None:
-            cls._fastmcp = FastMCP("Agent Kernel FastMCP Instance")
+            cls._fastmcp = FastMCP(
+                "Agent Kernel FastMCP Instance",
+                stateless_http=AKConfig.get().mcp.stateless_http,
+            )
         if AKConfig.get().mcp.expose_agents:
             agents: dict[str, Agent] = Runtime.current().agents()
             for name, agent in agents.items():

{agentkernel-0.4.0 → agentkernel-0.5.1}/src/agentkernel/auth/handler.py

@@ -7,7 +7,7 @@ from pydantic import BaseModel
 class ValidationContext(BaseModel):
     path: Optional[str] = None
     http_method: Optional[str] = None
-    headers: Dict[str, str]
+    headers: Optional[Dict[str, Any]] = None
 
 
 class ValidationResult(BaseModel):

{agentkernel-0.4.0 → agentkernel-0.5.1}/src/agentkernel/core/__init__.py

@@ -29,3 +29,4 @@ from .service import AgentService
 from .hooks import PreHook, PostHook
 from .tool import ToolContext, ToolBuilder
 from .util.key_value_cache import KeyValueCache
+from .chat_service import ChatService

{agentkernel-0.4.0 → agentkernel-0.5.1}/src/agentkernel/core/builder.py

@@ -90,6 +90,7 @@ class SessionStoreBuilder(Builder):
         REDIS = "REDIS"
         DYNAMODB = "DYNAMODB"
         COSMOSDB = "COSMOSDB"
+        FIRESTORE = "FIRESTORE"
 
         @classmethod
         def from_str(cls, type_str: str) -> Self:
@@ -141,6 +142,10 @@ class SessionStoreBuilder(Builder):
             from .session.cosmosdb import CosmosDBSessionStore
 
             return CosmosDBSessionStore(cache=SessionCacheBuilder.build())
+        elif session_store_type == SessionStoreBuilder.Types.FIRESTORE:
+            from .session.firestore import FirestoreSessionStore
+
+            return FirestoreSessionStore(cache=SessionCacheBuilder.build())
         else:
             from .session.in_memory import InMemorySessionStore