usecortex-ai 0.4.0__tar.gz → 0.5.0__tar.gz

{usecortex_ai-0.4.0 → usecortex_ai-0.5.0}/LICENSE

@@ -1,22 +1,22 @@
-Copyright (c) 2024 Cortex AI
-
-All Rights Reserved.
-
-PROPRIETARY AND CONFIDENTIAL
-
-This software is the proprietary and confidential property of Cortex AI ("the Company").
-Permission is hereby granted to users to install and use this software as part of the Cortex AI service, subject to the terms and conditions of the service agreement entered into with the Company.
-
-You may not, without the express written permission of the Company:
-
-1. Copy, modify, or create derivative works of the software.
-2. Distribute, sell, rent, lease, sublicense, or otherwise transfer the software to any third party.
-3. Reverse engineer, decompile, or disassemble the software, except and only to the extent that such activity is expressly permitted by applicable law notwithstanding this limitation.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+Copyright (c) 2024 Cortex AI
+
+All Rights Reserved.
+
+PROPRIETARY AND CONFIDENTIAL
+
+This software is the proprietary and confidential property of Cortex AI ("the Company").
+Permission is hereby granted to users to install and use this software as part of the Cortex AI service, subject to the terms and conditions of the service agreement entered into with the Company.
+
+You may not, without the express written permission of the Company:
+
+1. Copy, modify, or create derivative works of the software.
+2. Distribute, sell, rent, lease, sublicense, or otherwise transfer the software to any third party.
+3. Reverse engineer, decompile, or disassemble the software, except and only to the extent that such activity is expressly permitted by applicable law notwithstanding this limitation.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.

{usecortex_ai-0.4.0/src/usecortex_ai.egg-info → usecortex_ai-0.5.0}/PKG-INFO

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

{usecortex_ai-0.4.0 → usecortex_ai-0.5.0}/README.md

@@ -1,135 +1,135 @@
-# Cortex AI Python SDK - [usecortex.ai](https://www.usecortex.ai/)
-
-The official Python SDK for the Cortex AI platform. Build powerful, context-aware AI applications in your Python applications.
-
-**Cortex** is your plug-and-play memory infrastructure. It powers intelligent, context-aware retrieval for any AI app or agent. Whether you’re building a customer support bot, research copilot, or internal knowledge assistant.
-
-[Learn more about the SDK from our docs](https://docs.usecortex.ai/)
-
-## Core features
-
-* **Dynamic retrieval and querying** that always retrieve the most relevant context
-* **Built-in long-term memory** that evolves with every user interaction
-* **Personalization hooks** for user preferences, intent, and history
-* **Developer-first SDK** with the most flexible APIs and fine-grained controls
-
-## Getting started
-
-### Installation
-
-```bash
-pip install usecortex-ai
-```
-
-### Client setup
-
-We provide both synchronous and asynchronous clients. Use **`AsyncCortexAI`** when working with async/await patterns, and **`CortexAI`** for traditional synchronous workflows. Client initialization does not trigger any network requests, so you can safely create as many client instances as needed. Both clients expose the exact same set of methods.
-
-```python
-import os
-from usecortex_ai import CortexAI, AsyncCortexAI
-
-api_key = os.environ["CORTEX_API_KEY"]  # Set your Cortex API key in the environment variable CORTEX_API_KEY. Optional, but recommended.
-
-# Sync client
-client = CortexAI(token=api_key)
-
-# Async client (for async/await usage)
-async_client = AsyncCortexAI(token=api_key)
-```
-
-### Create a Tenant
-
-You can consider a `tenant` as a single database that can have internal isolated collections called `sub-tenants`. [Know more about the concept of tenant here](https://docs.usecortex.ai/essentials/multi-tenant)
-
-```python
-def create_tenant():
-    return client.user.create_tenant(tenant_id="my-company")
-```
-
-### Index Your Data
-
-When you index your data, you make it ready for retrieval from Cortex using natural language.
-
-```python
-# Upload text content
-def upload_text():
-    return client.upload.upload_text(
-        tenant_id="my-company-py-sync",
-        sub_tenant_id="engineering",
-        content="Our API rate limits are 1000 requests per minute for premium accounts.",
-        file_id="api-docs-rate-limits",
-        tenant_metadata={"sub_tenant_id": "engineering"}
-    )
-
-# Upload document file
-def upload_file():
-    with open("company-handbook.pdf", 'rb') as file_obj:
-        file_data = ("company-handbook.pdf", file_obj)
-        return client.upload.upload_document(
-            tenant_id="my-company",
-            file=file_data,
-            file_id="company-handbook.pdf"
-        )
-```
-
-**For a more detailed explanation** of document upload, including supported file formats, processing pipeline, metadata handling, and advanced configuration options, refer to the [Upload Document endpoint documentation](https://docs.usecortex.ai/api-reference/endpoint/upload-document).
-
-### Search and retrieval
-
-```python
-# Semantic search with retrieval
-results = client.search.retrieve(
-    query="What are the API rate limits?",
-    tenant_id="my-company",
-    sub_tenant_id="engineering",
-    max_chunks=10
-)
-
-# List all sources
-all_sources = client.sources.get_all(
-    tenant_id="my-company",
-    sub_tenant_id="engineering",
-)
-
-# Get specific sources by ID
-specific_sources = client.sources.get_by_ids(
-    tenant_id="my-company",
-    sub_tenant_id="engineering",
-    source_ids=["api-docs-rate-limits", "company-handbook"]
-)
-```
-
-**For a more detailed explanation** of search and retrieval, including query parameters, scoring mechanisms, result structure, and advanced search features, refer to the [Search endpoint documentation](https://docs.usecortex.ai/api-reference/endpoint/search).
-
-## SDK Method Structure & Type Safety
-
-Our SDKs follow a predictable pattern that mirrors the API structure while providing full type safety.
-
-> **Method Mapping** : `client.<group>.<function_name>` mirrors  `api.usecortex.ai/<group>/<function_name>`
->
-> For example: `client.upload.upload_text()` corresponds to `POST /upload/upload_text`
-
-The SDKs provide exact type parity with the API specification:
-
-- **Request Parameters** : Every field documented in the API reference (required, optional, types, validation rules) is reflected in the SDK method signatures
-- **Response Objects** : Return types match the exact JSON schema documented for each endpoint
-- **Error Types** : Exception structures mirror the error response formats from the API
-- **Nested Objects** : Complex nested parameters and responses maintain their full structure and typing
-
-> This means you can rely on your IDE’s autocomplete and type checking. If a parameter is optional in the API docs, it’s optional in the SDK. If a response contains a specific field, your IDE will know about it. Our SDKs are built in such a way that your IDE will automatically provide **autocompletion, type-checking, inline documentation with examples, and compile time validation** for each and every method.
->
-> Just hit **Cmd+Space/Ctrl+Space!**
-
-## Links
-
-- **Homepage:** [usecortex.ai](https://www.usecortex.ai/)
-- **Documentation:** [docs.usecortex.ai](https://docs.usecortex.ai/)
-
-## Our docs
-
-Please refer to our [API reference](https://docs.usecortex.ai/api-reference/introduction) for detailed explanations of every API endpoint, parameter options, and advanced use cases.
-
-## Support
-
-If you have any questions or need help, please reach out to our support team at [founders@usecortex.ai](mailto:founders@usecortex.ai).
+# Cortex AI Python SDK - [usecortex.ai](https://www.usecortex.ai/)
+
+The official Python SDK for the Cortex AI platform. Build powerful, context-aware AI applications in your Python applications.
+
+**Cortex** is your plug-and-play memory infrastructure. It powers intelligent, context-aware retrieval for any AI app or agent. Whether you’re building a customer support bot, research copilot, or internal knowledge assistant.
+
+[Learn more about the SDK from our docs](https://docs.usecortex.ai/)
+
+## Core features
+
+* **Dynamic retrieval and querying** that always retrieve the most relevant context
+* **Built-in long-term memory** that evolves with every user interaction
+* **Personalization hooks** for user preferences, intent, and history
+* **Developer-first SDK** with the most flexible APIs and fine-grained controls
+
+## Getting started
+
+### Installation
+
+```bash
+pip install usecortex-ai
+```
+
+### Client setup
+
+We provide both synchronous and asynchronous clients. Use **`AsyncCortexAI`** when working with async/await patterns, and **`CortexAI`** for traditional synchronous workflows. Client initialization does not trigger any network requests, so you can safely create as many client instances as needed. Both clients expose the exact same set of methods.
+
+```python
+import os
+from usecortex_ai import CortexAI, AsyncCortexAI
+
+api_key = os.environ["CORTEX_API_KEY"]  # Set your Cortex API key in the environment variable CORTEX_API_KEY. Optional, but recommended.
+
+# Sync client
+client = CortexAI(token=api_key)
+
+# Async client (for async/await usage)
+async_client = AsyncCortexAI(token=api_key)
+```
+
+### Create a Tenant
+
+You can consider a `tenant` as a single database that can have internal isolated collections called `sub-tenants`. [Know more about the concept of tenant here](https://docs.usecortex.ai/essentials/multi-tenant)
+
+```python
+def create_tenant():
+    return client.user.create_tenant(tenant_id="my-company")
+```
+
+### Index Your Data
+
+When you index your data, you make it ready for retrieval from Cortex using natural language.
+
+```python
+# Upload text content
+def upload_text():
+    return client.upload.upload_text(
+        tenant_id="my-company-py-sync",
+        sub_tenant_id="engineering",
+        content="Our API rate limits are 1000 requests per minute for premium accounts.",
+        file_id="api-docs-rate-limits",
+        tenant_metadata={"sub_tenant_id": "engineering"}
+    )
+
+# Upload document file
+def upload_file():
+    with open("company-handbook.pdf", 'rb') as file_obj:
+        file_data = ("company-handbook.pdf", file_obj)
+        return client.upload.upload_document(
+            tenant_id="my-company",
+            file=file_data,
+            file_id="company-handbook.pdf"
+        )
+```
+
+**For a more detailed explanation** of document upload, including supported file formats, processing pipeline, metadata handling, and advanced configuration options, refer to the [Upload Document endpoint documentation](https://docs.usecortex.ai/api-reference/endpoint/upload-document).
+
+### Search and retrieval
+
+```python
+# Semantic search with retrieval
+results = client.search.retrieve(
+    query="What are the API rate limits?",
+    tenant_id="my-company",
+    sub_tenant_id="engineering",
+    max_chunks=10
+)
+
+# List all sources
+all_sources = client.sources.get_all(
+    tenant_id="my-company",
+    sub_tenant_id="engineering",
+)
+
+# Get specific sources by ID
+specific_sources = client.sources.get_by_ids(
+    tenant_id="my-company",
+    sub_tenant_id="engineering",
+    source_ids=["api-docs-rate-limits", "company-handbook"]
+)
+```
+
+**For a more detailed explanation** of search and retrieval, including query parameters, scoring mechanisms, result structure, and advanced search features, refer to the [Search endpoint documentation](https://docs.usecortex.ai/api-reference/endpoint/search).
+
+## SDK Method Structure & Type Safety
+
+Our SDKs follow a predictable pattern that mirrors the API structure while providing full type safety.
+
+> **Method Mapping** : `client.<group>.<function_name>` mirrors  `api.usecortex.ai/<group>/<function_name>`
+>
+> For example: `client.upload.upload_text()` corresponds to `POST /upload/upload_text`
+
+The SDKs provide exact type parity with the API specification:
+
+- **Request Parameters** : Every field documented in the API reference (required, optional, types, validation rules) is reflected in the SDK method signatures
+- **Response Objects** : Return types match the exact JSON schema documented for each endpoint
+- **Error Types** : Exception structures mirror the error response formats from the API
+- **Nested Objects** : Complex nested parameters and responses maintain their full structure and typing
+
+> This means you can rely on your IDE’s autocomplete and type checking. If a parameter is optional in the API docs, it’s optional in the SDK. If a response contains a specific field, your IDE will know about it. Our SDKs are built in such a way that your IDE will automatically provide **autocompletion, type-checking, inline documentation with examples, and compile time validation** for each and every method.
+>
+> Just hit **Cmd+Space/Ctrl+Space!**
+
+## Links
+
+- **Homepage:** [usecortex.ai](https://www.usecortex.ai/)
+- **Documentation:** [docs.usecortex.ai](https://docs.usecortex.ai/)
+
+## Our docs
+
+Please refer to our [API reference](https://docs.usecortex.ai/api-reference/introduction) for detailed explanations of every API endpoint, parameter options, and advanced use cases.
+
+## Support
+
+If you have any questions or need help, please reach out to our support team at [founders@usecortex.ai](mailto:founders@usecortex.ai).

{usecortex_ai-0.4.0 → usecortex_ai-0.5.0}/pyproject.toml

@@ -1,40 +1,40 @@
-[build-system]
-requires = ["setuptools>=61.0"]
-build-backend = "setuptools.build_meta"
-
-[project]
-name = "usecortex-ai"
-
-version = "0.4.0"
-
-authors = [
-  { name = "Soham Ratnaparkhi", email = "soham@usecortex.ai" },
-]
-description = "The official Python SDK for the Cortex AI platform."
-readme = "README.md"
-requires-python = ">=3.10"
-license = { file = "LICENSE" }
-keywords = ["cortex", "ai", "sdk", "api", "generative ai", "rag"]
-
-# Add classifiers to help users find your package on PyPI
-classifiers = [
-    # See https://pypi.org/classifiers/
-    "Development Status :: 5 - Production/Stable",
-    "Programming Language :: Python :: 3",
-    "Programming Language :: Python :: 3.10",
-    "Programming Language :: Python :: 3.11",
-    "Programming Language :: Python :: 3.12",
-    "Intended Audience :: Developers",
-    "Topic :: Software Development :: Libraries :: Python Modules",
-    "Typing :: Typed",
-]
-
-dependencies = [
-  "httpx>=0.24",
-  "pydantic>=1.10,<3",
-]
-
-[project.urls]
-Homepage = "https://www.usecortex.ai/"
-Documentation = "https://docs.usecortex.ai/"
-
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "usecortex-ai"
+
+version = "0.5.0"
+
+authors = [
+  { name = "Soham Ratnaparkhi", email = "soham@usecortex.ai" },
+]
+description = "The official Python SDK for the Cortex AI platform."
+readme = "README.md"
+requires-python = ">=3.10"
+license = { file = "LICENSE" }
+keywords = ["cortex", "ai", "sdk", "api", "generative ai", "rag"]
+
+# Add classifiers to help users find your package on PyPI
+classifiers = [
+    # See https://pypi.org/classifiers/
+    "Development Status :: 5 - Production/Stable",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Intended Audience :: Developers",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+    "Typing :: Typed",
+]
+
+dependencies = [
+  "httpx>=0.24",
+  "pydantic>=1.10,<3",
+]
+
+[project.urls]
+Homepage = "https://www.usecortex.ai/"
+Documentation = "https://docs.usecortex.ai/"
+

{usecortex_ai-0.4.0 → usecortex_ai-0.5.0}/src/usecortex_ai/__init__.py

@@ -20,6 +20,7 @@ from .types import (
     Entity,
     ErrorResponse,
     FetchMode,
+    ForcefulRelationsPayload,
     GraphContext,
     HttpValidationError,
     Infra,
@@ -100,6 +101,7 @@ __all__ = [
     "ErrorResponse",
     "FetchMode",
     "ForbiddenError",
+    "ForcefulRelationsPayload",
     "GraphContext",
     "HttpValidationError",
     "Infra",

{usecortex_ai-0.4.0 → usecortex_ai-0.5.0}/src/usecortex_ai/search/client.py

@@ -47,6 +47,7 @@ class SearchClient:
         graph_context: typing.Optional[bool] = OMIT,
         extra_context: typing.Optional[str] = OMIT,
         search_mode: typing.Optional[SearchMode] = OMIT,
+        filters: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> RetrievalResult:
         """
@@ -63,6 +64,11 @@ class SearchClient:
             - "fast" (default): Single query, faster response
             - "accurate": Multi-query generation with reranking, higher quality
 
+            Use `filters` to narrow results by metadata:
+            - Provide key-value pairs matching fields defined in your tenant_metadata_schema
+            - Example: `{"category": "engineering", "priority": "high"}`
+            - Filters are validated against your tenant schema for type safety
+
         Parameters
         ----------
         tenant_id : str
@@ -101,6 +107,9 @@ class SearchClient:
         search_mode : typing.Optional[SearchMode]
             What to search: 'sources' for documents or 'memories' for user memories
 
+        filters : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+            Optional key-value pairs to filter search results by tenant metadata fields. Keys must match fields defined in tenant_metadata_schema during tenant creation. Supports exact match filtering on indexed metadata fields. Example: {'category': 'engineering', 'priority': 'high'}
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -129,6 +138,7 @@ class SearchClient:
             graph_context=graph_context,
             extra_context=extra_context,
             search_mode=search_mode,
+            filters=filters,
             request_options=request_options,
         )
         return _response.data
@@ -336,6 +346,7 @@ class AsyncSearchClient:
         graph_context: typing.Optional[bool] = OMIT,
         extra_context: typing.Optional[str] = OMIT,
         search_mode: typing.Optional[SearchMode] = OMIT,
+        filters: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> RetrievalResult:
         """
@@ -352,6 +363,11 @@ class AsyncSearchClient:
             - "fast" (default): Single query, faster response
             - "accurate": Multi-query generation with reranking, higher quality
 
+            Use `filters` to narrow results by metadata:
+            - Provide key-value pairs matching fields defined in your tenant_metadata_schema
+            - Example: `{"category": "engineering", "priority": "high"}`
+            - Filters are validated against your tenant schema for type safety
+
         Parameters
         ----------
         tenant_id : str
@@ -390,6 +406,9 @@ class AsyncSearchClient:
         search_mode : typing.Optional[SearchMode]
             What to search: 'sources' for documents or 'memories' for user memories
 
+        filters : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+            Optional key-value pairs to filter search results by tenant metadata fields. Keys must match fields defined in tenant_metadata_schema during tenant creation. Supports exact match filtering on indexed metadata fields. Example: {'category': 'engineering', 'priority': 'high'}
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -422,6 +441,7 @@ class AsyncSearchClient:
             graph_context=graph_context,
             extra_context=extra_context,
             search_mode=search_mode,
+            filters=filters,
             request_options=request_options,
         )
         return _response.data

{usecortex_ai-0.4.0 → usecortex_ai-0.5.0}/src/usecortex_ai/search/raw_client.py

@@ -49,6 +49,7 @@ class RawSearchClient:
         graph_context: typing.Optional[bool] = OMIT,
         extra_context: typing.Optional[str] = OMIT,
         search_mode: typing.Optional[SearchMode] = OMIT,
+        filters: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[RetrievalResult]:
         """
@@ -65,6 +66,11 @@ class RawSearchClient:
             - "fast" (default): Single query, faster response
             - "accurate": Multi-query generation with reranking, higher quality
 
+            Use `filters` to narrow results by metadata:
+            - Provide key-value pairs matching fields defined in your tenant_metadata_schema
+            - Example: `{"category": "engineering", "priority": "high"}`
+            - Filters are validated against your tenant schema for type safety
+
         Parameters
         ----------
         tenant_id : str
@@ -103,6 +109,9 @@ class RawSearchClient:
         search_mode : typing.Optional[SearchMode]
             What to search: 'sources' for documents or 'memories' for user memories
 
+        filters : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+            Optional key-value pairs to filter search results by tenant metadata fields. Keys must match fields defined in tenant_metadata_schema during tenant creation. Supports exact match filtering on indexed metadata fields. Example: {'category': 'engineering', 'priority': 'high'}
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -127,6 +136,7 @@ class RawSearchClient:
                 "graph_context": graph_context,
                 "extra_context": extra_context,
                 "search_mode": search_mode,
+                "filters": filters,
             },
             headers={
                 "content-type": "application/json",
@@ -633,6 +643,7 @@ class AsyncRawSearchClient:
         graph_context: typing.Optional[bool] = OMIT,
         extra_context: typing.Optional[str] = OMIT,
         search_mode: typing.Optional[SearchMode] = OMIT,
+        filters: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[RetrievalResult]:
         """
@@ -649,6 +660,11 @@ class AsyncRawSearchClient:
             - "fast" (default): Single query, faster response
             - "accurate": Multi-query generation with reranking, higher quality
 
+            Use `filters` to narrow results by metadata:
+            - Provide key-value pairs matching fields defined in your tenant_metadata_schema
+            - Example: `{"category": "engineering", "priority": "high"}`
+            - Filters are validated against your tenant schema for type safety
+
         Parameters
         ----------
         tenant_id : str
@@ -687,6 +703,9 @@ class AsyncRawSearchClient:
         search_mode : typing.Optional[SearchMode]
             What to search: 'sources' for documents or 'memories' for user memories
 
+        filters : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+            Optional key-value pairs to filter search results by tenant metadata fields. Keys must match fields defined in tenant_metadata_schema during tenant creation. Supports exact match filtering on indexed metadata fields. Example: {'category': 'engineering', 'priority': 'high'}
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -711,6 +730,7 @@ class AsyncRawSearchClient:
                 "graph_context": graph_context,
                 "extra_context": extra_context,
                 "search_mode": search_mode,
+                "filters": filters,
             },
             headers={
                 "content-type": "application/json",

{usecortex_ai-0.4.0 → usecortex_ai-0.5.0}/src/usecortex_ai/types/__init__.py

@@ -19,6 +19,7 @@ from .delete_user_memory_response import DeleteUserMemoryResponse
 from .entity import Entity
 from .error_response import ErrorResponse
 from .fetch_mode import FetchMode
+from .forceful_relations_payload import ForcefulRelationsPayload
 from .graph_context import GraphContext
 from .http_validation_error import HttpValidationError
 from .infra import Infra
@@ -76,6 +77,7 @@ __all__ = [
     "Entity",
     "ErrorResponse",
     "FetchMode",
+    "ForcefulRelationsPayload",
     "GraphContext",
     "HttpValidationError",
     "Infra",

usecortex_ai-0.5.0/src/usecortex_ai/types/forceful_relations_payload.py

@@ -0,0 +1,27 @@
+# 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 ForcefulRelationsPayload(UniversalBaseModel):
+    cortex_source_ids: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    Cortex source IDs to forcefully relate to the uploaded source.
+    """
+
+    properties: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Optional properties to attach to the forceful relation.
+    """
+
+    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.4.0 → usecortex_ai-0.5.0}/src/usecortex_ai/types/memory_item.py

@@ -4,6 +4,7 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .forceful_relations_payload import ForcefulRelationsPayload
 from .user_assistant_pair import UserAssistantPair
 
 
@@ -40,7 +41,7 @@ class MemoryItem(UniversalBaseModel):
 
     infer: typing.Optional[bool] = pydantic.Field(default=None)
     """
-    If true, process and extract additional insights/inferences from the content before indexing. Useful for extracting implicit information from conversations.
+    If true, process and extract additional insights/inferences from the contentbefore indexingUseful for extracting implicit information from conversations
     """
 
     custom_instructions: typing.Optional[str] = pydantic.Field(default=None)
@@ -72,6 +73,11 @@ class MemoryItem(UniversalBaseModel):
     Example: > "{"title":"Q1 Report.pdf","author":"Alice Smith","file_id":"custom_file_123"}"
     """
 
+    relations: typing.Optional[ForcefulRelationsPayload] = pydantic.Field(default=None)
+    """
+    Forcefully connect 2 sources based on cortex source ids or common properties.
+    """
+
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

{usecortex_ai-0.4.0 → usecortex_ai-0.5.0}/src/usecortex_ai/types/retrieval_result.py

@@ -15,6 +15,10 @@ class RetrievalResult(UniversalBaseModel):
 
     chunks: typing.Optional[typing.List[VectorStoreChunk]] = None
     graph_context: typing.Optional[GraphContext] = None
+    extra_context: typing.Optional[typing.Dict[str, VectorStoreChunk]] = pydantic.Field(default=None)
+    """
+    Map of chunk_uuid to VectorStoreChunk for extra context from forcefully related sources. Use chunk.extra_context_ids to look up chunks: extra_context[id] for id in chunk.extra_context_ids.
+    """
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

{usecortex_ai-0.4.0 → usecortex_ai-0.5.0}/src/usecortex_ai/types/vector_store_chunk.py

@@ -62,6 +62,11 @@ class VectorStoreChunk(UniversalBaseModel):
     Custom metadata associated with your tenant
     """
 
+    extra_context_ids: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    IDs of related chunks providing extra context (from forceful relations). Only present in accurate mode when sources have forceful relations.
+    """
+
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

{usecortex_ai-0.4.0 → usecortex_ai-0.5.0/src/usecortex_ai.egg-info}/PKG-INFO

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

{usecortex_ai-0.4.0 → usecortex_ai-0.5.0}/src/usecortex_ai.egg-info/SOURCES.txt

@@ -71,6 +71,7 @@ src/usecortex_ai/types/delete_user_memory_response.py
 src/usecortex_ai/types/entity.py
 src/usecortex_ai/types/error_response.py
 src/usecortex_ai/types/fetch_mode.py
+src/usecortex_ai/types/forceful_relations_payload.py
 src/usecortex_ai/types/graph_context.py
 src/usecortex_ai/types/http_validation_error.py
 src/usecortex_ai/types/infra.py