#!/usr/bin/env python3

import json

import sys

from pathlib import Path

from unittest.mock import patch

from minisweagent.models.test_models import DeterministicModel

from minisweagent.run.extra.github_issue import DEFAULT_CONFIG, main

def update_trajectory():

    traj_path = Path(__file__).parent / "github_issue.traj.json"

    trajectory = json.loads(traj_path.read_text())

    model_responses = [msg["content"] for msg in trajectory[2:] if msg["role"] == "assistant"]

    with patch("minisweagent.run.extra.github_issue.get_model") as mock_get_model:

        mock_get_model.return_value = DeterministicModel(outputs=model_responses)

        github_url = "https://github.com/SWE-agent/test-repo/issues/1"

        agent = main(issue_url=github_url, model="tardis", config=DEFAULT_CONFIG)

    traj_path.write_text(json.dumps(agent.messages, indent=2))

if __name__ == "__main__":

    update_trajectory()