mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-06-24 14:19:55 +02:00
28 lines
742 B
Python
28 lines
742 B
Python
from agentic_security.config import get_or_create_config
|
|
from agentic_security.core.app import set_secrets
|
|
|
|
|
|
class InMemorySecrets:
|
|
def __init__(self):
|
|
config = get_or_create_config()
|
|
self.secrets = config.get_config_value("secrets", {})
|
|
set_secrets(self.secrets)
|
|
|
|
def set_secret(self, key: str, value: str):
|
|
self.secrets[key] = value
|
|
|
|
def get_secret(self, key: str) -> str:
|
|
return self.secrets.get(key, None)
|
|
|
|
|
|
# Dependency
|
|
def get_in_memory_secrets() -> InMemorySecrets:
|
|
return InMemorySecrets()
|
|
|
|
|
|
# Example usage in a FastAPI route
|
|
# @app.get("/some-endpoint")
|
|
# async def some_endpoint(secrets: InMemorySecrets = Depends(get_in_memory_secrets)):
|
|
# # Use secrets here
|
|
# pass
|