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

everyrow/__init__.py

@@ -1,5 +1,12 @@
 from everyrow.api_utils import create_client
+from everyrow.billing import BillingResponse, get_billing_balance
 from everyrow.session import create_session
 from everyrow.task import fetch_task_data
 
-__all__ = ["create_client", "create_session", "fetch_task_data"]
+__all__ = [
+    "BillingResponse",
+    "create_client",
+    "create_session",
+    "fetch_task_data",
+    "get_billing_balance",
+]

everyrow/api_utils.py

@@ -3,6 +3,7 @@ from typing import TypeVar
 
 from everyrow.constants import DEFAULT_EVERYROW_API_URL, EveryrowError
 from everyrow.generated.client import AuthenticatedClient
+from everyrow.generated.models.error_response import ErrorResponse
 from everyrow.generated.models.http_validation_error import HTTPValidationError
 from everyrow.generated.models.insufficient_balance_error import (
     InsufficientBalanceError,
@@ -26,6 +27,7 @@ def create_client() -> AuthenticatedClient:
         base_url=os.environ.get("EVERYROW_API_URL", DEFAULT_EVERYROW_API_URL),
         token=os.environ["EVERYROW_API_KEY"],
         raise_on_unexpected_status=True,
+        follow_redirects=True,
     )
 
 
@@ -33,8 +35,10 @@ T = TypeVar("T")
 
 
 def handle_response[T](
-    response: T | HTTPValidationError | InsufficientBalanceError | None,
+    response: T | ErrorResponse | HTTPValidationError | InsufficientBalanceError | None,
 ) -> T:
+    if isinstance(response, ErrorResponse):
+        raise EveryrowError(response.message)
     if isinstance(response, HTTPValidationError):
         raise EveryrowError(response.detail)
     if isinstance(response, InsufficientBalanceError):

everyrow/billing.py

@@ -0,0 +1,29 @@
+"""Billing utilities for everyrow SDK."""
+
+from dataclasses import dataclass
+
+from everyrow.api_utils import create_client
+from everyrow.generated.api.billing import get_billing_balance_billing_get
+
+
+@dataclass
+class BillingResponse:
+    """Response containing the user's current billing balance."""
+
+    current_balance_dollars: float
+
+
+async def get_billing_balance() -> BillingResponse:
+    """Get the current billing balance for the authenticated user.
+
+    Returns:
+        BillingResponse containing the user's current balance.
+
+    Raises:
+        RuntimeError: If the request fails
+    """
+    client = create_client()
+    response = await get_billing_balance_billing_get.asyncio(client=client)
+    if response is None:
+        raise RuntimeError("Failed to get billing balance")
+    return BillingResponse(current_balance_dollars=response.current_balance_dollars)

everyrow/constants.py

@@ -1,4 +1,4 @@
-DEFAULT_EVERYROW_API_URL = "https://engine.futuresearch.ai"
+DEFAULT_EVERYROW_API_URL = "https://everyrow.io/api/v0"
 
 
 class EveryrowError(Exception): ...

everyrow/generated/__init__.py

@@ -1,4 +1,4 @@
-"""A client library for accessing FastAPI"""
+"""A client library for accessing everyrow API"""
 
 from .client import AuthenticatedClient, Client
 

everyrow/generated/api/{default/interrupt_chat_task_tasks_chat_interrupt_post.py → artifacts/create_artifact_artifacts_post.py}

@@ -5,20 +5,21 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.http_validation_error import HTTPValidationError
-from ...models.submit_chat_task_body import SubmitChatTaskBody
+from ...models.create_artifact_request import CreateArtifactRequest
+from ...models.create_artifact_response import CreateArtifactResponse
+from ...models.error_response import ErrorResponse
 from ...types import Response
 
 
 def _get_kwargs(
     *,
-    body: SubmitChatTaskBody,
+    body: CreateArtifactRequest,
 ) -> dict[str, Any]:
     headers: dict[str, Any] = {}
 
     _kwargs: dict[str, Any] = {
         "method": "post",
-        "url": "/tasks/chat/interrupt",
+        "url": "/artifacts",
     }
 
     _kwargs["json"] = body.to_dict()
@@ -31,13 +32,14 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: AuthenticatedClient | Client, response: httpx.Response
-) -> Any | HTTPValidationError | None:
+) -> CreateArtifactResponse | ErrorResponse | None:
     if response.status_code == 200:
-        response_200 = response.json()
+        response_200 = CreateArtifactResponse.from_dict(response.json())
+
         return response_200
 
     if response.status_code == 422:
-        response_422 = HTTPValidationError.from_dict(response.json())
+        response_422 = ErrorResponse.from_dict(response.json())
 
         return response_422
 
@@ -49,7 +51,7 @@ def _parse_response(
 
 def _build_response(
     *, client: AuthenticatedClient | Client, response: httpx.Response
-) -> Response[Any | HTTPValidationError]:
+) -> Response[CreateArtifactResponse | ErrorResponse]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -61,21 +63,22 @@ def _build_response(
 def sync_detailed(
     *,
     client: AuthenticatedClient,
-    body: SubmitChatTaskBody,
-) -> Response[Any | HTTPValidationError]:
-    """Interrupt Chat Task
+    body: CreateArtifactRequest,
+) -> Response[CreateArtifactResponse | ErrorResponse]:
+    """Create an artifact
 
-     Interrupt the current conversation's running tasks and process a new message.
+     Upload data as an artifact for use in operations. A list of JSON objects creates a table. A single
+    JSON object creates a scalar.
 
     Args:
-        body (SubmitChatTaskBody):
+        body (CreateArtifactRequest):
 
     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[Any | HTTPValidationError]
+        Response[CreateArtifactResponse | ErrorResponse]
     """
 
     kwargs = _get_kwargs(
@@ -92,21 +95,22 @@ def sync_detailed(
 def sync(
     *,
     client: AuthenticatedClient,
-    body: SubmitChatTaskBody,
-) -> Any | HTTPValidationError | None:
-    """Interrupt Chat Task
+    body: CreateArtifactRequest,
+) -> CreateArtifactResponse | ErrorResponse | None:
+    """Create an artifact
 
-     Interrupt the current conversation's running tasks and process a new message.
+     Upload data as an artifact for use in operations. A list of JSON objects creates a table. A single
+    JSON object creates a scalar.
 
     Args:
-        body (SubmitChatTaskBody):
+        body (CreateArtifactRequest):
 
     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:
-        Any | HTTPValidationError
+        CreateArtifactResponse | ErrorResponse
     """
 
     return sync_detailed(
@@ -118,21 +122,22 @@ def sync(
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
-    body: SubmitChatTaskBody,
-) -> Response[Any | HTTPValidationError]:
-    """Interrupt Chat Task
+    body: CreateArtifactRequest,
+) -> Response[CreateArtifactResponse | ErrorResponse]:
+    """Create an artifact
 
-     Interrupt the current conversation's running tasks and process a new message.
+     Upload data as an artifact for use in operations. A list of JSON objects creates a table. A single
+    JSON object creates a scalar.
 
     Args:
-        body (SubmitChatTaskBody):
+        body (CreateArtifactRequest):
 
     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[Any | HTTPValidationError]
+        Response[CreateArtifactResponse | ErrorResponse]
     """
 
     kwargs = _get_kwargs(
@@ -147,21 +152,22 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: AuthenticatedClient,
-    body: SubmitChatTaskBody,
-) -> Any | HTTPValidationError | None:
-    """Interrupt Chat Task
+    body: CreateArtifactRequest,
+) -> CreateArtifactResponse | ErrorResponse | None:
+    """Create an artifact
 
-     Interrupt the current conversation's running tasks and process a new message.
+     Upload data as an artifact for use in operations. A list of JSON objects creates a table. A single
+    JSON object creates a scalar.
 
     Args:
-        body (SubmitChatTaskBody):
+        body (CreateArtifactRequest):
 
     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:
-        Any | HTTPValidationError
+        CreateArtifactResponse | ErrorResponse
     """
 
     return (

everyrow/generated/api/billing/__init__.py

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

everyrow/generated/api/{default/get_queues_stats_jobs_queues_get.py → billing/get_billing_balance_billing_get.py}

@@ -5,22 +5,22 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.queue_stats import QueueStats
+from ...models.billing_response import BillingResponse
 from ...types import Response
 
 
 def _get_kwargs() -> dict[str, Any]:
     _kwargs: dict[str, Any] = {
         "method": "get",
-        "url": "/jobs/queues",
+        "url": "/billing",
     }
 
     return _kwargs
 
 
-def _parse_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> QueueStats | None:
+def _parse_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> BillingResponse | None:
     if response.status_code == 200:
-        response_200 = QueueStats.from_dict(response.json())
+        response_200 = BillingResponse.from_dict(response.json())
 
         return response_200
 
@@ -30,7 +30,7 @@ def _parse_response(*, client: AuthenticatedClient | Client, response: httpx.Res
         return None
 
 
-def _build_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> Response[QueueStats]:
+def _build_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> Response[BillingResponse]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -42,15 +42,17 @@ def _build_response(*, client: AuthenticatedClient | Client, response: httpx.Res
 def sync_detailed(
     *,
     client: AuthenticatedClient,
-) -> Response[QueueStats]:
-    """Get Queues Stats
+) -> Response[BillingResponse]:
+    """Get billing balance
+
+     Get the current billing balance for the authenticated user.
 
     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[QueueStats]
+        Response[BillingResponse]
     """
 
     kwargs = _get_kwargs()
@@ -65,15 +67,17 @@ def sync_detailed(
 def sync(
     *,
     client: AuthenticatedClient,
-) -> QueueStats | None:
-    """Get Queues Stats
+) -> BillingResponse | None:
+    """Get billing balance
+
+     Get the current billing balance for the authenticated user.
 
     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:
-        QueueStats
+        BillingResponse
     """
 
     return sync_detailed(
@@ -84,15 +88,17 @@ def sync(
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
-) -> Response[QueueStats]:
-    """Get Queues Stats
+) -> Response[BillingResponse]:
+    """Get billing balance
+
+     Get the current billing balance for the authenticated user.
 
     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[QueueStats]
+        Response[BillingResponse]
     """
 
     kwargs = _get_kwargs()
@@ -105,15 +111,17 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: AuthenticatedClient,
-) -> QueueStats | None:
-    """Get Queues Stats
+) -> BillingResponse | None:
+    """Get billing balance
+
+     Get the current billing balance for the authenticated user.
 
     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:
-        QueueStats
+        BillingResponse
     """
 
     return (

everyrow/generated/api/operations/__init__.py

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

everyrow/generated/api/{default/re_execute_task_endpoint_tasks_re_execute_post.py → operations/agent_map_operations_agent_map_post.py}

@@ -5,22 +5,22 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.http_validation_error import HTTPValidationError
+from ...models.agent_map_operation import AgentMapOperation
+from ...models.error_response import ErrorResponse
 from ...models.insufficient_balance_error import InsufficientBalanceError
-from ...models.task_id_request import TaskIdRequest
-from ...models.task_response import TaskResponse
+from ...models.operation_response import OperationResponse
 from ...types import Response
 
 
 def _get_kwargs(
     *,
-    body: TaskIdRequest,
+    body: AgentMapOperation,
 ) -> dict[str, Any]:
     headers: dict[str, Any] = {}
 
     _kwargs: dict[str, Any] = {
         "method": "post",
-        "url": "/tasks/re-execute",
+        "url": "/operations/agent-map",
     }
 
     _kwargs["json"] = body.to_dict()
@@ -33,9 +33,9 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: AuthenticatedClient | Client, response: httpx.Response
-) -> HTTPValidationError | InsufficientBalanceError | TaskResponse | None:
+) -> ErrorResponse | InsufficientBalanceError | OperationResponse | None:
     if response.status_code == 200:
-        response_200 = TaskResponse.from_dict(response.json())
+        response_200 = OperationResponse.from_dict(response.json())
 
         return response_200
 
@@ -45,7 +45,7 @@ def _parse_response(
         return response_402
 
     if response.status_code == 422:
-        response_422 = HTTPValidationError.from_dict(response.json())
+        response_422 = ErrorResponse.from_dict(response.json())
 
         return response_422
 
@@ -57,7 +57,7 @@ def _parse_response(
 
 def _build_response(
     *, client: AuthenticatedClient | Client, response: httpx.Response
-) -> Response[HTTPValidationError | InsufficientBalanceError | TaskResponse]:
+) -> Response[ErrorResponse | InsufficientBalanceError | OperationResponse]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -69,24 +69,27 @@ def _build_response(
 def sync_detailed(
     *,
     client: AuthenticatedClient,
-    body: TaskIdRequest,
-) -> Response[HTTPValidationError | InsufficientBalanceError | TaskResponse]:
-    """Re Execute Task Endpoint
+    body: AgentMapOperation,
+) -> Response[ErrorResponse | InsufficientBalanceError | OperationResponse]:
+    """Parallel AI research agents
 
-     Re-execute a task by creating a new task with the same parameters.
+     Run an AI agent on each row in parallel to perform research and generate responses.
 
-    This creates a completely fresh task from scratch. No data or state is
-    carried over from the original task.
+    **Configuration options** (mutually exclusive):
+
+    1. **Use a preset** - set `effort_level` to `low`, `medium`, or `high`
+    2. **Fully customize** - set `effort_level=null` and provide ALL of: `llm`, `iteration_budget`,
+    `include_research`
 
     Args:
-        body (TaskIdRequest):
+        body (AgentMapOperation):
 
     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[HTTPValidationError | InsufficientBalanceError | TaskResponse]
+        Response[ErrorResponse | InsufficientBalanceError | OperationResponse]
     """
 
     kwargs = _get_kwargs(
@@ -103,24 +106,27 @@ def sync_detailed(
 def sync(
     *,
     client: AuthenticatedClient,
-    body: TaskIdRequest,
-) -> HTTPValidationError | InsufficientBalanceError | TaskResponse | None:
-    """Re Execute Task Endpoint
+    body: AgentMapOperation,
+) -> ErrorResponse | InsufficientBalanceError | OperationResponse | None:
+    """Parallel AI research agents
+
+     Run an AI agent on each row in parallel to perform research and generate responses.
 
-     Re-execute a task by creating a new task with the same parameters.
+    **Configuration options** (mutually exclusive):
 
-    This creates a completely fresh task from scratch. No data or state is
-    carried over from the original task.
+    1. **Use a preset** - set `effort_level` to `low`, `medium`, or `high`
+    2. **Fully customize** - set `effort_level=null` and provide ALL of: `llm`, `iteration_budget`,
+    `include_research`
 
     Args:
-        body (TaskIdRequest):
+        body (AgentMapOperation):
 
     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:
-        HTTPValidationError | InsufficientBalanceError | TaskResponse
+        ErrorResponse | InsufficientBalanceError | OperationResponse
     """
 
     return sync_detailed(
@@ -132,24 +138,27 @@ def sync(
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
-    body: TaskIdRequest,
-) -> Response[HTTPValidationError | InsufficientBalanceError | TaskResponse]:
-    """Re Execute Task Endpoint
+    body: AgentMapOperation,
+) -> Response[ErrorResponse | InsufficientBalanceError | OperationResponse]:
+    """Parallel AI research agents
 
-     Re-execute a task by creating a new task with the same parameters.
+     Run an AI agent on each row in parallel to perform research and generate responses.
 
-    This creates a completely fresh task from scratch. No data or state is
-    carried over from the original task.
+    **Configuration options** (mutually exclusive):
+
+    1. **Use a preset** - set `effort_level` to `low`, `medium`, or `high`
+    2. **Fully customize** - set `effort_level=null` and provide ALL of: `llm`, `iteration_budget`,
+    `include_research`
 
     Args:
-        body (TaskIdRequest):
+        body (AgentMapOperation):
 
     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[HTTPValidationError | InsufficientBalanceError | TaskResponse]
+        Response[ErrorResponse | InsufficientBalanceError | OperationResponse]
     """
 
     kwargs = _get_kwargs(
@@ -164,24 +173,27 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: AuthenticatedClient,
-    body: TaskIdRequest,
-) -> HTTPValidationError | InsufficientBalanceError | TaskResponse | None:
-    """Re Execute Task Endpoint
+    body: AgentMapOperation,
+) -> ErrorResponse | InsufficientBalanceError | OperationResponse | None:
+    """Parallel AI research agents
+
+     Run an AI agent on each row in parallel to perform research and generate responses.
 
-     Re-execute a task by creating a new task with the same parameters.
+    **Configuration options** (mutually exclusive):
 
-    This creates a completely fresh task from scratch. No data or state is
-    carried over from the original task.
+    1. **Use a preset** - set `effort_level` to `low`, `medium`, or `high`
+    2. **Fully customize** - set `effort_level=null` and provide ALL of: `llm`, `iteration_budget`,
+    `include_research`
 
     Args:
-        body (TaskIdRequest):
+        body (AgentMapOperation):
 
     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:
-        HTTPValidationError | InsufficientBalanceError | TaskResponse
+        ErrorResponse | InsufficientBalanceError | OperationResponse
     """
 
     return (