fix: install api client from git repo

pyproject.toml

@@ -72,9 +72,10 @@ dependencies = [
   "datasets",
   "colorlog>=6.9.0",
   "codegen-sdk-pink>=0.1.0",
-  "dateutil",
+  "codegen-api-client",
 ]
 
+# renovate: datasource=python-version depName=python
 license = { text = "Apache-2.0" }
 classifiers = [
   "Development Status :: 4 - Beta",
@@ -161,11 +162,15 @@ dev-dependencies = [
   "modal>=0.73.25",
   "pytest-lsp>=1.0.0b1",
   "cython>=3.0.11",
+  "codegen-api-client",
 ]
 
 [tool.uv.workspace]
 exclude = ["codegen-examples"]
 
+[tool.uv.sources]
+codegen-api-client = { git = "https://github.com/codegen-sh/codegen-api-client" }
+
 
 [tool.cython-lint]
 max-line-length = 200

src/codegen/agents/agent.py

@@ -1,11 +1,12 @@
 import os
 from typing import Any, Optional
 
-from codegen.agents.client.openapi_client.api.agents_api import AgentsApi
-from codegen.agents.client.openapi_client.api_client import ApiClient
-from codegen.agents.client.openapi_client.configuration import Configuration
-from codegen.agents.client.openapi_client.models.agent_run_response import AgentRunResponse
-from codegen.agents.client.openapi_client.models.create_agent_run_input import CreateAgentRunInput
+from codegen_api_client.api.agents_api import AgentsApi
+from codegen_api_client.api_client import ApiClient
+from codegen_api_client.configuration import Configuration
+from codegen_api_client.models.agent_run_response import AgentRunResponse
+from codegen_api_client.models.create_agent_run_input import CreateAgentRunInput
+
 from codegen.agents.constants import CODEGEN_BASE_API_URL
 
 
@@ -78,7 +79,7 @@
         # Convert API response to dict for Job initialization
 
         job = AgentTask(agent_run_response, self.api_client, self.org_id)
         self.current_job = job
         return job
 
     def get_status(self) -> Optional[dict[str, Any]]: