usecortex-ai 0.4.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

usecortex_ai/upload/raw_client.py

@@ -18,12 +18,10 @@ from ..errors.service_unavailable_error import ServiceUnavailableError
 from ..errors.unauthorized_error import UnauthorizedError
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.add_memory_response import AddMemoryResponse
-from ..types.app_sources_upload_data import AppSourcesUploadData
 from ..types.delete_user_memory_response import DeleteUserMemoryResponse
 from ..types.memory_item import MemoryItem
 from ..types.processing_status import ProcessingStatus
 from ..types.source_upload_response import SourceUploadResponse
-from .types.body_upload_app_ingestion_upload_app_post_app_sources import BodyUploadAppIngestionUploadAppPostAppSources
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -33,27 +31,20 @@ class RawUploadClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def upload_document(
+    def upload_knowledge(
         self,
         *,
-        files: typing.List[core.File],
         tenant_id: str,
         sub_tenant_id: typing.Optional[str] = OMIT,
         upsert: typing.Optional[bool] = OMIT,
+        files: typing.Optional[typing.List[core.File]] = OMIT,
         file_metadata: typing.Optional[str] = OMIT,
+        app_sources: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[SourceUploadResponse]:
         """
-        Upload one or more documents for ingestion.
-        Supports both single and batch uploads.
-        For single file: send one file with the 'files' field.
-        For batch: send multiple files with the 'files' field (FastAPI will parse as list).
-
         Parameters
         ----------
-        files : typing.List[core.File]
-            See core.File for more documentation
-
         tenant_id : str
             Unique identifier for the tenant/organization
 
@@ -61,9 +52,16 @@ class RawUploadClient:
             Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
 
         upsert : typing.Optional[bool]
+            If true, update existing sources with the same id.
+
+        files : typing.Optional[typing.List[core.File]]
+            See core.File for more documentation
 
         file_metadata : typing.Optional[str]
-            JSON Array of file metadata objects
+            JSON array of file metadata objects; length must match files when provided.
+
+        app_sources : typing.Optional[str]
+            JSON: single source object or array of app-generated sources to index. Omit when only uploading files.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -74,16 +72,17 @@ class RawUploadClient:
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            "ingestion/upload-document",
+            "ingestion/upload_knowledge",
             method="POST",
             data={
                 "tenant_id": tenant_id,
                 "sub_tenant_id": sub_tenant_id,
                 "upsert": upsert,
                 "file_metadata": file_metadata,
+                "app_sources": app_sources,
             },
             files={
-                "files": files,
+                **({"files": files} if files is not None else {}),
             },
             request_options=request_options,
             omit=OMIT,
@@ -181,150 +180,6 @@ class RawUploadClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
-    def upload_app_sources(
-        self,
-        *,
-        app_sources: BodyUploadAppIngestionUploadAppPostAppSources,
-        tenant_id: str,
-        sub_tenant_id: typing.Optional[str] = OMIT,
-        upsert: typing.Optional[bool] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[AppSourcesUploadData]:
-        """
-        Upload structured sources generated by first-party or third-party apps.
-        Supports both single and batch uploads.
-
-        Parameters
-        ----------
-        app_sources : BodyUploadAppIngestionUploadAppPostAppSources
-            Single source object or list of structured source objects containing app-generated data to be indexed
-
-        tenant_id : str
-            Unique identifier for the tenant/organization
-
-        sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
-
-        upsert : typing.Optional[bool]
-            If true, update existing sources with the same source_id. Defaults to True.
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        HttpResponse[AppSourcesUploadData]
-            Successful Response
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "ingestion/upload-app",
-            method="POST",
-            json={
-                "app_sources": convert_and_respect_annotation_metadata(
-                    object_=app_sources, annotation=BodyUploadAppIngestionUploadAppPostAppSources, direction="write"
-                ),
-                "tenant_id": tenant_id,
-                "sub_tenant_id": sub_tenant_id,
-                "upsert": upsert,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    AppSourcesUploadData,
-                    parse_obj_as(
-                        type_=AppSourcesUploadData,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-                return HttpResponse(response=_response, data=_data)
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 401:
-                raise UnauthorizedError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 403:
-                raise ForbiddenError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 503:
-                raise ServiceUnavailableError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
-        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
-
     def verify_processing(
         self,
         *,
@@ -334,12 +189,6 @@ class RawUploadClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[ProcessingStatus]:
         """
-        Check the current processing status of your uploaded content.
-
-        This endpoint allows you to monitor the progress of documents, text, or other content you've uploaded. Simply provide the file ID to see whether processing is complete, still in progress, or if any errors occurred.
-
-        Use this to determine when your content is ready for search and retrieval, or to troubleshoot any processing issues.
-
         Parameters
         ----------
         file_id : str
@@ -360,7 +209,7 @@ class RawUploadClient:
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            "ingestion/verify-processing",
+            "ingestion/verify_processing",
             method="POST",
             params={
                 "file_id": file_id,
@@ -471,24 +320,6 @@ class RawUploadClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[AddMemoryResponse]:
         """
-        Add memories for indexing and retrieval.
-
-            This API provides a unified entry point for ingesting various types of content:
-
-            **Content Types Supported:**
-            - Raw text content (`text` field)
-            - Markdown formatted content (`text` + `is_markdown=true`)
-            - User/assistant conversation pairs (`user_assistant_pairs` field)
-
-            **Key Features:**
-            - **Inference Mode (`infer=true`)**: When enabled, the system extracts additional
-              insights and implicit information from your content. Useful for conversation
-              analysis and knowledge extraction.
-            - **Direct Mode (`infer=false`)**: Content is chunked and indexed as-is without
-              additional processing.
-
-            Use `/ingestion/verify-processing` endpoint to check the processing status.
-
         Parameters
         ----------
         memories : typing.Sequence[MemoryItem]
@@ -629,14 +460,6 @@ class RawUploadClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[DeleteUserMemoryResponse]:
         """
-        Permanently remove a specific memory from storage.
-
-            This API allows you to delete a memory by its unique identifier.
-            Once deleted, the memory cannot be recovered, so use this operation carefully.
-
-            The memory will be removed from your tenant's storage and will no longer
-            appear in search results or memory listings.
-
         Parameters
         ----------
         tenant_id : str
@@ -763,27 +586,20 @@ class AsyncRawUploadClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def upload_document(
+    async def upload_knowledge(
         self,
         *,
-        files: typing.List[core.File],
         tenant_id: str,
         sub_tenant_id: typing.Optional[str] = OMIT,
         upsert: typing.Optional[bool] = OMIT,
+        files: typing.Optional[typing.List[core.File]] = OMIT,
         file_metadata: typing.Optional[str] = OMIT,
+        app_sources: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[SourceUploadResponse]:
         """
-        Upload one or more documents for ingestion.
-        Supports both single and batch uploads.
-        For single file: send one file with the 'files' field.
-        For batch: send multiple files with the 'files' field (FastAPI will parse as list).
-
         Parameters
         ----------
-        files : typing.List[core.File]
-            See core.File for more documentation
-
         tenant_id : str
             Unique identifier for the tenant/organization
 
@@ -791,9 +607,16 @@ class AsyncRawUploadClient:
             Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
 
         upsert : typing.Optional[bool]
+            If true, update existing sources with the same id.
+
+        files : typing.Optional[typing.List[core.File]]
+            See core.File for more documentation
 
         file_metadata : typing.Optional[str]
-            JSON Array of file metadata objects
+            JSON array of file metadata objects; length must match files when provided.
+
+        app_sources : typing.Optional[str]
+            JSON: single source object or array of app-generated sources to index. Omit when only uploading files.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -804,16 +627,17 @@ class AsyncRawUploadClient:
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "ingestion/upload-document",
+            "ingestion/upload_knowledge",
             method="POST",
             data={
                 "tenant_id": tenant_id,
                 "sub_tenant_id": sub_tenant_id,
                 "upsert": upsert,
                 "file_metadata": file_metadata,
+                "app_sources": app_sources,
             },
             files={
-                "files": files,
+                **({"files": files} if files is not None else {}),
             },
             request_options=request_options,
             omit=OMIT,
@@ -911,150 +735,6 @@ class AsyncRawUploadClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
-    async def upload_app_sources(
-        self,
-        *,
-        app_sources: BodyUploadAppIngestionUploadAppPostAppSources,
-        tenant_id: str,
-        sub_tenant_id: typing.Optional[str] = OMIT,
-        upsert: typing.Optional[bool] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[AppSourcesUploadData]:
-        """
-        Upload structured sources generated by first-party or third-party apps.
-        Supports both single and batch uploads.
-
-        Parameters
-        ----------
-        app_sources : BodyUploadAppIngestionUploadAppPostAppSources
-            Single source object or list of structured source objects containing app-generated data to be indexed
-
-        tenant_id : str
-            Unique identifier for the tenant/organization
-
-        sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
-
-        upsert : typing.Optional[bool]
-            If true, update existing sources with the same source_id. Defaults to True.
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        AsyncHttpResponse[AppSourcesUploadData]
-            Successful Response
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "ingestion/upload-app",
-            method="POST",
-            json={
-                "app_sources": convert_and_respect_annotation_metadata(
-                    object_=app_sources, annotation=BodyUploadAppIngestionUploadAppPostAppSources, direction="write"
-                ),
-                "tenant_id": tenant_id,
-                "sub_tenant_id": sub_tenant_id,
-                "upsert": upsert,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    AppSourcesUploadData,
-                    parse_obj_as(
-                        type_=AppSourcesUploadData,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-                return AsyncHttpResponse(response=_response, data=_data)
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 401:
-                raise UnauthorizedError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 403:
-                raise ForbiddenError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 503:
-                raise ServiceUnavailableError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
-        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
-
     async def verify_processing(
         self,
         *,
@@ -1064,12 +744,6 @@ class AsyncRawUploadClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[ProcessingStatus]:
         """
-        Check the current processing status of your uploaded content.
-
-        This endpoint allows you to monitor the progress of documents, text, or other content you've uploaded. Simply provide the file ID to see whether processing is complete, still in progress, or if any errors occurred.
-
-        Use this to determine when your content is ready for search and retrieval, or to troubleshoot any processing issues.
-
         Parameters
         ----------
         file_id : str
@@ -1090,7 +764,7 @@ class AsyncRawUploadClient:
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "ingestion/verify-processing",
+            "ingestion/verify_processing",
             method="POST",
             params={
                 "file_id": file_id,
@@ -1201,24 +875,6 @@ class AsyncRawUploadClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[AddMemoryResponse]:
         """
-        Add memories for indexing and retrieval.
-
-            This API provides a unified entry point for ingesting various types of content:
-
-            **Content Types Supported:**
-            - Raw text content (`text` field)
-            - Markdown formatted content (`text` + `is_markdown=true`)
-            - User/assistant conversation pairs (`user_assistant_pairs` field)
-
-            **Key Features:**
-            - **Inference Mode (`infer=true`)**: When enabled, the system extracts additional
-              insights and implicit information from your content. Useful for conversation
-              analysis and knowledge extraction.
-            - **Direct Mode (`infer=false`)**: Content is chunked and indexed as-is without
-              additional processing.
-
-            Use `/ingestion/verify-processing` endpoint to check the processing status.
-
         Parameters
         ----------
         memories : typing.Sequence[MemoryItem]
@@ -1359,14 +1015,6 @@ class AsyncRawUploadClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[DeleteUserMemoryResponse]:
         """
-        Permanently remove a specific memory from storage.
-
-            This API allows you to delete a memory by its unique identifier.
-            Once deleted, the memory cannot be recovered, so use this operation carefully.
-
-            The memory will be removed from your tenant's storage and will no longer
-            appear in search results or memory listings.
-
         Parameters
         ----------
         tenant_id : str

{usecortex_ai-0.4.0.dist-info → usecortex_ai-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: usecortex-ai
-Version: 0.4.0
+Version: 0.5.1
 Summary: The official Python SDK for the Cortex AI platform.
 Author-email: Soham Ratnaparkhi <soham@usecortex.ai>
 License: Copyright (c) 2024 Cortex AI