Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .stats.yml
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
configured_endpoints: 14
configured_endpoints: 15
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/brainbase-egrigokhan%2Fbrainbase-a197a88d1abbdbd5c886875d2cf12ffd6abf79aec7d4dbd0439ed50197692eba.yml
5 changes: 3 additions & 2 deletions api.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,11 @@

Methods:

- <code title="post /api/workers/{id}">client.workers.<a href="./src/brainbase/resources/workers/workers.py">create</a>(id, \*\*<a href="src/brainbase/types/worker_create_params.py">params</a>) -> None</code>
- <code title="post /api/workers">client.workers.<a href="./src/brainbase/resources/workers/workers.py">create</a>(\*\*<a href="src/brainbase/types/worker_create_params.py">params</a>) -> None</code>
- <code title="get /api/workers/{id}">client.workers.<a href="./src/brainbase/resources/workers/workers.py">retrieve</a>(id) -> None</code>
- <code title="get /api/workers">client.workers.<a href="./src/brainbase/resources/workers/workers.py">list</a>() -> None</code>
- <code title="delete /api/workers/{id}">client.workers.<a href="./src/brainbase/resources/workers/workers.py">delete</a>(id) -> None</code>
- <code title="post /api/workers/{id}">client.workers.<a href="./src/brainbase/resources/workers/workers.py">create_id</a>(id, \*\*<a href="src/brainbase/types/worker_create_id_params.py">params</a>) -> None</code>
- <code title="delete /api/workers/{id}">client.workers.<a href="./src/brainbase/resources/workers/workers.py">delete_id</a>(id) -> None</code>

## Deployments

Expand Down
146 changes: 121 additions & 25 deletions src/brainbase/resources/workers/workers.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
FlowsResourceWithStreamingResponse,
AsyncFlowsResourceWithStreamingResponse,
)
from ...types import worker_create_params
from ...types import worker_create_params, worker_create_id_params
from ..._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
from ..._utils import (
maybe_transform,
Expand Down Expand Up @@ -69,10 +69,9 @@ def with_streaming_response(self) -> WorkersResourceWithStreamingResponse:

def create(
self,
id: str,
*,
name: str,
description: str | NotGiven = NOT_GIVEN,
name: str | NotGiven = NOT_GIVEN,
status: str | NotGiven = NOT_GIVEN,
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
# The extra values given here take precedence over values defined on the client or passed to this method.
Expand All @@ -82,7 +81,7 @@ def create(
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
) -> None:
"""
Update a worker
Create a new worker

Args:
extra_headers: Send extra headers
Expand All @@ -93,15 +92,13 @@ def create(

timeout: Override the client-level default timeout for this request, in seconds
"""
if not id:
raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
return self._post(
f"/api/workers/{id}",
"/api/workers",
body=maybe_transform(
{
"description": description,
"name": name,
"description": description,
"status": status,
},
worker_create_params.WorkerCreateParams,
Expand Down Expand Up @@ -166,7 +163,52 @@ def list(
cast_to=NoneType,
)

def delete(
def create_id(
self,
id: str,
*,
description: str | NotGiven = NOT_GIVEN,
name: str | NotGiven = NOT_GIVEN,
status: str | NotGiven = NOT_GIVEN,
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
# The extra values given here take precedence over values defined on the client or passed to this method.
extra_headers: Headers | None = None,
extra_query: Query | None = None,
extra_body: Body | None = None,
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
) -> None:
"""
Update a worker

Args:
extra_headers: Send extra headers

extra_query: Add additional query parameters to the request

extra_body: Add additional JSON properties to the request

timeout: Override the client-level default timeout for this request, in seconds
"""
if not id:
raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
return self._post(
f"/api/workers/{id}",
body=maybe_transform(
{
"description": description,
"name": name,
"status": status,
},
worker_create_id_params.WorkerCreateIDParams,
),
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
),
cast_to=NoneType,
)

def delete_id(
self,
id: str,
*,
Expand Down Expand Up @@ -231,10 +273,9 @@ def with_streaming_response(self) -> AsyncWorkersResourceWithStreamingResponse:

async def create(
self,
id: str,
*,
name: str,
description: str | NotGiven = NOT_GIVEN,
name: str | NotGiven = NOT_GIVEN,
status: str | NotGiven = NOT_GIVEN,
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
# The extra values given here take precedence over values defined on the client or passed to this method.
Expand All @@ -244,7 +285,7 @@ async def create(
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
) -> None:
"""
Update a worker
Create a new worker

Args:
extra_headers: Send extra headers
Expand All @@ -255,15 +296,13 @@ async def create(

timeout: Override the client-level default timeout for this request, in seconds
"""
if not id:
raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
return await self._post(
f"/api/workers/{id}",
"/api/workers",
body=await async_maybe_transform(
{
"description": description,
"name": name,
"description": description,
"status": status,
},
worker_create_params.WorkerCreateParams,
Expand Down Expand Up @@ -328,7 +367,52 @@ async def list(
cast_to=NoneType,
)

async def delete(
async def create_id(
self,
id: str,
*,
description: str | NotGiven = NOT_GIVEN,
name: str | NotGiven = NOT_GIVEN,
status: str | NotGiven = NOT_GIVEN,
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
# The extra values given here take precedence over values defined on the client or passed to this method.
extra_headers: Headers | None = None,
extra_query: Query | None = None,
extra_body: Body | None = None,
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
) -> None:
"""
Update a worker

Args:
extra_headers: Send extra headers

extra_query: Add additional query parameters to the request

extra_body: Add additional JSON properties to the request

timeout: Override the client-level default timeout for this request, in seconds
"""
if not id:
raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
return await self._post(
f"/api/workers/{id}",
body=await async_maybe_transform(
{
"description": description,
"name": name,
"status": status,
},
worker_create_id_params.WorkerCreateIDParams,
),
options=make_request_options(
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
),
cast_to=NoneType,
)

async def delete_id(
self,
id: str,
*,
Expand Down Expand Up @@ -376,8 +460,11 @@ def __init__(self, workers: WorkersResource) -> None:
self.list = to_raw_response_wrapper(
workers.list,
)
self.delete = to_raw_response_wrapper(
workers.delete,
self.create_id = to_raw_response_wrapper(
workers.create_id,
)
self.delete_id = to_raw_response_wrapper(
workers.delete_id,
)

@cached_property
Expand All @@ -402,8 +489,11 @@ def __init__(self, workers: AsyncWorkersResource) -> None:
self.list = async_to_raw_response_wrapper(
workers.list,
)
self.delete = async_to_raw_response_wrapper(
workers.delete,
self.create_id = async_to_raw_response_wrapper(
workers.create_id,
)
self.delete_id = async_to_raw_response_wrapper(
workers.delete_id,
)

@cached_property
Expand All @@ -428,8 +518,11 @@ def __init__(self, workers: WorkersResource) -> None:
self.list = to_streamed_response_wrapper(
workers.list,
)
self.delete = to_streamed_response_wrapper(
workers.delete,
self.create_id = to_streamed_response_wrapper(
workers.create_id,
)
self.delete_id = to_streamed_response_wrapper(
workers.delete_id,
)

@cached_property
Expand All @@ -454,8 +547,11 @@ def __init__(self, workers: AsyncWorkersResource) -> None:
self.list = async_to_streamed_response_wrapper(
workers.list,
)
self.delete = async_to_streamed_response_wrapper(
workers.delete,
self.create_id = async_to_streamed_response_wrapper(
workers.create_id,
)
self.delete_id = async_to_streamed_response_wrapper(
workers.delete_id,
)

@cached_property
Expand Down
1 change: 1 addition & 0 deletions src/brainbase/types/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,3 +3,4 @@
from __future__ import annotations

from .worker_create_params import WorkerCreateParams as WorkerCreateParams
from .worker_create_id_params import WorkerCreateIDParams as WorkerCreateIDParams
15 changes: 15 additions & 0 deletions src/brainbase/types/worker_create_id_params.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.

from __future__ import annotations

from typing_extensions import TypedDict

__all__ = ["WorkerCreateIDParams"]


class WorkerCreateIDParams(TypedDict, total=False):
description: str

name: str

status: str
6 changes: 3 additions & 3 deletions src/brainbase/types/worker_create_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,14 +2,14 @@

from __future__ import annotations

from typing_extensions import TypedDict
from typing_extensions import Required, TypedDict

__all__ = ["WorkerCreateParams"]


class WorkerCreateParams(TypedDict, total=False):
description: str
name: Required[str]

name: str
description: str

status: str
Loading