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

{digitalocean_genai_sdk → do_gradientai}/_exceptions.py

@@ -18,11 +18,11 @@ __all__ = [
 ]
 
 
-class DigitaloceanGenaiSDKError(Exception):
+class GradientAIError(Exception):
     pass
 
 
-class APIError(DigitaloceanGenaiSDKError):
+class APIError(GradientAIError):
     message: str
     request: httpx.Request
 

{digitalocean_genai_sdk → do_gradientai}/_resource.py

@@ -8,13 +8,13 @@ from typing import TYPE_CHECKING
 import anyio
 
 if TYPE_CHECKING:
-    from ._client import DigitaloceanGenaiSDK, AsyncDigitaloceanGenaiSDK
+    from ._client import GradientAI, AsyncGradientAI
 
 
 class SyncAPIResource:
-    _client: DigitaloceanGenaiSDK
+    _client: GradientAI
 
-    def __init__(self, client: DigitaloceanGenaiSDK) -> None:
+    def __init__(self, client: GradientAI) -> None:
         self._client = client
         self._get = client.get
         self._post = client.post
@@ -28,9 +28,9 @@ class SyncAPIResource:
 
 
 class AsyncAPIResource:
-    _client: AsyncDigitaloceanGenaiSDK
+    _client: AsyncGradientAI
 
-    def __init__(self, client: AsyncDigitaloceanGenaiSDK) -> None:
+    def __init__(self, client: AsyncGradientAI) -> None:
         self._client = client
         self._get = client.get
         self._post = client.post

{digitalocean_genai_sdk → do_gradientai}/_response.py

@@ -29,7 +29,7 @@ from ._utils import is_given, extract_type_arg, is_annotated_type, is_type_alias
 from ._models import BaseModel, is_basemodel
 from ._constants import RAW_RESPONSE_HEADER, OVERRIDE_CAST_TO_HEADER
 from ._streaming import Stream, AsyncStream, is_stream_class_type, extract_stream_chunk_type
-from ._exceptions import DigitaloceanGenaiSDKError, APIResponseValidationError
+from ._exceptions import GradientAIError, APIResponseValidationError
 
 if TYPE_CHECKING:
     from ._models import FinalRequestOptions
@@ -218,7 +218,7 @@ class BaseAPIResponse(Generic[R]):
             and issubclass(origin, pydantic.BaseModel)
         ):
             raise TypeError(
-                "Pydantic models must subclass our base model type, e.g. `from digitalocean_genai_sdk import BaseModel`"
+                "Pydantic models must subclass our base model type, e.g. `from do_gradientai import BaseModel`"
             )
 
         if (
@@ -285,7 +285,7 @@ class APIResponse(BaseAPIResponse[R]):
         the `to` argument, e.g.
 
         ```py
-        from digitalocean_genai_sdk import BaseModel
+        from do_gradientai import BaseModel
 
 
         class MyModel(BaseModel):
@@ -387,7 +387,7 @@ class AsyncAPIResponse(BaseAPIResponse[R]):
         the `to` argument, e.g.
 
         ```py
-        from digitalocean_genai_sdk import BaseModel
+        from do_gradientai import BaseModel
 
 
         class MyModel(BaseModel):
@@ -558,11 +558,11 @@ class AsyncStreamedBinaryAPIResponse(AsyncAPIResponse[bytes]):
 class MissingStreamClassError(TypeError):
     def __init__(self) -> None:
         super().__init__(
-            "The `stream` argument was set to `True` but the `stream_cls` argument was not given. See `digitalocean_genai_sdk._streaming` for reference",
+            "The `stream` argument was set to `True` but the `stream_cls` argument was not given. See `do_gradientai._streaming` for reference",
         )
 
 
-class StreamAlreadyConsumed(DigitaloceanGenaiSDKError):
+class StreamAlreadyConsumed(GradientAIError):
     """
     Attempted to read or stream content, but the content has already
     been streamed.

{digitalocean_genai_sdk → do_gradientai}/_streaming.py

@@ -12,7 +12,7 @@ import httpx
 from ._utils import extract_type_var_from_base
 
 if TYPE_CHECKING:
-    from ._client import DigitaloceanGenaiSDK, AsyncDigitaloceanGenaiSDK
+    from ._client import GradientAI, AsyncGradientAI
 
 
 _T = TypeVar("_T")
@@ -30,7 +30,7 @@ class Stream(Generic[_T]):
         *,
         cast_to: type[_T],
         response: httpx.Response,
-        client: DigitaloceanGenaiSDK,
+        client: GradientAI,
     ) -> None:
         self.response = response
         self._cast_to = cast_to
@@ -93,7 +93,7 @@ class AsyncStream(Generic[_T]):
         *,
         cast_to: type[_T],
         response: httpx.Response,
-        client: AsyncDigitaloceanGenaiSDK,
+        client: AsyncGradientAI,
     ) -> None:
         self.response = response
         self._cast_to = cast_to

{digitalocean_genai_sdk → do_gradientai}/_types.py

@@ -81,7 +81,7 @@ HttpxRequestFiles = Union[Mapping[str, HttpxFileTypes], Sequence[Tuple[str, Http
 # This unfortunately means that you will either have
 # to import this type and pass it explicitly:
 #
-# from digitalocean_genai_sdk import NoneType
+# from do_gradientai import NoneType
 # client.get('/foo', cast_to=NoneType)
 #
 # or build it yourself:

{digitalocean_genai_sdk → do_gradientai}/_utils/_logs.py

@@ -1,12 +1,12 @@
 import os
 import logging
 
-logger: logging.Logger = logging.getLogger("digitalocean_genai_sdk")
+logger: logging.Logger = logging.getLogger("do_gradientai")
 httpx_logger: logging.Logger = logging.getLogger("httpx")
 
 
 def _basic_config() -> None:
-    # e.g. [2023-10-05 14:12:26 - digitalocean_genai_sdk._base_client:818 - DEBUG] HTTP Request: POST http://127.0.0.1:4010/foo/bar "200 OK"
+    # e.g. [2023-10-05 14:12:26 - do_gradientai._base_client:818 - DEBUG] HTTP Request: POST http://127.0.0.1:4010/foo/bar "200 OK"
     logging.basicConfig(
         format="[%(asctime)s - %(name)s:%(lineno)d - %(levelname)s] %(message)s",
         datefmt="%Y-%m-%d %H:%M:%S",
@@ -14,7 +14,7 @@ def _basic_config() -> None:
 
 
 def setup_logging() -> None:
-    env = os.environ.get("DIGITALOCEAN_GENAI_SDK_LOG")
+    env = os.environ.get("GRADIENT_AI_LOG")
     if env == "debug":
         _basic_config()
         logger.setLevel(logging.DEBUG)

do_gradientai/_utils/_resources_proxy.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+from typing import Any
+from typing_extensions import override
+
+from ._proxy import LazyProxy
+
+
+class ResourcesProxy(LazyProxy[Any]):
+    """A proxy for the `do_gradientai.resources` module.
+
+    This is used so that we can lazily import `do_gradientai.resources` only when
+    needed *and* so that users can just import `do_gradientai` and reference `do_gradientai.resources`
+    """
+
+    @override
+    def __load__(self) -> Any:
+        import importlib
+
+        mod = importlib.import_module("do_gradientai.resources")
+        return mod
+
+
+resources = ResourcesProxy().__as_proxied__()

do_gradientai/_version.py

@@ -0,0 +1,4 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+__title__ = "do_gradientai"
+__version__ = "0.1.0-alpha.5"  # x-release-please-version

{digitalocean_genai_sdk → do_gradientai}/resources/__init__.py

@@ -1,13 +1,5 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from .auth import (
-    AuthResource,
-    AsyncAuthResource,
-    AuthResourceWithRawResponse,
-    AsyncAuthResourceWithRawResponse,
-    AuthResourceWithStreamingResponse,
-    AsyncAuthResourceWithStreamingResponse,
-)
 from .chat import (
     ChatResource,
     AsyncChatResource,
@@ -40,37 +32,13 @@ from .regions import (
     RegionsResourceWithStreamingResponse,
     AsyncRegionsResourceWithStreamingResponse,
 )
-from .api_keys import (
-    APIKeysResource,
-    AsyncAPIKeysResource,
-    APIKeysResourceWithRawResponse,
-    AsyncAPIKeysResourceWithRawResponse,
-    APIKeysResourceWithStreamingResponse,
-    AsyncAPIKeysResourceWithStreamingResponse,
-)
-from .providers import (
-    ProvidersResource,
-    AsyncProvidersResource,
-    ProvidersResourceWithRawResponse,
-    AsyncProvidersResourceWithRawResponse,
-    ProvidersResourceWithStreamingResponse,
-    AsyncProvidersResourceWithStreamingResponse,
-)
-from .embeddings import (
-    EmbeddingsResource,
-    AsyncEmbeddingsResource,
-    EmbeddingsResourceWithRawResponse,
-    AsyncEmbeddingsResourceWithRawResponse,
-    EmbeddingsResourceWithStreamingResponse,
-    AsyncEmbeddingsResourceWithStreamingResponse,
-)
-from .indexing_jobs import (
-    IndexingJobsResource,
-    AsyncIndexingJobsResource,
-    IndexingJobsResourceWithRawResponse,
-    AsyncIndexingJobsResourceWithRawResponse,
-    IndexingJobsResourceWithStreamingResponse,
-    AsyncIndexingJobsResourceWithStreamingResponse,
+from .inference import (
+    InferenceResource,
+    AsyncInferenceResource,
+    InferenceResourceWithRawResponse,
+    AsyncInferenceResourceWithRawResponse,
+    InferenceResourceWithStreamingResponse,
+    AsyncInferenceResourceWithStreamingResponse,
 )
 from .knowledge_bases import (
     KnowledgeBasesResource,
@@ -80,6 +48,14 @@ from .knowledge_bases import (
     KnowledgeBasesResourceWithStreamingResponse,
     AsyncKnowledgeBasesResourceWithStreamingResponse,
 )
+from .model_providers import (
+    ModelProvidersResource,
+    AsyncModelProvidersResource,
+    ModelProvidersResourceWithRawResponse,
+    AsyncModelProvidersResourceWithRawResponse,
+    ModelProvidersResourceWithStreamingResponse,
+    AsyncModelProvidersResourceWithStreamingResponse,
+)
 
 __all__ = [
     "AgentsResource",
@@ -88,54 +64,36 @@ __all__ = [
     "AsyncAgentsResourceWithRawResponse",
     "AgentsResourceWithStreamingResponse",
     "AsyncAgentsResourceWithStreamingResponse",
-    "ProvidersResource",
-    "AsyncProvidersResource",
-    "ProvidersResourceWithRawResponse",
-    "AsyncProvidersResourceWithRawResponse",
-    "ProvidersResourceWithStreamingResponse",
-    "AsyncProvidersResourceWithStreamingResponse",
-    "AuthResource",
-    "AsyncAuthResource",
-    "AuthResourceWithRawResponse",
-    "AsyncAuthResourceWithRawResponse",
-    "AuthResourceWithStreamingResponse",
-    "AsyncAuthResourceWithStreamingResponse",
+    "ModelProvidersResource",
+    "AsyncModelProvidersResource",
+    "ModelProvidersResourceWithRawResponse",
+    "AsyncModelProvidersResourceWithRawResponse",
+    "ModelProvidersResourceWithStreamingResponse",
+    "AsyncModelProvidersResourceWithStreamingResponse",
     "RegionsResource",
     "AsyncRegionsResource",
     "RegionsResourceWithRawResponse",
     "AsyncRegionsResourceWithRawResponse",
     "RegionsResourceWithStreamingResponse",
     "AsyncRegionsResourceWithStreamingResponse",
-    "IndexingJobsResource",
-    "AsyncIndexingJobsResource",
-    "IndexingJobsResourceWithRawResponse",
-    "AsyncIndexingJobsResourceWithRawResponse",
-    "IndexingJobsResourceWithStreamingResponse",
-    "AsyncIndexingJobsResourceWithStreamingResponse",
     "KnowledgeBasesResource",
     "AsyncKnowledgeBasesResource",
     "KnowledgeBasesResourceWithRawResponse",
     "AsyncKnowledgeBasesResourceWithRawResponse",
     "KnowledgeBasesResourceWithStreamingResponse",
     "AsyncKnowledgeBasesResourceWithStreamingResponse",
-    "APIKeysResource",
-    "AsyncAPIKeysResource",
-    "APIKeysResourceWithRawResponse",
-    "AsyncAPIKeysResourceWithRawResponse",
-    "APIKeysResourceWithStreamingResponse",
-    "AsyncAPIKeysResourceWithStreamingResponse",
     "ChatResource",
     "AsyncChatResource",
     "ChatResourceWithRawResponse",
     "AsyncChatResourceWithRawResponse",
     "ChatResourceWithStreamingResponse",
     "AsyncChatResourceWithStreamingResponse",
-    "EmbeddingsResource",
-    "AsyncEmbeddingsResource",
-    "EmbeddingsResourceWithRawResponse",
-    "AsyncEmbeddingsResourceWithRawResponse",
-    "EmbeddingsResourceWithStreamingResponse",
-    "AsyncEmbeddingsResourceWithStreamingResponse",
+    "InferenceResource",
+    "AsyncInferenceResource",
+    "InferenceResourceWithRawResponse",
+    "AsyncInferenceResourceWithRawResponse",
+    "InferenceResourceWithStreamingResponse",
+    "AsyncInferenceResourceWithStreamingResponse",
     "ModelsResource",
     "AsyncModelsResource",
     "ModelsResourceWithRawResponse",

do_gradientai/resources/agents/__init__.py

@@ -0,0 +1,145 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .agents import (
+    AgentsResource,
+    AsyncAgentsResource,
+    AgentsResourceWithRawResponse,
+    AsyncAgentsResourceWithRawResponse,
+    AgentsResourceWithStreamingResponse,
+    AsyncAgentsResourceWithStreamingResponse,
+)
+from .routes import (
+    RoutesResource,
+    AsyncRoutesResource,
+    RoutesResourceWithRawResponse,
+    AsyncRoutesResourceWithRawResponse,
+    RoutesResourceWithStreamingResponse,
+    AsyncRoutesResourceWithStreamingResponse,
+)
+from .api_keys import (
+    APIKeysResource,
+    AsyncAPIKeysResource,
+    APIKeysResourceWithRawResponse,
+    AsyncAPIKeysResourceWithRawResponse,
+    APIKeysResourceWithStreamingResponse,
+    AsyncAPIKeysResourceWithStreamingResponse,
+)
+from .versions import (
+    VersionsResource,
+    AsyncVersionsResource,
+    VersionsResourceWithRawResponse,
+    AsyncVersionsResourceWithRawResponse,
+    VersionsResourceWithStreamingResponse,
+    AsyncVersionsResourceWithStreamingResponse,
+)
+from .functions import (
+    FunctionsResource,
+    AsyncFunctionsResource,
+    FunctionsResourceWithRawResponse,
+    AsyncFunctionsResourceWithRawResponse,
+    FunctionsResourceWithStreamingResponse,
+    AsyncFunctionsResourceWithStreamingResponse,
+)
+from .evaluation_runs import (
+    EvaluationRunsResource,
+    AsyncEvaluationRunsResource,
+    EvaluationRunsResourceWithRawResponse,
+    AsyncEvaluationRunsResourceWithRawResponse,
+    EvaluationRunsResourceWithStreamingResponse,
+    AsyncEvaluationRunsResourceWithStreamingResponse,
+)
+from .knowledge_bases import (
+    KnowledgeBasesResource,
+    AsyncKnowledgeBasesResource,
+    KnowledgeBasesResourceWithRawResponse,
+    AsyncKnowledgeBasesResourceWithRawResponse,
+    KnowledgeBasesResourceWithStreamingResponse,
+    AsyncKnowledgeBasesResourceWithStreamingResponse,
+)
+from .evaluation_metrics import (
+    EvaluationMetricsResource,
+    AsyncEvaluationMetricsResource,
+    EvaluationMetricsResourceWithRawResponse,
+    AsyncEvaluationMetricsResourceWithRawResponse,
+    EvaluationMetricsResourceWithStreamingResponse,
+    AsyncEvaluationMetricsResourceWithStreamingResponse,
+)
+from .evaluation_datasets import (
+    EvaluationDatasetsResource,
+    AsyncEvaluationDatasetsResource,
+    EvaluationDatasetsResourceWithRawResponse,
+    AsyncEvaluationDatasetsResourceWithRawResponse,
+    EvaluationDatasetsResourceWithStreamingResponse,
+    AsyncEvaluationDatasetsResourceWithStreamingResponse,
+)
+from .evaluation_test_cases import (
+    EvaluationTestCasesResource,
+    AsyncEvaluationTestCasesResource,
+    EvaluationTestCasesResourceWithRawResponse,
+    AsyncEvaluationTestCasesResourceWithRawResponse,
+    EvaluationTestCasesResourceWithStreamingResponse,
+    AsyncEvaluationTestCasesResourceWithStreamingResponse,
+)
+
+__all__ = [
+    "APIKeysResource",
+    "AsyncAPIKeysResource",
+    "APIKeysResourceWithRawResponse",
+    "AsyncAPIKeysResourceWithRawResponse",
+    "APIKeysResourceWithStreamingResponse",
+    "AsyncAPIKeysResourceWithStreamingResponse",
+    "EvaluationMetricsResource",
+    "AsyncEvaluationMetricsResource",
+    "EvaluationMetricsResourceWithRawResponse",
+    "AsyncEvaluationMetricsResourceWithRawResponse",
+    "EvaluationMetricsResourceWithStreamingResponse",
+    "AsyncEvaluationMetricsResourceWithStreamingResponse",
+    "EvaluationRunsResource",
+    "AsyncEvaluationRunsResource",
+    "EvaluationRunsResourceWithRawResponse",
+    "AsyncEvaluationRunsResourceWithRawResponse",
+    "EvaluationRunsResourceWithStreamingResponse",
+    "AsyncEvaluationRunsResourceWithStreamingResponse",
+    "EvaluationTestCasesResource",
+    "AsyncEvaluationTestCasesResource",
+    "EvaluationTestCasesResourceWithRawResponse",
+    "AsyncEvaluationTestCasesResourceWithRawResponse",
+    "EvaluationTestCasesResourceWithStreamingResponse",
+    "AsyncEvaluationTestCasesResourceWithStreamingResponse",
+    "EvaluationDatasetsResource",
+    "AsyncEvaluationDatasetsResource",
+    "EvaluationDatasetsResourceWithRawResponse",
+    "AsyncEvaluationDatasetsResourceWithRawResponse",
+    "EvaluationDatasetsResourceWithStreamingResponse",
+    "AsyncEvaluationDatasetsResourceWithStreamingResponse",
+    "FunctionsResource",
+    "AsyncFunctionsResource",
+    "FunctionsResourceWithRawResponse",
+    "AsyncFunctionsResourceWithRawResponse",
+    "FunctionsResourceWithStreamingResponse",
+    "AsyncFunctionsResourceWithStreamingResponse",
+    "VersionsResource",
+    "AsyncVersionsResource",
+    "VersionsResourceWithRawResponse",
+    "AsyncVersionsResourceWithRawResponse",
+    "VersionsResourceWithStreamingResponse",
+    "AsyncVersionsResourceWithStreamingResponse",
+    "KnowledgeBasesResource",
+    "AsyncKnowledgeBasesResource",
+    "KnowledgeBasesResourceWithRawResponse",
+    "AsyncKnowledgeBasesResourceWithRawResponse",
+    "KnowledgeBasesResourceWithStreamingResponse",
+    "AsyncKnowledgeBasesResourceWithStreamingResponse",
+    "RoutesResource",
+    "AsyncRoutesResource",
+    "RoutesResourceWithRawResponse",
+    "AsyncRoutesResourceWithRawResponse",
+    "RoutesResourceWithStreamingResponse",
+    "AsyncRoutesResourceWithStreamingResponse",
+    "AgentsResource",
+    "AsyncAgentsResource",
+    "AgentsResourceWithRawResponse",
+    "AsyncAgentsResourceWithRawResponse",
+    "AgentsResourceWithStreamingResponse",
+    "AsyncAgentsResourceWithStreamingResponse",
+]