agentkernel 0.2.10__tar.gz → 0.2.11__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentkernel
-Version: 0.2.10
+Version: 0.2.11
 Summary: Agent Kernel - Unified AI Agents Runtime
 Author: Yaala Labs
 Author-email: Yaala Labs <agentkernel@yaalalabs.com>
@@ -28,14 +28,15 @@ 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'
@@ -52,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
@@ -516,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.
@@ -617,6 +750,15 @@ 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
@@ -686,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..."
@@ -748,6 +901,20 @@ gmail:
       "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..."

{agentkernel-0.2.10 → 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.
@@ -541,6 +673,15 @@ 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
@@ -610,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..."
@@ -672,6 +824,20 @@ gmail:
       "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..."

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

@@ -4,10 +4,10 @@ build-backend = "uv_build"
 
 [project]
 name = "agentkernel"
-version = "0.2.10"
+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",
@@ -116,7 +117,7 @@ package = true
 
 [tool.isort]
 profile = "black"
-line_length = 120
+line_length = 150
 
 [tool.mypy]
 python_version = "3.12"
@@ -126,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.10 → agentkernel-0.2.11}/src/agentkernel/api/a2a/a2a.py

@@ -46,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())

{agentkernel-0.2.10 → agentkernel-0.2.11}/src/agentkernel/api/handler.py

@@ -140,10 +140,7 @@ class AgentRESTRequestHandler(RESTRequestHandler):
             if req.images:
                 for image in req.images:
                     self._log.debug(f"Adding image: {image.name}")
-                    if (
-                        not image.image_data.startswith(("http://", "https://", "data:", "s3://"))
-                        and not image.mime_type
-                    ):
+                    if not image.image_data.startswith(("http://", "https://", "data:", "s3://")) and not image.mime_type:
                         raise ValueError("mime_type is missing for image input, either in the base64 or explicitly")
                     requests.append(
                         AgentRequestImage(
@@ -170,9 +167,7 @@ class AgentRESTRequestHandler(RESTRequestHandler):
 
             return {
                 "result": (
-                    str(result)
-                    if isinstance(result, (AgentReplyText, AgentReplyImage))
-                    else "Non textual result received"
+                    str(result) if isinstance(result, (AgentReplyText, AgentReplyImage)) else "Non textual result received"
                 ),  # sending image is not supported at the moment
                 "session_id": service.get_response_session_id(req.session_id),
             }
@@ -185,9 +180,7 @@ class AgentRESTRequestHandler(RESTRequestHandler):
                 status_code=HTTPStatus.BAD_REQUEST,
                 detail={
                     "error": str(e),
-                    "session_id": (
-                        service.get_response_session_id(req.session_id) if service is not None else req.session_id
-                    ),
+                    "session_id": (service.get_response_session_id(req.session_id) if service is not None else req.session_id),
                 },
             )
         except Exception as e:
@@ -295,11 +288,7 @@ class AgentRESTRequestHandler(RESTRequestHandler):
             self._log.debug(f"Result: {result}")
 
             return {
-                "result": (
-                    str(result)
-                    if isinstance(result, (AgentReplyText, AgentReplyImage))
-                    else "Non textual result received"
-                ),
+                "result": (str(result) if isinstance(result, (AgentReplyText, AgentReplyImage)) else "Non textual result received"),
                 "session_id": service.get_response_session_id(session_id),
             }
 

{agentkernel-0.2.10 → agentkernel-0.2.11}/src/agentkernel/core/base.py

@@ -171,13 +171,13 @@ class Runner(ABC):
     @abstractmethod
     async def run(self, agent: Any, session: Session, requests: list[AgentRequest]) -> AgentReply:
         """
-        Runs the agent with the provided multi modal inputs.
+        Runs the agent with the provided multi-modal inputs.
         :param agent: The agent to run.
         :param session: The session to use for the agent.
         :param requests: The list of requests to provide to the agent.
         :return: The result of the agent's execution.
         """
-        pass
+        raise NotImplementedError
 
 
 class Agent(ABC):

{agentkernel-0.2.10 → agentkernel-0.2.11}/src/agentkernel/core/config.py

@@ -137,6 +137,20 @@ class _TestConfig(BaseModel):
     judge: _JudgeConfig = Field(description="Judge configuration", default_factory=_JudgeConfig)
 
 
+class _GuardrailParamConfig(BaseModel):
+    enabled: bool = Field(default=False, description="Enable Guardrail")
+    type: str = Field(default="openai", pattern="^(openai|bedrock)$")
+    config_path: Optional[str] = Field(default=None, description="Path to guardrail configuration file (OpenAI only)")
+    model: Optional[str] = Field(default="gpt-4o-mini", description="LLM model name to use for guardrail (OpenAI only)")
+    id: Optional[str] = Field(default=None, description="AWS Bedrock guardrail ID (Bedrock only)")
+    version: Optional[str] = Field(default="DRAFT", description="AWS Bedrock guardrail version (Bedrock only)")
+
+
+class _GuardrailConfig(BaseModel):
+    input: _GuardrailParamConfig = Field(description="Input Guardrail configuration", default_factory=_GuardrailParamConfig)
+    output: _GuardrailParamConfig = Field(description="Output Guardrail configuration", default_factory=_GuardrailParamConfig)
+
+
 class AKConfig(YamlBaseSettingsModified):
     debug: bool = Field(default=False, description="Enable debug mode")
     session: _SessionStoreConfig = Field(
@@ -151,19 +165,14 @@ class AKConfig(YamlBaseSettingsModified):
     )
     slack: _SlackConfig = Field(description="Slack related configurations", default_factory=_SlackConfig)
     whatsapp: _WhatsAppConfig = Field(description="WhatsApp related configurations", default_factory=_WhatsAppConfig)
-    messenger: _MessengerConfig = Field(
-        description="Facebook Messenger related configurations", default_factory=_MessengerConfig
-    )
-    instagram: _InstagramConfig = Field(
-        description="Instagram Business API related configurations", default_factory=_InstagramConfig
-    )
-    telegram: _TelegramConfig = Field(
-        description="Telegram Bot related configurations", default_factory=_TelegramConfig
-    )
+    messenger: _MessengerConfig = Field(description="Facebook Messenger related configurations", default_factory=_MessengerConfig)
+    instagram: _InstagramConfig = Field(description="Instagram Business API related configurations", default_factory=_InstagramConfig)
+    telegram: _TelegramConfig = Field(description="Telegram Bot related configurations", default_factory=_TelegramConfig)
     gmail: _GmailConfig = Field(description="Gmail related configurations", default_factory=_GmailConfig)
 
     trace: _TraceConfig = Field(description="Tracing related configurations", default_factory=_TraceConfig)
     test: _TestConfig = Field(description="Test related configurations", default_factory=_TestConfig)
+    guardrail: _GuardrailConfig = Field(description="Guardrail related configurations", default_factory=_GuardrailConfig)
     library_version: str = Field(default=_get_ak_version(), description="Library version")
 
     @classmethod

{agentkernel-0.2.10 → agentkernel-0.2.11}/src/agentkernel/core/hooks.py

@@ -18,9 +18,7 @@ Currently, they will get only called for the initial execution of an agent when
 
 class PreHook(ABC):
     @abstractmethod
-    async def on_run(
-        self, session: "Session", agent: "Agent", requests: list[AgentRequest]
-    ) -> list[AgentRequest] | AgentReply:
+    async def on_run(self, session: "Session", agent: "Agent", requests: list[AgentRequest]) -> list[AgentRequest] | AgentReply:
         """
         Hook method called before an agent starts executing a prompt. These hooks can modify the prompt or halt execution.
         Some use cases:
@@ -49,9 +47,7 @@ class PreHook(ABC):
 
 class PostHook(ABC):
     @abstractmethod
-    async def on_run(
-        self, session: "Session", requests: list[AgentRequest], agent: "Agent", agent_reply: AgentReply
-    ) -> AgentReply:
+    async def on_run(self, session: "Session", requests: list[AgentRequest], agent: "Agent", agent_reply: AgentReply) -> AgentReply:
         """
         Hook method called after an agent finishes executing a prompt. These hooks can modify the agent's reply. Some use cases:
           - Moderation of agent replies. e.g. output guardrails

{agentkernel-0.2.10 → agentkernel-0.2.11}/src/agentkernel/core/runtime.py

@@ -9,6 +9,7 @@ from typing import Optional
 from singleton_type import Singleton
 
 from ..core.util.key_value_cache import KeyValueCache
+from ..guardrail.guardrail import InputGuardrailFactory, OutputGuardrailFactory
 from .base import Agent, Session
 from .builder import SessionStoreBuilder
 from .model import (
@@ -31,6 +32,8 @@ class Runtime:
 
     _current: Optional[Runtime] = None
     _lock: RLock = RLock()
+    _system_pre_hooks: list = [InputGuardrailFactory.get()]
+    _system_post_hooks: list = [OutputGuardrailFactory.get()]
 
     def __init__(self, sessions: SessionStore):
         """
@@ -134,13 +137,11 @@ class Runtime:
         """
         self._log.debug(f"Executing pre hooks with agent '{agent.name}' and requests: {requests}")
 
-        pre_hooks = agent.pre_hooks
+        pre_hooks = agent.pre_hooks + self._system_pre_hooks  # system pre-hooks are always executed last
         for hook in pre_hooks:
             reply = await hook.on_run(session, agent, requests)
             if isinstance(reply, (AgentReplyText, AgentReplyImage)):
-                self._log.debug(
-                    f"PreHook halted execution for agent '{agent.name}' by hook '{hook.name()}' with reply: {reply}"
-                )
+                self._log.debug(f"PreHook halted execution for agent '{agent.name}' by hook '{hook.name()}' with reply: {reply}")
                 return reply
 
             # Validation to ensure the correct type is returned from the hooks. This is important to avoid runtime errors.
@@ -151,22 +152,18 @@ class Runtime:
                             f"PreHook '{hook.name()}' returned an invalid type in the requests list. Expected AgentRequest, got {type(item)}"
                         )
             else:
-                raise TypeError(
-                    f"PreHook '{hook.name()}' returned an invalid type. Expected list[AgentRequest], got {type(reply)}"
-                )
+                raise TypeError(f"PreHook '{hook.name()}' returned an invalid type. Expected list[AgentRequest], got {type(reply)}")
             requests = reply
 
         self._log.debug(f"Running agent '{agent.name}' with requests: {requests}")
 
         reply = await agent.runner.run(agent, session, requests)
 
-        post_hooks = agent.post_hooks
+        post_hooks = self._system_post_hooks + agent.post_hooks  # system post-hooks are always executed first
         for hook in post_hooks:
             reply = await hook.on_run(session, requests, agent, reply)
             if not isinstance(reply, (AgentReplyText, AgentReplyImage)):
-                raise TypeError(
-                    f"PostHook '{hook.name()}' returned an invalid type. Expected AgentReply, got {type(reply)}"
-                )
+                raise TypeError(f"PostHook '{hook.name()}' returned an invalid type. Expected AgentReply, got {type(reply)}")
             self._log.debug(f"PostHook executed for agent '{agent.name}' by hook '{hook.name()}' reply: {reply}")
         return reply