add session id

This commit is contained in:
knielsen404
2025-07-15 11:17:16 +02:00
parent b8fbb37935
commit 094fd6e94e
+10
View File
@@ -17,12 +17,21 @@ from gateway.common.authorization import (
)
from gateway.common.guardrails import Guardrail
import uuid
# Timestamps of last API calls per guardrails string
_guardrails_cache = {}
# Locks per guardrails string
_guardrails_locks = {}
# Temporary session ID generation
def generate_session_id():
return str(uuid.uuid4())
session_id = generate_session_id()
def rate_limit(expiration_time: int = 3600):
"""
Decorator to limit API calls to once per expiration_time seconds
@@ -153,6 +162,7 @@ async def check_guardrails(
headers={
"Authorization": context.get_guardrailing_authorization(),
"Accept": CONTENT_TYPE_JSON,
"X-Session-Id": session_id,
},
timeout=5,
)