vasiuuu/DGX_AI

1from __future__ import annotations2 3from typing import TYPE_CHECKING4 5from codeforge.models import CodeForgeObservation6 7if TYPE_CHECKING:8    from codeforge.tasks import Task9 10 11def build_observation(12    *,13    episode_id: str,14    task: Task,15    current_files: dict[str, str],16    budget_remaining: int,17    previous_score: float,18    last_citations: tuple[dict[str, object], ...] = (),19    last_grounding: dict[str, object] | None = None,20    is_done: bool = False,21    last_reward: float = 0.0,22    last_cluster_hits: tuple[str, ...] = (),23    last_interrogation_questions: tuple[str, ...] = (),24    last_ralph_run_id: str | None = None,25    last_ralph_iterations: tuple[dict[str, object], ...] = (),26    cumulative_audit_summary: dict[str, object] | None = None,27    error: str | None = None,28) -> CodeForgeObservation:29    return CodeForgeObservation(30        episode_id=episode_id,31        task_id=task.task_id,32        task_level=task.task_level,33        task_brief=task.brief,34        initial_files=dict(task.initial_files),35        current_files=dict(current_files),36        budget_remaining=budget_remaining,37        previous_score=previous_score,38        last_citations=last_citations,39        last_grounding=last_grounding,40        is_done=is_done,41        last_reward=last_reward,42        last_cluster_hits=last_cluster_hits,43        last_interrogation_questions=last_interrogation_questions,44        last_ralph_run_id=last_ralph_run_id,45        last_ralph_iterations=last_ralph_iterations,46        cumulative_audit_summary=cumulative_audit_summary or {},47        error=error,48        reward=last_reward,49        done=is_done,50    )51