Skip to content

Commit 72da2a5

Browse files
stainless-app[bot]stainless-app[bot]
committed
feat: Add axon rest endpoints for raw SQL access (#8287)
1 parent 2405ac6 commit 72da2a5

18 files changed

Lines changed: 782 additions & 22 deletions

‎.stats.yml‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
configured_endpoints: 122
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/runloop-ai%2Frunloop-9e0cbe82902910e818ebad8e8633d6e8730c288fbecc105c26fe3f1a571928c1.yml
3-
openapi_spec_hash: 558fa609a2b40ebf526e47f7e1af4491
4-
config_hash: 2dbb07d90e3e2d37fa6645af119965d5
1+
configured_endpoints: 124
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/runloop-ai%2Frunloop-c576eb67f119a7eb5815d4a3bf413c652cd7e4c257095e3b6b51967fe72fc00e.yml
3+
openapi_spec_hash: 0a4d20adf725a121e39a3442afa34b32
4+
config_hash: a759c23a5a04ad26f8740acc7e094c01

‎api.md‎

Lines changed: 27 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -105,11 +105,33 @@ from runloop_api_client.types import (
105105

106106
Methods:
107107

108-
- <code title="post /v1/axons">client.axons.<a href="./src/runloop_api_client/resources/axons.py">create</a>(\*\*<a href="src/runloop_api_client/types/axon_create_params.py">params</a>) -> <a href="./src/runloop_api_client/types/axon_view.py">AxonView</a></code>
109-
- <code title="get /v1/axons/{id}">client.axons.<a href="./src/runloop_api_client/resources/axons.py">retrieve</a>(id) -> <a href="./src/runloop_api_client/types/axon_view.py">AxonView</a></code>
110-
- <code title="get /v1/axons">client.axons.<a href="./src/runloop_api_client/resources/axons.py">list</a>() -> <a href="./src/runloop_api_client/types/axon_list_view.py">AxonListView</a></code>
111-
- <code title="post /v1/axons/{id}/publish">client.axons.<a href="./src/runloop_api_client/resources/axons.py">publish</a>(id, \*\*<a href="src/runloop_api_client/types/axon_publish_params.py">params</a>) -> <a href="./src/runloop_api_client/types/publish_result_view.py">PublishResultView</a></code>
112-
- <code title="get /v1/axons/{id}/subscribe/sse">client.axons.<a href="./src/runloop_api_client/resources/axons.py">subscribe_sse</a>(id) -> <a href="./src/runloop_api_client/types/axon_event_view.py">AxonEventView</a></code>
108+
- <code title="post /v1/axons">client.axons.<a href="./src/runloop_api_client/resources/axons/axons.py">create</a>(\*\*<a href="src/runloop_api_client/types/axon_create_params.py">params</a>) -> <a href="./src/runloop_api_client/types/axon_view.py">AxonView</a></code>
109+
- <code title="get /v1/axons/{id}">client.axons.<a href="./src/runloop_api_client/resources/axons/axons.py">retrieve</a>(id) -> <a href="./src/runloop_api_client/types/axon_view.py">AxonView</a></code>
110+
- <code title="get /v1/axons">client.axons.<a href="./src/runloop_api_client/resources/axons/axons.py">list</a>() -> <a href="./src/runloop_api_client/types/axon_list_view.py">AxonListView</a></code>
111+
- <code title="post /v1/axons/{id}/publish">client.axons.<a href="./src/runloop_api_client/resources/axons/axons.py">publish</a>(id, \*\*<a href="src/runloop_api_client/types/axon_publish_params.py">params</a>) -> <a href="./src/runloop_api_client/types/publish_result_view.py">PublishResultView</a></code>
112+
- <code title="get /v1/axons/{id}/subscribe/sse">client.axons.<a href="./src/runloop_api_client/resources/axons/axons.py">subscribe_sse</a>(id) -> <a href="./src/runloop_api_client/types/axon_event_view.py">AxonEventView</a></code>
113+
114+
## Sql
115+
116+
Types:
117+
118+
```python
119+
from runloop_api_client.types.axons import (
120+
SqlBatchParams,
121+
SqlBatchResultView,
122+
SqlColumnMetaView,
123+
SqlQueryResultView,
124+
SqlResultMetaView,
125+
SqlStatementParams,
126+
SqlStepErrorView,
127+
SqlStepResultView,
128+
)
129+
```
130+
131+
Methods:
132+
133+
- <code title="post /v1/axons/{id}/sql/batch">client.axons.sql.<a href="./src/runloop_api_client/resources/axons/sql.py">batch</a>(id, \*\*<a href="src/runloop_api_client/types/axons/sql_batch_params.py">params</a>) -> <a href="./src/runloop_api_client/types/axons/sql_batch_result_view.py">SqlBatchResultView</a></code>
134+
- <code title="post /v1/axons/{id}/sql/query">client.axons.sql.<a href="./src/runloop_api_client/resources/axons/sql.py">query</a>(id, \*\*<a href="src/runloop_api_client/types/axons/sql_query_params.py">params</a>) -> <a href="./src/runloop_api_client/types/axons/sql_query_result_view.py">SqlQueryResultView</a></code>
113135

114136
# Blueprints
115137

‎src/runloop_api_client/_client.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,12 +47,12 @@
4747
gateway_configs,
4848
network_policies,
4949
)
50-
from .resources.axons import AxonsResource, AsyncAxonsResource
5150
from .resources.agents import AgentsResource, AsyncAgentsResource
5251
from .resources.objects import ObjectsResource, AsyncObjectsResource
5352
from .resources.secrets import SecretsResource, AsyncSecretsResource
5453
from .resources.benchmarks import BenchmarksResource, AsyncBenchmarksResource
5554
from .resources.blueprints import BlueprintsResource, AsyncBlueprintsResource
55+
from .resources.axons.axons import AxonsResource, AsyncAxonsResource
5656
from .resources.mcp_configs import McpConfigsResource, AsyncMcpConfigsResource
5757
from .resources.repositories import RepositoriesResource, AsyncRepositoriesResource
5858
from .resources.benchmark_jobs import BenchmarkJobsResource, AsyncBenchmarkJobsResource

‎src/runloop_api_client/resources/axons/__init__.py‎

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2+
3+
from .sql import (
4+
SqlResource,
5+
AsyncSqlResource,
6+
SqlResourceWithRawResponse,
7+
AsyncSqlResourceWithRawResponse,
8+
SqlResourceWithStreamingResponse,
9+
AsyncSqlResourceWithStreamingResponse,
10+
)
11+
from .axons import (
12+
AxonsResource,
13+
AsyncAxonsResource,
14+
AxonsResourceWithRawResponse,
15+
AsyncAxonsResourceWithRawResponse,
16+
AxonsResourceWithStreamingResponse,
17+
AsyncAxonsResourceWithStreamingResponse,
18+
)
19+
20+
__all__ = [
21+
"SqlResource",
22+
"AsyncSqlResource",
23+
"SqlResourceWithRawResponse",
24+
"AsyncSqlResourceWithRawResponse",
25+
"SqlResourceWithStreamingResponse",
26+
"AsyncSqlResourceWithStreamingResponse",
27+
"AxonsResource",
28+
"AsyncAxonsResource",
29+
"AxonsResourceWithRawResponse",
30+
"AsyncAxonsResourceWithRawResponse",
31+
"AxonsResourceWithStreamingResponse",
32+
"AsyncAxonsResourceWithStreamingResponse",
33+
]

src/runloop_api_client/resources/axons.py renamed to src/runloop_api_client/resources/axons/axons.py

Lines changed: 44 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -7,28 +7,40 @@
77

88
import httpx
99

10-
from ..types import axon_create_params, axon_publish_params
11-
from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
12-
from .._utils import path_template, maybe_transform, async_maybe_transform
13-
from .._compat import cached_property
14-
from .._resource import SyncAPIResource, AsyncAPIResource
15-
from .._response import (
10+
from .sql import (
11+
SqlResource,
12+
AsyncSqlResource,
13+
SqlResourceWithRawResponse,
14+
AsyncSqlResourceWithRawResponse,
15+
SqlResourceWithStreamingResponse,
16+
AsyncSqlResourceWithStreamingResponse,
17+
)
18+
from ...types import axon_create_params, axon_publish_params
19+
from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
20+
from ..._utils import path_template, maybe_transform, async_maybe_transform
21+
from ..._compat import cached_property
22+
from ..._resource import SyncAPIResource, AsyncAPIResource
23+
from ..._response import (
1624
to_raw_response_wrapper,
1725
to_streamed_response_wrapper,
1826
async_to_raw_response_wrapper,
1927
async_to_streamed_response_wrapper,
2028
)
21-
from .._streaming import Stream, AsyncStream
22-
from .._base_client import make_request_options
23-
from ..types.axon_view import AxonView
24-
from ..types.axon_list_view import AxonListView
25-
from ..types.axon_event_view import AxonEventView
26-
from ..types.publish_result_view import PublishResultView
29+
from ..._streaming import Stream, AsyncStream
30+
from ..._base_client import make_request_options
31+
from ...types.axon_view import AxonView
32+
from ...types.axon_list_view import AxonListView
33+
from ...types.axon_event_view import AxonEventView
34+
from ...types.publish_result_view import PublishResultView
2735

2836
__all__ = ["AxonsResource", "AsyncAxonsResource"]
2937

3038

3139
class AxonsResource(SyncAPIResource):
40+
@cached_property
41+
def sql(self) -> SqlResource:
42+
return SqlResource(self._client)
43+
3244
@cached_property
3345
def with_raw_response(self) -> AxonsResourceWithRawResponse:
3446
"""
@@ -239,6 +251,10 @@ def subscribe_sse(
239251

240252

241253
class AsyncAxonsResource(AsyncAPIResource):
254+
@cached_property
255+
def sql(self) -> AsyncSqlResource:
256+
return AsyncSqlResource(self._client)
257+
242258
@cached_property
243259
def with_raw_response(self) -> AsyncAxonsResourceWithRawResponse:
244260
"""
@@ -468,6 +484,10 @@ def __init__(self, axons: AxonsResource) -> None:
468484
axons.subscribe_sse,
469485
)
470486

487+
@cached_property
488+
def sql(self) -> SqlResourceWithRawResponse:
489+
return SqlResourceWithRawResponse(self._axons.sql)
490+
471491

472492
class AsyncAxonsResourceWithRawResponse:
473493
def __init__(self, axons: AsyncAxonsResource) -> None:
@@ -489,6 +509,10 @@ def __init__(self, axons: AsyncAxonsResource) -> None:
489509
axons.subscribe_sse,
490510
)
491511

512+
@cached_property
513+
def sql(self) -> AsyncSqlResourceWithRawResponse:
514+
return AsyncSqlResourceWithRawResponse(self._axons.sql)
515+
492516

493517
class AxonsResourceWithStreamingResponse:
494518
def __init__(self, axons: AxonsResource) -> None:
@@ -510,6 +534,10 @@ def __init__(self, axons: AxonsResource) -> None:
510534
axons.subscribe_sse,
511535
)
512536

537+
@cached_property
538+
def sql(self) -> SqlResourceWithStreamingResponse:
539+
return SqlResourceWithStreamingResponse(self._axons.sql)
540+
513541

514542
class AsyncAxonsResourceWithStreamingResponse:
515543
def __init__(self, axons: AsyncAxonsResource) -> None:
@@ -530,3 +558,7 @@ def __init__(self, axons: AsyncAxonsResource) -> None:
530558
self.subscribe_sse = async_to_streamed_response_wrapper(
531559
axons.subscribe_sse,
532560
)
561+
562+
@cached_property
563+
def sql(self) -> AsyncSqlResourceWithStreamingResponse:
564+
return AsyncSqlResourceWithStreamingResponse(self._axons.sql)

0 commit comments

Comments
 (0)