fix: install api client from git repo

Author: rushilpatel0

pyproject.toml

@@ -43,7 +43,7 @@ dependencies = [
   "hatch-vcs>=0.4.0",
   "hatchling>=1.25.0",
   "pyinstrument>=5.0.0",
-  "pip>=24.3.1",                       # This is needed for some NPM/YARN/PNPM post-install scripts to work!
+  "pip>=24.3.1", # This is needed for some NPM/YARN/PNPM post-install scripts to work!
   "rich-click>=1.8.5",
   "python-dotenv>=1.0.1",
   "giturlparse",
@@ -72,7 +72,11 @@ dependencies = [
   "datasets",
   "colorlog>=6.9.0",
   "codegen-sdk-pink>=0.1.0",
+<<<<<<< Updated upstream
   "dateutil",
+=======
+  "codegen-api-client",
+>>>>>>> Stashed changes
 ]
 
 license = { text = "Apache-2.0" }
@@ -161,11 +165,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.git" }
+
 
 [tool.cython-lint]
 max-line-length = 200

src/codegen/agents/agent.py

@@ -1,11 +1,10 @@
 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