hydra-db-python 0.1.0__py3-none-any.whl

hydra_db/environment.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+
+
+class HydraDBEnvironment(enum.Enum):
+    HYDRA_DB_PROD = "https://api.hydradb.com"

hydra_db/errors/__init__.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .bad_request_error import BadRequestError
+from .forbidden_error import ForbiddenError
+from .internal_server_error import InternalServerError
+from .not_found_error import NotFoundError
+from .service_unavailable_error import ServiceUnavailableError
+from .too_many_requests_error import TooManyRequestsError
+from .unauthorized_error import UnauthorizedError
+from .unprocessable_entity_error import UnprocessableEntityError
+
+__all__ = [
+    "BadRequestError",
+    "ForbiddenError",
+    "InternalServerError",
+    "NotFoundError",
+    "ServiceUnavailableError",
+    "TooManyRequestsError",
+    "UnauthorizedError",
+    "UnprocessableEntityError",
+]

hydra_db/errors/bad_request_error.py

@@ -0,0 +1,10 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.api_error import ApiError
+
+
+class BadRequestError(ApiError):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+        super().__init__(status_code=400, headers=headers, body=body)

hydra_db/errors/forbidden_error.py

@@ -0,0 +1,10 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.api_error import ApiError
+
+
+class ForbiddenError(ApiError):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+        super().__init__(status_code=403, headers=headers, body=body)

hydra_db/errors/internal_server_error.py

@@ -0,0 +1,10 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.api_error import ApiError
+
+
+class InternalServerError(ApiError):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+        super().__init__(status_code=500, headers=headers, body=body)

hydra_db/errors/not_found_error.py

@@ -0,0 +1,10 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.api_error import ApiError
+
+
+class NotFoundError(ApiError):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+        super().__init__(status_code=404, headers=headers, body=body)

hydra_db/errors/service_unavailable_error.py

@@ -0,0 +1,10 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.api_error import ApiError
+
+
+class ServiceUnavailableError(ApiError):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+        super().__init__(status_code=503, headers=headers, body=body)

hydra_db/errors/too_many_requests_error.py

@@ -0,0 +1,11 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.api_error import ApiError
+from ..types.actual_error_response import ActualErrorResponse
+
+
+class TooManyRequestsError(ApiError):
+    def __init__(self, body: ActualErrorResponse, headers: typing.Optional[typing.Dict[str, str]] = None):
+        super().__init__(status_code=429, headers=headers, body=body)

hydra_db/errors/unauthorized_error.py

@@ -0,0 +1,10 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.api_error import ApiError
+
+
+class UnauthorizedError(ApiError):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+        super().__init__(status_code=401, headers=headers, body=body)

hydra_db/errors/unprocessable_entity_error.py

@@ -0,0 +1,10 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.api_error import ApiError
+
+
+class UnprocessableEntityError(ApiError):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+        super().__init__(status_code=422, headers=headers, body=body)

hydra_db/fetch/__init__.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .types import FetchListDataResponse
+
+__all__ = ["FetchListDataResponse"]

hydra_db/fetch/client.py

@@ -0,0 +1,408 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from ..types.content_filter import ContentFilter
+from ..types.fetch_mode import FetchMode
+from ..types.list_content_kind import ListContentKind
+from ..types.source_fetch_response import SourceFetchResponse
+from ..types.source_graph_relations_response import SourceGraphRelationsResponse
+from .raw_client import AsyncRawFetchClient, RawFetchClient
+from .types.fetch_list_data_response import FetchListDataResponse
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class FetchClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawFetchClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawFetchClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawFetchClient
+        """
+        return self._raw_client
+
+    def list_data(
+        self,
+        *,
+        tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = OMIT,
+        kind: typing.Optional[ListContentKind] = OMIT,
+        source_ids: typing.Optional[typing.Sequence[str]] = OMIT,
+        page: typing.Optional[int] = OMIT,
+        page_size: typing.Optional[int] = OMIT,
+        filters: typing.Optional[ContentFilter] = OMIT,
+        include_fields: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> FetchListDataResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            Tenant ID
+
+        sub_tenant_id : typing.Optional[str]
+            Sub-tenant ID
+
+        kind : typing.Optional[ListContentKind]
+            Whether to list normal sources or user memories.
+
+        source_ids : typing.Optional[typing.Sequence[str]]
+            Optional list of IDs to fetch (max 100). If omitted, returns all.
+
+        page : typing.Optional[int]
+            Page number to retrieve (1-indexed). Defaults to 1.
+
+        page_size : typing.Optional[int]
+            Number of items per page (1-100). Defaults to 50.
+
+        filters : typing.Optional[ContentFilter]
+            Optional filters. Provide key-value pairs to match against tenant_metadata, document_metadata, and/or source-level fields (title, type, description, url, timestamp).
+
+        include_fields : typing.Optional[typing.Sequence[str]]
+            Optional list of source fields to include in the response. When provided, only the specified fields (plus id, tenant_id, sub_tenant_id which are always returned) will be populated; all other fields will have their default/empty values. This reduces payload size and improves performance. Allowed values: attachments, content, description, document_metadata, note, relations, tenant_metadata, timestamp, title, type, url. Omit or pass null to return all fields. Only applies to kind=knowledge; ignored for kind=memories.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        FetchListDataResponse
+            Successful Response
+
+        Examples
+        --------
+        from hydra-db import HydraDB
+
+        client = HydraDB(token="YOUR_TOKEN", )
+        client.fetch.list_data(tenant_id='tenant_1234', )
+        """
+        _response = self._raw_client.list_data(
+            tenant_id=tenant_id,
+            sub_tenant_id=sub_tenant_id,
+            kind=kind,
+            source_ids=source_ids,
+            page=page,
+            page_size=page_size,
+            filters=filters,
+            include_fields=include_fields,
+            request_options=request_options,
+        )
+        return _response.data
+
+    def graph_relations_by_source_id(
+        self,
+        *,
+        source_id: str,
+        tenant_id: typing.Optional[str] = None,
+        is_memory: typing.Optional[bool] = None,
+        sub_tenant_id: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SourceGraphRelationsResponse:
+        """
+        Parameters
+        ----------
+        source_id : str
+            The source ID to fetch relations for
+
+        tenant_id : typing.Optional[str]
+            Unique identifier for the tenant/organization
+
+        is_memory : typing.Optional[bool]
+            Whether to fetch relations for memories
+
+        sub_tenant_id : typing.Optional[str]
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
+
+        limit : typing.Optional[int]
+            Maximum number of relations to return
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SourceGraphRelationsResponse
+            Successful Response
+
+        Examples
+        --------
+        from hydra-db import HydraDB
+
+        client = HydraDB(token="YOUR_TOKEN", )
+        client.fetch.graph_relations_by_source_id(source_id='<str>', tenant_id='tenant_1234', )
+        """
+        _response = self._raw_client.graph_relations_by_source_id(
+            source_id=source_id,
+            tenant_id=tenant_id,
+            is_memory=is_memory,
+            sub_tenant_id=sub_tenant_id,
+            limit=limit,
+            request_options=request_options,
+        )
+        return _response.data
+
+    def content(
+        self,
+        *,
+        tenant_id: str,
+        source_id: str,
+        sub_tenant_id: typing.Optional[str] = OMIT,
+        mode: typing.Optional[FetchMode] = OMIT,
+        expiry_seconds: typing.Optional[int] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SourceFetchResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        source_id : str
+            Source ID of the file to fetch
+
+        sub_tenant_id : typing.Optional[str]
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
+
+        mode : typing.Optional[FetchMode]
+            Fetch mode: 'content' returns file content, 'url' returns presigned URL, 'both' returns both
+
+        expiry_seconds : typing.Optional[int]
+            Expiry time in seconds for presigned URL (60-604800, default: 3600)
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SourceFetchResponse
+            Successful Response
+
+        Examples
+        --------
+        from hydra-db import HydraDB
+
+        client = HydraDB(token="YOUR_TOKEN", )
+        client.fetch.content(tenant_id='tenant_1234', source_id='<source_id>', )
+        """
+        _response = self._raw_client.content(
+            tenant_id=tenant_id,
+            source_id=source_id,
+            sub_tenant_id=sub_tenant_id,
+            mode=mode,
+            expiry_seconds=expiry_seconds,
+            request_options=request_options,
+        )
+        return _response.data
+
+
+class AsyncFetchClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawFetchClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawFetchClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawFetchClient
+        """
+        return self._raw_client
+
+    async def list_data(
+        self,
+        *,
+        tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = OMIT,
+        kind: typing.Optional[ListContentKind] = OMIT,
+        source_ids: typing.Optional[typing.Sequence[str]] = OMIT,
+        page: typing.Optional[int] = OMIT,
+        page_size: typing.Optional[int] = OMIT,
+        filters: typing.Optional[ContentFilter] = OMIT,
+        include_fields: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> FetchListDataResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            Tenant ID
+
+        sub_tenant_id : typing.Optional[str]
+            Sub-tenant ID
+
+        kind : typing.Optional[ListContentKind]
+            Whether to list normal sources or user memories.
+
+        source_ids : typing.Optional[typing.Sequence[str]]
+            Optional list of IDs to fetch (max 100). If omitted, returns all.
+
+        page : typing.Optional[int]
+            Page number to retrieve (1-indexed). Defaults to 1.
+
+        page_size : typing.Optional[int]
+            Number of items per page (1-100). Defaults to 50.
+
+        filters : typing.Optional[ContentFilter]
+            Optional filters. Provide key-value pairs to match against tenant_metadata, document_metadata, and/or source-level fields (title, type, description, url, timestamp).
+
+        include_fields : typing.Optional[typing.Sequence[str]]
+            Optional list of source fields to include in the response. When provided, only the specified fields (plus id, tenant_id, sub_tenant_id which are always returned) will be populated; all other fields will have their default/empty values. This reduces payload size and improves performance. Allowed values: attachments, content, description, document_metadata, note, relations, tenant_metadata, timestamp, title, type, url. Omit or pass null to return all fields. Only applies to kind=knowledge; ignored for kind=memories.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        FetchListDataResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from hydra-db import AsyncHydraDB
+
+        client = AsyncHydraDB(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.fetch.list_data(tenant_id='tenant_1234', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.list_data(
+            tenant_id=tenant_id,
+            sub_tenant_id=sub_tenant_id,
+            kind=kind,
+            source_ids=source_ids,
+            page=page,
+            page_size=page_size,
+            filters=filters,
+            include_fields=include_fields,
+            request_options=request_options,
+        )
+        return _response.data
+
+    async def graph_relations_by_source_id(
+        self,
+        *,
+        source_id: str,
+        tenant_id: typing.Optional[str] = None,
+        is_memory: typing.Optional[bool] = None,
+        sub_tenant_id: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SourceGraphRelationsResponse:
+        """
+        Parameters
+        ----------
+        source_id : str
+            The source ID to fetch relations for
+
+        tenant_id : typing.Optional[str]
+            Unique identifier for the tenant/organization
+
+        is_memory : typing.Optional[bool]
+            Whether to fetch relations for memories
+
+        sub_tenant_id : typing.Optional[str]
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
+
+        limit : typing.Optional[int]
+            Maximum number of relations to return
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SourceGraphRelationsResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from hydra-db import AsyncHydraDB
+
+        client = AsyncHydraDB(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.fetch.graph_relations_by_source_id(source_id='<str>', tenant_id='tenant_1234', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.graph_relations_by_source_id(
+            source_id=source_id,
+            tenant_id=tenant_id,
+            is_memory=is_memory,
+            sub_tenant_id=sub_tenant_id,
+            limit=limit,
+            request_options=request_options,
+        )
+        return _response.data
+
+    async def content(
+        self,
+        *,
+        tenant_id: str,
+        source_id: str,
+        sub_tenant_id: typing.Optional[str] = OMIT,
+        mode: typing.Optional[FetchMode] = OMIT,
+        expiry_seconds: typing.Optional[int] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SourceFetchResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        source_id : str
+            Source ID of the file to fetch
+
+        sub_tenant_id : typing.Optional[str]
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
+
+        mode : typing.Optional[FetchMode]
+            Fetch mode: 'content' returns file content, 'url' returns presigned URL, 'both' returns both
+
+        expiry_seconds : typing.Optional[int]
+            Expiry time in seconds for presigned URL (60-604800, default: 3600)
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SourceFetchResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from hydra-db import AsyncHydraDB
+
+        client = AsyncHydraDB(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.fetch.content(tenant_id='tenant_1234', source_id='<source_id>', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.content(
+            tenant_id=tenant_id,
+            source_id=source_id,
+            sub_tenant_id=sub_tenant_id,
+            mode=mode,
+            expiry_seconds=expiry_seconds,
+            request_options=request_options,
+        )
+        return _response.data