c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python 0.1.0a3__py3-none-any.whl

digitalocean_genai_sdk/resources/agents/knowledge_bases.py

@@ -0,0 +1,346 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._compat import cached_property
+from ..._resource import SyncAPIResource, AsyncAPIResource
+from ..._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from ..._base_client import make_request_options
+from ...types.agents.api_link_knowledge_base_output import APILinkKnowledgeBaseOutput
+from ...types.agents.knowledge_base_detach_response import KnowledgeBaseDetachResponse
+
+__all__ = ["KnowledgeBasesResource", "AsyncKnowledgeBasesResource"]
+
+
+class KnowledgeBasesResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> KnowledgeBasesResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
+        """
+        return KnowledgeBasesResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> KnowledgeBasesResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        """
+        return KnowledgeBasesResourceWithStreamingResponse(self)
+
+    def attach(
+        self,
+        agent_uuid: str,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> APILinkKnowledgeBaseOutput:
+        """
+        To attach knowledge bases to an agent, send a POST request to
+        `/v2/gen-ai/agents/{agent_uuid}/knowledge_bases`
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not agent_uuid:
+            raise ValueError(f"Expected a non-empty value for `agent_uuid` but received {agent_uuid!r}")
+        return self._post(
+            f"/v2/genai/agents/{agent_uuid}/knowledge_bases",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=APILinkKnowledgeBaseOutput,
+        )
+
+    def attach_single(
+        self,
+        knowledge_base_uuid: str,
+        *,
+        agent_uuid: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> APILinkKnowledgeBaseOutput:
+        """
+        To attach a knowledge base to an agent, send a POST request to
+        `/v2/gen-ai/agents/{agent_uuid}/knowledge_bases/{knowledge_base_uuid}`
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not agent_uuid:
+            raise ValueError(f"Expected a non-empty value for `agent_uuid` but received {agent_uuid!r}")
+        if not knowledge_base_uuid:
+            raise ValueError(
+                f"Expected a non-empty value for `knowledge_base_uuid` but received {knowledge_base_uuid!r}"
+            )
+        return self._post(
+            f"/v2/genai/agents/{agent_uuid}/knowledge_bases/{knowledge_base_uuid}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=APILinkKnowledgeBaseOutput,
+        )
+
+    def detach(
+        self,
+        knowledge_base_uuid: str,
+        *,
+        agent_uuid: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> KnowledgeBaseDetachResponse:
+        """
+        To detach a knowledge base from an agent, send a DELETE request to
+        `/v2/gen-ai/agents/{agent_uuid}/knowledge_bases/{knowledge_base_uuid}`.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not agent_uuid:
+            raise ValueError(f"Expected a non-empty value for `agent_uuid` but received {agent_uuid!r}")
+        if not knowledge_base_uuid:
+            raise ValueError(
+                f"Expected a non-empty value for `knowledge_base_uuid` but received {knowledge_base_uuid!r}"
+            )
+        return self._delete(
+            f"/v2/genai/agents/{agent_uuid}/knowledge_bases/{knowledge_base_uuid}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=KnowledgeBaseDetachResponse,
+        )
+
+
+class AsyncKnowledgeBasesResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncKnowledgeBasesResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncKnowledgeBasesResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncKnowledgeBasesResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        """
+        return AsyncKnowledgeBasesResourceWithStreamingResponse(self)
+
+    async def attach(
+        self,
+        agent_uuid: str,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> APILinkKnowledgeBaseOutput:
+        """
+        To attach knowledge bases to an agent, send a POST request to
+        `/v2/gen-ai/agents/{agent_uuid}/knowledge_bases`
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not agent_uuid:
+            raise ValueError(f"Expected a non-empty value for `agent_uuid` but received {agent_uuid!r}")
+        return await self._post(
+            f"/v2/genai/agents/{agent_uuid}/knowledge_bases",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=APILinkKnowledgeBaseOutput,
+        )
+
+    async def attach_single(
+        self,
+        knowledge_base_uuid: str,
+        *,
+        agent_uuid: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> APILinkKnowledgeBaseOutput:
+        """
+        To attach a knowledge base to an agent, send a POST request to
+        `/v2/gen-ai/agents/{agent_uuid}/knowledge_bases/{knowledge_base_uuid}`
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not agent_uuid:
+            raise ValueError(f"Expected a non-empty value for `agent_uuid` but received {agent_uuid!r}")
+        if not knowledge_base_uuid:
+            raise ValueError(
+                f"Expected a non-empty value for `knowledge_base_uuid` but received {knowledge_base_uuid!r}"
+            )
+        return await self._post(
+            f"/v2/genai/agents/{agent_uuid}/knowledge_bases/{knowledge_base_uuid}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=APILinkKnowledgeBaseOutput,
+        )
+
+    async def detach(
+        self,
+        knowledge_base_uuid: str,
+        *,
+        agent_uuid: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> KnowledgeBaseDetachResponse:
+        """
+        To detach a knowledge base from an agent, send a DELETE request to
+        `/v2/gen-ai/agents/{agent_uuid}/knowledge_bases/{knowledge_base_uuid}`.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not agent_uuid:
+            raise ValueError(f"Expected a non-empty value for `agent_uuid` but received {agent_uuid!r}")
+        if not knowledge_base_uuid:
+            raise ValueError(
+                f"Expected a non-empty value for `knowledge_base_uuid` but received {knowledge_base_uuid!r}"
+            )
+        return await self._delete(
+            f"/v2/genai/agents/{agent_uuid}/knowledge_bases/{knowledge_base_uuid}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=KnowledgeBaseDetachResponse,
+        )
+
+
+class KnowledgeBasesResourceWithRawResponse:
+    def __init__(self, knowledge_bases: KnowledgeBasesResource) -> None:
+        self._knowledge_bases = knowledge_bases
+
+        self.attach = to_raw_response_wrapper(
+            knowledge_bases.attach,
+        )
+        self.attach_single = to_raw_response_wrapper(
+            knowledge_bases.attach_single,
+        )
+        self.detach = to_raw_response_wrapper(
+            knowledge_bases.detach,
+        )
+
+
+class AsyncKnowledgeBasesResourceWithRawResponse:
+    def __init__(self, knowledge_bases: AsyncKnowledgeBasesResource) -> None:
+        self._knowledge_bases = knowledge_bases
+
+        self.attach = async_to_raw_response_wrapper(
+            knowledge_bases.attach,
+        )
+        self.attach_single = async_to_raw_response_wrapper(
+            knowledge_bases.attach_single,
+        )
+        self.detach = async_to_raw_response_wrapper(
+            knowledge_bases.detach,
+        )
+
+
+class KnowledgeBasesResourceWithStreamingResponse:
+    def __init__(self, knowledge_bases: KnowledgeBasesResource) -> None:
+        self._knowledge_bases = knowledge_bases
+
+        self.attach = to_streamed_response_wrapper(
+            knowledge_bases.attach,
+        )
+        self.attach_single = to_streamed_response_wrapper(
+            knowledge_bases.attach_single,
+        )
+        self.detach = to_streamed_response_wrapper(
+            knowledge_bases.detach,
+        )
+
+
+class AsyncKnowledgeBasesResourceWithStreamingResponse:
+    def __init__(self, knowledge_bases: AsyncKnowledgeBasesResource) -> None:
+        self._knowledge_bases = knowledge_bases
+
+        self.attach = async_to_streamed_response_wrapper(
+            knowledge_bases.attach,
+        )
+        self.attach_single = async_to_streamed_response_wrapper(
+            knowledge_bases.attach_single,
+        )
+        self.detach = async_to_streamed_response_wrapper(
+            knowledge_bases.detach,
+        )

digitalocean_genai_sdk/resources/agents/versions.py

@@ -0,0 +1,298 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._utils import maybe_transform, async_maybe_transform
+from ..._compat import cached_property
+from ..._resource import SyncAPIResource, AsyncAPIResource
+from ..._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from ..._base_client import make_request_options
+from ...types.agents import version_list_params, version_update_params
+from ...types.agents.version_list_response import VersionListResponse
+from ...types.agents.version_update_response import VersionUpdateResponse
+
+__all__ = ["VersionsResource", "AsyncVersionsResource"]
+
+
+class VersionsResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> VersionsResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
+        """
+        return VersionsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> VersionsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        """
+        return VersionsResourceWithStreamingResponse(self)
+
+    def update(
+        self,
+        path_uuid: str,
+        *,
+        body_uuid: str | NotGiven = NOT_GIVEN,
+        version_hash: str | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> VersionUpdateResponse:
+        """
+        To update to a specific agent version, send a PUT request to
+        `/v2/gen-ai/agents/{uuid}/versions`.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not path_uuid:
+            raise ValueError(f"Expected a non-empty value for `path_uuid` but received {path_uuid!r}")
+        return self._put(
+            f"/v2/gen-ai/agents/{path_uuid}/versions",
+            body=maybe_transform(
+                {
+                    "body_uuid": body_uuid,
+                    "version_hash": version_hash,
+                },
+                version_update_params.VersionUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=VersionUpdateResponse,
+        )
+
+    def list(
+        self,
+        uuid: str,
+        *,
+        page: int | NotGiven = NOT_GIVEN,
+        per_page: int | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> VersionListResponse:
+        """
+        To list all agent versions, send a GET request to
+        `/v2/gen-ai/agents/{uuid}/versions`.
+
+        Args:
+          page: page number.
+
+          per_page: items per page.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not uuid:
+            raise ValueError(f"Expected a non-empty value for `uuid` but received {uuid!r}")
+        return self._get(
+            f"/v2/gen-ai/agents/{uuid}/versions",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "page": page,
+                        "per_page": per_page,
+                    },
+                    version_list_params.VersionListParams,
+                ),
+            ),
+            cast_to=VersionListResponse,
+        )
+
+
+class AsyncVersionsResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncVersionsResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncVersionsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncVersionsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        """
+        return AsyncVersionsResourceWithStreamingResponse(self)
+
+    async def update(
+        self,
+        path_uuid: str,
+        *,
+        body_uuid: str | NotGiven = NOT_GIVEN,
+        version_hash: str | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> VersionUpdateResponse:
+        """
+        To update to a specific agent version, send a PUT request to
+        `/v2/gen-ai/agents/{uuid}/versions`.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not path_uuid:
+            raise ValueError(f"Expected a non-empty value for `path_uuid` but received {path_uuid!r}")
+        return await self._put(
+            f"/v2/gen-ai/agents/{path_uuid}/versions",
+            body=await async_maybe_transform(
+                {
+                    "body_uuid": body_uuid,
+                    "version_hash": version_hash,
+                },
+                version_update_params.VersionUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=VersionUpdateResponse,
+        )
+
+    async def list(
+        self,
+        uuid: str,
+        *,
+        page: int | NotGiven = NOT_GIVEN,
+        per_page: int | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> VersionListResponse:
+        """
+        To list all agent versions, send a GET request to
+        `/v2/gen-ai/agents/{uuid}/versions`.
+
+        Args:
+          page: page number.
+
+          per_page: items per page.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not uuid:
+            raise ValueError(f"Expected a non-empty value for `uuid` but received {uuid!r}")
+        return await self._get(
+            f"/v2/gen-ai/agents/{uuid}/versions",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "page": page,
+                        "per_page": per_page,
+                    },
+                    version_list_params.VersionListParams,
+                ),
+            ),
+            cast_to=VersionListResponse,
+        )
+
+
+class VersionsResourceWithRawResponse:
+    def __init__(self, versions: VersionsResource) -> None:
+        self._versions = versions
+
+        self.update = to_raw_response_wrapper(
+            versions.update,
+        )
+        self.list = to_raw_response_wrapper(
+            versions.list,
+        )
+
+
+class AsyncVersionsResourceWithRawResponse:
+    def __init__(self, versions: AsyncVersionsResource) -> None:
+        self._versions = versions
+
+        self.update = async_to_raw_response_wrapper(
+            versions.update,
+        )
+        self.list = async_to_raw_response_wrapper(
+            versions.list,
+        )
+
+
+class VersionsResourceWithStreamingResponse:
+    def __init__(self, versions: VersionsResource) -> None:
+        self._versions = versions
+
+        self.update = to_streamed_response_wrapper(
+            versions.update,
+        )
+        self.list = to_streamed_response_wrapper(
+            versions.list,
+        )
+
+
+class AsyncVersionsResourceWithStreamingResponse:
+    def __init__(self, versions: AsyncVersionsResource) -> None:
+        self._versions = versions
+
+        self.update = async_to_streamed_response_wrapper(
+            versions.update,
+        )
+        self.list = async_to_streamed_response_wrapper(
+            versions.list,
+        )

digitalocean_genai_sdk/resources/api_keys/__init__.py

@@ -0,0 +1,19 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .api_keys import (
+    APIKeysResource,
+    AsyncAPIKeysResource,
+    APIKeysResourceWithRawResponse,
+    AsyncAPIKeysResourceWithRawResponse,
+    APIKeysResourceWithStreamingResponse,
+    AsyncAPIKeysResourceWithStreamingResponse,
+)
+
+__all__ = [
+    "APIKeysResource",
+    "AsyncAPIKeysResource",
+    "APIKeysResourceWithRawResponse",
+    "AsyncAPIKeysResourceWithRawResponse",
+    "APIKeysResourceWithStreamingResponse",
+    "AsyncAPIKeysResourceWithStreamingResponse",
+]