usecortex-ai 0.1.1__py3-none-any.whl → 0.2.1__py3-none-any.whl

{cortex_ai → usecortex_ai}/upload/client.py

@@ -7,7 +7,6 @@ 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
@@ -294,8 +293,10 @@ class UploadClient:
         self,
         *,
         tenant_id: str,
-        request: MarkdownUploadRequest,
+        content: str,
         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:
@@ -307,10 +308,14 @@ class UploadClient:
         ----------
         tenant_id : str
 
-        request : MarkdownUploadRequest
+        content : str
 
         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]
@@ -323,15 +328,17 @@ class UploadClient:
 
         Examples
         --------
-        from usecortex-ai import CortexAI, MarkdownUploadRequest
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.upload.upload_markdown(tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
+        client.upload.upload_markdown(tenant_id='tenant_id', content='content', )
         """
         _response = self._raw_client.upload_markdown(
             tenant_id=tenant_id,
-            request=request,
+            content=content,
             sub_tenant_id=sub_tenant_id,
+            tenant_metadata=tenant_metadata,
+            document_metadata=document_metadata,
             relations=relations,
             request_options=request_options,
         )
@@ -341,8 +348,10 @@ class UploadClient:
         self,
         *,
         tenant_id: str,
-        request: MarkdownUploadRequest,
+        content: str,
         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:
@@ -354,10 +363,14 @@ class UploadClient:
         ----------
         tenant_id : str
 
-        request : MarkdownUploadRequest
+        content : str
 
         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]
@@ -370,15 +383,17 @@ class UploadClient:
 
         Examples
         --------
-        from usecortex-ai import CortexAI, MarkdownUploadRequest
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.upload.upload_text(tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
+        client.upload.upload_text(tenant_id='tenant_id', content='content', )
         """
         _response = self._raw_client.upload_text(
             tenant_id=tenant_id,
-            request=request,
+            content=content,
             sub_tenant_id=sub_tenant_id,
+            tenant_metadata=tenant_metadata,
+            document_metadata=document_metadata,
             relations=relations,
             request_options=request_options,
         )
@@ -389,8 +404,10 @@ class UploadClient:
         *,
         source_id: str,
         tenant_id: str,
-        request: MarkdownUploadRequest,
+        content: str,
         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:
@@ -401,10 +418,14 @@ class UploadClient:
 
         tenant_id : str
 
-        request : MarkdownUploadRequest
+        content : str
 
         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]
@@ -417,16 +438,18 @@ class UploadClient:
 
         Examples
         --------
-        from usecortex-ai import CortexAI, MarkdownUploadRequest
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.upload.update_markdown(source_id='source_id', tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
+        client.upload.update_markdown(source_id='source_id', tenant_id='tenant_id', content='content', )
         """
         _response = self._raw_client.update_markdown(
             source_id=source_id,
             tenant_id=tenant_id,
-            request=request,
+            content=content,
             sub_tenant_id=sub_tenant_id,
+            tenant_metadata=tenant_metadata,
+            document_metadata=document_metadata,
             relations=relations,
             request_options=request_options,
         )
@@ -437,8 +460,10 @@ class UploadClient:
         *,
         source_id: str,
         tenant_id: str,
-        request: MarkdownUploadRequest,
+        content: str,
         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:
@@ -449,10 +474,14 @@ class UploadClient:
 
         tenant_id : str
 
-        request : MarkdownUploadRequest
+        content : str
 
         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]
@@ -465,16 +494,18 @@ class UploadClient:
 
         Examples
         --------
-        from usecortex-ai import CortexAI, MarkdownUploadRequest
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.upload.update_text(source_id='source_id', tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
+        client.upload.update_text(source_id='source_id', tenant_id='tenant_id', content='content', )
         """
         _response = self._raw_client.update_text(
             source_id=source_id,
             tenant_id=tenant_id,
-            request=request,
+            content=content,
             sub_tenant_id=sub_tenant_id,
+            tenant_metadata=tenant_metadata,
+            document_metadata=document_metadata,
             relations=relations,
             request_options=request_options,
         )
@@ -528,8 +559,8 @@ class UploadClient:
         self,
         *,
         tenant_id: str,
-        embeddings: typing.Dict[str, typing.Sequence[float]],
         sub_tenant_id: typing.Optional[str] = None,
+        embeddings: typing.Optional[typing.Dict[str, typing.Sequence[float]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -537,11 +568,10 @@ class UploadClient:
         ----------
         tenant_id : str
 
-        embeddings : typing.Dict[str, typing.Sequence[float]]
-            Dictionary with chunk_id as key and embedding array as value
-
         sub_tenant_id : typing.Optional[str]
 
+        embeddings : typing.Optional[typing.Dict[str, typing.Sequence[float]]]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -555,10 +585,10 @@ class UploadClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.upload.update_embeddings(tenant_id='tenant_id', embeddings={'key': [1.1]}, )
+        client.upload.update_embeddings(tenant_id='tenant_id', )
         """
         _response = self._raw_client.update_embeddings(
-            tenant_id=tenant_id, embeddings=embeddings, sub_tenant_id=sub_tenant_id, request_options=request_options
+            tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, embeddings=embeddings, request_options=request_options
         )
         return _response.data
 
@@ -1068,8 +1098,10 @@ class AsyncUploadClient:
         self,
         *,
         tenant_id: str,
-        request: MarkdownUploadRequest,
+        content: str,
         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:
@@ -1081,10 +1113,14 @@ class AsyncUploadClient:
         ----------
         tenant_id : str
 
-        request : MarkdownUploadRequest
+        content : str
 
         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]
@@ -1099,17 +1135,19 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from usecortex-ai import AsyncCortexAI, MarkdownUploadRequest
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.upload.upload_markdown(tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
+            await client.upload.upload_markdown(tenant_id='tenant_id', content='content', )
         asyncio.run(main())
         """
         _response = await self._raw_client.upload_markdown(
             tenant_id=tenant_id,
-            request=request,
+            content=content,
             sub_tenant_id=sub_tenant_id,
+            tenant_metadata=tenant_metadata,
+            document_metadata=document_metadata,
             relations=relations,
             request_options=request_options,
         )
@@ -1119,8 +1157,10 @@ class AsyncUploadClient:
         self,
         *,
         tenant_id: str,
-        request: MarkdownUploadRequest,
+        content: str,
         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:
@@ -1132,10 +1172,14 @@ class AsyncUploadClient:
         ----------
         tenant_id : str
 
-        request : MarkdownUploadRequest
+        content : str
 
         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]
@@ -1150,17 +1194,19 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from usecortex-ai import AsyncCortexAI, MarkdownUploadRequest
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.upload.upload_text(tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
+            await client.upload.upload_text(tenant_id='tenant_id', content='content', )
         asyncio.run(main())
         """
         _response = await self._raw_client.upload_text(
             tenant_id=tenant_id,
-            request=request,
+            content=content,
             sub_tenant_id=sub_tenant_id,
+            tenant_metadata=tenant_metadata,
+            document_metadata=document_metadata,
             relations=relations,
             request_options=request_options,
         )
@@ -1171,8 +1217,10 @@ class AsyncUploadClient:
         *,
         source_id: str,
         tenant_id: str,
-        request: MarkdownUploadRequest,
+        content: str,
         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:
@@ -1183,10 +1231,14 @@ class AsyncUploadClient:
 
         tenant_id : str
 
-        request : MarkdownUploadRequest
+        content : str
 
         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]
@@ -1201,18 +1253,20 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from usecortex-ai import AsyncCortexAI, MarkdownUploadRequest
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.upload.update_markdown(source_id='source_id', tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
+            await client.upload.update_markdown(source_id='source_id', tenant_id='tenant_id', content='content', )
         asyncio.run(main())
         """
         _response = await self._raw_client.update_markdown(
             source_id=source_id,
             tenant_id=tenant_id,
-            request=request,
+            content=content,
             sub_tenant_id=sub_tenant_id,
+            tenant_metadata=tenant_metadata,
+            document_metadata=document_metadata,
             relations=relations,
             request_options=request_options,
         )
@@ -1223,8 +1277,10 @@ class AsyncUploadClient:
         *,
         source_id: str,
         tenant_id: str,
-        request: MarkdownUploadRequest,
+        content: str,
         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:
@@ -1235,10 +1291,14 @@ class AsyncUploadClient:
 
         tenant_id : str
 
-        request : MarkdownUploadRequest
+        content : str
 
         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]
@@ -1253,18 +1313,20 @@ class AsyncUploadClient:
         --------
         import asyncio
 
-        from usecortex-ai import AsyncCortexAI, MarkdownUploadRequest
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.upload.update_text(source_id='source_id', tenant_id='tenant_id', request=MarkdownUploadRequest(content='content', ), )
+            await client.upload.update_text(source_id='source_id', tenant_id='tenant_id', content='content', )
         asyncio.run(main())
         """
         _response = await self._raw_client.update_text(
             source_id=source_id,
             tenant_id=tenant_id,
-            request=request,
+            content=content,
             sub_tenant_id=sub_tenant_id,
+            tenant_metadata=tenant_metadata,
+            document_metadata=document_metadata,
             relations=relations,
             request_options=request_options,
         )
@@ -1322,8 +1384,8 @@ class AsyncUploadClient:
         self,
         *,
         tenant_id: str,
-        embeddings: typing.Dict[str, typing.Sequence[float]],
         sub_tenant_id: typing.Optional[str] = None,
+        embeddings: typing.Optional[typing.Dict[str, typing.Sequence[float]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SingleUploadData:
         """
@@ -1331,11 +1393,10 @@ class AsyncUploadClient:
         ----------
         tenant_id : str
 
-        embeddings : typing.Dict[str, typing.Sequence[float]]
-            Dictionary with chunk_id as key and embedding array as value
-
         sub_tenant_id : typing.Optional[str]
 
+        embeddings : typing.Optional[typing.Dict[str, typing.Sequence[float]]]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1352,11 +1413,11 @@ class AsyncUploadClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.upload.update_embeddings(tenant_id='tenant_id', embeddings={'key': [1.1]}, )
+            await client.upload.update_embeddings(tenant_id='tenant_id', )
         asyncio.run(main())
         """
         _response = await self._raw_client.update_embeddings(
-            tenant_id=tenant_id, embeddings=embeddings, sub_tenant_id=sub_tenant_id, request_options=request_options
+            tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, embeddings=embeddings, request_options=request_options
         )
         return _response.data