diff --git a/src/backend/impress/settings.py b/src/backend/impress/settings.py index 8a6fa8e01..e82a5f495 100755 --- a/src/backend/impress/settings.py +++ b/src/backend/impress/settings.py @@ -458,16 +458,25 @@ class Base(Configuration): AI_BASE_URL = values.Value(None, environ_name="AI_BASE_URL", environ_prefix=None) AI_MODEL = values.Value(None, environ_name="AI_MODEL", environ_prefix=None) - AI_DOCUMENT_RATE_THROTTLE_RATES = { - "minute": 5, - "hour": 100, - "day": 500, - } - AI_USER_RATE_THROTTLE_RATES = { - "minute": 3, - "hour": 50, - "day": 200, - } + AI_DOCUMENT_RATE_THROTTLE_RATES = values.DictValue( + { + "minute": 5, + "hour": 100, + "day": 500, + }, + environ_name="AI_DOCUMENT_RATE_THROTTLE_RATES", + environ_prefix=None, + ) + + AI_USER_RATE_THROTTLE_RATES = values.DictValue( + { + "minute": 3, + "hour": 50, + "day": 200, + }, + environ_name="AI_USER_RATE_THROTTLE_RATES", + environ_prefix=None, + ) USER_OIDC_FIELDS_TO_FULLNAME = values.ListValue( default=["first_name", "last_name"], diff --git a/src/helm/env.d/docs-ia/values.impress.yaml.gotmpl b/src/helm/env.d/docs-ia/values.impress.yaml.gotmpl index b1782cdc1..33483c649 100644 --- a/src/helm/env.d/docs-ia/values.impress.yaml.gotmpl +++ b/src/helm/env.d/docs-ia/values.impress.yaml.gotmpl @@ -16,7 +16,9 @@ backend: secretKeyRef: name: backend key: AI_BASE_URL + AI_DOCUMENT_RATE_THROTTLE_RATES: "{'minute': 5000,'hour': 5000,'day': 5000}" AI_MODEL: meta-llama/Meta-Llama-3.1-70B-Instruct + AI_USER_RATE_THROTTLE_RATES: "{'minute': 5000,'hour': 5000,'day': 5000}" DJANGO_CSRF_TRUSTED_ORIGINS: http://docs-ia.beta.numerique.gouv.fr,https://docs-ia.beta.numerique.gouv.fr DJANGO_CONFIGURATION: Production DJANGO_ALLOWED_HOSTS: "*"