Commit Graph

49 Commits

Author SHA1 Message Date
Víctor Mayoral Vilches bfb1281096 Prompt injection mitigation (#249)
* Reproduce prompt injection flaw in CAI

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Mitigate prompt injection

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Add a second PoC, refactor and propose a message when guardrails are triggered

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Add CAI_GUARDRAILS and 2 additional PoCs

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Mitigate PoC 3

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Mitigate PoC 4

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Add and mitigate PoCs 5 and 6

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Add PoCs 7 to 10

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Test and mitigate PoCs 7 to 10

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Add PoCs 11 to 14

Inspired by 'Embrade The Red' research and also by 'The Month of AI bugs'

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

* Mitigate PoCs 11 to 14

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>

---------

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-08-28 20:25:24 +02:00
Víctor Mayoral Vilches eb97f54caa Add another article, v0.5.3
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-07-10 05:39:53 +00:00
Víctor Mayoral Vilches 6f76f5e972 v0.5.2
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-06-24 16:39:32 +00:00
Víctor Mayoral Vilches 3f9f2d6631 v0.5.1
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-06-20 11:57:49 +02:00
Víctor Mayoral Vilches 191b657c6a Minor fixes during discussion
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-06-16 11:24:36 +02:00
Víctor Mayoral Vilches 04359a07f3 Revert "Apply code formatting with make format" 2025-06-14 09:51:19 +00:00
Víctor Mayoral Vilches c14fd96bdb Merge branch 'main' of https://github.com/aliasrobotics/cai 2025-06-09 14:59:45 +02:00
Kaarel Kaarelson b727917222 Add missing PyPDF2 package fix (#194)
* Add missing PyPDF2 package

* Fix the missing quotation

---------

Co-authored-by: Kaarel Kaarelson <kaarelkaarelson@Mac.lan>
Co-authored-by: Kaarel Kaarelson <kaarelkaarelson@Kaarels-MacBook-Air.local>
Co-authored-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-06-09 14:55:44 +02:00
Kaarel Kaarelson b80b898941 Add missing PyPDF2 package (#193)
Co-authored-by: Kaarel Kaarelson <kaarelkaarelson@Mac.lan>
2025-06-09 10:52:47 +02:00
Víctor Mayoral Vilches 5e005a7ca8 Remove cai-logs
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-06-05 19:40:31 +02:00
Víctor Mayoral Vilches 9b4b0f6165 v0.4.0 release
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-05-28 12:00:06 +00:00
Víctor Mayoral Vilches 80039ec527 Fix bug with cai-replay, add cai-gif
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-05-28 12:11:07 +02:00
Víctor Mayoral Vilches 7c14522518 Add asciinema tool, fix outstanding issues with f-strings
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-05-28 11:48:19 +02:00
Víctor Mayoral Vilches f54d4f5460 Add missing dependency
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-05-26 12:58:08 +02:00
Víctor Mayoral Vilches 5d3a31f528 Add fixes for proper pentestperf func.
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-05-23 14:18:35 +00:00
luijait f3f521944b Fix requirement 2025-05-23 10:21:42 +02:00
luijait cdefd5fe1f Streaming finished 2025-05-11 12:54:18 +02:00
Víctor Mayoral Vilches f4ee934944 Add log viz capabilities
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-05-11 07:26:14 +02:00
Víctor Mayoral Vilches 0c2cae2066 Add cai-replay
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-05-09 17:03:48 +02:00
Víctor Mayoral Vilches 5b29fbe4db Fix docs
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-04-02 16:48:23 +02:00
Víctor Mayoral Vilches d1fd6623e5 Merge remote-tracking branch 'origin/main' into cai 2025-03-31 13:21:34 +02:00
Martin Jagusch a7a814d30d Remove duplicate dependency definition of pynput (#367)
`pynput` is in the pyproject.toml file twice, as you can see on line 54.

The uv.lock file doesn't change.
2025-03-27 13:40:06 -04:00
Víctor Mayoral Vilches 887c2fb319 Add tools, prompts and agents
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-03-27 15:06:12 +01:00
Víctor Mayoral Vilches ac4797cec6 Re-structure under sdk/agents, get CLI mockup
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-03-27 14:31:38 +01:00
Rohan Mehta fe42548cce v0.0.7 (#353)
Upgrade version to prepare for a new release.
2025-03-26 09:06:24 -07:00
Víctor Mayoral Vilches adb358baae Update docs to integrate CAI, minor refinements
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-03-26 12:41:45 +01:00
Rohan Mehta bde8969892 Update pyproject.toml 2025-03-25 13:37:01 -04:00
Martín Bravo 692dfe0327 Merge branch 'main' of https://github.com/openai/openai-agents-python into feat/draw_graph 2025-03-25 18:02:51 +01:00
Martín Bravo 5a0e6a9215 Rename visualization dependency to viz in pyproject.toml 2025-03-25 18:01:45 +01:00
Rohan Mehta f28848e850 [1/n] Add MCP types to the SDK
### Summary:
1. Add the MCP dep for python 3.10, since it doesn't support 3.9 and below
2. Create MCPServer, which is the agents SDK representation of an MCP server
3. Create implementations for HTTP-SSE and StdIO servers, directly copying the [MCP SDK example](https://github.com/modelcontextprotocol/python-sdk/blob/main/examples/clients/simple-chatbot/mcp_simple_chatbot/main.py)
4. Add a util to transform MCP tools into Agent SDK tools

Note: I added optional caching support to the servers. That way, if you happen to know a server's tools don't change, you can just cache them.

### Test Plan:

Checks pass. I added tests at the end of the stack.
2025-03-25 12:51:40 -04:00
Martín Bravo 2c98d79d60 Add visualization optional dependency for graphviz 2025-03-25 16:59:17 +01:00
Martín Bravo 1ec5c2e233 Merge branch 'main' of https://github.com/openai/openai-agents-python into feat/draw_graph 2025-03-25 16:58:01 +01:00
Víctor Mayoral Vilches 147fc8142e Initial re-structuring towards CAI
Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-03-24 19:47:41 +01:00
Víctor Mayoral Vilches 8189f331b6 Update openai library version to avoid version-related issue
In particular:
Traceback (most recent call last):
  File /Users/alias/Alias/research/openai-agents-python/examples/agent_patterns/litellm.py, line 4, in <module>
    from agents import OpenAIChatCompletionsModel,Agent,Runner
  File /Users/alias/Alias/research/openai-agents-python/src/agents/__init__.py, line 44, in <module>
    from .models.openai_chatcompletions import OpenAIChatCompletionsModel
  File /Users/alias/Alias/research/openai-agents-python/src/agents/models/openai_chatcompletions.py, line 57, in <module>
    from openai.types.responses.response_usage import InputTokensDetails, OutputTokensDetails
ImportError: cannot import name 'InputTokensDetails' from 'openai.types.responses.response_usage' (/Users/alias/Alias/research/openai-agents-python/env/lib/python3.12/site-packages/openai/types/responses/response_usage.py). Did you mean: 'OutputTokensDetails'?

Signed-off-by: Víctor Mayoral Vilches <v.mayoralv@gmail.com>
2025-03-24 18:11:38 +01:00
Martín Bravo 65b906a275 feat: add optional dependency for visualization using Graphviz 2025-03-24 09:08:29 +01:00
Rohan Mehta 1ed6e67080 v0.0.6 (voice support) 2025-03-20 13:10:54 -04:00
Rohan Mehta ec66b71419 update tests 2025-03-20 13:08:38 -04:00
Dominik Kundel 20f0793862 feat: add voice pipeline support
> Co-authored-by: rm@openai.com
2025-03-20 09:43:13 -07:00
Rohan Mehta a4cfa67160 v0.0.5 2025-03-19 12:32:36 -04:00
Rohan Mehta e7caa14c30 Fix breaking changes from openai 1.66.2 2025-03-19 12:30:45 -04:00
Martín Bravo cd96e745df refactor: move graphviz dependency to visualization section 2025-03-18 09:53:53 +01:00
Martín Bravo dba2d4bd23 Merge branch 'main' of https://github.com/openai/openai-agents-python into feat/draw_graph 2025-03-17 23:50:52 +01:00
Rohan Mehta 417cdc5efe Pretty print result classes 2025-03-17 11:11:39 -04:00
MartinEBravo 4dc9b6712e Merge branch 'main' of https://github.com/openai/openai-agents-python into feat/draw_graph 2025-03-17 10:17:36 +01:00
Martín Bravo a3d2673368 Add graphviz as a dependency in pyproject.toml 2025-03-13 18:33:08 +01:00
Rohan Mehta 62c12170e9 v0.0.4 2025-03-13 11:18:40 -04:00
Rohan Mehta c7058adab0 pin to openai 1.66.2, update tests 2025-03-11 15:26:06 -07:00
Rohan Mehta 72f443149a 0.0.2 2025-03-11 10:16:04 -07:00
Rohan Mehta aaec57a426 Initial commit 2025-03-11 09:42:28 -07:00