usecortex-ai 0.2.0__tar.gz → 0.2.1__tar.gz

{usecortex_ai-0.2.0/src/usecortex_ai.egg-info → usecortex_ai-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: usecortex-ai
-Version: 0.2.0
+Version: 0.2.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.2.0 → usecortex_ai-0.2.1}/pyproject.toml

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

{usecortex_ai-0.2.0 → usecortex_ai-0.2.1}/src/usecortex_ai/__init__.py

@@ -31,6 +31,7 @@ from .types import (
     Source,
     SourceContent,
     SourceModel,
+    SubTenantIdsData,
     TenantCreateData,
     TenantStats,
     ValidationError,
@@ -88,6 +89,7 @@ __all__ = [
     "Source",
     "SourceContent",
     "SourceModel",
+    "SubTenantIdsData",
     "TenantCreateData",
     "TenantStats",
     "UnauthorizedError",

{usecortex_ai-0.2.0 → usecortex_ai-0.2.1}/src/usecortex_ai/embeddings/client.py

@@ -84,8 +84,8 @@ class EmbeddingsClient:
     def search(
         self,
         *,
-        embeddings: typing.Sequence[float],
         tenant_id: str,
+        embeddings: typing.Optional[typing.Sequence[float]] = OMIT,
         sub_tenant_id: typing.Optional[str] = OMIT,
         max_chunks: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -109,11 +109,10 @@ class EmbeddingsClient:
 
         Parameters
         ----------
-        embeddings : typing.Sequence[float]
-            Single embedding vector for search
-
         tenant_id : str
 
+        embeddings : typing.Optional[typing.Sequence[float]]
+
         sub_tenant_id : typing.Optional[str]
 
         max_chunks : typing.Optional[int]
@@ -131,11 +130,11 @@ class EmbeddingsClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.embeddings.search(embeddings=[1.1], tenant_id='tenant_id', )
+        client.embeddings.search(tenant_id='tenant_id', )
         """
         _response = self._raw_client.search(
-            embeddings=embeddings,
             tenant_id=tenant_id,
+            embeddings=embeddings,
             sub_tenant_id=sub_tenant_id,
             max_chunks=max_chunks,
             request_options=request_options,
@@ -294,8 +293,8 @@ class AsyncEmbeddingsClient:
     async def search(
         self,
         *,
-        embeddings: typing.Sequence[float],
         tenant_id: str,
+        embeddings: typing.Optional[typing.Sequence[float]] = OMIT,
         sub_tenant_id: typing.Optional[str] = OMIT,
         max_chunks: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -319,11 +318,10 @@ class AsyncEmbeddingsClient:
 
         Parameters
         ----------
-        embeddings : typing.Sequence[float]
-            Single embedding vector for search
-
         tenant_id : str
 
+        embeddings : typing.Optional[typing.Sequence[float]]
+
         sub_tenant_id : typing.Optional[str]
 
         max_chunks : typing.Optional[int]
@@ -344,12 +342,12 @@ class AsyncEmbeddingsClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.embeddings.search(embeddings=[1.1], tenant_id='tenant_id', )
+            await client.embeddings.search(tenant_id='tenant_id', )
         asyncio.run(main())
         """
         _response = await self._raw_client.search(
-            embeddings=embeddings,
             tenant_id=tenant_id,
+            embeddings=embeddings,
             sub_tenant_id=sub_tenant_id,
             max_chunks=max_chunks,
             request_options=request_options,

{usecortex_ai-0.2.0 → usecortex_ai-0.2.1}/src/usecortex_ai/embeddings/raw_client.py

@@ -178,8 +178,8 @@ class RawEmbeddingsClient:
     def search(
         self,
         *,
-        embeddings: typing.Sequence[float],
         tenant_id: str,
+        embeddings: typing.Optional[typing.Sequence[float]] = OMIT,
         sub_tenant_id: typing.Optional[str] = OMIT,
         max_chunks: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -203,11 +203,10 @@ class RawEmbeddingsClient:
 
         Parameters
         ----------
-        embeddings : typing.Sequence[float]
-            Single embedding vector for search
-
         tenant_id : str
 
+        embeddings : typing.Optional[typing.Sequence[float]]
+
         sub_tenant_id : typing.Optional[str]
 
         max_chunks : typing.Optional[int]
@@ -224,8 +223,8 @@ class RawEmbeddingsClient:
             "embeddings/search",
             method="POST",
             json={
-                "embeddings": embeddings,
                 "tenant_id": tenant_id,
+                "embeddings": embeddings,
                 "sub_tenant_id": sub_tenant_id,
                 "max_chunks": max_chunks,
             },
@@ -742,8 +741,8 @@ class AsyncRawEmbeddingsClient:
     async def search(
         self,
         *,
-        embeddings: typing.Sequence[float],
         tenant_id: str,
+        embeddings: typing.Optional[typing.Sequence[float]] = OMIT,
         sub_tenant_id: typing.Optional[str] = OMIT,
         max_chunks: typing.Optional[int] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -767,11 +766,10 @@ class AsyncRawEmbeddingsClient:
 
         Parameters
         ----------
-        embeddings : typing.Sequence[float]
-            Single embedding vector for search
-
         tenant_id : str
 
+        embeddings : typing.Optional[typing.Sequence[float]]
+
         sub_tenant_id : typing.Optional[str]
 
         max_chunks : typing.Optional[int]
@@ -788,8 +786,8 @@ class AsyncRawEmbeddingsClient:
             "embeddings/search",
             method="POST",
             json={
-                "embeddings": embeddings,
                 "tenant_id": tenant_id,
+                "embeddings": embeddings,
                 "sub_tenant_id": sub_tenant_id,
                 "max_chunks": max_chunks,
             },

{usecortex_ai-0.2.0 → usecortex_ai-0.2.1}/src/usecortex_ai/sources/client.py

@@ -5,6 +5,7 @@ import typing
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
 from ..types.list_sources_response import ListSourcesResponse
+from ..types.sub_tenant_ids_data import SubTenantIdsData
 from .raw_client import AsyncRawSourcesClient, RawSourcesClient
 
 # this is used as the default value for optional parameters
@@ -94,6 +95,45 @@ class SourcesClient:
         )
         return _response.data
 
+    def get_sub_tenant_ids(
+        self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> SubTenantIdsData:
+        """
+        Get all sub-tenant IDs (tenant IDs) contained within a specific Weaviate collection.
+        Fetches the tenant IDs directly from Weaviate using default cluster credentials.
+
+        Args:
+            tenant_id: The tenant ID to fetch sub-tenant IDs for
+            api_details: Authentication dependency
+
+        Returns:
+            SubTenantIdsData: Contains collection_name, sub_tenant_ids list, count, and success message
+
+        Example:
+            GET /list/sub_tenant_ids?tenant_id=my_tenant_123
+
+        Parameters
+        ----------
+        tenant_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SubTenantIdsData
+            Successful Response
+
+        Examples
+        --------
+        from usecortex-ai import CortexAI
+
+        client = CortexAI(token="YOUR_TOKEN", )
+        client.sources.get_sub_tenant_ids(tenant_id='tenant_id', )
+        """
+        _response = self._raw_client.get_sub_tenant_ids(tenant_id=tenant_id, request_options=request_options)
+        return _response.data
+
 
 class AsyncSourcesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -185,3 +225,46 @@ class AsyncSourcesClient:
             tenant_id=tenant_id, source_ids=source_ids, request_options=request_options
         )
         return _response.data
+
+    async def get_sub_tenant_ids(
+        self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> SubTenantIdsData:
+        """
+        Get all sub-tenant IDs (tenant IDs) contained within a specific Weaviate collection.
+        Fetches the tenant IDs directly from Weaviate using default cluster credentials.
+
+        Args:
+            tenant_id: The tenant ID to fetch sub-tenant IDs for
+            api_details: Authentication dependency
+
+        Returns:
+            SubTenantIdsData: Contains collection_name, sub_tenant_ids list, count, and success message
+
+        Example:
+            GET /list/sub_tenant_ids?tenant_id=my_tenant_123
+
+        Parameters
+        ----------
+        tenant_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SubTenantIdsData
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from usecortex-ai import AsyncCortexAI
+
+        client = AsyncCortexAI(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.sources.get_sub_tenant_ids(tenant_id='tenant_id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_sub_tenant_ids(tenant_id=tenant_id, request_options=request_options)
+        return _response.data

{usecortex_ai-0.2.0 → usecortex_ai-0.2.1}/src/usecortex_ai/sources/raw_client.py

@@ -17,6 +17,7 @@ from ..errors.unauthorized_error import UnauthorizedError
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.actual_error_response import ActualErrorResponse
 from ..types.list_sources_response import ListSourcesResponse
+from ..types.sub_tenant_ids_data import SubTenantIdsData
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -276,6 +277,135 @@ class RawSourcesClient:
             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 get_sub_tenant_ids(
+        self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[SubTenantIdsData]:
+        """
+        Get all sub-tenant IDs (tenant IDs) contained within a specific Weaviate collection.
+        Fetches the tenant IDs directly from Weaviate using default cluster credentials.
+
+        Args:
+            tenant_id: The tenant ID to fetch sub-tenant IDs for
+            api_details: Authentication dependency
+
+        Returns:
+            SubTenantIdsData: Contains collection_name, sub_tenant_ids list, count, and success message
+
+        Example:
+            GET /list/sub_tenant_ids?tenant_id=my_tenant_123
+
+        Parameters
+        ----------
+        tenant_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[SubTenantIdsData]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "list/sub_tenant_ids",
+            method="GET",
+            params={
+                "tenant_id": tenant_id,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    SubTenantIdsData,
+                    parse_obj_as(
+                        type_=SubTenantIdsData,  # 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 AsyncRawSourcesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -530,3 +660,132 @@ class AsyncRawSourcesClient:
         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 get_sub_tenant_ids(
+        self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[SubTenantIdsData]:
+        """
+        Get all sub-tenant IDs (tenant IDs) contained within a specific Weaviate collection.
+        Fetches the tenant IDs directly from Weaviate using default cluster credentials.
+
+        Args:
+            tenant_id: The tenant ID to fetch sub-tenant IDs for
+            api_details: Authentication dependency
+
+        Returns:
+            SubTenantIdsData: Contains collection_name, sub_tenant_ids list, count, and success message
+
+        Example:
+            GET /list/sub_tenant_ids?tenant_id=my_tenant_123
+
+        Parameters
+        ----------
+        tenant_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[SubTenantIdsData]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "list/sub_tenant_ids",
+            method="GET",
+            params={
+                "tenant_id": tenant_id,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    SubTenantIdsData,
+                    parse_obj_as(
+                        type_=SubTenantIdsData,  # 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)

{usecortex_ai-0.2.0 → usecortex_ai-0.2.1}/src/usecortex_ai/types/__init__.py

@@ -30,6 +30,7 @@ from .single_upload_data import SingleUploadData
 from .source import Source
 from .source_content import SourceContent
 from .source_model import SourceModel
+from .sub_tenant_ids_data import SubTenantIdsData
 from .tenant_create_data import TenantCreateData
 from .tenant_stats import TenantStats
 from .validation_error import ValidationError
@@ -64,6 +65,7 @@ __all__ = [
     "Source",
     "SourceContent",
     "SourceModel",
+    "SubTenantIdsData",
     "TenantCreateData",
     "TenantStats",
     "ValidationError",

{usecortex_ai-0.2.0 → usecortex_ai-0.2.1}/src/usecortex_ai/types/markdown_upload_request.py

@@ -4,12 +4,14 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .relations import Relations
 
 
 class MarkdownUploadRequest(UniversalBaseModel):
     content: str
     tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
     document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    relations: typing.Optional[Relations] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

usecortex_ai-0.2.1/src/usecortex_ai/types/relations.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 Relations(UniversalBaseModel):
+    cortex_source_ids: typing.Optional[typing.List[str]] = None
+    auto_discovery: 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

{usecortex_ai-0.2.0 → usecortex_ai-0.2.1}/src/usecortex_ai/types/search_chunk.py

@@ -11,17 +11,18 @@ class SearchChunk(UniversalBaseModel):
     chunk_uuid: str
     source_id: str
     chunk_content: str
-    source_collection: typing.Optional[typing.List[str]] = None
-    layout: typing.Optional[str] = None
-    source_url: typing.Optional[str] = None
-    version: str
     source_type: typing.Optional[str] = None
     source_upload_time: typing.Optional[str] = None
     source_title: typing.Optional[str] = None
     source_last_updated_time: typing.Optional[str] = None
+    source_collection: typing.Optional[typing.List[str]] = None
+    layout: typing.Optional[str] = None
+    source_url: typing.Optional[str] = None
+    version: typing.Optional[str] = None
     relevancy_score: typing.Optional[float] = None
     cortex_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
     document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
     extra_context: typing.Optional[ExtendedContext] = None
 
     if IS_PYDANTIC_V2:

{usecortex_ai-0.2.0 → usecortex_ai-0.2.1}/src/usecortex_ai/types/source.py

@@ -8,17 +8,18 @@ from .source_content import SourceContent
 
 
 class Source(UniversalBaseModel):
-    collections: typing.Optional[typing.List[str]] = None
     id: str
     title: str
     type: str
+    timestamp: str
+    content: SourceContent
+    collections: typing.Optional[typing.List[str]] = None
     description: typing.Optional[str] = None
     note: typing.Optional[str] = None
     url: typing.Optional[str] = None
-    timestamp: str
-    content: SourceContent
     cortex_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
     document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
     meta: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
     attachments: typing.Optional[typing.List[typing.Optional[typing.Any]]] = None
 

{usecortex_ai-0.2.0 → usecortex_ai-0.2.1}/src/usecortex_ai/types/source_content.py

@@ -13,8 +13,8 @@ class SourceContent(UniversalBaseModel):
     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[str]] = None
-    layout: typing.Optional[typing.List[typing.Optional[typing.Any]]] = 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

{usecortex_ai-0.2.0 → usecortex_ai-0.2.1}/src/usecortex_ai/types/source_model.py

@@ -19,6 +19,8 @@ class SourceModel(UniversalBaseModel):
     timestamp: typing.Optional[str] = None
     content: typing.Optional[ContentModel] = None
     cortex_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
     meta: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
     attachments: typing.Optional[typing.List[AttachmentModel]] = None