feat(server): add global togle to disable managed provider downscaling (#1717)

Signed-off-by: Radek Ježek <[email protected]>

Author: jezekra1

apps/agentstack-server/src/agentstack_server/configuration.py

@@ -231,6 +231,7 @@ class DockerConfigJson(BaseModel):
 
 
 class ManagedProviderConfiguration(BaseModel):
+    disable_downscaling: bool = False
     manifest_template_dir: Path | None = None
     self_registration_use_local_network: bool = Field(
         default=False,

apps/agentstack-server/src/agentstack_server/jobs/crons/provider.py

@@ -31,7 +31,9 @@
 @blueprint.periodic(cron="*/1 * * * *")
 @blueprint.task(queueing_lock="scale_down_providers", queue=str(Queues.CRON_PROVIDER))
 @inject
-async def scale_down_providers(timestamp: int, service: ProviderService):
+async def scale_down_providers(timestamp: int, service: ProviderService, configuration: Configuration):
+    if configuration.provider.disable_downscaling:
+        return
     await service.scale_down_providers()
 
 

helm/templates/deployment.yaml

@@ -273,8 +273,8 @@ spec:
             - name: AUTH__DISABLE_AUTH
               value: "true"
             {{- end }}
-            {{- if .Values.features.selfRegistration }}
-            - name: PROVIDER__AUTO_REMOVE_ENABLED
+            {{- if .Values.disableProviderDownscaling }}
+            - name: PROVIDER__DISABLE_DOWNSCALING
               value: "true"
             {{- end }}
             - name: GENERATE_CONVERSATION_TITLE__ENABLED

helm/values.yaml

@@ -83,6 +83,8 @@ providers: []
 unmanagedProviders: [] # DEPRECATED: use providers instead
 variables: {} # DEPRECATED: use server API to manage variables instead
 
+disableProviderDownscaling: false
+
 # External registries in the format: [name: githubURL]
 # for example
 # github: "https://github.com/i-am-bee/[email protected]#path=agent-registry.yaml"