usecortex-ai 0.1.0__tar.gz → 0.1.1__tar.gz

{usecortex_ai-0.1.0/src/usecortex_ai.egg-info → usecortex_ai-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: usecortex-ai
-Version: 0.1.0
+Version: 0.1.1
 Summary: The official Python SDK for the Cortex AI platform.
 Author-email: Nishkarsh Shrivastava <nishkarsh@usecortex.ai>
 License: Copyright (c) 2024 Cortex AI

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/pyproject.toml

@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
 name = "usecortex-ai"
 
 # Start with an initial version
-version = "0.1.0"
+version = "0.1.1"
 
 authors = [
   { name = "Nishkarsh Shrivastava", email = "nishkarsh@usecortex.ai" },

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/__init__.py

@@ -8,6 +8,8 @@ from .types import (
     AttachmentModel,
     BatchUploadData,
     Bm25OperatorType,
+    BodyScrapeWebpageUploadScrapeWebpagePost,
+    BodyUpdateScrapeJobUploadUpdateWebpagePatch,
     ContentModel,
     DeleteMemoryRequest,
     EmbeddingsCreateCollectionData,
@@ -18,14 +20,13 @@ from .types import (
     ExtendedContext,
     FetchContentData,
     FileUploadResult,
-    FullTextSearchData,
     HttpValidationError,
     ListSourcesResponse,
     MarkdownUploadRequest,
     ProcessingStatus,
     RelatedChunk,
+    Relations,
     SearchChunk,
-    SearchData,
     SingleUploadData,
     Source,
     SourceContent,
@@ -58,6 +59,8 @@ __all__ = [
     "BadRequestError",
     "BatchUploadData",
     "Bm25OperatorType",
+    "BodyScrapeWebpageUploadScrapeWebpagePost",
+    "BodyUpdateScrapeJobUploadUpdateWebpagePatch",
     "ContentModel",
     "CortexAI",
     "CortexAIEnvironment",
@@ -71,7 +74,6 @@ __all__ = [
     "FetchContentData",
     "FileUploadResult",
     "ForbiddenError",
-    "FullTextSearchData",
     "HttpValidationError",
     "InternalServerError",
     "ListSourcesResponse",
@@ -79,8 +81,8 @@ __all__ = [
     "NotFoundError",
     "ProcessingStatus",
     "RelatedChunk",
+    "Relations",
     "SearchChunk",
-    "SearchData",
     "ServiceUnavailableError",
     "SingleUploadData",
     "Source",

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/client.py

@@ -50,7 +50,7 @@ class CortexAI:
 
     Examples
     --------
-    from cortex-ai import CortexAI
+    from usecortex-ai import CortexAI
 
     client = CortexAI(token="YOUR_TOKEN", )
     """
@@ -115,7 +115,7 @@ class CortexAI:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.root_get()
@@ -157,7 +157,7 @@ class AsyncCortexAI:
 
     Examples
     --------
-    from cortex-ai import AsyncCortexAI
+    from usecortex-ai import AsyncCortexAI
 
     client = AsyncCortexAI(token="YOUR_TOKEN", )
     """
@@ -224,7 +224,7 @@ class AsyncCortexAI:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/embeddings/client.py

@@ -71,7 +71,7 @@ class EmbeddingsClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.embeddings.delete(chunk_ids=['chunk_ids'], tenant_id='tenant_id', )
@@ -128,7 +128,7 @@ class EmbeddingsClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.embeddings.search(embeddings=[1.1], tenant_id='tenant_id', )
@@ -176,7 +176,7 @@ class EmbeddingsClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.embeddings.get_by_chunk_ids(chunk_ids=['chunk_ids'], tenant_id='tenant_id', )
@@ -211,7 +211,7 @@ class EmbeddingsClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.embeddings.create_collection(tenant_id='tenant_id', )
@@ -279,7 +279,7 @@ class AsyncEmbeddingsClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -340,7 +340,7 @@ class AsyncEmbeddingsClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -392,7 +392,7 @@ class AsyncEmbeddingsClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -431,7 +431,7 @@ class AsyncEmbeddingsClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/fetch/client.py

@@ -59,7 +59,7 @@ class FetchClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.fetch.fetch_content(file_id='file_id', file_type='file_type', tenant_id='tenant_id', )
@@ -125,7 +125,7 @@ class AsyncFetchClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/search/client.py

@@ -5,8 +5,7 @@ import typing
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
 from ..types.bm_25_operator_type import Bm25OperatorType
-from ..types.full_text_search_data import FullTextSearchData
-from ..types.search_data import SearchData
+from ..types.search_chunk import SearchChunk
 from .raw_client import AsyncRawSearchClient, RawSearchClient
 from .types.alpha import Alpha
 
@@ -98,7 +97,7 @@ class SearchClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.search.qna(question='question', session_id='session_id', tenant_id='tenant_id', )
@@ -136,7 +135,7 @@ class SearchClient:
         recency_bias: typing.Optional[float] = OMIT,
         num_related_chunks: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> SearchData:
+    ) -> typing.List[SearchChunk]:
         """
         Search for content within indexed sources using semantic and keyword search capabilities.
 
@@ -179,12 +178,12 @@ class SearchClient:
 
         Returns
         -------
-        SearchData
+        typing.List[SearchChunk]
             Successful Response
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.search.retrieve(query='query', tenant_id='tenant_id', )
@@ -210,7 +209,7 @@ class SearchClient:
         operator: typing.Optional[Bm25OperatorType] = OMIT,
         max_chunks: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> FullTextSearchData:
+    ) -> typing.List[SearchChunk]:
         """
         Full text search endpoint for Cortex customers.
         Performs full text search with configurable operators for precise text matching against the Findr backend.
@@ -248,12 +247,12 @@ class SearchClient:
 
         Returns
         -------
-        FullTextSearchData
+        typing.List[SearchChunk]
             Successful Response
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.search.full_text_search(query='query', tenant_id='tenant_id', )
@@ -355,7 +354,7 @@ class AsyncSearchClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -395,7 +394,7 @@ class AsyncSearchClient:
         recency_bias: typing.Optional[float] = OMIT,
         num_related_chunks: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> SearchData:
+    ) -> typing.List[SearchChunk]:
         """
         Search for content within indexed sources using semantic and keyword search capabilities.
 
@@ -438,14 +437,14 @@ class AsyncSearchClient:
 
         Returns
         -------
-        SearchData
+        typing.List[SearchChunk]
             Successful Response
 
         Examples
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -473,7 +472,7 @@ class AsyncSearchClient:
         operator: typing.Optional[Bm25OperatorType] = OMIT,
         max_chunks: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> FullTextSearchData:
+    ) -> typing.List[SearchChunk]:
         """
         Full text search endpoint for Cortex customers.
         Performs full text search with configurable operators for precise text matching against the Findr backend.
@@ -511,14 +510,14 @@ class AsyncSearchClient:
 
         Returns
         -------
-        FullTextSearchData
+        typing.List[SearchChunk]
             Successful Response
 
         Examples
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/search/raw_client.py

@@ -18,8 +18,7 @@ from ..errors.unauthorized_error import UnauthorizedError
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.actual_error_response import ActualErrorResponse
 from ..types.bm_25_operator_type import Bm25OperatorType
-from ..types.full_text_search_data import FullTextSearchData
-from ..types.search_data import SearchData
+from ..types.search_chunk import SearchChunk
 from .types.alpha import Alpha
 
 # this is used as the default value for optional parameters
@@ -230,7 +229,7 @@ class RawSearchClient:
         recency_bias: typing.Optional[float] = OMIT,
         num_related_chunks: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[SearchData]:
+    ) -> HttpResponse[typing.List[SearchChunk]]:
         """
         Search for content within indexed sources using semantic and keyword search capabilities.
 
@@ -273,7 +272,7 @@ class RawSearchClient:
 
         Returns
         -------
-        HttpResponse[SearchData]
+        HttpResponse[typing.List[SearchChunk]]
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -297,9 +296,9 @@ class RawSearchClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    SearchData,
+                    typing.List[SearchChunk],
                     parse_obj_as(
-                        type_=SearchData,  # type: ignore
+                        type_=typing.List[SearchChunk],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -395,7 +394,7 @@ class RawSearchClient:
         operator: typing.Optional[Bm25OperatorType] = OMIT,
         max_chunks: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[FullTextSearchData]:
+    ) -> HttpResponse[typing.List[SearchChunk]]:
         """
         Full text search endpoint for Cortex customers.
         Performs full text search with configurable operators for precise text matching against the Findr backend.
@@ -433,7 +432,7 @@ class RawSearchClient:
 
         Returns
         -------
-        HttpResponse[FullTextSearchData]
+        HttpResponse[typing.List[SearchChunk]]
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -455,9 +454,9 @@ class RawSearchClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    FullTextSearchData,
+                    typing.List[SearchChunk],
                     parse_obj_as(
-                        type_=FullTextSearchData,  # type: ignore
+                        type_=typing.List[SearchChunk],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -749,7 +748,7 @@ class AsyncRawSearchClient:
         recency_bias: typing.Optional[float] = OMIT,
         num_related_chunks: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[SearchData]:
+    ) -> AsyncHttpResponse[typing.List[SearchChunk]]:
         """
         Search for content within indexed sources using semantic and keyword search capabilities.
 
@@ -792,7 +791,7 @@ class AsyncRawSearchClient:
 
         Returns
         -------
-        AsyncHttpResponse[SearchData]
+        AsyncHttpResponse[typing.List[SearchChunk]]
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -816,9 +815,9 @@ class AsyncRawSearchClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    SearchData,
+                    typing.List[SearchChunk],
                     parse_obj_as(
-                        type_=SearchData,  # type: ignore
+                        type_=typing.List[SearchChunk],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -914,7 +913,7 @@ class AsyncRawSearchClient:
         operator: typing.Optional[Bm25OperatorType] = OMIT,
         max_chunks: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[FullTextSearchData]:
+    ) -> AsyncHttpResponse[typing.List[SearchChunk]]:
         """
         Full text search endpoint for Cortex customers.
         Performs full text search with configurable operators for precise text matching against the Findr backend.
@@ -952,7 +951,7 @@ class AsyncRawSearchClient:
 
         Returns
         -------
-        AsyncHttpResponse[FullTextSearchData]
+        AsyncHttpResponse[typing.List[SearchChunk]]
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -974,9 +973,9 @@ class AsyncRawSearchClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    FullTextSearchData,
+                    typing.List[SearchChunk],
                     parse_obj_as(
-                        type_=FullTextSearchData,  # type: ignore
+                        type_=typing.List[SearchChunk],  # type: ignore
                         object_=_response.json(),
                     ),
                 )

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/sources/client.py

@@ -50,7 +50,7 @@ class SourcesClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.sources.get_all(tenant_id='tenant_id', )
@@ -84,7 +84,7 @@ class SourcesClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.sources.get_by_ids(tenant_id='tenant_id', source_ids=['source_ids'], )
@@ -136,7 +136,7 @@ class AsyncSourcesClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
@@ -174,7 +174,7 @@ class AsyncSourcesClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/tenant/client.py

@@ -51,7 +51,7 @@ class TenantClient:
 
         Examples
         --------
-        from cortex-ai import CortexAI
+        from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
         client.tenant.stats(tenant_id='tenant_id', )
@@ -107,7 +107,7 @@ class AsyncTenantClient:
         --------
         import asyncio
 
-        from cortex-ai import AsyncCortexAI
+        from usecortex-ai import AsyncCortexAI
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/types/__init__.py

@@ -7,6 +7,8 @@ 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 .body_scrape_webpage_upload_scrape_webpage_post import BodyScrapeWebpageUploadScrapeWebpagePost
+from .body_update_scrape_job_upload_update_webpage_patch import BodyUpdateScrapeJobUploadUpdateWebpagePatch
 from .content_model import ContentModel
 from .delete_memory_request import DeleteMemoryRequest
 from .embeddings_create_collection_data import EmbeddingsCreateCollectionData
@@ -17,14 +19,13 @@ 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 .relations import Relations
 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
@@ -40,6 +41,8 @@ __all__ = [
     "AttachmentModel",
     "BatchUploadData",
     "Bm25OperatorType",
+    "BodyScrapeWebpageUploadScrapeWebpagePost",
+    "BodyUpdateScrapeJobUploadUpdateWebpagePatch",
     "ContentModel",
     "DeleteMemoryRequest",
     "EmbeddingsCreateCollectionData",
@@ -50,14 +53,13 @@ __all__ = [
     "ExtendedContext",
     "FetchContentData",
     "FileUploadResult",
-    "FullTextSearchData",
     "HttpValidationError",
     "ListSourcesResponse",
     "MarkdownUploadRequest",
     "ProcessingStatus",
     "RelatedChunk",
+    "Relations",
     "SearchChunk",
-    "SearchData",
     "SingleUploadData",
     "Source",
     "SourceContent",

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/types/app_sources_upload_data.py

@@ -9,7 +9,7 @@ from .file_upload_result import FileUploadResult
 
 class AppSourcesUploadData(UniversalBaseModel):
     success: typing.Optional[bool] = None
-    message: str
+    message: typing.Optional[str] = None
     uploaded: typing.List[FileUploadResult]
 
     if IS_PYDANTIC_V2:

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/types/batch_upload_data.py

@@ -9,7 +9,7 @@ from .file_upload_result import FileUploadResult
 
 class BatchUploadData(UniversalBaseModel):
     success: typing.Optional[bool] = None
-    message: str
+    message: typing.Optional[str] = None
     uploaded: typing.List[FileUploadResult]
 
     if IS_PYDANTIC_V2:

usecortex_ai-0.1.1/src/cortex_ai/types/body_scrape_webpage_upload_scrape_webpage_post.py

@@ -0,0 +1,19 @@
+# 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 BodyScrapeWebpageUploadScrapeWebpagePost(UniversalBaseModel):
+    relations: 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

usecortex_ai-0.1.1/src/cortex_ai/types/body_update_scrape_job_upload_update_webpage_patch.py

@@ -0,0 +1,19 @@
+# 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 BodyUpdateScrapeJobUploadUpdateWebpagePatch(UniversalBaseModel):
+    relations: 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

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/types/embeddings_create_collection_data.py

@@ -8,7 +8,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 class EmbeddingsCreateCollectionData(UniversalBaseModel):
     success: typing.Optional[bool] = None
-    message: str
+    message: typing.Optional[str] = None
     tenant_id: str
     sub_tenant_id: str
 

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/types/embeddings_delete_data.py

@@ -8,7 +8,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 class EmbeddingsDeleteData(UniversalBaseModel):
     success: typing.Optional[bool] = None
-    message: str
+    message: typing.Optional[str] = None
     total_deleted: int
     status: typing.Dict[str, bool]
 

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/types/embeddings_get_data.py

@@ -8,7 +8,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 class EmbeddingsGetData(UniversalBaseModel):
     success: typing.Optional[bool] = None
-    message: str
+    message: typing.Optional[str] = None
     embeddings: typing.Dict[str, typing.List[float]]
     not_found_chunk_ids: typing.List[str]
 

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/types/embeddings_search_data.py

@@ -8,7 +8,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 class EmbeddingsSearchData(UniversalBaseModel):
     success: typing.Optional[bool] = None
-    message: str
+    message: typing.Optional[str] = None
     chunk_ids: typing.List[str]
     scores: typing.List[float]
 

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/types/error_response.py

@@ -8,7 +8,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 class ErrorResponse(UniversalBaseModel):
     success: typing.Optional[bool] = None
-    message: str
+    message: typing.Optional[str] = None
     error_code: typing.Optional[str] = None
     data: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
 

{usecortex_ai-0.1.0 → usecortex_ai-0.1.1}/src/cortex_ai/types/extended_context.py

@@ -8,7 +8,7 @@ from .related_chunk import RelatedChunk
 
 
 class ExtendedContext(UniversalBaseModel):
-    related_chunks: typing.Optional[typing.List[RelatedChunk]] = None
+    relations: 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