From 900c52c494d12c52c9a63e715580ae0366d506e4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?V=C3=ADctor=20Mayoral=20Vilches?= Date: Sun, 30 Nov 2025 12:42:26 +0100 Subject: [PATCH] v0.5.7, fix in guardrails MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Víctor Mayoral Vilches --- pyproject.toml | 2 +- src/cai/agents/guardrails.py | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 4a1f3c54..db4ea30b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "cai-framework" -version = "0.5.6" +version = "0.5.7" description = "Cybersecurity AI Framework" readme = "README.md" requires-python = ">=3.9" diff --git a/src/cai/agents/guardrails.py b/src/cai/agents/guardrails.py index b2941b60..9fabee4b 100644 --- a/src/cai/agents/guardrails.py +++ b/src/cai/agents/guardrails.py @@ -24,6 +24,9 @@ from openai import AsyncOpenAI import os import unicodedata +# Determine API key +api_key = os.getenv("ALIAS_API_KEY", os.getenv("OPENAI_API_KEY", "sk-alias-1234567890")) + class PromptInjectionCheck(BaseModel): """Result of prompt injection detection""" @@ -221,7 +224,7 @@ injection_detector_agent = Agent( output_type=PromptInjectionCheck, model=OpenAIChatCompletionsModel( model=os.getenv('CAI_MODEL', 'alias0'), - openai_client=AsyncOpenAI(), + openai_client=AsyncOpenAI(api_key=api_key), ) )