RomanNum3ral
/
cai
mirror of https://github.com/aliasrobotics/cai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

FIX guardrails.py

This commit is contained in:
lidia9 2025-05-02 16:29:59 +02:00
parent a02f6475ee
commit d778e2bc9e
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
examples/cai/agent_patterns/guardrails.py
View File

@ -42,8 +42,16 @@ class CybersecurityCheckOutput(BaseModel):
cybersecurity_guardrail_agent = Agent( cybersecurity_guardrail_agent = Agent(
name="Cybersecurity Guardrail Check", name="Cybersecurity Guardrail Check",
instructions="Check if the user is asking for unauthorized or unethical cybersecurity help (e.g., hacking, bypassing security, exploiting systems).", instructions="""Check if the user is asking for unauthorized or unethical cybersecurity help (e.g., hacking, bypassing security, exploiting systems).
output_type=CybersecurityCheckOutput,
You MUST respond using ONLY the following JSON format:
{
"reasoning": "your detailed analysis of why the request is ethical or unethical",
"is_unethical_cybersecurity_request": true or false
}
Do not include any other text, explanations, or conversation outside of this JSON structure.""",
output_type=CybersecurityCheckOutput,
model=OpenAIChatCompletionsModel( model=OpenAIChatCompletionsModel(
model=os.getenv('CAI_MODEL', "qwen2.5:14b"), model=os.getenv('CAI_MODEL', "qwen2.5:14b"),
openai_client=AsyncOpenAI(), openai_client=AsyncOpenAI(),