2121)
2222from ..pagination import SyncCursorPage , AsyncCursorPage
2323from .._base_client import AsyncPaginator , make_request_options
24- from ..types .document import Document
25- from ..types .document_status import DocumentStatus
24+ from ..types .memory import Memory
25+ from ..types .memory_status import MemoryStatus
2626from ..types .memory_search_response import MemorySearchResponse
2727from ..types .memory_status_response import MemoryStatusResponse
2828
@@ -112,7 +112,7 @@ def list(
112112 extra_query : Query | None = None ,
113113 extra_body : Body | None = None ,
114114 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
115- ) -> SyncCursorPage [Document ]:
115+ ) -> SyncCursorPage [Memory ]:
116116 """This endpoint allows you to paginate through all documents in the index.
117117
118118 You can
@@ -133,7 +133,7 @@ def list(
133133 """
134134 return self ._get_api_list (
135135 "/memories/list" ,
136- page = SyncCursorPage [Document ],
136+ page = SyncCursorPage [Memory ],
137137 options = make_request_options (
138138 extra_headers = extra_headers ,
139139 extra_query = extra_query ,
@@ -149,7 +149,7 @@ def list(
149149 memory_list_params .MemoryListParams ,
150150 ),
151151 ),
152- model = Document ,
152+ model = Memory ,
153153 )
154154
155155 def add (
@@ -166,7 +166,7 @@ def add(
166166 extra_query : Query | None = None ,
167167 extra_body : Body | None = None ,
168168 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
169- ) -> DocumentStatus :
169+ ) -> MemoryStatus :
170170 """Adds an arbitrary document to the index.
171171
172172 This can be any text, email, call
@@ -211,7 +211,7 @@ def add(
211211 options = make_request_options (
212212 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
213213 ),
214- cast_to = DocumentStatus ,
214+ cast_to = MemoryStatus ,
215215 )
216216
217217 def get (
@@ -272,7 +272,7 @@ def get(
272272 extra_query : Query | None = None ,
273273 extra_body : Body | None = None ,
274274 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
275- ) -> Document :
275+ ) -> Memory :
276276 """
277277 Retrieves a document by provider and resource_id.
278278
@@ -294,7 +294,7 @@ def get(
294294 options = make_request_options (
295295 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
296296 ),
297- cast_to = Document ,
297+ cast_to = Memory ,
298298 )
299299
300300 def search (
@@ -439,7 +439,7 @@ def upload(
439439 extra_query : Query | None = None ,
440440 extra_body : Body | None = None ,
441441 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
442- ) -> DocumentStatus :
442+ ) -> MemoryStatus :
443443 """This endpoint will upload a file to the index and return a document ID.
444444
445445 The file
@@ -478,7 +478,7 @@ def upload(
478478 options = make_request_options (
479479 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
480480 ),
481- cast_to = DocumentStatus ,
481+ cast_to = MemoryStatus ,
482482 )
483483
484484
@@ -565,7 +565,7 @@ def list(
565565 extra_query : Query | None = None ,
566566 extra_body : Body | None = None ,
567567 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
568- ) -> AsyncPaginator [Document , AsyncCursorPage [Document ]]:
568+ ) -> AsyncPaginator [Memory , AsyncCursorPage [Memory ]]:
569569 """This endpoint allows you to paginate through all documents in the index.
570570
571571 You can
@@ -586,7 +586,7 @@ def list(
586586 """
587587 return self ._get_api_list (
588588 "/memories/list" ,
589- page = AsyncCursorPage [Document ],
589+ page = AsyncCursorPage [Memory ],
590590 options = make_request_options (
591591 extra_headers = extra_headers ,
592592 extra_query = extra_query ,
@@ -602,7 +602,7 @@ def list(
602602 memory_list_params .MemoryListParams ,
603603 ),
604604 ),
605- model = Document ,
605+ model = Memory ,
606606 )
607607
608608 async def add (
@@ -619,7 +619,7 @@ async def add(
619619 extra_query : Query | None = None ,
620620 extra_body : Body | None = None ,
621621 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
622- ) -> DocumentStatus :
622+ ) -> MemoryStatus :
623623 """Adds an arbitrary document to the index.
624624
625625 This can be any text, email, call
@@ -664,7 +664,7 @@ async def add(
664664 options = make_request_options (
665665 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
666666 ),
667- cast_to = DocumentStatus ,
667+ cast_to = MemoryStatus ,
668668 )
669669
670670 async def get (
@@ -725,7 +725,7 @@ async def get(
725725 extra_query : Query | None = None ,
726726 extra_body : Body | None = None ,
727727 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
728- ) -> Document :
728+ ) -> Memory :
729729 """
730730 Retrieves a document by provider and resource_id.
731731
@@ -747,7 +747,7 @@ async def get(
747747 options = make_request_options (
748748 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
749749 ),
750- cast_to = Document ,
750+ cast_to = Memory ,
751751 )
752752
753753 async def search (
@@ -892,7 +892,7 @@ async def upload(
892892 extra_query : Query | None = None ,
893893 extra_body : Body | None = None ,
894894 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
895- ) -> DocumentStatus :
895+ ) -> MemoryStatus :
896896 """This endpoint will upload a file to the index and return a document ID.
897897
898898 The file
@@ -931,7 +931,7 @@ async def upload(
931931 options = make_request_options (
932932 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
933933 ),
934- cast_to = DocumentStatus ,
934+ cast_to = MemoryStatus ,
935935 )
936936
937937
0 commit comments