usecortex-ai 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

{cortex_ai → usecortex_ai}/upload/client.py

@@ -7,7 +7,9 @@ from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
 from ..types.app_sources_upload_data import AppSourcesUploadData
 from ..types.batch_upload_data import BatchUploadData
+from ..types.markdown_upload_request import MarkdownUploadRequest
 from ..types.processing_status import ProcessingStatus
+from ..types.relations import Relations
 from ..types.single_upload_data import SingleUploadData
 from ..types.source_model import SourceModel
 from .raw_client import AsyncRawUploadClient, RawUploadClient
@@ -39,6 +41,7 @@ class UploadClient:
         sub_tenant_id: typing.Optional[str] = None,
         tenant_metadata: typing.Optional[str] = OMIT,
         document_metadata: typing.Optional[str] = OMIT,
+        relations: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> BatchUploadData:
         """
@@ -55,6 +58,8 @@ class UploadClient:
 
         document_metadata : typing.Optional[str]
 
+        relations : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -65,7 +70,7 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.upload.batch_upload(tenant_id='tenant_id', )
@@ -76,6 +81,7 @@ class UploadClient:
             sub_tenant_id=sub_tenant_id,
             tenant_metadata=tenant_metadata,
             document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -89,6 +95,7 @@ class UploadClient:
         source_ids: typing.Optional[typing.List[str]] = OMIT,
         tenant_metadata: typing.Optional[str] = OMIT,
         document_metadata: typing.Optional[str] = OMIT,
+        relations: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> BatchUploadData:
         """
@@ -107,6 +114,8 @@ class UploadClient:
 
         document_metadata : typing.Optional[str]
 
+        relations : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -117,7 +126,7 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.upload.batch_update(tenant_id='tenant_id', )
@@ -129,6 +138,7 @@ class UploadClient:
             source_ids=source_ids,
             tenant_metadata=tenant_metadata,
             document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -141,6 +151,7 @@ class UploadClient:
         sub_tenant_id: typing.Optional[str] = None,
         tenant_metadata: typing.Optional[str] = OMIT,
         document_metadata: typing.Optional[str] = OMIT,
+        relations: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -157,6 +168,8 @@ class UploadClient:
 
         document_metadata : typing.Optional[str]
 
+        relations : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -167,7 +180,7 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.upload.upload_document(tenant_id='tenant_id', )
@@ -178,6 +191,7 @@ class UploadClient:
             sub_tenant_id=sub_tenant_id,
             tenant_metadata=tenant_metadata,
             document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -191,6 +205,7 @@ class UploadClient:
         sub_tenant_id: typing.Optional[str] = None,
         tenant_metadata: typing.Optional[str] = OMIT,
         document_metadata: typing.Optional[str] = OMIT,
+        relations: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -209,6 +224,8 @@ class UploadClient:
 
         document_metadata : typing.Optional[str]
 
+        relations : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -219,7 +236,7 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.upload.update_document(source_id='source_id', tenant_id='tenant_id', )
@@ -231,6 +248,7 @@ class UploadClient:
             sub_tenant_id=sub_tenant_id,
             tenant_metadata=tenant_metadata,
             document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -262,7 +280,7 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI, SourceModel
+        from usecortex-ai import CortexAI, SourceModel
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.upload.upload_app_sources(tenant_id='tenant_id', request=[SourceModel()], )
@@ -276,10 +294,9 @@ class UploadClient:
         self,
         *,
         tenant_id: str,
-        content: str,
+        request: MarkdownUploadRequest,
         sub_tenant_id: typing.Optional[str] = None,
-        tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
-        document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        relations: typing.Optional[Relations] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -290,13 +307,11 @@ class UploadClient:
         ----------
         tenant_id : str
 
-        content : str
+        request : MarkdownUploadRequest
 
         sub_tenant_id : typing.Optional[str]
 
-        tenant_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
-
-        document_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+        relations : typing.Optional[Relations]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -308,17 +323,16 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI, MarkdownUploadRequest
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.upload.upload_markdown(tenant_id='tenant_id', content='content', )
+        client.upload.upload_markdown(tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
         """
         _response = self._raw_client.upload_markdown(
             tenant_id=tenant_id,
-            content=content,
+            request=request,
             sub_tenant_id=sub_tenant_id,
-            tenant_metadata=tenant_metadata,
-            document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -327,10 +341,9 @@ class UploadClient:
         self,
         *,
         tenant_id: str,
-        content: str,
+        request: MarkdownUploadRequest,
         sub_tenant_id: typing.Optional[str] = None,
-        tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
-        document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        relations: typing.Optional[Relations] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -341,13 +354,11 @@ class UploadClient:
         ----------
         tenant_id : str
 
-        content : str
+        request : MarkdownUploadRequest
 
         sub_tenant_id : typing.Optional[str]
 
-        tenant_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
-
-        document_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+        relations : typing.Optional[Relations]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -359,17 +370,16 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI, MarkdownUploadRequest
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.upload.upload_text(tenant_id='tenant_id', content='content', )
+        client.upload.upload_text(tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
         """
         _response = self._raw_client.upload_text(
             tenant_id=tenant_id,
-            content=content,
+            request=request,
             sub_tenant_id=sub_tenant_id,
-            tenant_metadata=tenant_metadata,
-            document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -379,10 +389,9 @@ class UploadClient:
         *,
         source_id: str,
         tenant_id: str,
-        content: str,
+        request: MarkdownUploadRequest,
         sub_tenant_id: typing.Optional[str] = None,
-        tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
-        document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        relations: typing.Optional[Relations] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -392,13 +401,11 @@ class UploadClient:
 
         tenant_id : str
 
-        content : str
+        request : MarkdownUploadRequest
 
         sub_tenant_id : typing.Optional[str]
 
-        tenant_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
-
-        document_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+        relations : typing.Optional[Relations]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -410,18 +417,17 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI, MarkdownUploadRequest
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.upload.update_markdown(source_id='source_id', tenant_id='tenant_id', content='content', )
+        client.upload.update_markdown(source_id='source_id', tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
         """
         _response = self._raw_client.update_markdown(
             source_id=source_id,
             tenant_id=tenant_id,
-            content=content,
+            request=request,
             sub_tenant_id=sub_tenant_id,
-            tenant_metadata=tenant_metadata,
-            document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -431,10 +437,9 @@ class UploadClient:
         *,
         source_id: str,
         tenant_id: str,
-        content: str,
+        request: MarkdownUploadRequest,
         sub_tenant_id: typing.Optional[str] = None,
-        tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
-        document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        relations: typing.Optional[Relations] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -444,13 +449,11 @@ class UploadClient:
 
         tenant_id : str
 
-        content : str
+        request : MarkdownUploadRequest
 
         sub_tenant_id : typing.Optional[str]
 
-        tenant_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
-
-        document_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+        relations : typing.Optional[Relations]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -462,18 +465,17 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI, MarkdownUploadRequest
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.upload.update_text(source_id='source_id', tenant_id='tenant_id', content='content', )
+        client.upload.update_text(source_id='source_id', tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
         """
         _response = self._raw_client.update_text(
             source_id=source_id,
             tenant_id=tenant_id,
-            content=content,
+            request=request,
             sub_tenant_id=sub_tenant_id,
-            tenant_metadata=tenant_metadata,
-            document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -508,7 +510,7 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.upload.upload_embeddings(tenant_id='tenant_id', embeddings=[[1.1]], )
@@ -550,7 +552,7 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.upload.update_embeddings(tenant_id='tenant_id', embeddings={'key': [1.1]}, )
@@ -567,6 +569,7 @@ class UploadClient:
         tenant_id: str,
         sub_tenant_id: typing.Optional[str] = None,
         file_id: typing.Optional[str] = None,
+        relations: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -580,6 +583,8 @@ class UploadClient:
 
         file_id : typing.Optional[str]
 
+        relations : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -590,7 +595,7 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.upload.scrape_webpage(web_url='web_url', tenant_id='tenant_id', )
@@ -600,6 +605,7 @@ class UploadClient:
             tenant_id=tenant_id,
             sub_tenant_id=sub_tenant_id,
             file_id=file_id,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -611,6 +617,7 @@ class UploadClient:
         source_id: str,
         tenant_id: str,
         sub_tenant_id: typing.Optional[str] = None,
+        relations: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -624,6 +631,8 @@ class UploadClient:
 
         sub_tenant_id : typing.Optional[str]
 
+        relations : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -634,7 +643,7 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.upload.update_webpage(web_url='web_url', source_id='source_id', tenant_id='tenant_id', )
@@ -644,6 +653,7 @@ class UploadClient:
             source_id=source_id,
             tenant_id=tenant_id,
             sub_tenant_id=sub_tenant_id,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -675,7 +685,7 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.upload.delete_source(tenant_id='tenant_id', source_ids=['source_ids'], )
@@ -712,7 +722,7 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.upload.delete_memory(tenant_id='tenant_id', source_ids=['source_ids'], )
@@ -751,7 +761,7 @@ class UploadClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.upload.verify_processing(file_id='file_id', )
@@ -785,6 +795,7 @@ class AsyncUploadClient:
         sub_tenant_id: typing.Optional[str] = None,
         tenant_metadata: typing.Optional[str] = OMIT,
         document_metadata: typing.Optional[str] = OMIT,
+        relations: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> BatchUploadData:
         """
@@ -801,6 +812,8 @@ class AsyncUploadClient:
 
         document_metadata : typing.Optional[str]
 
+        relations : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -813,7 +826,7 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -826,6 +839,7 @@ class AsyncUploadClient:
             sub_tenant_id=sub_tenant_id,
             tenant_metadata=tenant_metadata,
             document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -839,6 +853,7 @@ class AsyncUploadClient:
         source_ids: typing.Optional[typing.List[str]] = OMIT,
         tenant_metadata: typing.Optional[str] = OMIT,
         document_metadata: typing.Optional[str] = OMIT,
+        relations: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> BatchUploadData:
         """
@@ -857,6 +872,8 @@ class AsyncUploadClient:
 
         document_metadata : typing.Optional[str]
 
+        relations : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -869,7 +886,7 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -883,6 +900,7 @@ class AsyncUploadClient:
             source_ids=source_ids,
             tenant_metadata=tenant_metadata,
             document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -895,6 +913,7 @@ class AsyncUploadClient:
         sub_tenant_id: typing.Optional[str] = None,
         tenant_metadata: typing.Optional[str] = OMIT,
         document_metadata: typing.Optional[str] = OMIT,
+        relations: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -911,6 +930,8 @@ class AsyncUploadClient:
 
         document_metadata : typing.Optional[str]
 
+        relations : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -923,7 +944,7 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -936,6 +957,7 @@ class AsyncUploadClient:
             sub_tenant_id=sub_tenant_id,
             tenant_metadata=tenant_metadata,
             document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -949,6 +971,7 @@ class AsyncUploadClient:
         sub_tenant_id: typing.Optional[str] = None,
         tenant_metadata: typing.Optional[str] = OMIT,
         document_metadata: typing.Optional[str] = OMIT,
+        relations: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -967,6 +990,8 @@ class AsyncUploadClient:
 
         document_metadata : typing.Optional[str]
 
+        relations : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -979,7 +1004,7 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -993,6 +1018,7 @@ class AsyncUploadClient:
             sub_tenant_id=sub_tenant_id,
             tenant_metadata=tenant_metadata,
             document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -1026,7 +1052,7 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI, SourceModel
+        from usecortex-ai import AsyncCortexAI, SourceModel
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -1042,10 +1068,9 @@ class AsyncUploadClient:
         self,
         *,
         tenant_id: str,
-        content: str,
+        request: MarkdownUploadRequest,
         sub_tenant_id: typing.Optional[str] = None,
-        tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
-        document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        relations: typing.Optional[Relations] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -1056,13 +1081,11 @@ class AsyncUploadClient:
         ----------
         tenant_id : str
 
-        content : str
+        request : MarkdownUploadRequest
 
         sub_tenant_id : typing.Optional[str]
 
-        tenant_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
-
-        document_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+        relations : typing.Optional[Relations]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1076,19 +1099,18 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI, MarkdownUploadRequest
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.upload.upload_markdown(tenant_id='tenant_id', content='content', )
+            await client.upload.upload_markdown(tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
         asyncio.run(main())
         """
         _response = await self._raw_client.upload_markdown(
             tenant_id=tenant_id,
-            content=content,
+            request=request,
             sub_tenant_id=sub_tenant_id,
-            tenant_metadata=tenant_metadata,
-            document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -1097,10 +1119,9 @@ class AsyncUploadClient:
         self,
         *,
         tenant_id: str,
-        content: str,
+        request: MarkdownUploadRequest,
         sub_tenant_id: typing.Optional[str] = None,
-        tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
-        document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        relations: typing.Optional[Relations] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -1111,13 +1132,11 @@ class AsyncUploadClient:
         ----------
         tenant_id : str
 
-        content : str
+        request : MarkdownUploadRequest
 
         sub_tenant_id : typing.Optional[str]
 
-        tenant_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
-
-        document_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+        relations : typing.Optional[Relations]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1131,19 +1150,18 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI, MarkdownUploadRequest
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.upload.upload_text(tenant_id='tenant_id', content='content', )
+            await client.upload.upload_text(tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
         asyncio.run(main())
         """
         _response = await self._raw_client.upload_text(
             tenant_id=tenant_id,
-            content=content,
+            request=request,
             sub_tenant_id=sub_tenant_id,
-            tenant_metadata=tenant_metadata,
-            document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -1153,10 +1171,9 @@ class AsyncUploadClient:
         *,
         source_id: str,
         tenant_id: str,
-        content: str,
+        request: MarkdownUploadRequest,
         sub_tenant_id: typing.Optional[str] = None,
-        tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
-        document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        relations: typing.Optional[Relations] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -1166,13 +1183,11 @@ class AsyncUploadClient:
 
         tenant_id : str
 
-        content : str
+        request : MarkdownUploadRequest
 
         sub_tenant_id : typing.Optional[str]
 
-        tenant_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
-
-        document_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+        relations : typing.Optional[Relations]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1186,20 +1201,19 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI, MarkdownUploadRequest
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.upload.update_markdown(source_id='source_id', tenant_id='tenant_id', content='content', )
+            await client.upload.update_markdown(source_id='source_id', tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
         asyncio.run(main())
         """
         _response = await self._raw_client.update_markdown(
             source_id=source_id,
             tenant_id=tenant_id,
-            content=content,
+            request=request,
             sub_tenant_id=sub_tenant_id,
-            tenant_metadata=tenant_metadata,
-            document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -1209,10 +1223,9 @@ class AsyncUploadClient:
         *,
         source_id: str,
         tenant_id: str,
-        content: str,
+        request: MarkdownUploadRequest,
         sub_tenant_id: typing.Optional[str] = None,
-        tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
-        document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        relations: typing.Optional[Relations] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -1222,13 +1235,11 @@ class AsyncUploadClient:
 
         tenant_id : str
 
-        content : str
+        request : MarkdownUploadRequest
 
         sub_tenant_id : typing.Optional[str]
 
-        tenant_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
-
-        document_metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+        relations : typing.Optional[Relations]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1242,20 +1253,19 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI, MarkdownUploadRequest
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.upload.update_text(source_id='source_id', tenant_id='tenant_id', content='content', )
+            await client.upload.update_text(source_id='source_id', tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
         asyncio.run(main())
         """
         _response = await self._raw_client.update_text(
             source_id=source_id,
             tenant_id=tenant_id,
-            content=content,
+            request=request,
             sub_tenant_id=sub_tenant_id,
-            tenant_metadata=tenant_metadata,
-            document_metadata=document_metadata,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -1292,7 +1302,7 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -1338,7 +1348,7 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -1357,6 +1367,7 @@ class AsyncUploadClient:
         tenant_id: str,
         sub_tenant_id: typing.Optional[str] = None,
         file_id: typing.Optional[str] = None,
+        relations: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -1370,6 +1381,8 @@ class AsyncUploadClient:
 
         file_id : typing.Optional[str]
 
+        relations : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1382,7 +1395,7 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -1394,6 +1407,7 @@ class AsyncUploadClient:
             tenant_id=tenant_id,
             sub_tenant_id=sub_tenant_id,
             file_id=file_id,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -1405,6 +1419,7 @@ class AsyncUploadClient:
         source_id: str,
         tenant_id: str,
         sub_tenant_id: typing.Optional[str] = None,
+        relations: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -1418,6 +1433,8 @@ class AsyncUploadClient:
 
         sub_tenant_id : typing.Optional[str]
 
+        relations : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1430,7 +1447,7 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -1442,6 +1459,7 @@ class AsyncUploadClient:
             source_id=source_id,
             tenant_id=tenant_id,
             sub_tenant_id=sub_tenant_id,
+            relations=relations,
             request_options=request_options,
         )
         return _response.data
@@ -1475,7 +1493,7 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -1516,7 +1534,7 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -1559,7 +1577,7 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None: