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

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

@@ -27,12 +27,13 @@
 
 blueprint = Blueprint()
 
+if not get_configuration().provider.disable_downscaling:
 
-@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):
-    await service.scale_down_providers()
+    @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):
+        await service.scale_down_providers()
 
 
 # TODO: Can't use DI here because it's not initialized yet

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"