44
55from ..core .client_wrapper import AsyncClientWrapper , SyncClientWrapper
66from ..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
107from ..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
1212from ..types .success_response import SuccessResponse
1313from .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