everyrow 0.1.10__py3-none-any.whl → 0.2.0__py3-none-any.whl

everyrow/generated/api/operations/single_agent_operations_single_agent_post.py

@@ -0,0 +1,236 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.error_response import ErrorResponse
+from ...models.insufficient_balance_error import InsufficientBalanceError
+from ...models.operation_response import OperationResponse
+from ...models.single_agent_operation import SingleAgentOperation
+from ...types import Response
+
+
+def _get_kwargs(
+    *,
+    body: SingleAgentOperation,
+) -> dict[str, Any]:
+    headers: dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/operations/single-agent",
+    }
+
+    _kwargs["json"] = body.to_dict()
+
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> ErrorResponse | InsufficientBalanceError | OperationResponse | None:
+    if response.status_code == 200:
+        response_200 = OperationResponse.from_dict(response.json())
+
+        return response_200
+
+    if response.status_code == 402:
+        response_402 = InsufficientBalanceError.from_dict(response.json())
+
+        return response_402
+
+    if response.status_code == 422:
+        response_422 = ErrorResponse.from_dict(response.json())
+
+        return response_422
+
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[ErrorResponse | InsufficientBalanceError | OperationResponse]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient,
+    body: SingleAgentOperation,
+) -> Response[ErrorResponse | InsufficientBalanceError | OperationResponse]:
+    """Single AI research agent
+
+     Run a single AI agent to perform research and generate a response.
+
+    **Configuration options** (mutually exclusive):
+
+    1. **Use a preset** - set `effort_level` to one of:
+       - `low`: Fast, minimal research (Gemini Flash, 0 iterations, no provenance)
+       - `medium`: Balanced (Gemini Flash High, 5 iterations, with provenance)
+       - `high`: Thorough research (Claude Opus, 10 iterations, with provenance)
+
+    2. **Fully customize** - set `effort_level=null` and provide ALL of:
+       - `llm`: The LLM model to use
+       - `iteration_budget`: Number of agent iterations (0-20)
+       - `include_research`: Whether to include research notes
+
+    You cannot mix these approaches - either use a preset OR specify all custom parameters.
+
+    Args:
+        body (SingleAgentOperation):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[ErrorResponse | InsufficientBalanceError | OperationResponse]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient,
+    body: SingleAgentOperation,
+) -> ErrorResponse | InsufficientBalanceError | OperationResponse | None:
+    """Single AI research agent
+
+     Run a single AI agent to perform research and generate a response.
+
+    **Configuration options** (mutually exclusive):
+
+    1. **Use a preset** - set `effort_level` to one of:
+       - `low`: Fast, minimal research (Gemini Flash, 0 iterations, no provenance)
+       - `medium`: Balanced (Gemini Flash High, 5 iterations, with provenance)
+       - `high`: Thorough research (Claude Opus, 10 iterations, with provenance)
+
+    2. **Fully customize** - set `effort_level=null` and provide ALL of:
+       - `llm`: The LLM model to use
+       - `iteration_budget`: Number of agent iterations (0-20)
+       - `include_research`: Whether to include research notes
+
+    You cannot mix these approaches - either use a preset OR specify all custom parameters.
+
+    Args:
+        body (SingleAgentOperation):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        ErrorResponse | InsufficientBalanceError | OperationResponse
+    """
+
+    return sync_detailed(
+        client=client,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient,
+    body: SingleAgentOperation,
+) -> Response[ErrorResponse | InsufficientBalanceError | OperationResponse]:
+    """Single AI research agent
+
+     Run a single AI agent to perform research and generate a response.
+
+    **Configuration options** (mutually exclusive):
+
+    1. **Use a preset** - set `effort_level` to one of:
+       - `low`: Fast, minimal research (Gemini Flash, 0 iterations, no provenance)
+       - `medium`: Balanced (Gemini Flash High, 5 iterations, with provenance)
+       - `high`: Thorough research (Claude Opus, 10 iterations, with provenance)
+
+    2. **Fully customize** - set `effort_level=null` and provide ALL of:
+       - `llm`: The LLM model to use
+       - `iteration_budget`: Number of agent iterations (0-20)
+       - `include_research`: Whether to include research notes
+
+    You cannot mix these approaches - either use a preset OR specify all custom parameters.
+
+    Args:
+        body (SingleAgentOperation):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[ErrorResponse | InsufficientBalanceError | OperationResponse]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient,
+    body: SingleAgentOperation,
+) -> ErrorResponse | InsufficientBalanceError | OperationResponse | None:
+    """Single AI research agent
+
+     Run a single AI agent to perform research and generate a response.
+
+    **Configuration options** (mutually exclusive):
+
+    1. **Use a preset** - set `effort_level` to one of:
+       - `low`: Fast, minimal research (Gemini Flash, 0 iterations, no provenance)
+       - `medium`: Balanced (Gemini Flash High, 5 iterations, with provenance)
+       - `high`: Thorough research (Claude Opus, 10 iterations, with provenance)
+
+    2. **Fully customize** - set `effort_level=null` and provide ALL of:
+       - `llm`: The LLM model to use
+       - `iteration_budget`: Number of agent iterations (0-20)
+       - `include_research`: Whether to include research notes
+
+    You cannot mix these approaches - either use a preset OR specify all custom parameters.
+
+    Args:
+        body (SingleAgentOperation):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        ErrorResponse | InsufficientBalanceError | OperationResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+            body=body,
+        )
+    ).parsed

everyrow/generated/api/sessions/__init__.py

@@ -0,0 +1 @@
+"""Contains endpoint functions for accessing the API"""

everyrow/generated/api/{default/copy_artifacts_artifacts_copy_post.py → sessions/create_session_endpoint_sessions_post.py}

@@ -5,21 +5,21 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.copy_artifacts_request import CopyArtifactsRequest
-from ...models.copy_artifacts_response import CopyArtifactsResponse
+from ...models.create_session import CreateSession
 from ...models.http_validation_error import HTTPValidationError
+from ...models.session_response import SessionResponse
 from ...types import Response
 
 
 def _get_kwargs(
     *,
-    body: CopyArtifactsRequest,
+    body: CreateSession,
 ) -> dict[str, Any]:
     headers: dict[str, Any] = {}
 
     _kwargs: dict[str, Any] = {
         "method": "post",
-        "url": "/artifacts/copy",
+        "url": "/sessions",
     }
 
     _kwargs["json"] = body.to_dict()
@@ -32,9 +32,9 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: AuthenticatedClient | Client, response: httpx.Response
-) -> CopyArtifactsResponse | HTTPValidationError | None:
+) -> HTTPValidationError | SessionResponse | None:
     if response.status_code == 200:
-        response_200 = CopyArtifactsResponse.from_dict(response.json())
+        response_200 = SessionResponse.from_dict(response.json())
 
         return response_200
 
@@ -51,7 +51,7 @@ def _parse_response(
 
 def _build_response(
     *, client: AuthenticatedClient | Client, response: httpx.Response
-) -> Response[CopyArtifactsResponse | HTTPValidationError]:
+) -> Response[HTTPValidationError | SessionResponse]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -63,19 +63,21 @@ def _build_response(
 def sync_detailed(
     *,
     client: AuthenticatedClient,
-    body: CopyArtifactsRequest,
-) -> Response[CopyArtifactsResponse | HTTPValidationError]:
-    """Copy Artifacts
+    body: CreateSession,
+) -> Response[HTTPValidationError | SessionResponse]:
+    """Create a session
+
+     Create a new session for organizing tasks.
 
     Args:
-        body (CopyArtifactsRequest):
+        body (CreateSession):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[CopyArtifactsResponse | HTTPValidationError]
+        Response[HTTPValidationError | SessionResponse]
     """
 
     kwargs = _get_kwargs(
@@ -92,19 +94,21 @@ def sync_detailed(
 def sync(
     *,
     client: AuthenticatedClient,
-    body: CopyArtifactsRequest,
-) -> CopyArtifactsResponse | HTTPValidationError | None:
-    """Copy Artifacts
+    body: CreateSession,
+) -> HTTPValidationError | SessionResponse | None:
+    """Create a session
+
+     Create a new session for organizing tasks.
 
     Args:
-        body (CopyArtifactsRequest):
+        body (CreateSession):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        CopyArtifactsResponse | HTTPValidationError
+        HTTPValidationError | SessionResponse
     """
 
     return sync_detailed(
@@ -116,19 +120,21 @@ def sync(
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
-    body: CopyArtifactsRequest,
-) -> Response[CopyArtifactsResponse | HTTPValidationError]:
-    """Copy Artifacts
+    body: CreateSession,
+) -> Response[HTTPValidationError | SessionResponse]:
+    """Create a session
+
+     Create a new session for organizing tasks.
 
     Args:
-        body (CopyArtifactsRequest):
+        body (CreateSession):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[CopyArtifactsResponse | HTTPValidationError]
+        Response[HTTPValidationError | SessionResponse]
     """
 
     kwargs = _get_kwargs(
@@ -143,19 +149,21 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: AuthenticatedClient,
-    body: CopyArtifactsRequest,
-) -> CopyArtifactsResponse | HTTPValidationError | None:
-    """Copy Artifacts
+    body: CreateSession,
+) -> HTTPValidationError | SessionResponse | None:
+    """Create a session
+
+     Create a new session for organizing tasks.
 
     Args:
-        body (CopyArtifactsRequest):
+        body (CreateSession):
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        CopyArtifactsResponse | HTTPValidationError
+        HTTPValidationError | SessionResponse
     """
 
     return (

everyrow/generated/api/tasks/__init__.py

@@ -0,0 +1 @@
+"""Contains endpoint functions for accessing the API"""

everyrow/generated/api/{default/get_job_progress_for_task_jobs_progress_get.py → tasks/get_task_result_tasks_task_id_result_get.py}

@@ -1,31 +1,26 @@
 from http import HTTPStatus
 from typing import Any
+from urllib.parse import quote
 from uuid import UUID
 
 import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
+from ...models.error_response import ErrorResponse
 from ...models.http_validation_error import HTTPValidationError
-from ...models.progress_status import ProgressStatus
-from ...types import UNSET, Response
+from ...models.task_result_response import TaskResultResponse
+from ...types import Response
 
 
 def _get_kwargs(
-    *,
     task_id: UUID,
 ) -> dict[str, Any]:
-    params: dict[str, Any] = {}
-
-    json_task_id = str(task_id)
-    params["task_id"] = json_task_id
-
-    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
-
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": "/jobs/progress",
-        "params": params,
+        "url": "/tasks/{task_id}/result".format(
+            task_id=quote(str(task_id), safe=""),
+        ),
     }
 
     return _kwargs
@@ -33,12 +28,17 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: AuthenticatedClient | Client, response: httpx.Response
-) -> HTTPValidationError | ProgressStatus | None:
+) -> ErrorResponse | HTTPValidationError | TaskResultResponse | None:
     if response.status_code == 200:
-        response_200 = ProgressStatus.from_dict(response.json())
+        response_200 = TaskResultResponse.from_dict(response.json())
 
         return response_200
 
+    if response.status_code == 404:
+        response_404 = ErrorResponse.from_dict(response.json())
+
+        return response_404
+
     if response.status_code == 422:
         response_422 = HTTPValidationError.from_dict(response.json())
 
@@ -52,7 +52,7 @@ def _parse_response(
 
 def _build_response(
     *, client: AuthenticatedClient | Client, response: httpx.Response
-) -> Response[HTTPValidationError | ProgressStatus]:
+) -> Response[ErrorResponse | HTTPValidationError | TaskResultResponse]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -62,11 +62,14 @@ def _build_response(
 
 
 def sync_detailed(
+    task_id: UUID,
     *,
     client: AuthenticatedClient,
-    task_id: UUID,
-) -> Response[HTTPValidationError | ProgressStatus]:
-    """Get Job Progress For Task
+) -> Response[ErrorResponse | HTTPValidationError | TaskResultResponse]:
+    """Get task result data
+
+     Get the result data of a completed task. Returns the artifact data as a list of records (for tables)
+    or a single record (for scalars).
 
     Args:
         task_id (UUID):
@@ -76,7 +79,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[HTTPValidationError | ProgressStatus]
+        Response[ErrorResponse | HTTPValidationError | TaskResultResponse]
     """
 
     kwargs = _get_kwargs(
@@ -91,11 +94,14 @@ def sync_detailed(
 
 
 def sync(
+    task_id: UUID,
     *,
     client: AuthenticatedClient,
-    task_id: UUID,
-) -> HTTPValidationError | ProgressStatus | None:
-    """Get Job Progress For Task
+) -> ErrorResponse | HTTPValidationError | TaskResultResponse | None:
+    """Get task result data
+
+     Get the result data of a completed task. Returns the artifact data as a list of records (for tables)
+    or a single record (for scalars).
 
     Args:
         task_id (UUID):
@@ -105,21 +111,24 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        HTTPValidationError | ProgressStatus
+        ErrorResponse | HTTPValidationError | TaskResultResponse
     """
 
     return sync_detailed(
-        client=client,
         task_id=task_id,
+        client=client,
     ).parsed
 
 
 async def asyncio_detailed(
+    task_id: UUID,
     *,
     client: AuthenticatedClient,
-    task_id: UUID,
-) -> Response[HTTPValidationError | ProgressStatus]:
-    """Get Job Progress For Task
+) -> Response[ErrorResponse | HTTPValidationError | TaskResultResponse]:
+    """Get task result data
+
+     Get the result data of a completed task. Returns the artifact data as a list of records (for tables)
+    or a single record (for scalars).
 
     Args:
         task_id (UUID):
@@ -129,7 +138,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[HTTPValidationError | ProgressStatus]
+        Response[ErrorResponse | HTTPValidationError | TaskResultResponse]
     """
 
     kwargs = _get_kwargs(
@@ -142,11 +151,14 @@ async def asyncio_detailed(
 
 
 async def asyncio(
+    task_id: UUID,
     *,
     client: AuthenticatedClient,
-    task_id: UUID,
-) -> HTTPValidationError | ProgressStatus | None:
-    """Get Job Progress For Task
+) -> ErrorResponse | HTTPValidationError | TaskResultResponse | None:
+    """Get task result data
+
+     Get the result data of a completed task. Returns the artifact data as a list of records (for tables)
+    or a single record (for scalars).
 
     Args:
         task_id (UUID):
@@ -156,12 +168,12 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        HTTPValidationError | ProgressStatus
+        ErrorResponse | HTTPValidationError | TaskResultResponse
     """
 
     return (
         await asyncio_detailed(
-            client=client,
             task_id=task_id,
+            client=client,
         )
     ).parsed