Skip to content

Commit ef4ee07

Browse files
SDK regeneration
1 parent ab09c31 commit ef4ee07

11 files changed

Lines changed: 116 additions & 122 deletions

‎reference.md‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -420,15 +420,15 @@ Add graph episodes and thread messages to a draft batch. Items are appended in r
420420
<dd>
421421

422422
```python
423-
from zep_cloud import ApidataBatchAddItem, Zep
423+
from zep_cloud import BatchAddItem, Zep
424424

425425
client = Zep(
426426
api_key="YOUR_API_KEY",
427427
)
428428
client.batch.add(
429429
batch_id="batchId",
430430
items=[
431-
ApidataBatchAddItem(
431+
BatchAddItem(
432432
type="graph_episode",
433433
)
434434
],
@@ -456,7 +456,7 @@ client.batch.add(
456456
<dl>
457457
<dd>
458458

459-
**items:** `typing.Sequence[ApidataBatchAddItem]`
459+
**items:** `typing.Sequence[BatchAddItem]`
460460

461461
</dd>
462462
</dl>

‎src/zep_cloud/__init__.py‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -7,14 +7,14 @@
77
AddThreadMessagesResponse,
88
AddTripleResponse,
99
ApiError,
10-
ApidataBatchAddItem,
1110
ApidataBatchAddItemRole,
1211
ApidataBatchAddItemType,
13-
ApidataBatchItemDetail,
14-
ApidataBatchItemListResponse,
1512
ApidataBatchListResponse,
16-
ApidataBatchProgress,
17-
ApidataBatchSummary,
13+
BatchAddItem,
14+
BatchItemDetail,
15+
BatchItemListResponse,
16+
BatchProgress,
17+
BatchSummary,
1818
CloneGraphResponse,
1919
ClusterDetectConfig,
2020
CoOccurrenceDetectConfig,
@@ -95,16 +95,16 @@
9595
"AddThreadMessagesResponse",
9696
"AddTripleResponse",
9797
"ApiError",
98-
"ApidataBatchAddItem",
9998
"ApidataBatchAddItemRole",
10099
"ApidataBatchAddItemType",
101-
"ApidataBatchItemDetail",
102-
"ApidataBatchItemListResponse",
103100
"ApidataBatchListResponse",
104-
"ApidataBatchProgress",
105-
"ApidataBatchSummary",
106101
"AsyncZep",
107102
"BadRequestError",
103+
"BatchAddItem",
104+
"BatchItemDetail",
105+
"BatchItemListResponse",
106+
"BatchProgress",
107+
"BatchSummary",
108108
"CloneGraphResponse",
109109
"ClusterDetectConfig",
110110
"CoOccurrenceDetectConfig",

‎src/zep_cloud/batch/client.py‎

Lines changed: 32 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,11 @@
44

55
from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
66
from ..core.request_options import RequestOptions
7-
from ..types.apidata_batch_add_item import ApidataBatchAddItem
8-
from ..types.apidata_batch_item_detail import ApidataBatchItemDetail
9-
from ..types.apidata_batch_item_list_response import ApidataBatchItemListResponse
107
from ..types.apidata_batch_list_response import ApidataBatchListResponse
11-
from ..types.apidata_batch_summary import ApidataBatchSummary
8+
from ..types.batch_add_item import BatchAddItem
9+
from ..types.batch_item_detail import BatchItemDetail
10+
from ..types.batch_item_list_response import BatchItemListResponse
11+
from ..types.batch_summary import BatchSummary
1212
from ..types.success_response import SuccessResponse
1313
from .raw_client import AsyncRawBatchClient, RawBatchClient
1414

@@ -82,7 +82,7 @@ def create(
8282
*,
8383
metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
8484
request_options: typing.Optional[RequestOptions] = None,
85-
) -> ApidataBatchSummary:
85+
) -> BatchSummary:
8686
"""
8787
Create a draft batch that can be filled with graph episodes and thread messages.
8888
@@ -95,7 +95,7 @@ def create(
9595
9696
Returns
9797
-------
98-
ApidataBatchSummary
98+
BatchSummary
9999
Created batch
100100
101101
Examples
@@ -110,7 +110,7 @@ def create(
110110
_response = self._raw_client.create(metadata=metadata, request_options=request_options)
111111
return _response.data
112112

113-
def get(self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> ApidataBatchSummary:
113+
def get(self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> BatchSummary:
114114
"""
115115
Get a batch summary, including runtime progress when the batch has been processed.
116116
@@ -124,7 +124,7 @@ def get(self, batch_id: str, *, request_options: typing.Optional[RequestOptions]
124124
125125
Returns
126126
-------
127-
ApidataBatchSummary
127+
BatchSummary
128128
Batch summary
129129
130130
Examples
@@ -180,7 +180,7 @@ def list_items(
180180
cursor: typing.Optional[int] = None,
181181
status: typing.Optional[str] = None,
182182
request_options: typing.Optional[RequestOptions] = None,
183-
) -> ApidataBatchItemListResponse:
183+
) -> BatchItemListResponse:
184184
"""
185185
List items in a batch, including derived runtime status when the batch has been processed.
186186
@@ -203,7 +203,7 @@ def list_items(
203203
204204
Returns
205205
-------
206-
ApidataBatchItemListResponse
206+
BatchItemListResponse
207207
Batch item list
208208
209209
Examples
@@ -229,9 +229,9 @@ def add(
229229
self,
230230
batch_id: str,
231231
*,
232-
items: typing.Sequence[ApidataBatchAddItem],
232+
items: typing.Sequence[BatchAddItem],
233233
request_options: typing.Optional[RequestOptions] = None,
234-
) -> typing.List[ApidataBatchItemDetail]:
234+
) -> typing.List[BatchItemDetail]:
235235
"""
236236
Add graph episodes and thread messages to a draft batch. Items are appended in request order.
237237
@@ -240,27 +240,27 @@ def add(
240240
batch_id : str
241241
The batch ID.
242242
243-
items : typing.Sequence[ApidataBatchAddItem]
243+
items : typing.Sequence[BatchAddItem]
244244
245245
request_options : typing.Optional[RequestOptions]
246246
Request-specific configuration.
247247
248248
Returns
249249
-------
250-
typing.List[ApidataBatchItemDetail]
250+
typing.List[BatchItemDetail]
251251
Added batch items
252252
253253
Examples
254254
--------
255-
from zep_cloud import ApidataBatchAddItem, Zep
255+
from zep_cloud import BatchAddItem, Zep
256256
257257
client = Zep(
258258
api_key="YOUR_API_KEY",
259259
)
260260
client.batch.add(
261261
batch_id="batchId",
262262
items=[
263-
ApidataBatchAddItem(
263+
BatchAddItem(
264264
type="graph_episode",
265265
)
266266
],
@@ -269,7 +269,7 @@ def add(
269269
_response = self._raw_client.add(batch_id, items=items, request_options=request_options)
270270
return _response.data
271271

272-
def process(self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> ApidataBatchSummary:
272+
def process(self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> BatchSummary:
273273
"""
274274
Start processing a filled batch. Repeated calls return the existing batch run.
275275
@@ -283,7 +283,7 @@ def process(self, batch_id: str, *, request_options: typing.Optional[RequestOpti
283283
284284
Returns
285285
-------
286-
ApidataBatchSummary
286+
BatchSummary
287287
Batch processing state
288288
289289
Examples
@@ -377,7 +377,7 @@ async def create(
377377
*,
378378
metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
379379
request_options: typing.Optional[RequestOptions] = None,
380-
) -> ApidataBatchSummary:
380+
) -> BatchSummary:
381381
"""
382382
Create a draft batch that can be filled with graph episodes and thread messages.
383383
@@ -390,7 +390,7 @@ async def create(
390390
391391
Returns
392392
-------
393-
ApidataBatchSummary
393+
BatchSummary
394394
Created batch
395395
396396
Examples
@@ -413,9 +413,7 @@ async def main() -> None:
413413
_response = await self._raw_client.create(metadata=metadata, request_options=request_options)
414414
return _response.data
415415

416-
async def get(
417-
self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None
418-
) -> ApidataBatchSummary:
416+
async def get(self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> BatchSummary:
419417
"""
420418
Get a batch summary, including runtime progress when the batch has been processed.
421419
@@ -429,7 +427,7 @@ async def get(
429427
430428
Returns
431429
-------
432-
ApidataBatchSummary
430+
BatchSummary
433431
Batch summary
434432
435433
Examples
@@ -503,7 +501,7 @@ async def list_items(
503501
cursor: typing.Optional[int] = None,
504502
status: typing.Optional[str] = None,
505503
request_options: typing.Optional[RequestOptions] = None,
506-
) -> ApidataBatchItemListResponse:
504+
) -> BatchItemListResponse:
507505
"""
508506
List items in a batch, including derived runtime status when the batch has been processed.
509507
@@ -526,7 +524,7 @@ async def list_items(
526524
527525
Returns
528526
-------
529-
ApidataBatchItemListResponse
527+
BatchItemListResponse
530528
Batch item list
531529
532530
Examples
@@ -560,9 +558,9 @@ async def add(
560558
self,
561559
batch_id: str,
562560
*,
563-
items: typing.Sequence[ApidataBatchAddItem],
561+
items: typing.Sequence[BatchAddItem],
564562
request_options: typing.Optional[RequestOptions] = None,
565-
) -> typing.List[ApidataBatchItemDetail]:
563+
) -> typing.List[BatchItemDetail]:
566564
"""
567565
Add graph episodes and thread messages to a draft batch. Items are appended in request order.
568566
@@ -571,21 +569,21 @@ async def add(
571569
batch_id : str
572570
The batch ID.
573571
574-
items : typing.Sequence[ApidataBatchAddItem]
572+
items : typing.Sequence[BatchAddItem]
575573
576574
request_options : typing.Optional[RequestOptions]
577575
Request-specific configuration.
578576
579577
Returns
580578
-------
581-
typing.List[ApidataBatchItemDetail]
579+
typing.List[BatchItemDetail]
582580
Added batch items
583581
584582
Examples
585583
--------
586584
import asyncio
587585
588-
from zep_cloud import ApidataBatchAddItem, AsyncZep
586+
from zep_cloud import AsyncZep, BatchAddItem
589587
590588
client = AsyncZep(
591589
api_key="YOUR_API_KEY",
@@ -596,7 +594,7 @@ async def main() -> None:
596594
await client.batch.add(
597595
batch_id="batchId",
598596
items=[
599-
ApidataBatchAddItem(
597+
BatchAddItem(
600598
type="graph_episode",
601599
)
602600
],
@@ -608,9 +606,7 @@ async def main() -> None:
608606
_response = await self._raw_client.add(batch_id, items=items, request_options=request_options)
609607
return _response.data
610608

611-
async def process(
612-
self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None
613-
) -> ApidataBatchSummary:
609+
async def process(self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> BatchSummary:
614610
"""
615611
Start processing a filled batch. Repeated calls return the existing batch run.
616612
@@ -624,7 +620,7 @@ async def process(
624620
625621
Returns
626622
-------
627-
ApidataBatchSummary
623+
BatchSummary
628624
Batch processing state
629625
630626
Examples

0 commit comments

Comments
 (0)