3131)
3232from ..pagination import SyncCursorPage , AsyncCursorPage
3333from .._base_client import AsyncPaginator , make_request_options
34- from ..types .document import Document
3534from ..types .document_status import DocumentStatus
35+ from ..types .document_get_response import DocumentGetResponse
3636from ..types .document_list_response import DocumentListResponse
3737
3838__all__ = ["DocumentsResource" , "AsyncDocumentsResource" ]
@@ -111,7 +111,9 @@ def add(
111111 text : str ,
112112 collection : Optional [str ] | NotGiven = NOT_GIVEN ,
113113 date : Union [str , datetime ] | NotGiven = NOT_GIVEN ,
114- source : Literal ["generic" , "mcp" , "slack" , "s3" , "gmail" , "notion" , "google_docs" , "hubspot" ]
114+ source : Literal [
115+ "generic" , "mcp" , "slack" , "s3" , "gmail" , "notion" , "google_docs" , "hubspot" , "reddit" , "google-calendar"
116+ ]
115117 | NotGiven = NOT_GIVEN ,
116118 title : Optional [str ] | NotGiven = NOT_GIVEN ,
117119 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -228,7 +230,7 @@ def get(
228230 extra_query : Query | None = None ,
229231 extra_body : Body | None = None ,
230232 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
231- ) -> Document :
233+ ) -> DocumentGetResponse :
232234 """
233235 Retrieves a document by ID, including its collection name and sections.
234236
@@ -246,7 +248,7 @@ def get(
246248 options = make_request_options (
247249 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
248250 ),
249- cast_to = Document ,
251+ cast_to = DocumentGetResponse ,
250252 )
251253
252254 def upload (
@@ -376,7 +378,9 @@ async def add(
376378 text : str ,
377379 collection : Optional [str ] | NotGiven = NOT_GIVEN ,
378380 date : Union [str , datetime ] | NotGiven = NOT_GIVEN ,
379- source : Literal ["generic" , "mcp" , "slack" , "s3" , "gmail" , "notion" , "google_docs" , "hubspot" ]
381+ source : Literal [
382+ "generic" , "mcp" , "slack" , "s3" , "gmail" , "notion" , "google_docs" , "hubspot" , "reddit" , "google-calendar"
383+ ]
380384 | NotGiven = NOT_GIVEN ,
381385 title : Optional [str ] | NotGiven = NOT_GIVEN ,
382386 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -493,7 +497,7 @@ async def get(
493497 extra_query : Query | None = None ,
494498 extra_body : Body | None = None ,
495499 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
496- ) -> Document :
500+ ) -> DocumentGetResponse :
497501 """
498502 Retrieves a document by ID, including its collection name and sections.
499503
@@ -511,7 +515,7 @@ async def get(
511515 options = make_request_options (
512516 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
513517 ),
514- cast_to = Document ,
518+ cast_to = DocumentGetResponse ,
515519 )
516520
517521 async def upload (
0 commit comments