fix: update schema

Author: rushilpatel0

src/codegen/agents/README.md

@@ -0,0 +1,120 @@
+# Codegen Agents - Python SDK
+
+This module provides a Python client for interacting with the Codegen AI agents API.
+
+## Installation
+
+The Codegen Agent SDK is included as part of the Codegen package. Ensure you have the latest version installed:
+
+```bash
+pip install codegen
+```
+
+## Usage
+
+### Basic Example
+
+```python
+from codegen.agents.agent import Agent
+
+# Initialize the Agent with your organization ID and API token
+agent = Agent(
+    org_id="11",  # Your organization ID
+    token="your_api_token_here",  # Your API authentication token
+    base_url="https://codegen-sh-rest-api.modal.run"  # Optional - defaults to this URL
+)
+
+# Run an agent with a prompt
+task = agent.run(prompt="Which github repos can you currently access?")
+
+# Check the initial status
+print(task.status)  # Returns the current status of the task (e.g., "queued", "in_progress", etc.)
+
+# Refresh the task to get updated status
+task.refresh()
+
+# Check the updated status
+print(task.status)
+
+# Once task is complete, you can access the result
+if task.status == "completed":
+    print(task.result)
+```
+
+### Agent Class
+
+The `Agent` class is the main entry point for interacting with Codegen AI agents:
+
+```python
+Agent(token: str, org_id: Optional[int] = None, base_url: Optional[str] = CODEGEN_BASE_API_URL)
+```
+
+Parameters:
+- `token` (required): Your API authentication token
+- `org_id` (optional): Your organization ID. If not provided, defaults to environment variable `CODEGEN_ORG_ID` or "1"
+- `base_url` (optional): API base URL. Defaults to "https://codegen-sh-rest-api.modal.run"
+
+### Methods
+
+#### run()
+
+```python
+run(prompt: str) -> AgentTask
+```
+
+Runs an agent with the given prompt.
+
+Parameters:
+- `prompt` (required): The instruction for the agent to execute
+
+Returns:
+- An `AgentTask` object representing the running task
+
+#### get_status()
+
+```python
+get_status() -> Optional[Dict[str, Any]]
+```
+
+Gets the status of the current task.
+
+Returns:
+- A dictionary containing task status information (`id`, `status`, `result`), or `None` if no task has been run
+
+### AgentTask Class
+
+The `AgentTask` class represents a running or completed agent task:
+
+#### Attributes
+
+- `id`: The unique identifier for the task
+- `org_id`: The organization ID
+- `status`: Current status of the task (e.g., "queued", "in_progress", "completed", "failed")
+- `result`: The task result (available when status is "completed")
+
+#### Methods
+
+##### refresh()
+
+```python
+refresh() -> None
+```
+
+Refreshes the task status from the API.
+
+## Environment Variables
+
+- `CODEGEN_ORG_ID`: Default organization ID (used if `org_id` is not provided)
+
+## Error Handling
+
+Handle potential API errors using standard try/except blocks:
+
+```python
+try:
+    task = agent.run(prompt="Your prompt here")
+    task.refresh()
+    print(task.status)
+except Exception as e:
+    print(f"Error: {e}")
+```

src/codegen/agents/agent.py

@@ -17,9 +17,10 @@ def __init__(self, task_data: AgentRunResponse, api_client: ApiClient, org_id: i
         self.org_id = org_id
         self.status = task_data.status
         self.result = task_data.result
+        self.web_url = task_data.web_url
         self._api_client = api_client
         self._agents_api = AgentsApi(api_client)
-    
+        
     def refresh(self) -> None:
         """Refresh the job status from the API."""
         if self.id is None:
@@ -81,7 +82,6 @@ def run(self, prompt: str) -> AgentTask:
             authorization=f"Bearer {self.token}",
             _headers={"Content-Type": "application/json"}
         )
-        
         # Convert API response to dict for Job initialization
 
         job = AgentTask(agent_run_response, self.api_client, self.org_id)
@@ -101,6 +101,7 @@ def get_status(self) -> Optional[Dict[str, Any]]:
             return {
                 "id": self.current_job.id,
                 "status": self.current_job.status,
-                "result": self.current_job.result
+                "result": self.current_job.result,
+                "web_url": self.current_job.web_url
             }
         return None

src/codegen/agents/client/openapi_client/models/agent_run_response.py

@@ -31,7 +31,8 @@ class AgentRunResponse(BaseModel):
     status: Optional[StrictStr] = None
     created_at: Optional[StrictStr] = None
     result: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["id", "organization_id", "status", "created_at", "result"]
+    web_url: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["id", "organization_id", "status", "created_at", "result", "web_url"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -103,7 +104,8 @@ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
             "organization_id": obj.get("organization_id"),
             "status": obj.get("status"),
             "created_at": obj.get("created_at"),
-            "result": obj.get("result")
+            "result": obj.get("result"),
+            "web_url": obj.get("web_url")
         })
         return _obj