usecortex-ai 0.1.0__py3-none-any.whl

cortex_ai/tenant/raw_client.py

@@ -0,0 +1,283 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.http_response import AsyncHttpResponse, HttpResponse
+from ..core.pydantic_utilities import parse_obj_as
+from ..core.request_options import RequestOptions
+from ..errors.bad_request_error import BadRequestError
+from ..errors.forbidden_error import ForbiddenError
+from ..errors.internal_server_error import InternalServerError
+from ..errors.not_found_error import NotFoundError
+from ..errors.service_unavailable_error import ServiceUnavailableError
+from ..errors.unauthorized_error import UnauthorizedError
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..types.actual_error_response import ActualErrorResponse
+from ..types.tenant_stats import TenantStats
+
+
+class RawTenantClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def stats(
+        self,
+        *,
+        tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[TenantStats]:
+        """
+        Get tenant stats.
+
+        This endpoint returns stats for tenant.
+
+        Parameters
+        ----------
+        tenant_id : str
+
+        sub_tenant_id : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[TenantStats]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "tenant/stats",
+            method="GET",
+            params={
+                "tenant_id": tenant_id,
+                "sub_tenant_id": sub_tenant_id,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    TenantStats,
+                    parse_obj_as(
+                        type_=TenantStats,  # 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(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # 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(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # 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)
+
+
+class AsyncRawTenantClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def stats(
+        self,
+        *,
+        tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[TenantStats]:
+        """
+        Get tenant stats.
+
+        This endpoint returns stats for tenant.
+
+        Parameters
+        ----------
+        tenant_id : str
+
+        sub_tenant_id : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[TenantStats]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "tenant/stats",
+            method="GET",
+            params={
+                "tenant_id": tenant_id,
+                "sub_tenant_id": sub_tenant_id,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    TenantStats,
+                    parse_obj_as(
+                        type_=TenantStats,  # 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(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # 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(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # 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)

cortex_ai/types/__init__.py

@@ -0,0 +1,69 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .actual_error_response import ActualErrorResponse
+from .app_sources_upload_data import AppSourcesUploadData
+from .attachment_model import AttachmentModel
+from .batch_upload_data import BatchUploadData
+from .bm_25_operator_type import Bm25OperatorType
+from .content_model import ContentModel
+from .delete_memory_request import DeleteMemoryRequest
+from .embeddings_create_collection_data import EmbeddingsCreateCollectionData
+from .embeddings_delete_data import EmbeddingsDeleteData
+from .embeddings_get_data import EmbeddingsGetData
+from .embeddings_search_data import EmbeddingsSearchData
+from .error_response import ErrorResponse
+from .extended_context import ExtendedContext
+from .fetch_content_data import FetchContentData
+from .file_upload_result import FileUploadResult
+from .full_text_search_data import FullTextSearchData
+from .http_validation_error import HttpValidationError
+from .list_sources_response import ListSourcesResponse
+from .markdown_upload_request import MarkdownUploadRequest
+from .processing_status import ProcessingStatus
+from .related_chunk import RelatedChunk
+from .search_chunk import SearchChunk
+from .search_data import SearchData
+from .single_upload_data import SingleUploadData
+from .source import Source
+from .source_content import SourceContent
+from .source_model import SourceModel
+from .tenant_create_data import TenantCreateData
+from .tenant_stats import TenantStats
+from .validation_error import ValidationError
+from .validation_error_loc_item import ValidationErrorLocItem
+
+__all__ = [
+    "ActualErrorResponse",
+    "AppSourcesUploadData",
+    "AttachmentModel",
+    "BatchUploadData",
+    "Bm25OperatorType",
+    "ContentModel",
+    "DeleteMemoryRequest",
+    "EmbeddingsCreateCollectionData",
+    "EmbeddingsDeleteData",
+    "EmbeddingsGetData",
+    "EmbeddingsSearchData",
+    "ErrorResponse",
+    "ExtendedContext",
+    "FetchContentData",
+    "FileUploadResult",
+    "FullTextSearchData",
+    "HttpValidationError",
+    "ListSourcesResponse",
+    "MarkdownUploadRequest",
+    "ProcessingStatus",
+    "RelatedChunk",
+    "SearchChunk",
+    "SearchData",
+    "SingleUploadData",
+    "Source",
+    "SourceContent",
+    "SourceModel",
+    "TenantCreateData",
+    "TenantStats",
+    "ValidationError",
+    "ValidationErrorLocItem",
+]

cortex_ai/types/actual_error_response.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .error_response import ErrorResponse
+
+
+class ActualErrorResponse(UniversalBaseModel):
+    detail: ErrorResponse
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/app_sources_upload_data.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .file_upload_result import FileUploadResult
+
+
+class AppSourcesUploadData(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: str
+    uploaded: typing.List[FileUploadResult]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/attachment_model.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .content_model import ContentModel
+
+
+class AttachmentModel(UniversalBaseModel):
+    id: typing.Optional[str] = None
+    url: typing.Optional[str] = None
+    title: typing.Optional[str] = None
+    content_type: typing.Optional[str] = None
+    content_url: typing.Optional[str] = None
+    misc: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    content: typing.Optional[ContentModel] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/batch_upload_data.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .file_upload_result import FileUploadResult
+
+
+class BatchUploadData(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: str
+    uploaded: typing.List[FileUploadResult]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/bm_25_operator_type.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+Bm25OperatorType = typing.Union[typing.Literal["or", "and"], typing.Any]

cortex_ai/types/content_model.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ..core.serialization import FieldMetadata
+
+
+class ContentModel(UniversalBaseModel):
+    text: typing.Optional[str] = None
+    html_base_64: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="html_base64")] = None
+    csv_base_64: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="csv_base64")] = None
+    markdown: typing.Optional[str] = None
+    files: typing.Optional[typing.List[typing.Dict[str, typing.Optional[typing.Any]]]] = None
+    layout: typing.Optional[typing.List[typing.Dict[str, typing.Optional[typing.Any]]]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/delete_memory_request.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class DeleteMemoryRequest(UniversalBaseModel):
+    tenant_id: str
+    source_ids: typing.List[str]
+    sub_tenant_id: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/embeddings_create_collection_data.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class EmbeddingsCreateCollectionData(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: str
+    tenant_id: str
+    sub_tenant_id: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/embeddings_delete_data.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class EmbeddingsDeleteData(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: str
+    total_deleted: int
+    status: typing.Dict[str, bool]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/embeddings_get_data.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class EmbeddingsGetData(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: str
+    embeddings: typing.Dict[str, typing.List[float]]
+    not_found_chunk_ids: typing.List[str]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/embeddings_search_data.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class EmbeddingsSearchData(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: str
+    chunk_ids: typing.List[str]
+    scores: typing.List[float]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/error_response.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class ErrorResponse(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: str
+    error_code: typing.Optional[str] = None
+    data: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/extended_context.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .related_chunk import RelatedChunk
+
+
+class ExtendedContext(UniversalBaseModel):
+    related_chunks: typing.Optional[typing.List[RelatedChunk]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/fetch_content_data.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class FetchContentData(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: str
+    file_id: str
+    url: str
+    file_content: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

cortex_ai/types/file_upload_result.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class FileUploadResult(UniversalBaseModel):
+    file_id: str
+    filename: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow