hydradb-sdk 0.0.1__py3-none-any.whl

hydra_db/__init__.py

@@ -0,0 +1,302 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .types import (
+        ActualErrorResponse,
+        AddMemoryResponse,
+        ApiKeyCreateResponse,
+        AttachmentModel,
+        BatchProcessingStatus,
+        Bm25OperatorType,
+        CollectionStats,
+        ContentFilter,
+        ContentModel,
+        CortexModelsResponseCommonsActualErrorResponse,
+        CortexModelsResponseCommonsErrorResponse,
+        CustomPropertyDefinition,
+        DeleteResult,
+        DeleteUserMemoryResponse,
+        Entity,
+        ErrorResponse,
+        FetchMode,
+        ForcefulRelationsPayload,
+        GraphContext,
+        HttpValidationError,
+        Infra,
+        InfraStatusResponse,
+        IngestResponse,
+        InsertResult,
+        ListContentKind,
+        ListUserMemoriesResponse,
+        MemoryItem,
+        MemoryResultItem,
+        MilvusDataType,
+        PaginationMeta,
+        PathTriplet,
+        ProcessingStatus,
+        ProcessingStatusIndexingStatus,
+        QnASearchResponse,
+        RawEmbeddingDocument,
+        RawEmbeddingSearchResult,
+        RawEmbeddingVector,
+        RecallSearchRequest,
+        RecallSearchRequestAlpha,
+        RelationEvidence,
+        RetrievalResult,
+        RetrieveMode,
+        ScoredPathResponse,
+        SearchMode,
+        SourceDeleteResponse,
+        SourceDeleteResultItem,
+        SourceFetchResponse,
+        SourceGraphRelationsResponse,
+        SourceInfo,
+        SourceListResponse,
+        SourceModel,
+        SourceStatus,
+        SourceUploadResponse,
+        SourceUploadResultItem,
+        SubTenantIdsResponse,
+        SuperNodeItem,
+        SuperNodeResponse,
+        SupportedLlmProviders,
+        TenantCreateAcceptedResponse,
+        TenantDeleteResponse,
+        TenantIdsResponse,
+        TenantStatsResponse,
+        TripletWithEvidence,
+        UserAssistantPair,
+        UserMemory,
+        ValidationError,
+        ValidationErrorLocItem,
+        VectorStoreChunk,
+    )
+    from .errors import (
+        BadRequestError,
+        ForbiddenError,
+        InternalServerError,
+        NotFoundError,
+        ServiceUnavailableError,
+        TooManyRequestsError,
+        UnauthorizedError,
+        UnprocessableEntityError,
+    )
+    from . import data, embeddings, fetch, graph_health, ingestion_pipeline, key, recall, tenant, upload
+    from ._default_clients import DefaultAioHttpClient, DefaultAsyncHttpxClient
+    from .client import AsyncHydraDB, HydraDB
+    from .environment import HydraDBEnvironment
+    from .fetch import ListDataFetchResponse
+_dynamic_imports: typing.Dict[str, str] = {
+    "ActualErrorResponse": ".types",
+    "AddMemoryResponse": ".types",
+    "ApiKeyCreateResponse": ".types",
+    "AsyncHydraDB": ".client",
+    "AttachmentModel": ".types",
+    "BadRequestError": ".errors",
+    "BatchProcessingStatus": ".types",
+    "Bm25OperatorType": ".types",
+    "CollectionStats": ".types",
+    "ContentFilter": ".types",
+    "ContentModel": ".types",
+    "CortexModelsResponseCommonsActualErrorResponse": ".types",
+    "CortexModelsResponseCommonsErrorResponse": ".types",
+    "CustomPropertyDefinition": ".types",
+    "DefaultAioHttpClient": "._default_clients",
+    "DefaultAsyncHttpxClient": "._default_clients",
+    "DeleteResult": ".types",
+    "DeleteUserMemoryResponse": ".types",
+    "Entity": ".types",
+    "ErrorResponse": ".types",
+    "FetchMode": ".types",
+    "ForbiddenError": ".errors",
+    "ForcefulRelationsPayload": ".types",
+    "GraphContext": ".types",
+    "HttpValidationError": ".types",
+    "HydraDB": ".client",
+    "HydraDBEnvironment": ".environment",
+    "Infra": ".types",
+    "InfraStatusResponse": ".types",
+    "IngestResponse": ".types",
+    "InsertResult": ".types",
+    "InternalServerError": ".errors",
+    "ListContentKind": ".types",
+    "ListDataFetchResponse": ".fetch",
+    "ListUserMemoriesResponse": ".types",
+    "MemoryItem": ".types",
+    "MemoryResultItem": ".types",
+    "MilvusDataType": ".types",
+    "NotFoundError": ".errors",
+    "PaginationMeta": ".types",
+    "PathTriplet": ".types",
+    "ProcessingStatus": ".types",
+    "ProcessingStatusIndexingStatus": ".types",
+    "QnASearchResponse": ".types",
+    "RawEmbeddingDocument": ".types",
+    "RawEmbeddingSearchResult": ".types",
+    "RawEmbeddingVector": ".types",
+    "RecallSearchRequest": ".types",
+    "RecallSearchRequestAlpha": ".types",
+    "RelationEvidence": ".types",
+    "RetrievalResult": ".types",
+    "RetrieveMode": ".types",
+    "ScoredPathResponse": ".types",
+    "SearchMode": ".types",
+    "ServiceUnavailableError": ".errors",
+    "SourceDeleteResponse": ".types",
+    "SourceDeleteResultItem": ".types",
+    "SourceFetchResponse": ".types",
+    "SourceGraphRelationsResponse": ".types",
+    "SourceInfo": ".types",
+    "SourceListResponse": ".types",
+    "SourceModel": ".types",
+    "SourceStatus": ".types",
+    "SourceUploadResponse": ".types",
+    "SourceUploadResultItem": ".types",
+    "SubTenantIdsResponse": ".types",
+    "SuperNodeItem": ".types",
+    "SuperNodeResponse": ".types",
+    "SupportedLlmProviders": ".types",
+    "TenantCreateAcceptedResponse": ".types",
+    "TenantDeleteResponse": ".types",
+    "TenantIdsResponse": ".types",
+    "TenantStatsResponse": ".types",
+    "TooManyRequestsError": ".errors",
+    "TripletWithEvidence": ".types",
+    "UnauthorizedError": ".errors",
+    "UnprocessableEntityError": ".errors",
+    "UserAssistantPair": ".types",
+    "UserMemory": ".types",
+    "ValidationError": ".types",
+    "ValidationErrorLocItem": ".types",
+    "VectorStoreChunk": ".types",
+    "data": ".data",
+    "embeddings": ".embeddings",
+    "fetch": ".fetch",
+    "graph_health": ".graph_health",
+    "ingestion_pipeline": ".ingestion_pipeline",
+    "key": ".key",
+    "recall": ".recall",
+    "tenant": ".tenant",
+    "upload": ".upload",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = [
+    "ActualErrorResponse",
+    "AddMemoryResponse",
+    "ApiKeyCreateResponse",
+    "AsyncHydraDB",
+    "AttachmentModel",
+    "BadRequestError",
+    "BatchProcessingStatus",
+    "Bm25OperatorType",
+    "CollectionStats",
+    "ContentFilter",
+    "ContentModel",
+    "CortexModelsResponseCommonsActualErrorResponse",
+    "CortexModelsResponseCommonsErrorResponse",
+    "CustomPropertyDefinition",
+    "DefaultAioHttpClient",
+    "DefaultAsyncHttpxClient",
+    "DeleteResult",
+    "DeleteUserMemoryResponse",
+    "Entity",
+    "ErrorResponse",
+    "FetchMode",
+    "ForbiddenError",
+    "ForcefulRelationsPayload",
+    "GraphContext",
+    "HttpValidationError",
+    "HydraDB",
+    "HydraDBEnvironment",
+    "Infra",
+    "InfraStatusResponse",
+    "IngestResponse",
+    "InsertResult",
+    "InternalServerError",
+    "ListContentKind",
+    "ListDataFetchResponse",
+    "ListUserMemoriesResponse",
+    "MemoryItem",
+    "MemoryResultItem",
+    "MilvusDataType",
+    "NotFoundError",
+    "PaginationMeta",
+    "PathTriplet",
+    "ProcessingStatus",
+    "ProcessingStatusIndexingStatus",
+    "QnASearchResponse",
+    "RawEmbeddingDocument",
+    "RawEmbeddingSearchResult",
+    "RawEmbeddingVector",
+    "RecallSearchRequest",
+    "RecallSearchRequestAlpha",
+    "RelationEvidence",
+    "RetrievalResult",
+    "RetrieveMode",
+    "ScoredPathResponse",
+    "SearchMode",
+    "ServiceUnavailableError",
+    "SourceDeleteResponse",
+    "SourceDeleteResultItem",
+    "SourceFetchResponse",
+    "SourceGraphRelationsResponse",
+    "SourceInfo",
+    "SourceListResponse",
+    "SourceModel",
+    "SourceStatus",
+    "SourceUploadResponse",
+    "SourceUploadResultItem",
+    "SubTenantIdsResponse",
+    "SuperNodeItem",
+    "SuperNodeResponse",
+    "SupportedLlmProviders",
+    "TenantCreateAcceptedResponse",
+    "TenantDeleteResponse",
+    "TenantIdsResponse",
+    "TenantStatsResponse",
+    "TooManyRequestsError",
+    "TripletWithEvidence",
+    "UnauthorizedError",
+    "UnprocessableEntityError",
+    "UserAssistantPair",
+    "UserMemory",
+    "ValidationError",
+    "ValidationErrorLocItem",
+    "VectorStoreChunk",
+    "data",
+    "embeddings",
+    "fetch",
+    "graph_health",
+    "ingestion_pipeline",
+    "key",
+    "recall",
+    "tenant",
+    "upload",
+]

hydra_db/_default_clients.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import httpx
+
+SDK_DEFAULT_TIMEOUT = 60
+
+try:
+    import httpx_aiohttp  # type: ignore[import-not-found]
+except ImportError:
+
+    class DefaultAioHttpClient(httpx.AsyncClient):  # type: ignore
+        def __init__(self, **kwargs: typing.Any) -> None:
+            raise RuntimeError("To use the aiohttp client, install the aiohttp extra: pip install package[aiohttp]")
+
+else:
+
+    class DefaultAioHttpClient(httpx_aiohttp.HttpxAiohttpClient):  # type: ignore
+        def __init__(self, **kwargs: typing.Any) -> None:
+            kwargs.setdefault("timeout", SDK_DEFAULT_TIMEOUT)
+            kwargs.setdefault("follow_redirects", True)
+            super().__init__(**kwargs)
+
+
+class DefaultAsyncHttpxClient(httpx.AsyncClient):
+    def __init__(self, **kwargs: typing.Any) -> None:
+        kwargs.setdefault("timeout", SDK_DEFAULT_TIMEOUT)
+        kwargs.setdefault("follow_redirects", True)
+        super().__init__(**kwargs)