From 2a97d6fdee8887173de10e5d1bfeb3fc559b1d80 Mon Sep 17 00:00:00 2001 From: "codegen-sh[bot]" <131295404+codegen-sh[bot]@users.noreply.github.com> Date: Mon, 14 Apr 2025 14:14:38 +0000 Subject: [PATCH 1/3] CG-17302: Add job_url property to Agent class --- src/codegen/agents/agent.py | 11 +++++++++++ tests/unit/codegen/agents/test_agent.py | 18 ++++++++++++++++++ 2 files changed, 29 insertions(+) diff --git a/src/codegen/agents/agent.py b/src/codegen/agents/agent.py index 3f50bb87f..83409a9db 100644 --- a/src/codegen/agents/agent.py +++ b/src/codegen/agents/agent.py @@ -62,6 +62,17 @@ def __init__(self, token: str, org_id: Optional[int] = None, base_url: Optional[ # Current job self.current_job = None + @property + def job_url(self) -> Optional[str]: + """Get the URL for the current job. + + Returns: + str: The web URL for the current job, or None if no job has been run. + """ + if self.current_job: + return self.current_job.web_url + return None + def run(self, prompt: str) -> AgentTask: """Run an agent with the given prompt. diff --git a/tests/unit/codegen/agents/test_agent.py b/tests/unit/codegen/agents/test_agent.py index a41a9fea5..43f919405 100644 --- a/tests/unit/codegen/agents/test_agent.py +++ b/tests/unit/codegen/agents/test_agent.py @@ -146,6 +146,21 @@ def test_init_with_custom_base_url(self, mock_api_client): agent = Agent(token="test-token", org_id=42, base_url=custom_url) mock_config.assert_called_once_with(host=custom_url, access_token="test-token") + def test_job_url_with_no_job(self, agent): + """Test job_url property when no job has been run""" + assert agent.job_url is None + + def test_job_url_with_job(self, agent): + """Test job_url property returns current job's web_url""" + # Setup mock job + mock_job = MagicMock(spec=AgentTask) + mock_job.web_url = "https://example.com/run/456" + + agent.current_job = mock_job + + # Check job_url + assert agent.job_url == "https://example.com/run/456" + def test_run(self, agent, mock_agents_api): """Test run method creates and returns job""" # Setup mock API response @@ -250,6 +265,9 @@ def test_full_workflow(self, mock_response, mock_updated_response): assert job.status == "running" assert job.result is None + # Verify job_url property + assert agent.job_url == "https://example.com/run/987" + # Check status status = agent.get_status() From 8cdcc97fcd1df4fe85dbb7e14164cc44ab94c507 Mon Sep 17 00:00:00 2001 From: "codegen-sh[bot]" <131295404+codegen-sh[bot]@users.noreply.github.com> Date: Mon, 14 Apr 2025 14:15:45 +0000 Subject: [PATCH 2/3] Automated pre-commit update --- tests/unit/codegen/agents/test_agent.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/codegen/agents/test_agent.py b/tests/unit/codegen/agents/test_agent.py index 43f919405..3178bea2b 100644 --- a/tests/unit/codegen/agents/test_agent.py +++ b/tests/unit/codegen/agents/test_agent.py @@ -155,9 +155,9 @@ def test_job_url_with_job(self, agent): # Setup mock job mock_job = MagicMock(spec=AgentTask) mock_job.web_url = "https://example.com/run/456" - + agent.current_job = mock_job - + # Check job_url assert agent.job_url == "https://example.com/run/456" From af52c58c6e06c1b48b890f51464de514059f03f7 Mon Sep 17 00:00:00 2001 From: "codegen-sh[bot]" <131295404+codegen-sh[bot]@users.noreply.github.com> Date: Mon, 14 Apr 2025 14:19:20 +0000 Subject: [PATCH 3/3] CG-17302: Rename job_url property to web_url --- src/codegen/agents/agent.py | 2 +- tests/unit/codegen/agents/test_agent.py | 18 +++++++++--------- 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/src/codegen/agents/agent.py b/src/codegen/agents/agent.py index 83409a9db..2bbee8e87 100644 --- a/src/codegen/agents/agent.py +++ b/src/codegen/agents/agent.py @@ -63,7 +63,7 @@ def __init__(self, token: str, org_id: Optional[int] = None, base_url: Optional[ self.current_job = None @property - def job_url(self) -> Optional[str]: + def web_url(self) -> Optional[str]: """Get the URL for the current job. Returns: diff --git a/tests/unit/codegen/agents/test_agent.py b/tests/unit/codegen/agents/test_agent.py index 3178bea2b..ae3a008e3 100644 --- a/tests/unit/codegen/agents/test_agent.py +++ b/tests/unit/codegen/agents/test_agent.py @@ -146,20 +146,20 @@ def test_init_with_custom_base_url(self, mock_api_client): agent = Agent(token="test-token", org_id=42, base_url=custom_url) mock_config.assert_called_once_with(host=custom_url, access_token="test-token") - def test_job_url_with_no_job(self, agent): - """Test job_url property when no job has been run""" - assert agent.job_url is None + def test_web_url_with_no_job(self, agent): + """Test web_url property when no job has been run""" + assert agent.web_url is None - def test_job_url_with_job(self, agent): - """Test job_url property returns current job's web_url""" + def test_web_url_with_job(self, agent): + """Test web_url property returns current job's web_url""" # Setup mock job mock_job = MagicMock(spec=AgentTask) mock_job.web_url = "https://example.com/run/456" agent.current_job = mock_job - # Check job_url - assert agent.job_url == "https://example.com/run/456" + # Check web_url + assert agent.web_url == "https://example.com/run/456" def test_run(self, agent, mock_agents_api): """Test run method creates and returns job""" @@ -265,8 +265,8 @@ def test_full_workflow(self, mock_response, mock_updated_response): assert job.status == "running" assert job.result is None - # Verify job_url property - assert agent.job_url == "https://example.com/run/987" + # Verify web_url property + assert agent.web_url == "https://example.com/run/987" # Check status status = agent.get_status()