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

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

@@ -1,8 +1,8 @@
 Metadata-Version: 2.4
 Name: usecortex-ai
-Version: 0.3.6
+Version: 0.4.0
 Summary: The official Python SDK for the Cortex AI platform.
-Author-email: Nishkarsh Shrivastava <nishkarsh@usecortex.ai>
+Author-email: Soham Ratnaparkhi <soham@usecortex.ai>
 License: Copyright (c) 2024 Cortex AI
         
         All Rights Reserved.

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

@@ -5,10 +5,10 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "usecortex-ai"
 
-version = "0.3.6"
+version = "0.4.0"
 
 authors = [
-  { name = "Nishkarsh Shrivastava", email = "nishkarsh@usecortex.ai" },
+  { name = "Soham Ratnaparkhi", email = "soham@usecortex.ai" },
 ]
 description = "The official Python SDK for the Cortex AI platform."
 readme = "README.md"

usecortex_ai-0.4.0/src/usecortex_ai/__init__.py

@@ -0,0 +1,154 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .types import (
+    ActualErrorResponse,
+    AddMemoryResponse,
+    ApiKeyInfo,
+    AppSourcesUploadData,
+    AttachmentModel,
+    Bm25OperatorType,
+    CollectionStats,
+    ContentModel,
+    CustomPropertyDefinition,
+    DashboardApisResponse,
+    DashboardSourcesResponse,
+    DashboardTenantsResponse,
+    DeleteResult,
+    DeleteUserMemoryResponse,
+    Entity,
+    ErrorResponse,
+    FetchMode,
+    GraphContext,
+    HttpValidationError,
+    Infra,
+    InsertResult,
+    MemoryItem,
+    MemoryResultItem,
+    MilvusDataType,
+    PathTriplet,
+    ProcessingStatus,
+    ProcessingStatusIndexingStatus,
+    QnASearchResponse,
+    RawEmbeddingDocument,
+    RawEmbeddingSearchResult,
+    RawEmbeddingVector,
+    RelationEvidence,
+    RetrievalResult,
+    RetrieveMode,
+    ScoredPathResponse,
+    SearchMode,
+    SourceDeleteResponse,
+    SourceDeleteResultItem,
+    SourceFetchResponse,
+    SourceGraphRelationsResponse,
+    SourceListResponse,
+    SourceModel,
+    SourceStatus,
+    SourceUploadResponse,
+    SourceUploadResultItem,
+    SupportedLlmProviders,
+    TenantCreateResponse,
+    TenantInfo,
+    TenantMetadataSchemaInfo,
+    TenantStatsResponse,
+    TripletWithEvidence,
+    UserAssistantPair,
+    ValidationError,
+    ValidationErrorLocItem,
+    VectorStoreChunk,
+)
+from .errors import (
+    BadRequestError,
+    ForbiddenError,
+    InternalServerError,
+    NotFoundError,
+    ServiceUnavailableError,
+    TooManyRequestsError,
+    UnauthorizedError,
+    UnprocessableEntityError,
+)
+from . import dashboard, embeddings, fetch, search, sources, tenant, upload
+from .client import AsyncCortexAI, CortexAI
+from .environment import CortexAIEnvironment
+from .search import Alpha
+from .upload import BodyUploadAppIngestionUploadAppPostAppSources
+
+__all__ = [
+    "ActualErrorResponse",
+    "AddMemoryResponse",
+    "Alpha",
+    "ApiKeyInfo",
+    "AppSourcesUploadData",
+    "AsyncCortexAI",
+    "AttachmentModel",
+    "BadRequestError",
+    "Bm25OperatorType",
+    "BodyUploadAppIngestionUploadAppPostAppSources",
+    "CollectionStats",
+    "ContentModel",
+    "CortexAI",
+    "CortexAIEnvironment",
+    "CustomPropertyDefinition",
+    "DashboardApisResponse",
+    "DashboardSourcesResponse",
+    "DashboardTenantsResponse",
+    "DeleteResult",
+    "DeleteUserMemoryResponse",
+    "Entity",
+    "ErrorResponse",
+    "FetchMode",
+    "ForbiddenError",
+    "GraphContext",
+    "HttpValidationError",
+    "Infra",
+    "InsertResult",
+    "InternalServerError",
+    "MemoryItem",
+    "MemoryResultItem",
+    "MilvusDataType",
+    "NotFoundError",
+    "PathTriplet",
+    "ProcessingStatus",
+    "ProcessingStatusIndexingStatus",
+    "QnASearchResponse",
+    "RawEmbeddingDocument",
+    "RawEmbeddingSearchResult",
+    "RawEmbeddingVector",
+    "RelationEvidence",
+    "RetrievalResult",
+    "RetrieveMode",
+    "ScoredPathResponse",
+    "SearchMode",
+    "ServiceUnavailableError",
+    "SourceDeleteResponse",
+    "SourceDeleteResultItem",
+    "SourceFetchResponse",
+    "SourceGraphRelationsResponse",
+    "SourceListResponse",
+    "SourceModel",
+    "SourceStatus",
+    "SourceUploadResponse",
+    "SourceUploadResultItem",
+    "SupportedLlmProviders",
+    "TenantCreateResponse",
+    "TenantInfo",
+    "TenantMetadataSchemaInfo",
+    "TenantStatsResponse",
+    "TooManyRequestsError",
+    "TripletWithEvidence",
+    "UnauthorizedError",
+    "UnprocessableEntityError",
+    "UserAssistantPair",
+    "ValidationError",
+    "ValidationErrorLocItem",
+    "VectorStoreChunk",
+    "dashboard",
+    "embeddings",
+    "fetch",
+    "search",
+    "sources",
+    "tenant",
+    "upload",
+]

{usecortex_ai-0.3.6 → usecortex_ai-0.4.0}/src/usecortex_ai/client.py

@@ -5,7 +5,7 @@ import typing
 import httpx
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .core.request_options import RequestOptions
-from .document.client import AsyncDocumentClient, DocumentClient
+from .dashboard.client import AsyncDashboardClient, DashboardClient
 from .embeddings.client import AsyncEmbeddingsClient, EmbeddingsClient
 from .environment import CortexAIEnvironment
 from .fetch.client import AsyncFetchClient, FetchClient
@@ -14,8 +14,6 @@ from .search.client import AsyncSearchClient, SearchClient
 from .sources.client import AsyncSourcesClient, SourcesClient
 from .tenant.client import AsyncTenantClient, TenantClient
 from .upload.client import AsyncUploadClient, UploadClient
-from .user.client import AsyncUserClient, UserClient
-from .user_memory.client import AsyncUserMemoryClient, UserMemoryClient
 
 
 class CortexAI:
@@ -82,15 +80,13 @@ class CortexAI:
             timeout=_defaulted_timeout,
         )
         self._raw_client = RawCortexAI(client_wrapper=self._client_wrapper)
-        self.search = SearchClient(client_wrapper=self._client_wrapper)
-        self.sources = SourcesClient(client_wrapper=self._client_wrapper)
-        self.user_memory = UserMemoryClient(client_wrapper=self._client_wrapper)
-        self.fetch = FetchClient(client_wrapper=self._client_wrapper)
         self.upload = UploadClient(client_wrapper=self._client_wrapper)
-        self.embeddings = EmbeddingsClient(client_wrapper=self._client_wrapper)
-        self.user = UserClient(client_wrapper=self._client_wrapper)
         self.tenant = TenantClient(client_wrapper=self._client_wrapper)
-        self.document = DocumentClient(client_wrapper=self._client_wrapper)
+        self.embeddings = EmbeddingsClient(client_wrapper=self._client_wrapper)
+        self.fetch = FetchClient(client_wrapper=self._client_wrapper)
+        self.sources = SourcesClient(client_wrapper=self._client_wrapper)
+        self.search = SearchClient(client_wrapper=self._client_wrapper)
+        self.dashboard = DashboardClient(client_wrapper=self._client_wrapper)
 
     @property
     def with_raw_response(self) -> RawCortexAI:
@@ -103,8 +99,12 @@ class CortexAI:
         """
         return self._raw_client
 
-    def root_get(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Optional[typing.Any]:
+    def metrics_metrics_get(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.Optional[typing.Any]:
         """
+        Endpoint that serves Prometheus metrics.
+
         Parameters
         ----------
         request_options : typing.Optional[RequestOptions]
@@ -120,9 +120,9 @@ class CortexAI:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.root_get()
+        client.metrics_metrics_get()
         """
-        _response = self._raw_client.root_get(request_options=request_options)
+        _response = self._raw_client.metrics_metrics_get(request_options=request_options)
         return _response.data
 
 
@@ -190,15 +190,13 @@ class AsyncCortexAI:
             timeout=_defaulted_timeout,
         )
         self._raw_client = AsyncRawCortexAI(client_wrapper=self._client_wrapper)
-        self.search = AsyncSearchClient(client_wrapper=self._client_wrapper)
-        self.sources = AsyncSourcesClient(client_wrapper=self._client_wrapper)
-        self.user_memory = AsyncUserMemoryClient(client_wrapper=self._client_wrapper)
-        self.fetch = AsyncFetchClient(client_wrapper=self._client_wrapper)
         self.upload = AsyncUploadClient(client_wrapper=self._client_wrapper)
-        self.embeddings = AsyncEmbeddingsClient(client_wrapper=self._client_wrapper)
-        self.user = AsyncUserClient(client_wrapper=self._client_wrapper)
         self.tenant = AsyncTenantClient(client_wrapper=self._client_wrapper)
-        self.document = AsyncDocumentClient(client_wrapper=self._client_wrapper)
+        self.embeddings = AsyncEmbeddingsClient(client_wrapper=self._client_wrapper)
+        self.fetch = AsyncFetchClient(client_wrapper=self._client_wrapper)
+        self.sources = AsyncSourcesClient(client_wrapper=self._client_wrapper)
+        self.search = AsyncSearchClient(client_wrapper=self._client_wrapper)
+        self.dashboard = AsyncDashboardClient(client_wrapper=self._client_wrapper)
 
     @property
     def with_raw_response(self) -> AsyncRawCortexAI:
@@ -211,8 +209,12 @@ class AsyncCortexAI:
         """
         return self._raw_client
 
-    async def root_get(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Optional[typing.Any]:
+    async def metrics_metrics_get(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.Optional[typing.Any]:
         """
+        Endpoint that serves Prometheus metrics.
+
         Parameters
         ----------
         request_options : typing.Optional[RequestOptions]
@@ -231,10 +233,10 @@ class AsyncCortexAI:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.root_get()
+            await client.metrics_metrics_get()
         asyncio.run(main())
         """
-        _response = await self._raw_client.root_get(request_options=request_options)
+        _response = await self._raw_client.metrics_metrics_get(request_options=request_options)
         return _response.data