Added docker support

video_references.py

@@ -1,11 +1,11 @@
 import json
-import os
 import traceback
 from typing import Literal
 
 from pydantic import BaseModel, ConfigDict, Field, ValidationError, model_validator
 
 from agent import Agent, AgentMessage
+from paths import debug_path, prompt_path
 from utils import Timeline
 from windowizer import Window, Windowizer
 
@@ -71,8 +71,10 @@ class VideoReferenceBuilder:
     ) -> None:
         self._agent = agent
         self._windowizer = windowizer
-        with open("prompts/video_references.md", "r", encoding="utf-8") as f:
-            self._system_prompt = AgentMessage(content=f.read(), role="system")
+        self._system_prompt = AgentMessage(
+            content=prompt_path("video_references.md").read_text(encoding="utf-8"),
+            role="system",
+        )
 
     @staticmethod
     def _validate_result(
@@ -113,13 +115,10 @@ class VideoReferenceBuilder:
             ),
         ]
 
-        debug_dir = None
-        if os.path.isdir("debug"):
-            debug_dir = f"debug/VideoReferenceBuilder/{self.DEBUG_ID}"
+        debug_dir = debug_path("VideoReferenceBuilder", self.DEBUG_ID)
+        if debug_dir:
             VideoReferenceBuilder.DEBUG_ID += 1
-            os.makedirs(debug_dir, exist_ok=True)
-            with open(f"{debug_dir}/request.txt", "w", encoding="utf-8") as f:
-                f.write(messages[1].content)
+            (debug_dir / "request.txt").write_text(messages[1].content, encoding="utf-8")
 
         retries_left = self.MAX_RETRIES
         while retries_left > 0:
@@ -129,12 +128,10 @@ class VideoReferenceBuilder:
                 response_format=_WindowResult,
             )
             if debug_dir:
-                with open(
-                    f"{debug_dir}/{retries_left}-retries-left.txt",
-                    "w",
+                (debug_dir / f"{retries_left}-retries-left.txt").write_text(
+                    response,
                     encoding="utf-8",
-                ) as f:
-                    f.write(response)
+                )
 
             try:
                 result = _WindowResult.model_validate_json(response)