clean up

Author: arsenyinfo

agent/compiler/core.py

@@ -1,4 +1,5 @@
 from typing import TypedDict, overload
+import os
 import time
 import uuid
 import shlex
@@ -99,7 +100,7 @@ def exec_demux(container: docker.models.containers.Container, command: list[str]
         exit_code, (stdout, stderr) = container.exec_run(
             command,
             demux=True,
-            environment={"NO_COLOR": "1", "FORCE_COLOR": "0", "APP_DATABASE_URL": "postgres://postgres:postgres@postgres:5432/postgres"},
+            environment={"NO_COLOR": "1", "FORCE_COLOR": "0", **dict(os.environ)},
         )
         return CompileResult(
             exit_code=exit_code,

agent/policies/drizzle.py

@@ -118,15 +118,11 @@ def run(input: list[MessageParam], *args, init: bool = False, **kwargs) -> Drizz
 
     @property
     def is_successful(self) -> bool:
-        res = (
+        return (
             not isinstance(self.data.output, Exception)
             and self.data.output.feedback["exit_code"] == 0
             and self.data.output.feedback["stderr"] is None
         )
-        if not res:
-            print(f"DrizzleTaskNode failed: {self.data.output} {self.data.output.feedback}")
-        # FixMe: clean up after debugging
-        return res
 
     @staticmethod
     @contextmanager