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

usecortex_ai/types/source_upload_response.py

@@ -0,0 +1,35 @@
+# 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 .source_upload_result_item import SourceUploadResultItem
+
+
+class SourceUploadResponse(UniversalBaseModel):
+    success: typing.Optional[bool] = None
+    message: typing.Optional[str] = None
+    results: typing.Optional[typing.List[SourceUploadResultItem]] = pydantic.Field(default=None)
+    """
+    List of upload results for each source.
+    """
+
+    success_count: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Number of sources successfully queued.
+    """
+
+    failed_count: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Number of sources that failed to upload.
+    """
+
+    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/types/source_upload_result_item.py

@@ -0,0 +1,38 @@
+# 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 .source_status import SourceStatus
+
+
+class SourceUploadResultItem(UniversalBaseModel):
+    source_id: str = pydantic.Field()
+    """
+    Unique identifier for the uploaded source.
+    """
+
+    filename: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Original filename if present.
+    """
+
+    status: typing.Optional[SourceStatus] = pydantic.Field(default=None)
+    """
+    Initial processing status.
+    """
+
+    error: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Error message if upload failed.
+    """
+
+    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/types/supported_llm_providers.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+SupportedLlmProviders = typing.Union[typing.Literal["groq", "cerebras", "openai", "anthropic", "gemini"], typing.Any]

usecortex_ai/types/{embeddings_create_collection_data.py → tenant_create_response.py}

@@ -4,27 +4,29 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .infra import Infra
+from .tenant_metadata_schema_info import TenantMetadataSchemaInfo
 
 
-class EmbeddingsCreateCollectionData(UniversalBaseModel):
+class TenantCreateResponse(UniversalBaseModel):
     tenant_id: str = pydantic.Field()
     """
-    Identifier for the tenant/organization
+    Identifier provided by user
     """
 
-    sub_tenant_id: str = pydantic.Field()
+    infra: Infra = pydantic.Field()
     """
-    Identifier for the sub-tenant within the tenant
+    Infra status
     """
 
-    success: typing.Optional[bool] = pydantic.Field(default=None)
+    metadata_schema: typing.Optional[TenantMetadataSchemaInfo] = pydantic.Field(default=None)
     """
-    Indicates whether the embeddings tenant creation completed successfully
+    Summary of configured tenant metadata schema (if provided)
     """
 
     message: typing.Optional[str] = pydantic.Field(default=None)
     """
-    Status message about the tenant creation operation
+    Summary message
     """
 
     if IS_PYDANTIC_V2:

usecortex_ai/types/{extended_context.py → tenant_info.py}

@@ -4,13 +4,22 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
-from .triple_with_evidence import TripleWithEvidence
 
 
-class ExtendedContext(UniversalBaseModel):
-    chunk_relations: typing.Optional[typing.List[TripleWithEvidence]] = pydantic.Field(default=None)
+class TenantInfo(UniversalBaseModel):
+    tenant_id: str = pydantic.Field()
     """
-    Relations linked with this chunk
+    Organization tenant ID
+    """
+
+    organisation: str = pydantic.Field()
+    """
+    Organization name
+    """
+
+    timestamp: str = pydantic.Field()
+    """
+    Creation timestamp
     """
 
     if IS_PYDANTIC_V2:

usecortex_ai/types/{embeddings_search_data.py → tenant_metadata_schema_info.py}

@@ -6,25 +6,24 @@ import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
-class EmbeddingsSearchData(UniversalBaseModel):
-    chunk_ids: typing.List[str] = pydantic.Field()
+class TenantMetadataSchemaInfo(UniversalBaseModel):
     """
-    List of chunk IDs that match the search query
+    Summary of configured tenant metadata schema fields.
     """
 
-    scores: typing.List[float] = pydantic.Field()
+    field_count: int = pydantic.Field()
     """
-    Similarity scores for each matching chunk (higher is more similar)
+    Number of custom metadata fields configured
     """
 
-    success: typing.Optional[bool] = pydantic.Field(default=None)
+    dense_embedding_fields: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
     """
-    Indicates whether the embeddings search operation completed successfully
+    Fields with dense embeddings enabled for semantic search
     """
 
-    message: typing.Optional[str] = pydantic.Field(default=None)
+    sparse_embedding_fields: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
     """
-    Status message about the search operation
+    Fields with sparse embeddings enabled for keyword search
     """
 
     if IS_PYDANTIC_V2:

usecortex_ai/types/{tenant_create_data.py → tenant_stats_response.py}

@@ -4,27 +4,28 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .collection_stats import CollectionStats
 
 
-class TenantCreateData(UniversalBaseModel):
-    status: str = pydantic.Field()
+class TenantStatsResponse(UniversalBaseModel):
+    tenant_id: str = pydantic.Field()
     """
-    Current status of the tenant creation process
+    Tenant identifier
     """
 
-    tenant_id: str = pydantic.Field()
+    normal_collection: CollectionStats = pydantic.Field()
     """
-    Unique identifier assigned to the new tenant
+    Statistics for the normal (context) collection
     """
 
-    success: typing.Optional[bool] = pydantic.Field(default=None)
+    memory_collection: CollectionStats = pydantic.Field()
     """
-    Indicates whether the tenant creation was successful
+    Statistics for the memory collection
     """
 
     message: typing.Optional[str] = pydantic.Field(default=None)
     """
-    Response message describing the creation result
+    Summary message
     """
 
     if IS_PYDANTIC_V2:

usecortex_ai/types/{triple_with_evidence.py → triplet_with_evidence.py}

@@ -8,7 +8,7 @@ from .entity import Entity
 from .relation_evidence import RelationEvidence
 
 
-class TripleWithEvidence(UniversalBaseModel):
+class TripletWithEvidence(UniversalBaseModel):
     """
     Triple with multiple evidence items from different chunks
     """

usecortex_ai/types/user_assistant_pair.py

@@ -7,6 +7,10 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class UserAssistantPair(UniversalBaseModel):
+    """
+    Represents a user-assistant conversation pair.
+    """
+
     user: str = pydantic.Field()
     """
     User's message in the conversation

usecortex_ai/types/{search_chunk.py → vector_store_chunk.py}

@@ -4,10 +4,9 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
-from .extended_context import ExtendedContext
 
 
-class SearchChunk(UniversalBaseModel):
+class VectorStoreChunk(UniversalBaseModel):
     chunk_uuid: str = pydantic.Field()
     """
     Unique identifier for this content chunk
@@ -45,12 +44,12 @@ class SearchChunk(UniversalBaseModel):
 
     layout: typing.Optional[str] = pydantic.Field(default=None)
     """
-    Layout of the chunk in original document. You will generally receive a stringified dict with 2 keys, `offsets` and `page`(optional). Offsets will have `document_level_start_index` and `page_level_start_index`(optional)
+    Layout of the chunk in original document. You will generally receive        a stringified dict with 2 keys, `offsets` and `page`(optional). Offsets will have        `document_level_start_index` and `page_level_start_index`(optional)
     """
 
     relevancy_score: typing.Optional[float] = pydantic.Field(default=None)
     """
-    Score indicating how relevant this chunk is to your search query, with higher values indicating better matches
+    Score indicating how relevant this chunk is to your search query,         with higher values indicating better matches
     """
 
     document_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
@@ -63,11 +62,6 @@ class SearchChunk(UniversalBaseModel):
     Custom metadata associated with your tenant
     """
 
-    extra_context: typing.Optional[ExtendedContext] = pydantic.Field(default=None)
-    """
-    Additional context for this chunk
-    """
-
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

usecortex_ai/upload/__init__.py

@@ -2,3 +2,6 @@
 
 # isort: skip_file
 
+from .types import BodyUploadAppIngestionUploadAppPostAppSources
+
+__all__ = ["BodyUploadAppIngestionUploadAppPostAppSources"]