ローカルLLMにワークフローを作らせる

autonomous_windmill/job_poller.py

@@ -0,0 +1,48 @@
+"""ジョブ完了待ちポーリング。Windmill の success フィールドで判定する。"""
+import time
+from windmill_client import get_job, get_job_logs
+from config import POLL_INTERVAL, POLL_MAX_COUNT
+
+
+class JobTimeout(Exception):
+    pass
+
+
+def poll_until_done(job_id: str) -> tuple[bool, str]:
+    """
+    ジョブが完了するまでポーリングする。
+
+    判定優先順位:
+      1. success is False → 失敗（即返却）
+      2. success is True  → 成功（即返却）
+      3. それ以外          → 継続待機
+
+    ログ文字列は主判定に使わない（誤検知防止）。
+
+    Returns:
+        (success: bool, logs: str)
+    Raises:
+        JobTimeout: POLL_MAX_COUNT * POLL_INTERVAL 秒以内に完了しなかった場合
+    """
+    for _ in range(POLL_MAX_COUNT):
+        job = get_job(job_id)
+        success = job.get("success")
+
+        if success is False:
+            logs = get_job_logs(job_id)
+            # result.error があればログに付加（ログが空でもエラー詳細を取得できる）
+            result_error = job.get("result", {}) or {}
+            error_detail = result_error.get("error", {}) or {}
+            error_msg = error_detail.get("message", "")
+            if error_msg and error_msg not in (logs or ""):
+                logs = f"{logs}\n[result.error] {error_msg}".strip()
+            return False, logs
+
+        if success is True:
+            logs = get_job_logs(job_id)
+            return True, logs
+
+        time.sleep(POLL_INTERVAL)
+
+    timeout_sec = POLL_MAX_COUNT * POLL_INTERVAL
+    raise JobTimeout(f"ジョブ {job_id} が {timeout_sec} 秒以内に完了しませんでした")