Skip to content

Commit 2b909d8

Browse files
stainless-app[bot]stainless-app[bot]
committed
feat(api): api update (#267)
1 parent 2409f77 commit 2b909d8

31 files changed

+2411
-380
lines changed

‎.stats.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
11
configured_endpoints: 52
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/julep-ai-inc-dash%2Fjulep-26f60e516a24d68eda4282b612924fe4f9520489a5a114ecefe8d86c5479c0cc.yml
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/julep-ai-inc-dash%2Fjulep-f87931a479bdd5dadb2e21b8eaa9734c884099061f4179dbca01e5830d70b03c.yml

‎README.md‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,12 +33,12 @@ client = Julep(
3333
environment="dev",
3434
)
3535

36-
resource_created = client.tasks.create(
36+
task = client.tasks.create(
3737
agent_id="dad00000-0000-4000-a000-000000000000",
3838
main=[{"evaluate": {"foo": "string"}}],
3939
name="x",
4040
)
41-
print(resource_created.id)
41+
print(task.id)
4242
```
4343

4444
While you can provide an `api_key` keyword argument,
@@ -63,12 +63,12 @@ client = AsyncJulep(
6363

6464

6565
async def main() -> None:
66-
resource_created = await client.tasks.create(
66+
task = await client.tasks.create(
6767
agent_id="dad00000-0000-4000-a000-000000000000",
6868
main=[{"evaluate": {"foo": "string"}}],
6969
name="x",
7070
)
71-
print(resource_created.id)
71+
print(task.id)
7272

7373

7474
asyncio.run(main())

‎api.md‎

Lines changed: 28 additions & 23 deletions
Large diffs are not rendered by default.

‎src/julep/resources/agents/agents.py‎

Lines changed: 16 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -46,9 +46,7 @@
4646
from ...pagination import SyncOffsetPagination, AsyncOffsetPagination
4747
from ...types.agent import Agent
4848
from ..._base_client import AsyncPaginator, make_request_options
49-
from ...types.shared.resource_created import ResourceCreated
5049
from ...types.shared.resource_deleted import ResourceDeleted
51-
from ...types.shared.resource_updated import ResourceUpdated
5250

5351
__all__ = ["AgentsResource", "AsyncAgentsResource"]
5452

@@ -97,7 +95,7 @@ def create(
9795
extra_query: Query | None = None,
9896
extra_body: Body | None = None,
9997
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
100-
) -> ResourceCreated:
98+
) -> Agent:
10199
"""
102100
Create Agent
103101
@@ -129,7 +127,7 @@ def create(
129127
options=make_request_options(
130128
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
131129
),
132-
cast_to=ResourceCreated,
130+
cast_to=Agent,
133131
)
134132

135133
def update(
@@ -149,7 +147,7 @@ def update(
149147
extra_query: Query | None = None,
150148
extra_body: Body | None = None,
151149
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
152-
) -> ResourceUpdated:
150+
) -> Agent:
153151
"""
154152
Patch Agent
155153
@@ -183,7 +181,7 @@ def update(
183181
options=make_request_options(
184182
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
185183
),
186-
cast_to=ResourceUpdated,
184+
cast_to=Agent,
187185
)
188186

189187
def list(
@@ -285,7 +283,7 @@ def create_or_update(
285283
extra_query: Query | None = None,
286284
extra_body: Body | None = None,
287285
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
288-
) -> ResourceCreated:
286+
) -> Agent:
289287
"""
290288
Create Or Update Agent
291289
@@ -319,7 +317,7 @@ def create_or_update(
319317
options=make_request_options(
320318
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
321319
),
322-
cast_to=ResourceCreated,
320+
cast_to=Agent,
323321
)
324322

325323
def get(
@@ -372,7 +370,7 @@ def reset(
372370
extra_query: Query | None = None,
373371
extra_body: Body | None = None,
374372
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
375-
) -> ResourceUpdated:
373+
) -> Agent:
376374
"""
377375
Update Agent
378376
@@ -406,7 +404,7 @@ def reset(
406404
options=make_request_options(
407405
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
408406
),
409-
cast_to=ResourceUpdated,
407+
cast_to=Agent,
410408
)
411409

412410

@@ -454,7 +452,7 @@ async def create(
454452
extra_query: Query | None = None,
455453
extra_body: Body | None = None,
456454
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
457-
) -> ResourceCreated:
455+
) -> Agent:
458456
"""
459457
Create Agent
460458
@@ -486,7 +484,7 @@ async def create(
486484
options=make_request_options(
487485
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
488486
),
489-
cast_to=ResourceCreated,
487+
cast_to=Agent,
490488
)
491489

492490
async def update(
@@ -506,7 +504,7 @@ async def update(
506504
extra_query: Query | None = None,
507505
extra_body: Body | None = None,
508506
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
509-
) -> ResourceUpdated:
507+
) -> Agent:
510508
"""
511509
Patch Agent
512510
@@ -540,7 +538,7 @@ async def update(
540538
options=make_request_options(
541539
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
542540
),
543-
cast_to=ResourceUpdated,
541+
cast_to=Agent,
544542
)
545543

546544
def list(
@@ -642,7 +640,7 @@ async def create_or_update(
642640
extra_query: Query | None = None,
643641
extra_body: Body | None = None,
644642
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
645-
) -> ResourceCreated:
643+
) -> Agent:
646644
"""
647645
Create Or Update Agent
648646
@@ -676,7 +674,7 @@ async def create_or_update(
676674
options=make_request_options(
677675
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
678676
),
679-
cast_to=ResourceCreated,
677+
cast_to=Agent,
680678
)
681679

682680
async def get(
@@ -729,7 +727,7 @@ async def reset(
729727
extra_query: Query | None = None,
730728
extra_body: Body | None = None,
731729
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
732-
) -> ResourceUpdated:
730+
) -> Agent:
733731
"""
734732
Update Agent
735733
@@ -763,7 +761,7 @@ async def reset(
763761
options=make_request_options(
764762
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
765763
),
766-
cast_to=ResourceUpdated,
764+
cast_to=Agent,
767765
)
768766

769767

‎src/julep/resources/agents/docs.py‎

Lines changed: 6 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
from ...pagination import SyncOffsetPagination, AsyncOffsetPagination
2626
from ..._base_client import AsyncPaginator, make_request_options
2727
from ...types.agents import doc_list_params, doc_create_params, doc_search_params
28-
from ...types.shared.resource_created import ResourceCreated
2928
from ...types.shared.resource_deleted import ResourceDeleted
3029
from ...types.agents.doc_search_response import DocSearchResponse
3130

@@ -58,7 +57,6 @@ def create(
5857
*,
5958
content: Union[str, List[str]],
6059
title: str,
61-
connection_pool: object | NotGiven = NOT_GIVEN,
6260
embed_instruction: Optional[str] | NotGiven = NOT_GIVEN,
6361
metadata: Optional[object] | NotGiven = NOT_GIVEN,
6462
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -67,7 +65,7 @@ def create(
6765
extra_query: Query | None = None,
6866
extra_body: Body | None = None,
6967
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
70-
) -> ResourceCreated:
68+
) -> Doc:
7169
"""
7270
Create Agent Doc
7371
@@ -94,13 +92,9 @@ def create(
9492
doc_create_params.DocCreateParams,
9593
),
9694
options=make_request_options(
97-
extra_headers=extra_headers,
98-
extra_query=extra_query,
99-
extra_body=extra_body,
100-
timeout=timeout,
101-
query=maybe_transform({"connection_pool": connection_pool}, doc_create_params.DocCreateParams),
95+
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
10296
),
103-
cast_to=ResourceCreated,
97+
cast_to=Doc,
10498
)
10599

106100
def list(
@@ -390,7 +384,6 @@ async def create(
390384
*,
391385
content: Union[str, List[str]],
392386
title: str,
393-
connection_pool: object | NotGiven = NOT_GIVEN,
394387
embed_instruction: Optional[str] | NotGiven = NOT_GIVEN,
395388
metadata: Optional[object] | NotGiven = NOT_GIVEN,
396389
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -399,7 +392,7 @@ async def create(
399392
extra_query: Query | None = None,
400393
extra_body: Body | None = None,
401394
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
402-
) -> ResourceCreated:
395+
) -> Doc:
403396
"""
404397
Create Agent Doc
405398
@@ -426,15 +419,9 @@ async def create(
426419
doc_create_params.DocCreateParams,
427420
),
428421
options=make_request_options(
429-
extra_headers=extra_headers,
430-
extra_query=extra_query,
431-
extra_body=extra_body,
432-
timeout=timeout,
433-
query=await async_maybe_transform(
434-
{"connection_pool": connection_pool}, doc_create_params.DocCreateParams
435-
),
422+
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
436423
),
437-
cast_to=ResourceCreated,
424+
cast_to=Doc,
438425
)
439426

440427
def list(

‎src/julep/resources/agents/tools.py‎

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,11 @@
2323
from ...pagination import SyncOffsetPagination, AsyncOffsetPagination
2424
from ..._base_client import AsyncPaginator, make_request_options
2525
from ...types.agents import tool_list_params, tool_reset_params, tool_create_params, tool_update_params
26-
from ...types.shared.resource_created import ResourceCreated
2726
from ...types.shared.resource_deleted import ResourceDeleted
28-
from ...types.shared.resource_updated import ResourceUpdated
2927
from ...types.agents.tool_list_response import ToolListResponse
28+
from ...types.agents.tool_reset_response import ToolResetResponse
29+
from ...types.agents.tool_create_response import ToolCreateResponse
30+
from ...types.agents.tool_update_response import ToolUpdateResponse
3031

3132
__all__ = ["ToolsResource", "AsyncToolsResource"]
3233

@@ -79,7 +80,7 @@ def create(
7980
extra_query: Query | None = None,
8081
extra_body: Body | None = None,
8182
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
82-
) -> ResourceCreated:
83+
) -> ToolCreateResponse:
8384
"""
8485
Create Agent Tool
8586
@@ -124,7 +125,7 @@ def create(
124125
options=make_request_options(
125126
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
126127
),
127-
cast_to=ResourceCreated,
128+
cast_to=ToolCreateResponse,
128129
)
129130

130131
def update(
@@ -159,7 +160,7 @@ def update(
159160
extra_query: Query | None = None,
160161
extra_body: Body | None = None,
161162
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
162-
) -> ResourceUpdated:
163+
) -> ToolUpdateResponse:
163164
"""
164165
Patch Agent Tool
165166
@@ -206,7 +207,7 @@ def update(
206207
options=make_request_options(
207208
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
208209
),
209-
cast_to=ResourceUpdated,
210+
cast_to=ToolUpdateResponse,
210211
)
211212

212213
def list(
@@ -324,7 +325,7 @@ def reset(
324325
extra_query: Query | None = None,
325326
extra_body: Body | None = None,
326327
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
327-
) -> ResourceUpdated:
328+
) -> ToolResetResponse:
328329
"""
329330
Update Agent Tool
330331
@@ -371,7 +372,7 @@ def reset(
371372
options=make_request_options(
372373
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
373374
),
374-
cast_to=ResourceUpdated,
375+
cast_to=ToolResetResponse,
375376
)
376377

377378

@@ -423,7 +424,7 @@ async def create(
423424
extra_query: Query | None = None,
424425
extra_body: Body | None = None,
425426
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
426-
) -> ResourceCreated:
427+
) -> ToolCreateResponse:
427428
"""
428429
Create Agent Tool
429430
@@ -468,7 +469,7 @@ async def create(
468469
options=make_request_options(
469470
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
470471
),
471-
cast_to=ResourceCreated,
472+
cast_to=ToolCreateResponse,
472473
)
473474

474475
async def update(
@@ -503,7 +504,7 @@ async def update(
503504
extra_query: Query | None = None,
504505
extra_body: Body | None = None,
505506
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
506-
) -> ResourceUpdated:
507+
) -> ToolUpdateResponse:
507508
"""
508509
Patch Agent Tool
509510
@@ -550,7 +551,7 @@ async def update(
550551
options=make_request_options(
551552
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
552553
),
553-
cast_to=ResourceUpdated,
554+
cast_to=ToolUpdateResponse,
554555
)
555556

556557
def list(
@@ -668,7 +669,7 @@ async def reset(
668669
extra_query: Query | None = None,
669670
extra_body: Body | None = None,
670671
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
671-
) -> ResourceUpdated:
672+
) -> ToolResetResponse:
672673
"""
673674
Update Agent Tool
674675
@@ -715,7 +716,7 @@ async def reset(
715716
options=make_request_options(
716717
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
717718
),
718-
cast_to=ResourceUpdated,
719+
cast_to=ToolResetResponse,
719720
)
720721

721722

0 commit comments

Comments
 (0)