fix: single component run functionality (#102)

BasedLabs · Nov 19, 2024 · d0a7587 · d0a7587
1 parent 7d9d4aa
commit d0a7587
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/nolabs/infrastructure/.env.template b/nolabs/infrastructure/.env.template
@@ -15,7 +15,8 @@ NOLABS_REDIS_URL=redis://localhost:6379/0
 NOLABS_CELERY_EAGER=False
 
 # LOGGING
-NOLABS_LOGGING_LEVEL=ERROR
+NOLABS_LOGGING_LEVEL=INFO
+NOLABS_STRUCTURED_LOGGING=True
 
 # UVICORN
 UVICORN_HOST=0.0.0.0

diff --git a/nolabs/workflow/application/use_cases.py b/nolabs/workflow/application/use_cases.py
@@ -353,7 +353,7 @@ async def handle(self, request: StartWorkflowComponentRequest):
             graph = Graph(experiment_id=request.experiment_id)
 
             components = []
-            for component_data in ComponentData.objects(experiment_id=request.experiment_id):
+            for component_data in ComponentData.objects(experiment=request.experiment_id):
                 component = Component.restore(component_data)
                 components.append(component)