Regenerate SDK code based on OpenAPI spec (#5141)

Sourced from commit 537b4d397425b874666c1ffad335ec17ca24851e

Author: rgambee

src/futuresearch/generated/__init__.py

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

src/futuresearch/generated/api/billing/get_subscription_status_billing_subscription_get.py

@@ -0,0 +1,135 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.subscription_status_response import SubscriptionStatusResponse
+from ...types import Response
+
+
+def _get_kwargs() -> dict[str, Any]:
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": "/billing/subscription",
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> SubscriptionStatusResponse | None:
+    if response.status_code == 200:
+        response_200 = SubscriptionStatusResponse.from_dict(response.json())
+
+        return response_200
+
+    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[SubscriptionStatusResponse]:
+    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,
+) -> Response[SubscriptionStatusResponse]:
+    """Get subscription status
+
+     Get the current subscription tier and details 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[SubscriptionStatusResponse]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient,
+) -> SubscriptionStatusResponse | None:
+    """Get subscription status
+
+     Get the current subscription tier and details 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:
+        SubscriptionStatusResponse
+    """
+
+    return sync_detailed(
+        client=client,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient,
+) -> Response[SubscriptionStatusResponse]:
+    """Get subscription status
+
+     Get the current subscription tier and details 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[SubscriptionStatusResponse]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient,
+) -> SubscriptionStatusResponse | None:
+    """Get subscription status
+
+     Get the current subscription tier and details 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:
+        SubscriptionStatusResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+        )
+    ).parsed

src/futuresearch/generated/api/operations/forecast_operations_forecast_post.py

@@ -75,9 +75,10 @@ def sync_detailed(
     body: ForecastOperation,
     x_cohort_source: None | str | Unset = UNSET,
 ) -> Response[ErrorResponse | InsufficientBalanceResponse | OperationResponse]:
-    """AI-powered probability forecast
+    """AI-powered forecast
 
-     Run 6 parallel research agents per row, then synthesize into a probability forecast with rationale.
+     Run 6 parallel research agents per row, then synthesize into a forecast. Supports binary
+    (probability) and numeric (percentile) modes.
 
     Args:
         x_cohort_source (None | str | Unset):
@@ -109,9 +110,10 @@ def sync(
     body: ForecastOperation,
     x_cohort_source: None | str | Unset = UNSET,
 ) -> ErrorResponse | InsufficientBalanceResponse | OperationResponse | None:
-    """AI-powered probability forecast
+    """AI-powered forecast
 
-     Run 6 parallel research agents per row, then synthesize into a probability forecast with rationale.
+     Run 6 parallel research agents per row, then synthesize into a forecast. Supports binary
+    (probability) and numeric (percentile) modes.
 
     Args:
         x_cohort_source (None | str | Unset):
@@ -138,9 +140,10 @@ async def asyncio_detailed(
     body: ForecastOperation,
     x_cohort_source: None | str | Unset = UNSET,
 ) -> Response[ErrorResponse | InsufficientBalanceResponse | OperationResponse]:
-    """AI-powered probability forecast
+    """AI-powered forecast
 
-     Run 6 parallel research agents per row, then synthesize into a probability forecast with rationale.
+     Run 6 parallel research agents per row, then synthesize into a forecast. Supports binary
+    (probability) and numeric (percentile) modes.
 
     Args:
         x_cohort_source (None | str | Unset):
@@ -170,9 +173,10 @@ async def asyncio(
     body: ForecastOperation,
     x_cohort_source: None | str | Unset = UNSET,
 ) -> ErrorResponse | InsufficientBalanceResponse | OperationResponse | None:
-    """AI-powered probability forecast
+    """AI-powered forecast
 
-     Run 6 parallel research agents per row, then synthesize into a probability forecast with rationale.
+     Run 6 parallel research agents per row, then synthesize into a forecast. Supports binary
+    (probability) and numeric (percentile) modes.
 
     Args:
         x_cohort_source (None | str | Unset):

src/futuresearch/generated/api/sessions/update_session_endpoint_sessions_session_id_patch.py

@@ -1,5 +1,6 @@
 from http import HTTPStatus
 from typing import Any
+from urllib.parse import quote
 from uuid import UUID
 
 import httpx
@@ -21,7 +22,9 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "patch",
-        "url": f"/sessions/{session_id}",
+        "url": "/sessions/{session_id}".format(
+            session_id=quote(str(session_id), safe=""),
+        ),
     }
 
     _kwargs["json"] = body.to_dict()
@@ -85,7 +88,7 @@ def sync_detailed(
     """
 
     kwargs = _get_kwargs(
-        session_id,
+        session_id=session_id,
         body=body,
     )
 
@@ -119,7 +122,7 @@ def sync(
     """
 
     return sync_detailed(
-        session_id,
+        session_id=session_id,
         client=client,
         body=body,
     ).parsed
@@ -148,7 +151,7 @@ async def asyncio_detailed(
     """
 
     kwargs = _get_kwargs(
-        session_id,
+        session_id=session_id,
         body=body,
     )
 
@@ -181,7 +184,7 @@ async def asyncio(
 
     return (
         await asyncio_detailed(
-            session_id,
+            session_id=session_id,
             client=client,
             body=body,
         )