knielsen404
a7a79b31c2
remove timestamp
2025-07-16 16:59:44 +02:00
knielsen404
094fd6e94e
add session id
2025-07-15 11:17:16 +02:00
knielsen404
b8fbb37935
add dataset field to check request
2025-07-15 11:00:36 +02:00
knielsen404
31b128950b
add changes
2025-07-11 17:49:00 +02:00
Hemang
1825f543a7
Remove unnecessary methods.
2025-06-12 14:07:11 +02:00
Hemang
9f564a0401
Refactor the LLM provider routes to move common functionalities into a BaseInstrumentedResponse class and move provier specific implementations in the corresponding BaseProvider implementations.
2025-06-12 14:07:11 +02:00
Hemang
42a9c1cc30
Readability changes.
2025-06-05 11:58:12 +02:00
Hemang
cd6d6a50b0
Small changes related to constants and sorting order of imports.
2025-06-05 10:17:34 +02:00
Hemang
24d47c4585
Update gemini route to include streamGenerateContent in allowed endpoints response.
2025-06-04 15:04:56 +02:00
Hemang
e8106776b4
Refactor stdio implementation to use McpSession class.
2025-06-04 11:46:35 +02:00
Hemang
169eb066b9
Small cleanups in sse and stdio MCP implementation.
2025-05-21 15:58:22 +02:00
Luca Beurer-Kellner
e18c6b5bdb
Add an option to add extra metadata that is pushed and passed to Guardrails during an MCP session ( #47 )
...
* use select() before readline
* support for setting static metadata for MCP sessions
* nest extra mcp metadata in metadata object
* unify session metadata
* extra metadata tests
* use empty object as parameters, if None
* list_tools as tool call
* offset indices in tests
* test: adjust addresses
* mcp: make error reporting configurable
* line logging
* log version
* verbose logging + loud exception failure
* add server and client name to policy get
* append trace even if not pushing
* port tools/list message support to SSE
* use python -m build
* adjust guardrail failure address
* support for blocking tools/list in SSE
* use error-based failure response format by default
* tools/list test
* don't list_tools in stdio connect
* flaky test: handle second possible result in anthropic streaming case
---------
Co-authored-by: knielsen404 <kristian@invariantlabs.ai >
2025-05-19 13:44:37 +02:00
Hemang
e2e004b7b1
Move dockerfiles inside gateway/ and update main CLI script to be able to run build, up, down and logs on a local gateway server instance.
2025-05-14 14:21:48 +05:30
Kristian Nielsen
ed63a4b592
fix typo ( #42 )
2025-04-23 09:28:45 +01:00
Hemang
2c34205c4c
Fix asyncio and threading. Dedupe annotations before pushing. Add README.
2025-04-17 07:15:28 +02:00
Hemang
5bf121bbda
Use pyproject.toml instead of requirements.txt and fix some broken tests.
2025-04-17 07:15:28 +02:00
Luca Beurer-Kellner
6b6f33bde6
fix: don't hardcode preview
2025-04-04 10:38:14 +02:00
Luca Beurer-Kellner
c4dd3f3b19
Allow to specify different API keys for the guardrailing service ( #36 )
...
* minor refactor for getting invariant api keys for guardrailing
* allow different guardrailing api key
* tests
* fix comment + import
* improved unauthorized handling
2025-04-03 12:15:30 +02:00
Luca Beurer-Kellner
e17b53b927
Extract guardrails from header if provided ( #33 )
...
* guardrails from header
* use in-file guardrails in client.py
* support case without request
* remove client script
* tests: guardrailing rule passed in header
* include checked guardrails in annotation extra metadata
* include guardrailing action
* update guardrail metadata
2025-04-03 08:56:15 +02:00
Hemang
760b739894
Correct user API GET call to fetch policies.
2025-04-03 00:02:19 +02:00
Hemang
55f0f741c0
Add tests for guardrails integration with explorer.
2025-04-02 13:40:52 +02:00
Hemang
9aef873a74
Correct header name before calling explorer to fetch guardrails.
2025-04-02 13:40:52 +02:00
Hemang
750c83d3f8
Add calls to execute logging guardrails before pushing to explorer.
2025-04-02 13:40:52 +02:00
Hemang
050ec1ba58
Fetch guardrails from explorer. These have higher precedence than than the guardrails from file.
2025-04-02 13:40:52 +02:00
Luca Beurer-Kellner
7c0bb957fb
Pipelined Guardrails ( #32 )
...
* initial draft: pipelined guardrails
* documentation on stream instrumentation
* more comments
* fix: return earlier
* non-streaming case
* handle non-streaming case
* fix more cases
* simplify request instrumentation
* improve comments
* fix import issues
* extend tests for input guardrailing
* anthropic integration of pipelined and pre-guardrailing
* fix gemini streamed refusal
2025-03-31 14:13:58 +02:00
Hemang
e55df73487
Add guardrails for gemini integration.
2025-03-27 12:26:17 +01:00
Hemang
8191c83c12
Revert "Add debug logs for metadata push."
...
This reverts commit f247682fa3 .
2025-03-26 19:57:00 +01:00
Hemang
f247682fa3
Add debug logs for metadata push.
2025-03-26 15:20:26 +01:00
Hemang
a4ee5f0c47
Add metadata before pushing to Explorer.
2025-03-20 10:59:27 +01:00
Hemang
44988c3535
Add a preload guardrails API call.
2025-03-19 09:39:23 +01:00
Hemang
a5ea86a64e
Show logs if the docker container doesn't start on doing up(). Tested when the guardrails file contained unparseable policy.
2025-03-17 07:45:45 +01:00
Hemang
e773cc9f2d
Add Guardrails verification logic for openai route.
2025-03-17 07:45:45 +01:00
Hemang
38504b3e45
Rename utils.explorer to integrations.explorer. The integrations module will contain API calls.
2025-03-10 15:18:13 +01:00