agentkernel 0.2.9__tar.gz → 0.2.11__tar.gz

{agentkernel-0.2.9 → agentkernel-0.2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentkernel
-Version: 0.2.9
+Version: 0.2.11
 Summary: Agent Kernel - Unified AI Agents Runtime
 Author: Yaala Labs
 Author-email: Yaala Labs <agentkernel@yaalalabs.com>
@@ -28,17 +28,19 @@ Requires-Dist: google-api-python-client>=2.0.0 ; extra == 'gmail'
 Requires-Dist: httpx>=0.27.0 ; extra == 'instagram'
 Requires-Dist: langfuse>=3.9.2 ; extra == 'langfuse'
 Requires-Dist: nest-asyncio>=1.6.0 ; extra == 'langfuse'
-Requires-Dist: langgraph~=0.6.2 ; extra == 'langgraph'
-Requires-Dist: langchain~=0.3.27 ; extra == 'langgraph'
-Requires-Dist: langchain-community~=0.3.27 ; extra == 'langgraph'
-Requires-Dist: litellm~=1.74.3 ; extra == 'langgraph'
+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: httpx>=0.27.0 ; extra == 'messenger'
-Requires-Dist: openai-agents>=0.6.3 ; extra == 'openai'
+Requires-Dist: openai-agents>=0.6.5 ; extra == 'openai'
 Requires-Dist: openinference-instrumentation-openai-agents>=1.4.0 ; extra == 'openai'
+Requires-Dist: openai-guardrails>=0.2.1 ; extra == 'openai'
 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: httpx>=0.27.0 ; extra == 'telegram'
 Requires-Dist: pytest>=8.4.1 ; extra == 'test'
 Requires-Dist: pytest-asyncio>=1.2.0 ; extra == 'test'
@@ -51,7 +53,7 @@ Requires-Dist: datasets>=2.14.0 ; extra == 'test'
 Requires-Dist: pandas>=2.0.0 ; extra == 'test'
 Requires-Dist: litellm>=1.74.9 ; extra == 'test'
 Requires-Dist: httpx>=0.27.0 ; extra == 'whatsapp'
-Requires-Python: >=3.12
+Requires-Python: >=3.12, <3.14
 Provides-Extra: a2a
 Provides-Extra: adk
 Provides-Extra: api
@@ -515,6 +517,138 @@ test:
     embedding_model: text-embedding-3-small
 ```
 
+#### Guardrails Configuration
+
+Configure input and output guardrails to validate agent requests and responses for safety and compliance.
+
+- **Input Guardrails**
+  - **Enabled**
+    - **Field**: `guardrail.input.enabled`
+    - **Default**: `false`
+    - **Description**: Enable input validation guardrails
+    - **Environment Variable**: `AK_GUARDRAIL__INPUT__ENABLED`
+
+  - **Type**
+    - **Field**: `guardrail.input.type`
+    - **Default**: `openai`
+    - **Options**: `openai`, `bedrock`
+    - **Description**: Guardrail provider type
+    - **Environment Variable**: `AK_GUARDRAIL__INPUT__TYPE`
+
+  - **Config Path**
+    - **Field**: `guardrail.input.config_path`
+    - **Default**: `None`
+    - **Description**: Path to guardrail configuration JSON file (OpenAI only)
+    - **Environment Variable**: `AK_GUARDRAIL__INPUT__CONFIG_PATH`
+
+  - **Model**
+    - **Field**: `guardrail.input.model`
+    - **Default**: `gpt-4o-mini`
+    - **Description**: LLM model to use for guardrail validation (OpenAI only)
+    - **Environment Variable**: `AK_GUARDRAIL__INPUT__MODEL`
+
+  - **ID**
+    - **Field**: `guardrail.input.id`
+    - **Default**: `None`
+    - **Description**: AWS Bedrock guardrail ID (Bedrock only)
+    - **Environment Variable**: `AK_GUARDRAIL__INPUT__ID`
+
+  - **Version**
+    - **Field**: `guardrail.input.version`
+    - **Default**: `DRAFT`
+    - **Description**: AWS Bedrock guardrail version (Bedrock only)
+    - **Environment Variable**: `AK_GUARDRAIL__INPUT__VERSION`
+
+- **Output Guardrails**
+  - **Enabled**
+    - **Field**: `guardrail.output.enabled`
+    - **Default**: `false`
+    - **Description**: Enable output validation guardrails
+    - **Environment Variable**: `AK_GUARDRAIL__OUTPUT__ENABLED`
+
+  - **Type**
+    - **Field**: `guardrail.output.type`
+    - **Default**: `openai`
+    - **Options**: `openai`, `bedrock`
+    - **Description**: Guardrail provider type
+    - **Environment Variable**: `AK_GUARDRAIL__OUTPUT__TYPE`
+
+  - **Config Path**
+    - **Field**: `guardrail.output.config_path`
+    - **Default**: `None`
+    - **Description**: Path to guardrail configuration JSON file (OpenAI only)
+    - **Environment Variable**: `AK_GUARDRAIL__OUTPUT__CONFIG_PATH`
+
+  - **Model**
+    - **Field**: `guardrail.output.model`
+    - **Default**: `gpt-4o-mini`
+    - **Description**: LLM model to use for guardrail validation (OpenAI only)
+    - **Environment Variable**: `AK_GUARDRAIL__OUTPUT__MODEL`
+
+  - **ID**
+    - **Field**: `guardrail.output.id`
+    - **Default**: `None`
+    - **Description**: AWS Bedrock guardrail ID (Bedrock only)
+    - **Environment Variable**: `AK_GUARDRAIL__OUTPUT__ID`
+
+  - **Version**
+    - **Field**: `guardrail.output.version`
+    - **Default**: `DRAFT`
+    - **Description**: AWS Bedrock guardrail version (Bedrock only)
+    - **Environment Variable**: `AK_GUARDRAIL__OUTPUT__VERSION`
+
+**Guardrail Setup:**
+
+To use OpenAI guardrails, install the openai-guardrails package:
+
+```bash
+pip install agentkernel[openai]
+```
+
+To use AWS Bedrock guardrails, install the AWS package:
+
+```bash
+pip install agentkernel[aws]
+```
+
+Create guardrail configuration:
+
+**For OpenAI:** Create configuration files following the [OpenAI Guardrails format](https://guardrails.openai.com/).
+
+**For Bedrock:** Create a guardrail in AWS Bedrock and note the guardrail ID and version.
+
+Configure guardrails in your configuration:
+
+**OpenAI Example:**
+```yaml
+guardrail:
+  input:
+    enabled: true
+    type: openai
+    model: gpt-4o-mini
+    config_path: /path/to/guardrails_input.json
+  output:
+    enabled: true
+    type: openai
+    model: gpt-4o-mini
+    config_path: /path/to/guardrails_output.json
+```
+
+**Bedrock Example:**
+```yaml
+guardrail:
+  input:
+    enabled: true
+    type: bedrock
+    id: your-guardrail-id
+    version: "1"  # or "DRAFT"
+  output:
+    enabled: true
+    type: bedrock
+    id: your-guardrail-id
+    version: "1"
+```
+
 #### Messaging Platform Integrations
 
 Configure integrations with messaging platforms.
@@ -577,6 +711,17 @@ Configure integrations with messaging platforms.
 - **Bot Token**, **Webhook Secret**, **API Version**
   - **Environment Variables**: `AK_TELEGRAM__BOT_TOKEN`, `AK_TELEGRAM__WEBHOOK_SECRET`, `AK_TELEGRAM__API_VERSION`
 
+##### Gmail
+
+- **Agent**
+  - **Field**: `gmail.agent`
+  - **Default**: `"general"`
+  - **Description**: Default agent for Gmail interactions
+  - **Environment Variable**: `AK_GMAIL__AGENT`
+
+- **Client ID**, **Client Secret**, **Token File**, **Poll Interval**, **Label Filter**
+  - **Environment Variables**: `AK_GMAIL__CLIENT_ID`, `AK_GMAIL__CLIENT_SECRET`, `AK_GMAIL__TOKEN_FILE`, `AK_GMAIL__POLL_INTERVAL`, `AK_GMAIL__LABEL_FILTER`
+
 ### Configuration Examples
 
 #### Environment Variables
@@ -605,12 +750,24 @@ export AK_TEST__MODE=fallback  # Options: fuzzy, judge, fallback
 export AK_TEST__JUDGE__MODEL=gpt-4o-mini
 export AK_TEST__JUDGE__PROVIDER=openai
 export AK_TEST__JUDGE__EMBEDDING_MODEL=text-embedding-3-small
+# Guardrails configuration
+export AK_GUARDRAIL__INPUT__ENABLED=false
+export AK_GUARDRAIL__INPUT__TYPE=openai
+export AK_GUARDRAIL__INPUT__MODEL=gpt-4o-mini
+export AK_GUARDRAIL__INPUT__CONFIG_PATH=/path/to/guardrails_input.json
+export AK_GUARDRAIL__OUTPUT__ENABLED=false
+export AK_GUARDRAIL__OUTPUT__TYPE=openai
+export AK_GUARDRAIL__OUTPUT__MODEL=gpt-4o-mini
+export AK_GUARDRAIL__OUTPUT__CONFIG_PATH=/path/to/guardrails_output.json
 # Messaging platforms (optional)
 export AK_SLACK__AGENT=my-agent
 export AK_WHATSAPP__AGENT=my-agent
 export AK_MESSENGER__AGENT=my-agent
 export AK_INSTAGRAM__AGENT=my-agent
 export AK_TELEGRAM__AGENT=my-agent
+export AK_GMAIL__AGENT=my-agent
+export AK_GMAIL__CLIENT_ID=your-google-client-id
+export AK_GMAIL__CLIENT_SECRET=your-google-client-secret
 ```
 
 #### .env File
@@ -671,6 +828,17 @@ test:
     model: gpt-4o-mini
     provider: openai
     embedding_model: text-embedding-3-small
+guardrail:
+  input:
+    enabled: false
+    type: openai
+    model: gpt-4o-mini
+    config_path: /path/to/guardrails_input.json
+  output:
+    enabled: false
+    type: openai
+    model: gpt-4o-mini
+    config_path: /path/to/guardrails_output.json
 slack:
   agent: my-agent
   agent_acknowledgement: "Processing your request..."
@@ -683,6 +851,10 @@ instagram:
   agent: my-agent
 telegram:
   agent: my-agent
+gmail:
+  agent: my-agent
+  poll_interval: 30
+  label_filter: "INBOX"
 ```
 
 #### config.json
@@ -729,6 +901,20 @@ telegram:
       "embedding_model": "text-embedding-3-small"
     }
   },
+  "guardrail": {
+    "input": {
+      "enabled": false,
+      "type": "openai",
+      "model": "gpt-4o-mini",
+      "config_path": "/path/to/guardrails_input.json"
+    },
+    "output": {
+      "enabled": false,
+      "type": "openai",
+      "model": "gpt-4o-mini",
+      "config_path": "/path/to/guardrails_output.json"
+    }
+  },
   "slack": {
     "agent": "my-agent",
     "agent_acknowledgement": "Processing your request..."
@@ -745,6 +931,11 @@ telegram:
   },
   "telegram": {
     "agent": "my-agent"
+  },
+  "gmail": {
+    "agent": "my-agent",
+    "poll_interval": 30,
+    "label_filter": "INBOX"
   }
 }
 ```

{agentkernel-0.2.9 → agentkernel-0.2.11}/README.md

@@ -440,6 +440,138 @@ test:
     embedding_model: text-embedding-3-small
 ```
 
+#### Guardrails Configuration
+
+Configure input and output guardrails to validate agent requests and responses for safety and compliance.
+
+- **Input Guardrails**
+  - **Enabled**
+    - **Field**: `guardrail.input.enabled`
+    - **Default**: `false`
+    - **Description**: Enable input validation guardrails
+    - **Environment Variable**: `AK_GUARDRAIL__INPUT__ENABLED`
+
+  - **Type**
+    - **Field**: `guardrail.input.type`
+    - **Default**: `openai`
+    - **Options**: `openai`, `bedrock`
+    - **Description**: Guardrail provider type
+    - **Environment Variable**: `AK_GUARDRAIL__INPUT__TYPE`
+
+  - **Config Path**
+    - **Field**: `guardrail.input.config_path`
+    - **Default**: `None`
+    - **Description**: Path to guardrail configuration JSON file (OpenAI only)
+    - **Environment Variable**: `AK_GUARDRAIL__INPUT__CONFIG_PATH`
+
+  - **Model**
+    - **Field**: `guardrail.input.model`
+    - **Default**: `gpt-4o-mini`
+    - **Description**: LLM model to use for guardrail validation (OpenAI only)
+    - **Environment Variable**: `AK_GUARDRAIL__INPUT__MODEL`
+
+  - **ID**
+    - **Field**: `guardrail.input.id`
+    - **Default**: `None`
+    - **Description**: AWS Bedrock guardrail ID (Bedrock only)
+    - **Environment Variable**: `AK_GUARDRAIL__INPUT__ID`
+
+  - **Version**
+    - **Field**: `guardrail.input.version`
+    - **Default**: `DRAFT`
+    - **Description**: AWS Bedrock guardrail version (Bedrock only)
+    - **Environment Variable**: `AK_GUARDRAIL__INPUT__VERSION`
+
+- **Output Guardrails**
+  - **Enabled**
+    - **Field**: `guardrail.output.enabled`
+    - **Default**: `false`
+    - **Description**: Enable output validation guardrails
+    - **Environment Variable**: `AK_GUARDRAIL__OUTPUT__ENABLED`
+
+  - **Type**
+    - **Field**: `guardrail.output.type`
+    - **Default**: `openai`
+    - **Options**: `openai`, `bedrock`
+    - **Description**: Guardrail provider type
+    - **Environment Variable**: `AK_GUARDRAIL__OUTPUT__TYPE`
+
+  - **Config Path**
+    - **Field**: `guardrail.output.config_path`
+    - **Default**: `None`
+    - **Description**: Path to guardrail configuration JSON file (OpenAI only)
+    - **Environment Variable**: `AK_GUARDRAIL__OUTPUT__CONFIG_PATH`
+
+  - **Model**
+    - **Field**: `guardrail.output.model`
+    - **Default**: `gpt-4o-mini`
+    - **Description**: LLM model to use for guardrail validation (OpenAI only)
+    - **Environment Variable**: `AK_GUARDRAIL__OUTPUT__MODEL`
+
+  - **ID**
+    - **Field**: `guardrail.output.id`
+    - **Default**: `None`
+    - **Description**: AWS Bedrock guardrail ID (Bedrock only)
+    - **Environment Variable**: `AK_GUARDRAIL__OUTPUT__ID`
+
+  - **Version**
+    - **Field**: `guardrail.output.version`
+    - **Default**: `DRAFT`
+    - **Description**: AWS Bedrock guardrail version (Bedrock only)
+    - **Environment Variable**: `AK_GUARDRAIL__OUTPUT__VERSION`
+
+**Guardrail Setup:**
+
+To use OpenAI guardrails, install the openai-guardrails package:
+
+```bash
+pip install agentkernel[openai]
+```
+
+To use AWS Bedrock guardrails, install the AWS package:
+
+```bash
+pip install agentkernel[aws]
+```
+
+Create guardrail configuration:
+
+**For OpenAI:** Create configuration files following the [OpenAI Guardrails format](https://guardrails.openai.com/).
+
+**For Bedrock:** Create a guardrail in AWS Bedrock and note the guardrail ID and version.
+
+Configure guardrails in your configuration:
+
+**OpenAI Example:**
+```yaml
+guardrail:
+  input:
+    enabled: true
+    type: openai
+    model: gpt-4o-mini
+    config_path: /path/to/guardrails_input.json
+  output:
+    enabled: true
+    type: openai
+    model: gpt-4o-mini
+    config_path: /path/to/guardrails_output.json
+```
+
+**Bedrock Example:**
+```yaml
+guardrail:
+  input:
+    enabled: true
+    type: bedrock
+    id: your-guardrail-id
+    version: "1"  # or "DRAFT"
+  output:
+    enabled: true
+    type: bedrock
+    id: your-guardrail-id
+    version: "1"
+```
+
 #### Messaging Platform Integrations
 
 Configure integrations with messaging platforms.
@@ -502,6 +634,17 @@ Configure integrations with messaging platforms.
 - **Bot Token**, **Webhook Secret**, **API Version**
   - **Environment Variables**: `AK_TELEGRAM__BOT_TOKEN`, `AK_TELEGRAM__WEBHOOK_SECRET`, `AK_TELEGRAM__API_VERSION`
 
+##### Gmail
+
+- **Agent**
+  - **Field**: `gmail.agent`
+  - **Default**: `"general"`
+  - **Description**: Default agent for Gmail interactions
+  - **Environment Variable**: `AK_GMAIL__AGENT`
+
+- **Client ID**, **Client Secret**, **Token File**, **Poll Interval**, **Label Filter**
+  - **Environment Variables**: `AK_GMAIL__CLIENT_ID`, `AK_GMAIL__CLIENT_SECRET`, `AK_GMAIL__TOKEN_FILE`, `AK_GMAIL__POLL_INTERVAL`, `AK_GMAIL__LABEL_FILTER`
+
 ### Configuration Examples
 
 #### Environment Variables
@@ -530,12 +673,24 @@ export AK_TEST__MODE=fallback  # Options: fuzzy, judge, fallback
 export AK_TEST__JUDGE__MODEL=gpt-4o-mini
 export AK_TEST__JUDGE__PROVIDER=openai
 export AK_TEST__JUDGE__EMBEDDING_MODEL=text-embedding-3-small
+# Guardrails configuration
+export AK_GUARDRAIL__INPUT__ENABLED=false
+export AK_GUARDRAIL__INPUT__TYPE=openai
+export AK_GUARDRAIL__INPUT__MODEL=gpt-4o-mini
+export AK_GUARDRAIL__INPUT__CONFIG_PATH=/path/to/guardrails_input.json
+export AK_GUARDRAIL__OUTPUT__ENABLED=false
+export AK_GUARDRAIL__OUTPUT__TYPE=openai
+export AK_GUARDRAIL__OUTPUT__MODEL=gpt-4o-mini
+export AK_GUARDRAIL__OUTPUT__CONFIG_PATH=/path/to/guardrails_output.json
 # Messaging platforms (optional)
 export AK_SLACK__AGENT=my-agent
 export AK_WHATSAPP__AGENT=my-agent
 export AK_MESSENGER__AGENT=my-agent
 export AK_INSTAGRAM__AGENT=my-agent
 export AK_TELEGRAM__AGENT=my-agent
+export AK_GMAIL__AGENT=my-agent
+export AK_GMAIL__CLIENT_ID=your-google-client-id
+export AK_GMAIL__CLIENT_SECRET=your-google-client-secret
 ```
 
 #### .env File
@@ -596,6 +751,17 @@ test:
     model: gpt-4o-mini
     provider: openai
     embedding_model: text-embedding-3-small
+guardrail:
+  input:
+    enabled: false
+    type: openai
+    model: gpt-4o-mini
+    config_path: /path/to/guardrails_input.json
+  output:
+    enabled: false
+    type: openai
+    model: gpt-4o-mini
+    config_path: /path/to/guardrails_output.json
 slack:
   agent: my-agent
   agent_acknowledgement: "Processing your request..."
@@ -608,6 +774,10 @@ instagram:
   agent: my-agent
 telegram:
   agent: my-agent
+gmail:
+  agent: my-agent
+  poll_interval: 30
+  label_filter: "INBOX"
 ```
 
 #### config.json
@@ -654,6 +824,20 @@ telegram:
       "embedding_model": "text-embedding-3-small"
     }
   },
+  "guardrail": {
+    "input": {
+      "enabled": false,
+      "type": "openai",
+      "model": "gpt-4o-mini",
+      "config_path": "/path/to/guardrails_input.json"
+    },
+    "output": {
+      "enabled": false,
+      "type": "openai",
+      "model": "gpt-4o-mini",
+      "config_path": "/path/to/guardrails_output.json"
+    }
+  },
   "slack": {
     "agent": "my-agent",
     "agent_acknowledgement": "Processing your request..."
@@ -670,6 +854,11 @@ telegram:
   },
   "telegram": {
     "agent": "my-agent"
+  },
+  "gmail": {
+    "agent": "my-agent",
+    "poll_interval": 30,
+    "label_filter": "INBOX"
   }
 }
 ```

{agentkernel-0.2.9 → agentkernel-0.2.11}/pyproject.toml

@@ -4,10 +4,10 @@ build-backend = "uv_build"
 
 [project]
 name = "agentkernel"
-version = "0.2.9"
+version = "0.2.11"
 description = "Agent Kernel - Unified AI Agents Runtime"
 readme = "README.md"
-requires-python = ">=3.12"
+requires-python = ">=3.12,<3.14"
 license = { text = "MIT" }
 authors = [
     { name = "Yaala Labs", email = "agentkernel@yaalalabs.com" }
@@ -34,10 +34,10 @@ crewai = [
     "openinference-instrumentation-litellm>=0.1.28",
 ]
 langgraph = [
-    "langgraph~=0.6.2",
-    "langchain~=0.3.27",
-    "langchain_community~=0.3.27",
-    "litellm~=1.74.3"
+    "langgraph~=1.0.5",
+    "langchain~=1.2.3",
+    "langchain_community~=0.4.1",
+    "litellm~=1.74.9"
 ]
 cli = [
 
@@ -49,8 +49,9 @@ redis = [
     "redis>=7.1.0",
 ]
 openai = [
-    "openai-agents>=0.6.3",
+    "openai-agents>=0.6.5",
     "openinference-instrumentation-openai-agents>=1.4.0",
+    "openai-guardrails>=0.2.1",
 ]
 adk = [
     "google-adk>=1.14.1",
@@ -63,7 +64,8 @@ api = [
     "gunicorn>=23.0.0"
 ]
 slack = [
-    "slack-bolt==1.22.0"
+    "slack-bolt==1.22.0",
+    "httpx>=0.27.0"
 ]
 whatsapp = [
     "httpx>=0.27.0"
@@ -115,7 +117,7 @@ package = true
 
 [tool.isort]
 profile = "black"
-line_length = 120
+line_length = 150
 
 [tool.mypy]
 python_version = "3.12"
@@ -125,7 +127,7 @@ disallow_untyped_defs = true
 disallow_incomplete_defs = true
 
 [tool.black]
-line-length = 120
+line-length = 150
 target-version = ["py312"]
 
 [tool.pytest.ini_options]

{agentkernel-0.2.9 → agentkernel-0.2.11}/src/agentkernel/api/a2a/a2a.py

@@ -9,8 +9,9 @@ from a2a.types import AgentCard, InternalError, UnsupportedOperationError
 from a2a.utils import new_agent_text_message
 from a2a.utils.errors import ServerError
 
-from ...core import Agent, AgentService, GlobalRuntime
+from ...core import Agent, AgentService
 from ...core.config import AKConfig
+from ...core.runtime import Runtime
 
 
 class A2A:
@@ -45,9 +46,7 @@ class A2A:
                 raise ValueError("RequestContext must have a message")
             try:
                 response = await self._execute_agent(context.context_id, context.get_user_input())
-                await event_queue.enqueue_event(
-                    new_agent_text_message(str(response), context.context_id, context.task_id)
-                )
+                await event_queue.enqueue_event(new_agent_text_message(str(response), context.context_id, context.task_id))
             except Exception as e:
                 error = "Sorry, Agent Kernel encountered an error while processing your request"
                 self.log.error(traceback.format_exc())
@@ -69,7 +68,7 @@ class A2A:
             return
         if not AKConfig.get().a2a.enabled:
             return
-        agents: dict[str, Agent] = GlobalRuntime.instance().agents()
+        agents: dict[str, Agent] = Runtime.current().agents()
         for name, agent in agents.items():
             whitelisted = AKConfig.get().a2a.agents == ["*"] or name in AKConfig.get().a2a.agents
             if not whitelisted: