diff --git a/.release-please-manifest.json b/.release-please-manifest.json index e8285b71..4f9005ea 100644 --- a/.release-please-manifest.json +++ b/.release-please-manifest.json @@ -1,3 +1,3 @@ { - ".": "0.1.0-alpha.5" + ".": "0.1.0-alpha.6" } \ No newline at end of file diff --git a/.stats.yml b/.stats.yml index 291ef184..a1e73eb0 100644 --- a/.stats.yml +++ b/.stats.yml @@ -1,4 +1,4 @@ configured_endpoints: 77 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradientai-e8b3cbc80e18e4f7f277010349f25e1319156704f359911dc464cc21a0d077a6.yml openapi_spec_hash: c773d792724f5647ae25a5ae4ccec208 -config_hash: ecf128ea21a8fead9dabb9609c4dbce8 +config_hash: 9c2519464cf5de240e34bd89b9f65706 diff --git a/CHANGELOG.md b/CHANGELOG.md index 18fcce2a..d9b29735 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,5 +1,13 @@ # Changelog +## 0.1.0-alpha.6 (2025-06-27) + +Full Changelog: [v0.1.0-alpha.5...v0.1.0-alpha.6](https://github.com/digitalocean/gradientai-python/compare/v0.1.0-alpha.5...v0.1.0-alpha.6) + +### Features + +* **api:** manual updates ([04eb1be](https://github.com/digitalocean/gradientai-python/commit/04eb1be35de7db04e1f0d4e1da8719b54a353bb5)) + ## 0.1.0-alpha.5 (2025-06-27) Full Changelog: [v0.1.0-alpha.4...v0.1.0-alpha.5](https://github.com/digitalocean/gradientai-python/compare/v0.1.0-alpha.4...v0.1.0-alpha.5) diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index 4f59c83a..086907ef 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -36,7 +36,7 @@ $ pip install -r requirements-dev.lock Most of the SDK is generated code. Modifications to code will be persisted between generations, but may result in merge conflicts between manual patches and changes from the generator. The generator will never -modify the contents of the `src/do_gradientai/lib/` and `examples/` directories. +modify the contents of the `src/gradientai/lib/` and `examples/` directories. ## Adding and running examples diff --git a/README.md b/README.md index 7e25c206..24b0975b 100644 --- a/README.md +++ b/README.md @@ -25,7 +25,7 @@ The full API of this library can be found in [api.md](api.md). ```python import os -from do_gradientai import GradientAI +from gradientai import GradientAI client = GradientAI( api_key=os.environ.get("GRADIENTAI_API_KEY"), # This is the default and can be omitted @@ -55,7 +55,7 @@ Simply import `AsyncGradientAI` instead of `GradientAI` and use `await` with eac ```python import os import asyncio -from do_gradientai import AsyncGradientAI +from gradientai import AsyncGradientAI client = AsyncGradientAI( api_key=os.environ.get("GRADIENTAI_API_KEY"), # This is the default and can be omitted @@ -96,8 +96,8 @@ Then you can enable it by instantiating the client with `http_client=DefaultAioH ```python import os import asyncio -from do_gradientai import DefaultAioHttpClient -from do_gradientai import AsyncGradientAI +from gradientai import DefaultAioHttpClient +from gradientai import AsyncGradientAI async def main() -> None: @@ -134,7 +134,7 @@ Typed requests and responses provide autocomplete and documentation within your Nested parameters are dictionaries, typed using `TypedDict`, for example: ```python -from do_gradientai import GradientAI +from gradientai import GradientAI client = GradientAI() @@ -153,16 +153,16 @@ print(completion.stream_options) ## Handling errors -When the library is unable to connect to the API (for example, due to network connection problems or a timeout), a subclass of `do_gradientai.APIConnectionError` is raised. +When the library is unable to connect to the API (for example, due to network connection problems or a timeout), a subclass of `gradientai.APIConnectionError` is raised. When the API returns a non-success status code (that is, 4xx or 5xx -response), a subclass of `do_gradientai.APIStatusError` is raised, containing `status_code` and `response` properties. +response), a subclass of `gradientai.APIStatusError` is raised, containing `status_code` and `response` properties. -All errors inherit from `do_gradientai.APIError`. +All errors inherit from `gradientai.APIError`. ```python -import do_gradientai -from do_gradientai import GradientAI +import gradientai +from gradientai import GradientAI client = GradientAI() @@ -170,12 +170,12 @@ try: client.agents.versions.list( uuid="REPLACE_ME", ) -except do_gradientai.APIConnectionError as e: +except gradientai.APIConnectionError as e: print("The server could not be reached") print(e.__cause__) # an underlying Exception, likely raised within httpx. -except do_gradientai.RateLimitError as e: +except gradientai.RateLimitError as e: print("A 429 status code was received; we should back off a bit.") -except do_gradientai.APIStatusError as e: +except gradientai.APIStatusError as e: print("Another non-200-range status code was received") print(e.status_code) print(e.response) @@ -203,7 +203,7 @@ Connection errors (for example, due to a network connectivity problem), 408 Requ You can use the `max_retries` option to configure or disable retry settings: ```python -from do_gradientai import GradientAI +from gradientai import GradientAI # Configure the default for all requests: client = GradientAI( @@ -223,7 +223,7 @@ By default requests time out after 1 minute. You can configure this with a `time which accepts a float or an [`httpx.Timeout`](https://www.python-httpx.org/advanced/timeouts/#fine-tuning-the-configuration) object: ```python -from do_gradientai import GradientAI +from gradientai import GradientAI # Configure the default for all requests: client = GradientAI( @@ -277,7 +277,7 @@ if response.my_field is None: The "raw" Response object can be accessed by prefixing `.with_raw_response.` to any HTTP method call, e.g., ```py -from do_gradientai import GradientAI +from gradientai import GradientAI client = GradientAI() response = client.agents.versions.with_raw_response.list( @@ -289,9 +289,9 @@ version = response.parse() # get the object that `agents.versions.list()` would print(version.agent_versions) ``` -These methods return an [`APIResponse`](https://github.com/digitalocean/gradientai-python/tree/main/src/do_gradientai/_response.py) object. +These methods return an [`APIResponse`](https://github.com/digitalocean/gradientai-python/tree/main/src/gradientai/_response.py) object. -The async client returns an [`AsyncAPIResponse`](https://github.com/digitalocean/gradientai-python/tree/main/src/do_gradientai/_response.py) with the same structure, the only difference being `await`able methods for reading the response content. +The async client returns an [`AsyncAPIResponse`](https://github.com/digitalocean/gradientai-python/tree/main/src/gradientai/_response.py) with the same structure, the only difference being `await`able methods for reading the response content. #### `.with_streaming_response` @@ -355,7 +355,7 @@ You can directly override the [httpx client](https://www.python-httpx.org/api/#c ```python import httpx -from do_gradientai import GradientAI, DefaultHttpxClient +from gradientai import GradientAI, DefaultHttpxClient client = GradientAI( # Or use the `GRADIENT_AI_BASE_URL` env var @@ -378,7 +378,7 @@ client.with_options(http_client=DefaultHttpxClient(...)) By default the library closes underlying HTTP connections whenever the client is [garbage collected](https://docs.python.org/3/reference/datamodel.html#object.__del__). You can manually close the client using the `.close()` method if desired, or with a context manager that closes when exiting. ```py -from do_gradientai import GradientAI +from gradientai import GradientAI with GradientAI() as client: # make requests here @@ -406,8 +406,8 @@ If you've upgraded to the latest version but aren't seeing any new features you You can determine the version that is being used at runtime with: ```py -import do_gradientai -print(do_gradientai.__version__) +import gradientai +print(gradientai.__version__) ``` ## Requirements diff --git a/api.md b/api.md index 78a81061..9a2dd757 100644 --- a/api.md +++ b/api.md @@ -1,7 +1,7 @@ # Shared Types ```python -from do_gradientai.types import APILinks, APIMeta +from gradientai.types import APILinks, APIMeta ``` # Agents @@ -9,7 +9,7 @@ from do_gradientai.types import APILinks, APIMeta Types: ```python -from do_gradientai.types import ( +from gradientai.types import ( APIAgent, APIAgentAPIKeyInfo, APIAgentModel, @@ -29,19 +29,19 @@ from do_gradientai.types import ( Methods: -- client.agents.create(\*\*params) -> AgentCreateResponse -- client.agents.retrieve(uuid) -> AgentRetrieveResponse -- client.agents.update(path_uuid, \*\*params) -> AgentUpdateResponse -- client.agents.list(\*\*params) -> AgentListResponse -- client.agents.delete(uuid) -> AgentDeleteResponse -- client.agents.update_status(path_uuid, \*\*params) -> AgentUpdateStatusResponse +- client.agents.create(\*\*params) -> AgentCreateResponse +- client.agents.retrieve(uuid) -> AgentRetrieveResponse +- client.agents.update(path_uuid, \*\*params) -> AgentUpdateResponse +- client.agents.list(\*\*params) -> AgentListResponse +- client.agents.delete(uuid) -> AgentDeleteResponse +- client.agents.update_status(path_uuid, \*\*params) -> AgentUpdateStatusResponse ## APIKeys Types: ```python -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( APIKeyCreateResponse, APIKeyUpdateResponse, APIKeyListResponse, @@ -52,30 +52,30 @@ from do_gradientai.types.agents import ( Methods: -- client.agents.api_keys.create(path_agent_uuid, \*\*params) -> APIKeyCreateResponse -- client.agents.api_keys.update(path_api_key_uuid, \*, path_agent_uuid, \*\*params) -> APIKeyUpdateResponse -- client.agents.api_keys.list(agent_uuid, \*\*params) -> APIKeyListResponse -- client.agents.api_keys.delete(api_key_uuid, \*, agent_uuid) -> APIKeyDeleteResponse -- client.agents.api_keys.regenerate(api_key_uuid, \*, agent_uuid) -> APIKeyRegenerateResponse +- client.agents.api_keys.create(path_agent_uuid, \*\*params) -> APIKeyCreateResponse +- client.agents.api_keys.update(path_api_key_uuid, \*, path_agent_uuid, \*\*params) -> APIKeyUpdateResponse +- client.agents.api_keys.list(agent_uuid, \*\*params) -> APIKeyListResponse +- client.agents.api_keys.delete(api_key_uuid, \*, agent_uuid) -> APIKeyDeleteResponse +- client.agents.api_keys.regenerate(api_key_uuid, \*, agent_uuid) -> APIKeyRegenerateResponse ## EvaluationMetrics Types: ```python -from do_gradientai.types.agents import EvaluationMetricListResponse +from gradientai.types.agents import EvaluationMetricListResponse ``` Methods: -- client.agents.evaluation_metrics.list() -> EvaluationMetricListResponse +- client.agents.evaluation_metrics.list() -> EvaluationMetricListResponse ### Workspaces Types: ```python -from do_gradientai.types.agents.evaluation_metrics import ( +from gradientai.types.agents.evaluation_metrics import ( WorkspaceCreateResponse, WorkspaceRetrieveResponse, WorkspaceUpdateResponse, @@ -87,19 +87,19 @@ from do_gradientai.types.agents.evaluation_metrics import ( Methods: -- client.agents.evaluation_metrics.workspaces.create(\*\*params) -> WorkspaceCreateResponse -- client.agents.evaluation_metrics.workspaces.retrieve(workspace_uuid) -> WorkspaceRetrieveResponse -- client.agents.evaluation_metrics.workspaces.update(path_workspace_uuid, \*\*params) -> WorkspaceUpdateResponse -- client.agents.evaluation_metrics.workspaces.list() -> WorkspaceListResponse -- client.agents.evaluation_metrics.workspaces.delete(workspace_uuid) -> WorkspaceDeleteResponse -- client.agents.evaluation_metrics.workspaces.list_evaluation_test_cases(workspace_uuid) -> WorkspaceListEvaluationTestCasesResponse +- client.agents.evaluation_metrics.workspaces.create(\*\*params) -> WorkspaceCreateResponse +- client.agents.evaluation_metrics.workspaces.retrieve(workspace_uuid) -> WorkspaceRetrieveResponse +- client.agents.evaluation_metrics.workspaces.update(path_workspace_uuid, \*\*params) -> WorkspaceUpdateResponse +- client.agents.evaluation_metrics.workspaces.list() -> WorkspaceListResponse +- client.agents.evaluation_metrics.workspaces.delete(workspace_uuid) -> WorkspaceDeleteResponse +- client.agents.evaluation_metrics.workspaces.list_evaluation_test_cases(workspace_uuid) -> WorkspaceListEvaluationTestCasesResponse #### Agents Types: ```python -from do_gradientai.types.agents.evaluation_metrics.workspaces import ( +from gradientai.types.agents.evaluation_metrics.workspaces import ( AgentListResponse, AgentMoveResponse, ) @@ -107,15 +107,15 @@ from do_gradientai.types.agents.evaluation_metrics.workspaces import ( Methods: -- client.agents.evaluation_metrics.workspaces.agents.list(workspace_uuid, \*\*params) -> AgentListResponse -- client.agents.evaluation_metrics.workspaces.agents.move(path_workspace_uuid, \*\*params) -> AgentMoveResponse +- client.agents.evaluation_metrics.workspaces.agents.list(workspace_uuid, \*\*params) -> AgentListResponse +- client.agents.evaluation_metrics.workspaces.agents.move(path_workspace_uuid, \*\*params) -> AgentMoveResponse ## EvaluationRuns Types: ```python -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( APIEvaluationMetric, APIEvaluationMetricResult, APIEvaluationPrompt, @@ -129,17 +129,17 @@ from do_gradientai.types.agents import ( Methods: -- client.agents.evaluation_runs.create(\*\*params) -> EvaluationRunCreateResponse -- client.agents.evaluation_runs.retrieve(evaluation_run_uuid) -> EvaluationRunRetrieveResponse -- client.agents.evaluation_runs.list_results(evaluation_run_uuid) -> EvaluationRunListResultsResponse -- client.agents.evaluation_runs.retrieve_results(prompt_id, \*, evaluation_run_uuid) -> EvaluationRunRetrieveResultsResponse +- client.agents.evaluation_runs.create(\*\*params) -> EvaluationRunCreateResponse +- client.agents.evaluation_runs.retrieve(evaluation_run_uuid) -> EvaluationRunRetrieveResponse +- client.agents.evaluation_runs.list_results(evaluation_run_uuid) -> EvaluationRunListResultsResponse +- client.agents.evaluation_runs.retrieve_results(prompt_id, \*, evaluation_run_uuid) -> EvaluationRunRetrieveResultsResponse ## EvaluationTestCases Types: ```python -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( APIEvaluationTestCase, APIStarMetric, EvaluationTestCaseCreateResponse, @@ -152,18 +152,18 @@ from do_gradientai.types.agents import ( Methods: -- client.agents.evaluation_test_cases.create(\*\*params) -> EvaluationTestCaseCreateResponse -- client.agents.evaluation_test_cases.retrieve(test_case_uuid, \*\*params) -> EvaluationTestCaseRetrieveResponse -- client.agents.evaluation_test_cases.update(path_test_case_uuid, \*\*params) -> EvaluationTestCaseUpdateResponse -- client.agents.evaluation_test_cases.list() -> EvaluationTestCaseListResponse -- client.agents.evaluation_test_cases.list_evaluation_runs(evaluation_test_case_uuid, \*\*params) -> EvaluationTestCaseListEvaluationRunsResponse +- client.agents.evaluation_test_cases.create(\*\*params) -> EvaluationTestCaseCreateResponse +- client.agents.evaluation_test_cases.retrieve(test_case_uuid, \*\*params) -> EvaluationTestCaseRetrieveResponse +- client.agents.evaluation_test_cases.update(path_test_case_uuid, \*\*params) -> EvaluationTestCaseUpdateResponse +- client.agents.evaluation_test_cases.list() -> EvaluationTestCaseListResponse +- client.agents.evaluation_test_cases.list_evaluation_runs(evaluation_test_case_uuid, \*\*params) -> EvaluationTestCaseListEvaluationRunsResponse ## EvaluationDatasets Types: ```python -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( EvaluationDatasetCreateResponse, EvaluationDatasetCreateFileUploadPresignedURLsResponse, ) @@ -171,15 +171,15 @@ from do_gradientai.types.agents import ( Methods: -- client.agents.evaluation_datasets.create(\*\*params) -> EvaluationDatasetCreateResponse -- client.agents.evaluation_datasets.create_file_upload_presigned_urls(\*\*params) -> EvaluationDatasetCreateFileUploadPresignedURLsResponse +- client.agents.evaluation_datasets.create(\*\*params) -> EvaluationDatasetCreateResponse +- client.agents.evaluation_datasets.create_file_upload_presigned_urls(\*\*params) -> EvaluationDatasetCreateFileUploadPresignedURLsResponse ## Functions Types: ```python -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( FunctionCreateResponse, FunctionUpdateResponse, FunctionDeleteResponse, @@ -188,43 +188,43 @@ from do_gradientai.types.agents import ( Methods: -- client.agents.functions.create(path_agent_uuid, \*\*params) -> FunctionCreateResponse -- client.agents.functions.update(path_function_uuid, \*, path_agent_uuid, \*\*params) -> FunctionUpdateResponse -- client.agents.functions.delete(function_uuid, \*, agent_uuid) -> FunctionDeleteResponse +- client.agents.functions.create(path_agent_uuid, \*\*params) -> FunctionCreateResponse +- client.agents.functions.update(path_function_uuid, \*, path_agent_uuid, \*\*params) -> FunctionUpdateResponse +- client.agents.functions.delete(function_uuid, \*, agent_uuid) -> FunctionDeleteResponse ## Versions Types: ```python -from do_gradientai.types.agents import VersionUpdateResponse, VersionListResponse +from gradientai.types.agents import VersionUpdateResponse, VersionListResponse ``` Methods: -- client.agents.versions.update(path_uuid, \*\*params) -> VersionUpdateResponse -- client.agents.versions.list(uuid, \*\*params) -> VersionListResponse +- client.agents.versions.update(path_uuid, \*\*params) -> VersionUpdateResponse +- client.agents.versions.list(uuid, \*\*params) -> VersionListResponse ## KnowledgeBases Types: ```python -from do_gradientai.types.agents import APILinkKnowledgeBaseOutput, KnowledgeBaseDetachResponse +from gradientai.types.agents import APILinkKnowledgeBaseOutput, KnowledgeBaseDetachResponse ``` Methods: -- client.agents.knowledge_bases.attach(agent_uuid) -> APILinkKnowledgeBaseOutput -- client.agents.knowledge_bases.attach_single(knowledge_base_uuid, \*, agent_uuid) -> APILinkKnowledgeBaseOutput -- client.agents.knowledge_bases.detach(knowledge_base_uuid, \*, agent_uuid) -> KnowledgeBaseDetachResponse +- client.agents.knowledge_bases.attach(agent_uuid) -> APILinkKnowledgeBaseOutput +- client.agents.knowledge_bases.attach_single(knowledge_base_uuid, \*, agent_uuid) -> APILinkKnowledgeBaseOutput +- client.agents.knowledge_bases.detach(knowledge_base_uuid, \*, agent_uuid) -> KnowledgeBaseDetachResponse ## Routes Types: ```python -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( RouteUpdateResponse, RouteDeleteResponse, RouteAddResponse, @@ -234,10 +234,10 @@ from do_gradientai.types.agents import ( Methods: -- client.agents.routes.update(path_child_agent_uuid, \*, path_parent_agent_uuid, \*\*params) -> RouteUpdateResponse -- client.agents.routes.delete(child_agent_uuid, \*, parent_agent_uuid) -> RouteDeleteResponse -- client.agents.routes.add(path_child_agent_uuid, \*, path_parent_agent_uuid, \*\*params) -> RouteAddResponse -- client.agents.routes.view(uuid) -> RouteViewResponse +- client.agents.routes.update(path_child_agent_uuid, \*, path_parent_agent_uuid, \*\*params) -> RouteUpdateResponse +- client.agents.routes.delete(child_agent_uuid, \*, parent_agent_uuid) -> RouteDeleteResponse +- client.agents.routes.add(path_child_agent_uuid, \*, path_parent_agent_uuid, \*\*params) -> RouteAddResponse +- client.agents.routes.view(uuid) -> RouteViewResponse # ModelProviders @@ -248,7 +248,7 @@ Methods: Types: ```python -from do_gradientai.types.model_providers.anthropic import ( +from gradientai.types.model_providers.anthropic import ( KeyCreateResponse, KeyRetrieveResponse, KeyUpdateResponse, @@ -260,12 +260,12 @@ from do_gradientai.types.model_providers.anthropic import ( Methods: -- client.model_providers.anthropic.keys.create(\*\*params) -> KeyCreateResponse -- client.model_providers.anthropic.keys.retrieve(api_key_uuid) -> KeyRetrieveResponse -- client.model_providers.anthropic.keys.update(path_api_key_uuid, \*\*params) -> KeyUpdateResponse -- client.model_providers.anthropic.keys.list(\*\*params) -> KeyListResponse -- client.model_providers.anthropic.keys.delete(api_key_uuid) -> KeyDeleteResponse -- client.model_providers.anthropic.keys.list_agents(uuid, \*\*params) -> KeyListAgentsResponse +- client.model_providers.anthropic.keys.create(\*\*params) -> KeyCreateResponse +- client.model_providers.anthropic.keys.retrieve(api_key_uuid) -> KeyRetrieveResponse +- client.model_providers.anthropic.keys.update(path_api_key_uuid, \*\*params) -> KeyUpdateResponse +- client.model_providers.anthropic.keys.list(\*\*params) -> KeyListResponse +- client.model_providers.anthropic.keys.delete(api_key_uuid) -> KeyDeleteResponse +- client.model_providers.anthropic.keys.list_agents(uuid, \*\*params) -> KeyListAgentsResponse ## OpenAI @@ -274,7 +274,7 @@ Methods: Types: ```python -from do_gradientai.types.model_providers.openai import ( +from gradientai.types.model_providers.openai import ( KeyCreateResponse, KeyRetrieveResponse, KeyUpdateResponse, @@ -286,31 +286,31 @@ from do_gradientai.types.model_providers.openai import ( Methods: -- client.model_providers.openai.keys.create(\*\*params) -> KeyCreateResponse -- client.model_providers.openai.keys.retrieve(api_key_uuid) -> KeyRetrieveResponse -- client.model_providers.openai.keys.update(path_api_key_uuid, \*\*params) -> KeyUpdateResponse -- client.model_providers.openai.keys.list(\*\*params) -> KeyListResponse -- client.model_providers.openai.keys.delete(api_key_uuid) -> KeyDeleteResponse -- client.model_providers.openai.keys.retrieve_agents(uuid, \*\*params) -> KeyRetrieveAgentsResponse +- client.model_providers.openai.keys.create(\*\*params) -> KeyCreateResponse +- client.model_providers.openai.keys.retrieve(api_key_uuid) -> KeyRetrieveResponse +- client.model_providers.openai.keys.update(path_api_key_uuid, \*\*params) -> KeyUpdateResponse +- client.model_providers.openai.keys.list(\*\*params) -> KeyListResponse +- client.model_providers.openai.keys.delete(api_key_uuid) -> KeyDeleteResponse +- client.model_providers.openai.keys.retrieve_agents(uuid, \*\*params) -> KeyRetrieveAgentsResponse # Regions Types: ```python -from do_gradientai.types import RegionListResponse +from gradientai.types import RegionListResponse ``` Methods: -- client.regions.list(\*\*params) -> RegionListResponse +- client.regions.list(\*\*params) -> RegionListResponse # KnowledgeBases Types: ```python -from do_gradientai.types import ( +from gradientai.types import ( APIKnowledgeBase, KnowledgeBaseCreateResponse, KnowledgeBaseRetrieveResponse, @@ -322,18 +322,18 @@ from do_gradientai.types import ( Methods: -- client.knowledge_bases.create(\*\*params) -> KnowledgeBaseCreateResponse -- client.knowledge_bases.retrieve(uuid) -> KnowledgeBaseRetrieveResponse -- client.knowledge_bases.update(path_uuid, \*\*params) -> KnowledgeBaseUpdateResponse -- client.knowledge_bases.list(\*\*params) -> KnowledgeBaseListResponse -- client.knowledge_bases.delete(uuid) -> KnowledgeBaseDeleteResponse +- client.knowledge_bases.create(\*\*params) -> KnowledgeBaseCreateResponse +- client.knowledge_bases.retrieve(uuid) -> KnowledgeBaseRetrieveResponse +- client.knowledge_bases.update(path_uuid, \*\*params) -> KnowledgeBaseUpdateResponse +- client.knowledge_bases.list(\*\*params) -> KnowledgeBaseListResponse +- client.knowledge_bases.delete(uuid) -> KnowledgeBaseDeleteResponse ## DataSources Types: ```python -from do_gradientai.types.knowledge_bases import ( +from gradientai.types.knowledge_bases import ( APIFileUploadDataSource, APIKnowledgeBaseDataSource, APISpacesDataSource, @@ -347,16 +347,16 @@ from do_gradientai.types.knowledge_bases import ( Methods: -- client.knowledge_bases.data_sources.create(path_knowledge_base_uuid, \*\*params) -> DataSourceCreateResponse -- client.knowledge_bases.data_sources.list(knowledge_base_uuid, \*\*params) -> DataSourceListResponse -- client.knowledge_bases.data_sources.delete(data_source_uuid, \*, knowledge_base_uuid) -> DataSourceDeleteResponse +- client.knowledge_bases.data_sources.create(path_knowledge_base_uuid, \*\*params) -> DataSourceCreateResponse +- client.knowledge_bases.data_sources.list(knowledge_base_uuid, \*\*params) -> DataSourceListResponse +- client.knowledge_bases.data_sources.delete(data_source_uuid, \*, knowledge_base_uuid) -> DataSourceDeleteResponse ## IndexingJobs Types: ```python -from do_gradientai.types.knowledge_bases import ( +from gradientai.types.knowledge_bases import ( APIIndexedDataSource, APIIndexingJob, IndexingJobCreateResponse, @@ -369,11 +369,11 @@ from do_gradientai.types.knowledge_bases import ( Methods: -- client.knowledge_bases.indexing_jobs.create(\*\*params) -> IndexingJobCreateResponse -- client.knowledge_bases.indexing_jobs.retrieve(uuid) -> IndexingJobRetrieveResponse -- client.knowledge_bases.indexing_jobs.list(\*\*params) -> IndexingJobListResponse -- client.knowledge_bases.indexing_jobs.retrieve_data_sources(indexing_job_uuid) -> IndexingJobRetrieveDataSourcesResponse -- client.knowledge_bases.indexing_jobs.update_cancel(path_uuid, \*\*params) -> IndexingJobUpdateCancelResponse +- client.knowledge_bases.indexing_jobs.create(\*\*params) -> IndexingJobCreateResponse +- client.knowledge_bases.indexing_jobs.retrieve(uuid) -> IndexingJobRetrieveResponse +- client.knowledge_bases.indexing_jobs.list(\*\*params) -> IndexingJobListResponse +- client.knowledge_bases.indexing_jobs.retrieve_data_sources(indexing_job_uuid) -> IndexingJobRetrieveDataSourcesResponse +- client.knowledge_bases.indexing_jobs.update_cancel(path_uuid, \*\*params) -> IndexingJobUpdateCancelResponse # Chat @@ -382,12 +382,12 @@ Methods: Types: ```python -from do_gradientai.types.chat import ChatCompletionTokenLogprob, CompletionCreateResponse +from gradientai.types.chat import ChatCompletionTokenLogprob, CompletionCreateResponse ``` Methods: -- client.chat.completions.create(\*\*params) -> CompletionCreateResponse +- client.chat.completions.create(\*\*params) -> CompletionCreateResponse # Inference @@ -396,7 +396,7 @@ Methods: Types: ```python -from do_gradientai.types.inference import ( +from gradientai.types.inference import ( APIModelAPIKeyInfo, APIKeyCreateResponse, APIKeyUpdateResponse, @@ -408,21 +408,21 @@ from do_gradientai.types.inference import ( Methods: -- client.inference.api_keys.create(\*\*params) -> APIKeyCreateResponse -- client.inference.api_keys.update(path_api_key_uuid, \*\*params) -> APIKeyUpdateResponse -- client.inference.api_keys.list(\*\*params) -> APIKeyListResponse -- client.inference.api_keys.delete(api_key_uuid) -> APIKeyDeleteResponse -- client.inference.api_keys.update_regenerate(api_key_uuid) -> APIKeyUpdateRegenerateResponse +- client.inference.api_keys.create(\*\*params) -> APIKeyCreateResponse +- client.inference.api_keys.update(path_api_key_uuid, \*\*params) -> APIKeyUpdateResponse +- client.inference.api_keys.list(\*\*params) -> APIKeyListResponse +- client.inference.api_keys.delete(api_key_uuid) -> APIKeyDeleteResponse +- client.inference.api_keys.update_regenerate(api_key_uuid) -> APIKeyUpdateRegenerateResponse # Models Types: ```python -from do_gradientai.types import APIAgreement, APIModel, APIModelVersion, Model, ModelListResponse +from gradientai.types import APIAgreement, APIModel, APIModelVersion, Model, ModelListResponse ``` Methods: -- client.models.retrieve(model) -> Model -- client.models.list() -> ModelListResponse +- client.models.retrieve(model) -> Model +- client.models.list() -> ModelListResponse diff --git a/mypy.ini b/mypy.ini index 82b0c891..748d8234 100644 --- a/mypy.ini +++ b/mypy.ini @@ -8,7 +8,7 @@ show_error_codes = True # # We also exclude our `tests` as mypy doesn't always infer # types correctly and Pyright will still catch any type errors. -exclude = ^(src/do_gradientai/_files\.py|_dev/.*\.py|tests/.*)$ +exclude = ^(src/gradientai/_files\.py|_dev/.*\.py|tests/.*)$ strict_equality = True implicit_reexport = True diff --git a/pyproject.toml b/pyproject.toml index 768cf340..0f04322b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python" -version = "0.1.0-alpha.5" +version = "0.1.0-alpha.6" description = "The official Python library for GradientAI" dynamic = ["readme"] license = "Apache-2.0" @@ -78,14 +78,14 @@ format = { chain = [ "check:ruff" = "ruff check ." "fix:ruff" = "ruff check --fix ." -"check:importable" = "python -c 'import do_gradientai'" +"check:importable" = "python -c 'import gradientai'" typecheck = { chain = [ "typecheck:pyright", "typecheck:mypy" ]} "typecheck:pyright" = "pyright" -"typecheck:verify-types" = "pyright --verifytypes do_gradientai --ignoreexternal" +"typecheck:verify-types" = "pyright --verifytypes gradientai --ignoreexternal" "typecheck:mypy" = "mypy ." [build-system] @@ -98,7 +98,7 @@ include = [ ] [tool.hatch.build.targets.wheel] -packages = ["src/do_gradientai"] +packages = ["src/gradientai"] [tool.hatch.build.targets.sdist] # Basically everything except hidden files/directories (such as .github, .devcontainers, .python-version, etc) @@ -201,7 +201,7 @@ length-sort = true length-sort-straight = true combine-as-imports = true extra-standard-library = ["typing_extensions"] -known-first-party = ["do_gradientai", "tests"] +known-first-party = ["gradientai", "tests"] [tool.ruff.lint.per-file-ignores] "bin/**.py" = ["T201", "T203"] diff --git a/release-please-config.json b/release-please-config.json index a320c1a8..2ff9a58c 100644 --- a/release-please-config.json +++ b/release-please-config.json @@ -61,6 +61,6 @@ ], "release-type": "python", "extra-files": [ - "src/do_gradientai/_version.py" + "src/gradientai/_version.py" ] } \ No newline at end of file diff --git a/scripts/lint b/scripts/lint index e46e909b..37b38f6f 100755 --- a/scripts/lint +++ b/scripts/lint @@ -8,4 +8,4 @@ echo "==> Running lints" rye run lint echo "==> Making sure it imports" -rye run python -c 'import do_gradientai' +rye run python -c 'import gradientai' diff --git a/src/do_gradientai/__init__.py b/src/gradientai/__init__.py similarity index 95% rename from src/do_gradientai/__init__.py rename to src/gradientai/__init__.py index 41b943b2..3316fe47 100644 --- a/src/do_gradientai/__init__.py +++ b/src/gradientai/__init__.py @@ -89,12 +89,12 @@ # Update the __module__ attribute for exported symbols so that # error messages point to this module instead of the module # it was originally defined in, e.g. -# do_gradientai._exceptions.NotFoundError -> do_gradientai.NotFoundError +# gradientai._exceptions.NotFoundError -> gradientai.NotFoundError __locals = locals() for __name in __all__: if not __name.startswith("__"): try: - __locals[__name].__module__ = "do_gradientai" + __locals[__name].__module__ = "gradientai" except (TypeError, AttributeError): # Some of our exported symbols are builtins which we can't set attributes for. pass diff --git a/src/do_gradientai/_base_client.py b/src/gradientai/_base_client.py similarity index 99% rename from src/do_gradientai/_base_client.py rename to src/gradientai/_base_client.py index 30108c9d..6dce600b 100644 --- a/src/do_gradientai/_base_client.py +++ b/src/gradientai/_base_client.py @@ -389,7 +389,7 @@ def __init__( if max_retries is None: # pyright: ignore[reportUnnecessaryComparison] raise TypeError( - "max_retries cannot be None. If you want to disable retries, pass `0`; if you want unlimited retries, pass `math.inf` or a very high number; if you want the default behavior, pass `do_gradientai.DEFAULT_MAX_RETRIES`" + "max_retries cannot be None. If you want to disable retries, pass `0`; if you want unlimited retries, pass `math.inf` or a very high number; if you want the default behavior, pass `gradientai.DEFAULT_MAX_RETRIES`" ) def _enforce_trailing_slash(self, url: URL) -> URL: diff --git a/src/do_gradientai/_client.py b/src/gradientai/_client.py similarity index 100% rename from src/do_gradientai/_client.py rename to src/gradientai/_client.py diff --git a/src/do_gradientai/_compat.py b/src/gradientai/_compat.py similarity index 100% rename from src/do_gradientai/_compat.py rename to src/gradientai/_compat.py diff --git a/src/do_gradientai/_constants.py b/src/gradientai/_constants.py similarity index 100% rename from src/do_gradientai/_constants.py rename to src/gradientai/_constants.py diff --git a/src/do_gradientai/_exceptions.py b/src/gradientai/_exceptions.py similarity index 100% rename from src/do_gradientai/_exceptions.py rename to src/gradientai/_exceptions.py diff --git a/src/do_gradientai/_files.py b/src/gradientai/_files.py similarity index 100% rename from src/do_gradientai/_files.py rename to src/gradientai/_files.py diff --git a/src/do_gradientai/_models.py b/src/gradientai/_models.py similarity index 100% rename from src/do_gradientai/_models.py rename to src/gradientai/_models.py diff --git a/src/do_gradientai/_qs.py b/src/gradientai/_qs.py similarity index 100% rename from src/do_gradientai/_qs.py rename to src/gradientai/_qs.py diff --git a/src/do_gradientai/_resource.py b/src/gradientai/_resource.py similarity index 100% rename from src/do_gradientai/_resource.py rename to src/gradientai/_resource.py diff --git a/src/do_gradientai/_response.py b/src/gradientai/_response.py similarity index 99% rename from src/do_gradientai/_response.py rename to src/gradientai/_response.py index 8ca43971..2037e4ca 100644 --- a/src/do_gradientai/_response.py +++ b/src/gradientai/_response.py @@ -218,7 +218,7 @@ def _parse(self, *, to: type[_T] | None = None) -> R | _T: and issubclass(origin, pydantic.BaseModel) ): raise TypeError( - "Pydantic models must subclass our base model type, e.g. `from do_gradientai import BaseModel`" + "Pydantic models must subclass our base model type, e.g. `from gradientai import BaseModel`" ) if ( @@ -285,7 +285,7 @@ def parse(self, *, to: type[_T] | None = None) -> R | _T: the `to` argument, e.g. ```py - from do_gradientai import BaseModel + from gradientai import BaseModel class MyModel(BaseModel): @@ -387,7 +387,7 @@ async def parse(self, *, to: type[_T] | None = None) -> R | _T: the `to` argument, e.g. ```py - from do_gradientai import BaseModel + from gradientai import BaseModel class MyModel(BaseModel): @@ -558,7 +558,7 @@ async def stream_to_file( 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 `do_gradientai._streaming` for reference", + "The `stream` argument was set to `True` but the `stream_cls` argument was not given. See `gradientai._streaming` for reference", ) diff --git a/src/do_gradientai/_streaming.py b/src/gradientai/_streaming.py similarity index 100% rename from src/do_gradientai/_streaming.py rename to src/gradientai/_streaming.py diff --git a/src/do_gradientai/_types.py b/src/gradientai/_types.py similarity index 99% rename from src/do_gradientai/_types.py rename to src/gradientai/_types.py index c356c700..1bac876d 100644 --- a/src/do_gradientai/_types.py +++ b/src/gradientai/_types.py @@ -81,7 +81,7 @@ # This unfortunately means that you will either have # to import this type and pass it explicitly: # -# from do_gradientai import NoneType +# from gradientai import NoneType # client.get('/foo', cast_to=NoneType) # # or build it yourself: diff --git a/src/do_gradientai/_utils/__init__.py b/src/gradientai/_utils/__init__.py similarity index 100% rename from src/do_gradientai/_utils/__init__.py rename to src/gradientai/_utils/__init__.py diff --git a/src/do_gradientai/_utils/_logs.py b/src/gradientai/_utils/_logs.py similarity index 75% rename from src/do_gradientai/_utils/_logs.py rename to src/gradientai/_utils/_logs.py index ac45b1a5..9047e5c8 100644 --- a/src/do_gradientai/_utils/_logs.py +++ b/src/gradientai/_utils/_logs.py @@ -1,12 +1,12 @@ import os import logging -logger: logging.Logger = logging.getLogger("do_gradientai") +logger: logging.Logger = logging.getLogger("gradientai") httpx_logger: logging.Logger = logging.getLogger("httpx") def _basic_config() -> None: - # 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" + # e.g. [2023-10-05 14:12:26 - 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", diff --git a/src/do_gradientai/_utils/_proxy.py b/src/gradientai/_utils/_proxy.py similarity index 100% rename from src/do_gradientai/_utils/_proxy.py rename to src/gradientai/_utils/_proxy.py diff --git a/src/do_gradientai/_utils/_reflection.py b/src/gradientai/_utils/_reflection.py similarity index 100% rename from src/do_gradientai/_utils/_reflection.py rename to src/gradientai/_utils/_reflection.py diff --git a/src/do_gradientai/_utils/_resources_proxy.py b/src/gradientai/_utils/_resources_proxy.py similarity index 50% rename from src/do_gradientai/_utils/_resources_proxy.py rename to src/gradientai/_utils/_resources_proxy.py index 03763c3b..b3bc4931 100644 --- a/src/do_gradientai/_utils/_resources_proxy.py +++ b/src/gradientai/_utils/_resources_proxy.py @@ -7,17 +7,17 @@ class ResourcesProxy(LazyProxy[Any]): - """A proxy for the `do_gradientai.resources` module. + """A proxy for the `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` + This is used so that we can lazily import `gradientai.resources` only when + needed *and* so that users can just import `gradientai` and reference `gradientai.resources` """ @override def __load__(self) -> Any: import importlib - mod = importlib.import_module("do_gradientai.resources") + mod = importlib.import_module("gradientai.resources") return mod diff --git a/src/do_gradientai/_utils/_streams.py b/src/gradientai/_utils/_streams.py similarity index 100% rename from src/do_gradientai/_utils/_streams.py rename to src/gradientai/_utils/_streams.py diff --git a/src/do_gradientai/_utils/_sync.py b/src/gradientai/_utils/_sync.py similarity index 100% rename from src/do_gradientai/_utils/_sync.py rename to src/gradientai/_utils/_sync.py diff --git a/src/do_gradientai/_utils/_transform.py b/src/gradientai/_utils/_transform.py similarity index 100% rename from src/do_gradientai/_utils/_transform.py rename to src/gradientai/_utils/_transform.py diff --git a/src/do_gradientai/_utils/_typing.py b/src/gradientai/_utils/_typing.py similarity index 100% rename from src/do_gradientai/_utils/_typing.py rename to src/gradientai/_utils/_typing.py diff --git a/src/do_gradientai/_utils/_utils.py b/src/gradientai/_utils/_utils.py similarity index 100% rename from src/do_gradientai/_utils/_utils.py rename to src/gradientai/_utils/_utils.py diff --git a/src/do_gradientai/_version.py b/src/gradientai/_version.py similarity index 50% rename from src/do_gradientai/_version.py rename to src/gradientai/_version.py index 12e8e17e..b8ef5fc0 100644 --- a/src/do_gradientai/_version.py +++ b/src/gradientai/_version.py @@ -1,4 +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 +__title__ = "gradientai" +__version__ = "0.1.0-alpha.6" # x-release-please-version diff --git a/src/do_gradientai/py.typed b/src/gradientai/py.typed similarity index 100% rename from src/do_gradientai/py.typed rename to src/gradientai/py.typed diff --git a/src/do_gradientai/resources/__init__.py b/src/gradientai/resources/__init__.py similarity index 100% rename from src/do_gradientai/resources/__init__.py rename to src/gradientai/resources/__init__.py diff --git a/src/do_gradientai/resources/agents/__init__.py b/src/gradientai/resources/agents/__init__.py similarity index 100% rename from src/do_gradientai/resources/agents/__init__.py rename to src/gradientai/resources/agents/__init__.py diff --git a/src/do_gradientai/resources/agents/agents.py b/src/gradientai/resources/agents/agents.py similarity index 100% rename from src/do_gradientai/resources/agents/agents.py rename to src/gradientai/resources/agents/agents.py diff --git a/src/do_gradientai/resources/agents/api_keys.py b/src/gradientai/resources/agents/api_keys.py similarity index 100% rename from src/do_gradientai/resources/agents/api_keys.py rename to src/gradientai/resources/agents/api_keys.py diff --git a/src/do_gradientai/resources/agents/evaluation_datasets.py b/src/gradientai/resources/agents/evaluation_datasets.py similarity index 100% rename from src/do_gradientai/resources/agents/evaluation_datasets.py rename to src/gradientai/resources/agents/evaluation_datasets.py diff --git a/src/do_gradientai/resources/agents/evaluation_metrics/__init__.py b/src/gradientai/resources/agents/evaluation_metrics/__init__.py similarity index 100% rename from src/do_gradientai/resources/agents/evaluation_metrics/__init__.py rename to src/gradientai/resources/agents/evaluation_metrics/__init__.py diff --git a/src/do_gradientai/resources/agents/evaluation_metrics/evaluation_metrics.py b/src/gradientai/resources/agents/evaluation_metrics/evaluation_metrics.py similarity index 100% rename from src/do_gradientai/resources/agents/evaluation_metrics/evaluation_metrics.py rename to src/gradientai/resources/agents/evaluation_metrics/evaluation_metrics.py diff --git a/src/do_gradientai/resources/agents/evaluation_metrics/workspaces/__init__.py b/src/gradientai/resources/agents/evaluation_metrics/workspaces/__init__.py similarity index 100% rename from src/do_gradientai/resources/agents/evaluation_metrics/workspaces/__init__.py rename to src/gradientai/resources/agents/evaluation_metrics/workspaces/__init__.py diff --git a/src/do_gradientai/resources/agents/evaluation_metrics/workspaces/agents.py b/src/gradientai/resources/agents/evaluation_metrics/workspaces/agents.py similarity index 100% rename from src/do_gradientai/resources/agents/evaluation_metrics/workspaces/agents.py rename to src/gradientai/resources/agents/evaluation_metrics/workspaces/agents.py diff --git a/src/do_gradientai/resources/agents/evaluation_metrics/workspaces/workspaces.py b/src/gradientai/resources/agents/evaluation_metrics/workspaces/workspaces.py similarity index 100% rename from src/do_gradientai/resources/agents/evaluation_metrics/workspaces/workspaces.py rename to src/gradientai/resources/agents/evaluation_metrics/workspaces/workspaces.py diff --git a/src/do_gradientai/resources/agents/evaluation_runs.py b/src/gradientai/resources/agents/evaluation_runs.py similarity index 100% rename from src/do_gradientai/resources/agents/evaluation_runs.py rename to src/gradientai/resources/agents/evaluation_runs.py diff --git a/src/do_gradientai/resources/agents/evaluation_test_cases.py b/src/gradientai/resources/agents/evaluation_test_cases.py similarity index 100% rename from src/do_gradientai/resources/agents/evaluation_test_cases.py rename to src/gradientai/resources/agents/evaluation_test_cases.py diff --git a/src/do_gradientai/resources/agents/functions.py b/src/gradientai/resources/agents/functions.py similarity index 100% rename from src/do_gradientai/resources/agents/functions.py rename to src/gradientai/resources/agents/functions.py diff --git a/src/do_gradientai/resources/agents/knowledge_bases.py b/src/gradientai/resources/agents/knowledge_bases.py similarity index 100% rename from src/do_gradientai/resources/agents/knowledge_bases.py rename to src/gradientai/resources/agents/knowledge_bases.py diff --git a/src/do_gradientai/resources/agents/routes.py b/src/gradientai/resources/agents/routes.py similarity index 100% rename from src/do_gradientai/resources/agents/routes.py rename to src/gradientai/resources/agents/routes.py diff --git a/src/do_gradientai/resources/agents/versions.py b/src/gradientai/resources/agents/versions.py similarity index 100% rename from src/do_gradientai/resources/agents/versions.py rename to src/gradientai/resources/agents/versions.py diff --git a/src/do_gradientai/resources/chat/__init__.py b/src/gradientai/resources/chat/__init__.py similarity index 100% rename from src/do_gradientai/resources/chat/__init__.py rename to src/gradientai/resources/chat/__init__.py diff --git a/src/do_gradientai/resources/chat/chat.py b/src/gradientai/resources/chat/chat.py similarity index 100% rename from src/do_gradientai/resources/chat/chat.py rename to src/gradientai/resources/chat/chat.py diff --git a/src/do_gradientai/resources/chat/completions.py b/src/gradientai/resources/chat/completions.py similarity index 100% rename from src/do_gradientai/resources/chat/completions.py rename to src/gradientai/resources/chat/completions.py diff --git a/src/do_gradientai/resources/inference/__init__.py b/src/gradientai/resources/inference/__init__.py similarity index 100% rename from src/do_gradientai/resources/inference/__init__.py rename to src/gradientai/resources/inference/__init__.py diff --git a/src/do_gradientai/resources/inference/api_keys.py b/src/gradientai/resources/inference/api_keys.py similarity index 100% rename from src/do_gradientai/resources/inference/api_keys.py rename to src/gradientai/resources/inference/api_keys.py diff --git a/src/do_gradientai/resources/inference/inference.py b/src/gradientai/resources/inference/inference.py similarity index 100% rename from src/do_gradientai/resources/inference/inference.py rename to src/gradientai/resources/inference/inference.py diff --git a/src/do_gradientai/resources/knowledge_bases/__init__.py b/src/gradientai/resources/knowledge_bases/__init__.py similarity index 100% rename from src/do_gradientai/resources/knowledge_bases/__init__.py rename to src/gradientai/resources/knowledge_bases/__init__.py diff --git a/src/do_gradientai/resources/knowledge_bases/data_sources.py b/src/gradientai/resources/knowledge_bases/data_sources.py similarity index 100% rename from src/do_gradientai/resources/knowledge_bases/data_sources.py rename to src/gradientai/resources/knowledge_bases/data_sources.py diff --git a/src/do_gradientai/resources/knowledge_bases/indexing_jobs.py b/src/gradientai/resources/knowledge_bases/indexing_jobs.py similarity index 100% rename from src/do_gradientai/resources/knowledge_bases/indexing_jobs.py rename to src/gradientai/resources/knowledge_bases/indexing_jobs.py diff --git a/src/do_gradientai/resources/knowledge_bases/knowledge_bases.py b/src/gradientai/resources/knowledge_bases/knowledge_bases.py similarity index 100% rename from src/do_gradientai/resources/knowledge_bases/knowledge_bases.py rename to src/gradientai/resources/knowledge_bases/knowledge_bases.py diff --git a/src/do_gradientai/resources/model_providers/__init__.py b/src/gradientai/resources/model_providers/__init__.py similarity index 100% rename from src/do_gradientai/resources/model_providers/__init__.py rename to src/gradientai/resources/model_providers/__init__.py diff --git a/src/do_gradientai/resources/model_providers/anthropic/__init__.py b/src/gradientai/resources/model_providers/anthropic/__init__.py similarity index 100% rename from src/do_gradientai/resources/model_providers/anthropic/__init__.py rename to src/gradientai/resources/model_providers/anthropic/__init__.py diff --git a/src/do_gradientai/resources/model_providers/anthropic/anthropic.py b/src/gradientai/resources/model_providers/anthropic/anthropic.py similarity index 100% rename from src/do_gradientai/resources/model_providers/anthropic/anthropic.py rename to src/gradientai/resources/model_providers/anthropic/anthropic.py diff --git a/src/do_gradientai/resources/model_providers/anthropic/keys.py b/src/gradientai/resources/model_providers/anthropic/keys.py similarity index 100% rename from src/do_gradientai/resources/model_providers/anthropic/keys.py rename to src/gradientai/resources/model_providers/anthropic/keys.py diff --git a/src/do_gradientai/resources/model_providers/model_providers.py b/src/gradientai/resources/model_providers/model_providers.py similarity index 100% rename from src/do_gradientai/resources/model_providers/model_providers.py rename to src/gradientai/resources/model_providers/model_providers.py diff --git a/src/do_gradientai/resources/model_providers/openai/__init__.py b/src/gradientai/resources/model_providers/openai/__init__.py similarity index 100% rename from src/do_gradientai/resources/model_providers/openai/__init__.py rename to src/gradientai/resources/model_providers/openai/__init__.py diff --git a/src/do_gradientai/resources/model_providers/openai/keys.py b/src/gradientai/resources/model_providers/openai/keys.py similarity index 100% rename from src/do_gradientai/resources/model_providers/openai/keys.py rename to src/gradientai/resources/model_providers/openai/keys.py diff --git a/src/do_gradientai/resources/model_providers/openai/openai.py b/src/gradientai/resources/model_providers/openai/openai.py similarity index 100% rename from src/do_gradientai/resources/model_providers/openai/openai.py rename to src/gradientai/resources/model_providers/openai/openai.py diff --git a/src/do_gradientai/resources/models.py b/src/gradientai/resources/models.py similarity index 100% rename from src/do_gradientai/resources/models.py rename to src/gradientai/resources/models.py diff --git a/src/do_gradientai/resources/regions.py b/src/gradientai/resources/regions.py similarity index 100% rename from src/do_gradientai/resources/regions.py rename to src/gradientai/resources/regions.py diff --git a/src/do_gradientai/types/__init__.py b/src/gradientai/types/__init__.py similarity index 100% rename from src/do_gradientai/types/__init__.py rename to src/gradientai/types/__init__.py diff --git a/src/do_gradientai/types/agent_create_params.py b/src/gradientai/types/agent_create_params.py similarity index 100% rename from src/do_gradientai/types/agent_create_params.py rename to src/gradientai/types/agent_create_params.py diff --git a/src/do_gradientai/types/agent_create_response.py b/src/gradientai/types/agent_create_response.py similarity index 100% rename from src/do_gradientai/types/agent_create_response.py rename to src/gradientai/types/agent_create_response.py diff --git a/src/do_gradientai/types/agent_delete_response.py b/src/gradientai/types/agent_delete_response.py similarity index 100% rename from src/do_gradientai/types/agent_delete_response.py rename to src/gradientai/types/agent_delete_response.py diff --git a/src/do_gradientai/types/agent_list_params.py b/src/gradientai/types/agent_list_params.py similarity index 100% rename from src/do_gradientai/types/agent_list_params.py rename to src/gradientai/types/agent_list_params.py diff --git a/src/do_gradientai/types/agent_list_response.py b/src/gradientai/types/agent_list_response.py similarity index 100% rename from src/do_gradientai/types/agent_list_response.py rename to src/gradientai/types/agent_list_response.py diff --git a/src/do_gradientai/types/agent_retrieve_response.py b/src/gradientai/types/agent_retrieve_response.py similarity index 100% rename from src/do_gradientai/types/agent_retrieve_response.py rename to src/gradientai/types/agent_retrieve_response.py diff --git a/src/do_gradientai/types/agent_update_params.py b/src/gradientai/types/agent_update_params.py similarity index 100% rename from src/do_gradientai/types/agent_update_params.py rename to src/gradientai/types/agent_update_params.py diff --git a/src/do_gradientai/types/agent_update_response.py b/src/gradientai/types/agent_update_response.py similarity index 100% rename from src/do_gradientai/types/agent_update_response.py rename to src/gradientai/types/agent_update_response.py diff --git a/src/do_gradientai/types/agent_update_status_params.py b/src/gradientai/types/agent_update_status_params.py similarity index 100% rename from src/do_gradientai/types/agent_update_status_params.py rename to src/gradientai/types/agent_update_status_params.py diff --git a/src/do_gradientai/types/agent_update_status_response.py b/src/gradientai/types/agent_update_status_response.py similarity index 100% rename from src/do_gradientai/types/agent_update_status_response.py rename to src/gradientai/types/agent_update_status_response.py diff --git a/src/do_gradientai/types/agents/__init__.py b/src/gradientai/types/agents/__init__.py similarity index 100% rename from src/do_gradientai/types/agents/__init__.py rename to src/gradientai/types/agents/__init__.py diff --git a/src/do_gradientai/types/agents/api_evaluation_metric.py b/src/gradientai/types/agents/api_evaluation_metric.py similarity index 100% rename from src/do_gradientai/types/agents/api_evaluation_metric.py rename to src/gradientai/types/agents/api_evaluation_metric.py diff --git a/src/do_gradientai/types/agents/api_evaluation_metric_result.py b/src/gradientai/types/agents/api_evaluation_metric_result.py similarity index 100% rename from src/do_gradientai/types/agents/api_evaluation_metric_result.py rename to src/gradientai/types/agents/api_evaluation_metric_result.py diff --git a/src/do_gradientai/types/agents/api_evaluation_prompt.py b/src/gradientai/types/agents/api_evaluation_prompt.py similarity index 100% rename from src/do_gradientai/types/agents/api_evaluation_prompt.py rename to src/gradientai/types/agents/api_evaluation_prompt.py diff --git a/src/do_gradientai/types/agents/api_evaluation_run.py b/src/gradientai/types/agents/api_evaluation_run.py similarity index 100% rename from src/do_gradientai/types/agents/api_evaluation_run.py rename to src/gradientai/types/agents/api_evaluation_run.py diff --git a/src/do_gradientai/types/agents/api_evaluation_test_case.py b/src/gradientai/types/agents/api_evaluation_test_case.py similarity index 100% rename from src/do_gradientai/types/agents/api_evaluation_test_case.py rename to src/gradientai/types/agents/api_evaluation_test_case.py diff --git a/src/do_gradientai/types/agents/api_key_create_params.py b/src/gradientai/types/agents/api_key_create_params.py similarity index 100% rename from src/do_gradientai/types/agents/api_key_create_params.py rename to src/gradientai/types/agents/api_key_create_params.py diff --git a/src/do_gradientai/types/agents/api_key_create_response.py b/src/gradientai/types/agents/api_key_create_response.py similarity index 100% rename from src/do_gradientai/types/agents/api_key_create_response.py rename to src/gradientai/types/agents/api_key_create_response.py diff --git a/src/do_gradientai/types/agents/api_key_delete_response.py b/src/gradientai/types/agents/api_key_delete_response.py similarity index 100% rename from src/do_gradientai/types/agents/api_key_delete_response.py rename to src/gradientai/types/agents/api_key_delete_response.py diff --git a/src/do_gradientai/types/agents/api_key_list_params.py b/src/gradientai/types/agents/api_key_list_params.py similarity index 100% rename from src/do_gradientai/types/agents/api_key_list_params.py rename to src/gradientai/types/agents/api_key_list_params.py diff --git a/src/do_gradientai/types/agents/api_key_list_response.py b/src/gradientai/types/agents/api_key_list_response.py similarity index 100% rename from src/do_gradientai/types/agents/api_key_list_response.py rename to src/gradientai/types/agents/api_key_list_response.py diff --git a/src/do_gradientai/types/agents/api_key_regenerate_response.py b/src/gradientai/types/agents/api_key_regenerate_response.py similarity index 100% rename from src/do_gradientai/types/agents/api_key_regenerate_response.py rename to src/gradientai/types/agents/api_key_regenerate_response.py diff --git a/src/do_gradientai/types/agents/api_key_update_params.py b/src/gradientai/types/agents/api_key_update_params.py similarity index 100% rename from src/do_gradientai/types/agents/api_key_update_params.py rename to src/gradientai/types/agents/api_key_update_params.py diff --git a/src/do_gradientai/types/agents/api_key_update_response.py b/src/gradientai/types/agents/api_key_update_response.py similarity index 100% rename from src/do_gradientai/types/agents/api_key_update_response.py rename to src/gradientai/types/agents/api_key_update_response.py diff --git a/src/do_gradientai/types/agents/api_link_knowledge_base_output.py b/src/gradientai/types/agents/api_link_knowledge_base_output.py similarity index 100% rename from src/do_gradientai/types/agents/api_link_knowledge_base_output.py rename to src/gradientai/types/agents/api_link_knowledge_base_output.py diff --git a/src/do_gradientai/types/agents/api_star_metric.py b/src/gradientai/types/agents/api_star_metric.py similarity index 100% rename from src/do_gradientai/types/agents/api_star_metric.py rename to src/gradientai/types/agents/api_star_metric.py diff --git a/src/do_gradientai/types/agents/api_star_metric_param.py b/src/gradientai/types/agents/api_star_metric_param.py similarity index 100% rename from src/do_gradientai/types/agents/api_star_metric_param.py rename to src/gradientai/types/agents/api_star_metric_param.py diff --git a/src/do_gradientai/types/agents/evaluation_dataset_create_file_upload_presigned_urls_params.py b/src/gradientai/types/agents/evaluation_dataset_create_file_upload_presigned_urls_params.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_dataset_create_file_upload_presigned_urls_params.py rename to src/gradientai/types/agents/evaluation_dataset_create_file_upload_presigned_urls_params.py diff --git a/src/do_gradientai/types/agents/evaluation_dataset_create_file_upload_presigned_urls_response.py b/src/gradientai/types/agents/evaluation_dataset_create_file_upload_presigned_urls_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_dataset_create_file_upload_presigned_urls_response.py rename to src/gradientai/types/agents/evaluation_dataset_create_file_upload_presigned_urls_response.py diff --git a/src/do_gradientai/types/agents/evaluation_dataset_create_params.py b/src/gradientai/types/agents/evaluation_dataset_create_params.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_dataset_create_params.py rename to src/gradientai/types/agents/evaluation_dataset_create_params.py diff --git a/src/do_gradientai/types/agents/evaluation_dataset_create_response.py b/src/gradientai/types/agents/evaluation_dataset_create_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_dataset_create_response.py rename to src/gradientai/types/agents/evaluation_dataset_create_response.py diff --git a/src/do_gradientai/types/agents/evaluation_metric_list_response.py b/src/gradientai/types/agents/evaluation_metric_list_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metric_list_response.py rename to src/gradientai/types/agents/evaluation_metric_list_response.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/__init__.py b/src/gradientai/types/agents/evaluation_metrics/__init__.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/__init__.py rename to src/gradientai/types/agents/evaluation_metrics/__init__.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspace_create_params.py b/src/gradientai/types/agents/evaluation_metrics/workspace_create_params.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspace_create_params.py rename to src/gradientai/types/agents/evaluation_metrics/workspace_create_params.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspace_create_response.py b/src/gradientai/types/agents/evaluation_metrics/workspace_create_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspace_create_response.py rename to src/gradientai/types/agents/evaluation_metrics/workspace_create_response.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspace_delete_response.py b/src/gradientai/types/agents/evaluation_metrics/workspace_delete_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspace_delete_response.py rename to src/gradientai/types/agents/evaluation_metrics/workspace_delete_response.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspace_list_evaluation_test_cases_response.py b/src/gradientai/types/agents/evaluation_metrics/workspace_list_evaluation_test_cases_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspace_list_evaluation_test_cases_response.py rename to src/gradientai/types/agents/evaluation_metrics/workspace_list_evaluation_test_cases_response.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspace_list_response.py b/src/gradientai/types/agents/evaluation_metrics/workspace_list_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspace_list_response.py rename to src/gradientai/types/agents/evaluation_metrics/workspace_list_response.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspace_retrieve_response.py b/src/gradientai/types/agents/evaluation_metrics/workspace_retrieve_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspace_retrieve_response.py rename to src/gradientai/types/agents/evaluation_metrics/workspace_retrieve_response.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspace_update_params.py b/src/gradientai/types/agents/evaluation_metrics/workspace_update_params.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspace_update_params.py rename to src/gradientai/types/agents/evaluation_metrics/workspace_update_params.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspace_update_response.py b/src/gradientai/types/agents/evaluation_metrics/workspace_update_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspace_update_response.py rename to src/gradientai/types/agents/evaluation_metrics/workspace_update_response.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspaces/__init__.py b/src/gradientai/types/agents/evaluation_metrics/workspaces/__init__.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspaces/__init__.py rename to src/gradientai/types/agents/evaluation_metrics/workspaces/__init__.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspaces/agent_list_params.py b/src/gradientai/types/agents/evaluation_metrics/workspaces/agent_list_params.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspaces/agent_list_params.py rename to src/gradientai/types/agents/evaluation_metrics/workspaces/agent_list_params.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspaces/agent_list_response.py b/src/gradientai/types/agents/evaluation_metrics/workspaces/agent_list_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspaces/agent_list_response.py rename to src/gradientai/types/agents/evaluation_metrics/workspaces/agent_list_response.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspaces/agent_move_params.py b/src/gradientai/types/agents/evaluation_metrics/workspaces/agent_move_params.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspaces/agent_move_params.py rename to src/gradientai/types/agents/evaluation_metrics/workspaces/agent_move_params.py diff --git a/src/do_gradientai/types/agents/evaluation_metrics/workspaces/agent_move_response.py b/src/gradientai/types/agents/evaluation_metrics/workspaces/agent_move_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_metrics/workspaces/agent_move_response.py rename to src/gradientai/types/agents/evaluation_metrics/workspaces/agent_move_response.py diff --git a/src/do_gradientai/types/agents/evaluation_run_create_params.py b/src/gradientai/types/agents/evaluation_run_create_params.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_run_create_params.py rename to src/gradientai/types/agents/evaluation_run_create_params.py diff --git a/src/do_gradientai/types/agents/evaluation_run_create_response.py b/src/gradientai/types/agents/evaluation_run_create_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_run_create_response.py rename to src/gradientai/types/agents/evaluation_run_create_response.py diff --git a/src/do_gradientai/types/agents/evaluation_run_list_results_response.py b/src/gradientai/types/agents/evaluation_run_list_results_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_run_list_results_response.py rename to src/gradientai/types/agents/evaluation_run_list_results_response.py diff --git a/src/do_gradientai/types/agents/evaluation_run_retrieve_response.py b/src/gradientai/types/agents/evaluation_run_retrieve_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_run_retrieve_response.py rename to src/gradientai/types/agents/evaluation_run_retrieve_response.py diff --git a/src/do_gradientai/types/agents/evaluation_run_retrieve_results_response.py b/src/gradientai/types/agents/evaluation_run_retrieve_results_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_run_retrieve_results_response.py rename to src/gradientai/types/agents/evaluation_run_retrieve_results_response.py diff --git a/src/do_gradientai/types/agents/evaluation_test_case_create_params.py b/src/gradientai/types/agents/evaluation_test_case_create_params.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_test_case_create_params.py rename to src/gradientai/types/agents/evaluation_test_case_create_params.py diff --git a/src/do_gradientai/types/agents/evaluation_test_case_create_response.py b/src/gradientai/types/agents/evaluation_test_case_create_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_test_case_create_response.py rename to src/gradientai/types/agents/evaluation_test_case_create_response.py diff --git a/src/do_gradientai/types/agents/evaluation_test_case_list_evaluation_runs_params.py b/src/gradientai/types/agents/evaluation_test_case_list_evaluation_runs_params.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_test_case_list_evaluation_runs_params.py rename to src/gradientai/types/agents/evaluation_test_case_list_evaluation_runs_params.py diff --git a/src/do_gradientai/types/agents/evaluation_test_case_list_evaluation_runs_response.py b/src/gradientai/types/agents/evaluation_test_case_list_evaluation_runs_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_test_case_list_evaluation_runs_response.py rename to src/gradientai/types/agents/evaluation_test_case_list_evaluation_runs_response.py diff --git a/src/do_gradientai/types/agents/evaluation_test_case_list_response.py b/src/gradientai/types/agents/evaluation_test_case_list_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_test_case_list_response.py rename to src/gradientai/types/agents/evaluation_test_case_list_response.py diff --git a/src/do_gradientai/types/agents/evaluation_test_case_retrieve_params.py b/src/gradientai/types/agents/evaluation_test_case_retrieve_params.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_test_case_retrieve_params.py rename to src/gradientai/types/agents/evaluation_test_case_retrieve_params.py diff --git a/src/do_gradientai/types/agents/evaluation_test_case_retrieve_response.py b/src/gradientai/types/agents/evaluation_test_case_retrieve_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_test_case_retrieve_response.py rename to src/gradientai/types/agents/evaluation_test_case_retrieve_response.py diff --git a/src/do_gradientai/types/agents/evaluation_test_case_update_params.py b/src/gradientai/types/agents/evaluation_test_case_update_params.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_test_case_update_params.py rename to src/gradientai/types/agents/evaluation_test_case_update_params.py diff --git a/src/do_gradientai/types/agents/evaluation_test_case_update_response.py b/src/gradientai/types/agents/evaluation_test_case_update_response.py similarity index 100% rename from src/do_gradientai/types/agents/evaluation_test_case_update_response.py rename to src/gradientai/types/agents/evaluation_test_case_update_response.py diff --git a/src/do_gradientai/types/agents/function_create_params.py b/src/gradientai/types/agents/function_create_params.py similarity index 100% rename from src/do_gradientai/types/agents/function_create_params.py rename to src/gradientai/types/agents/function_create_params.py diff --git a/src/do_gradientai/types/agents/function_create_response.py b/src/gradientai/types/agents/function_create_response.py similarity index 100% rename from src/do_gradientai/types/agents/function_create_response.py rename to src/gradientai/types/agents/function_create_response.py diff --git a/src/do_gradientai/types/agents/function_delete_response.py b/src/gradientai/types/agents/function_delete_response.py similarity index 100% rename from src/do_gradientai/types/agents/function_delete_response.py rename to src/gradientai/types/agents/function_delete_response.py diff --git a/src/do_gradientai/types/agents/function_update_params.py b/src/gradientai/types/agents/function_update_params.py similarity index 100% rename from src/do_gradientai/types/agents/function_update_params.py rename to src/gradientai/types/agents/function_update_params.py diff --git a/src/do_gradientai/types/agents/function_update_response.py b/src/gradientai/types/agents/function_update_response.py similarity index 100% rename from src/do_gradientai/types/agents/function_update_response.py rename to src/gradientai/types/agents/function_update_response.py diff --git a/src/do_gradientai/types/agents/knowledge_base_detach_response.py b/src/gradientai/types/agents/knowledge_base_detach_response.py similarity index 100% rename from src/do_gradientai/types/agents/knowledge_base_detach_response.py rename to src/gradientai/types/agents/knowledge_base_detach_response.py diff --git a/src/do_gradientai/types/agents/route_add_params.py b/src/gradientai/types/agents/route_add_params.py similarity index 100% rename from src/do_gradientai/types/agents/route_add_params.py rename to src/gradientai/types/agents/route_add_params.py diff --git a/src/do_gradientai/types/agents/route_add_response.py b/src/gradientai/types/agents/route_add_response.py similarity index 100% rename from src/do_gradientai/types/agents/route_add_response.py rename to src/gradientai/types/agents/route_add_response.py diff --git a/src/do_gradientai/types/agents/route_delete_response.py b/src/gradientai/types/agents/route_delete_response.py similarity index 100% rename from src/do_gradientai/types/agents/route_delete_response.py rename to src/gradientai/types/agents/route_delete_response.py diff --git a/src/do_gradientai/types/agents/route_update_params.py b/src/gradientai/types/agents/route_update_params.py similarity index 100% rename from src/do_gradientai/types/agents/route_update_params.py rename to src/gradientai/types/agents/route_update_params.py diff --git a/src/do_gradientai/types/agents/route_update_response.py b/src/gradientai/types/agents/route_update_response.py similarity index 100% rename from src/do_gradientai/types/agents/route_update_response.py rename to src/gradientai/types/agents/route_update_response.py diff --git a/src/do_gradientai/types/agents/route_view_response.py b/src/gradientai/types/agents/route_view_response.py similarity index 100% rename from src/do_gradientai/types/agents/route_view_response.py rename to src/gradientai/types/agents/route_view_response.py diff --git a/src/do_gradientai/types/agents/version_list_params.py b/src/gradientai/types/agents/version_list_params.py similarity index 100% rename from src/do_gradientai/types/agents/version_list_params.py rename to src/gradientai/types/agents/version_list_params.py diff --git a/src/do_gradientai/types/agents/version_list_response.py b/src/gradientai/types/agents/version_list_response.py similarity index 100% rename from src/do_gradientai/types/agents/version_list_response.py rename to src/gradientai/types/agents/version_list_response.py diff --git a/src/do_gradientai/types/agents/version_update_params.py b/src/gradientai/types/agents/version_update_params.py similarity index 100% rename from src/do_gradientai/types/agents/version_update_params.py rename to src/gradientai/types/agents/version_update_params.py diff --git a/src/do_gradientai/types/agents/version_update_response.py b/src/gradientai/types/agents/version_update_response.py similarity index 100% rename from src/do_gradientai/types/agents/version_update_response.py rename to src/gradientai/types/agents/version_update_response.py diff --git a/src/do_gradientai/types/api_agent.py b/src/gradientai/types/api_agent.py similarity index 100% rename from src/do_gradientai/types/api_agent.py rename to src/gradientai/types/api_agent.py diff --git a/src/do_gradientai/types/api_agent_api_key_info.py b/src/gradientai/types/api_agent_api_key_info.py similarity index 100% rename from src/do_gradientai/types/api_agent_api_key_info.py rename to src/gradientai/types/api_agent_api_key_info.py diff --git a/src/do_gradientai/types/api_agent_model.py b/src/gradientai/types/api_agent_model.py similarity index 100% rename from src/do_gradientai/types/api_agent_model.py rename to src/gradientai/types/api_agent_model.py diff --git a/src/do_gradientai/types/api_agreement.py b/src/gradientai/types/api_agreement.py similarity index 100% rename from src/do_gradientai/types/api_agreement.py rename to src/gradientai/types/api_agreement.py diff --git a/src/do_gradientai/types/api_anthropic_api_key_info.py b/src/gradientai/types/api_anthropic_api_key_info.py similarity index 100% rename from src/do_gradientai/types/api_anthropic_api_key_info.py rename to src/gradientai/types/api_anthropic_api_key_info.py diff --git a/src/do_gradientai/types/api_deployment_visibility.py b/src/gradientai/types/api_deployment_visibility.py similarity index 100% rename from src/do_gradientai/types/api_deployment_visibility.py rename to src/gradientai/types/api_deployment_visibility.py diff --git a/src/do_gradientai/types/api_knowledge_base.py b/src/gradientai/types/api_knowledge_base.py similarity index 100% rename from src/do_gradientai/types/api_knowledge_base.py rename to src/gradientai/types/api_knowledge_base.py diff --git a/src/do_gradientai/types/api_model_version.py b/src/gradientai/types/api_model_version.py similarity index 100% rename from src/do_gradientai/types/api_model_version.py rename to src/gradientai/types/api_model_version.py diff --git a/src/do_gradientai/types/api_openai_api_key_info.py b/src/gradientai/types/api_openai_api_key_info.py similarity index 100% rename from src/do_gradientai/types/api_openai_api_key_info.py rename to src/gradientai/types/api_openai_api_key_info.py diff --git a/src/do_gradientai/types/api_retrieval_method.py b/src/gradientai/types/api_retrieval_method.py similarity index 100% rename from src/do_gradientai/types/api_retrieval_method.py rename to src/gradientai/types/api_retrieval_method.py diff --git a/src/do_gradientai/types/api_workspace.py b/src/gradientai/types/api_workspace.py similarity index 100% rename from src/do_gradientai/types/api_workspace.py rename to src/gradientai/types/api_workspace.py diff --git a/src/do_gradientai/types/chat/__init__.py b/src/gradientai/types/chat/__init__.py similarity index 100% rename from src/do_gradientai/types/chat/__init__.py rename to src/gradientai/types/chat/__init__.py diff --git a/src/do_gradientai/types/chat/chat_completion_token_logprob.py b/src/gradientai/types/chat/chat_completion_token_logprob.py similarity index 100% rename from src/do_gradientai/types/chat/chat_completion_token_logprob.py rename to src/gradientai/types/chat/chat_completion_token_logprob.py diff --git a/src/do_gradientai/types/chat/completion_create_params.py b/src/gradientai/types/chat/completion_create_params.py similarity index 100% rename from src/do_gradientai/types/chat/completion_create_params.py rename to src/gradientai/types/chat/completion_create_params.py diff --git a/src/do_gradientai/types/chat/completion_create_response.py b/src/gradientai/types/chat/completion_create_response.py similarity index 100% rename from src/do_gradientai/types/chat/completion_create_response.py rename to src/gradientai/types/chat/completion_create_response.py diff --git a/src/do_gradientai/types/inference/__init__.py b/src/gradientai/types/inference/__init__.py similarity index 100% rename from src/do_gradientai/types/inference/__init__.py rename to src/gradientai/types/inference/__init__.py diff --git a/src/do_gradientai/types/inference/api_key_create_params.py b/src/gradientai/types/inference/api_key_create_params.py similarity index 100% rename from src/do_gradientai/types/inference/api_key_create_params.py rename to src/gradientai/types/inference/api_key_create_params.py diff --git a/src/do_gradientai/types/inference/api_key_create_response.py b/src/gradientai/types/inference/api_key_create_response.py similarity index 100% rename from src/do_gradientai/types/inference/api_key_create_response.py rename to src/gradientai/types/inference/api_key_create_response.py diff --git a/src/do_gradientai/types/inference/api_key_delete_response.py b/src/gradientai/types/inference/api_key_delete_response.py similarity index 100% rename from src/do_gradientai/types/inference/api_key_delete_response.py rename to src/gradientai/types/inference/api_key_delete_response.py diff --git a/src/do_gradientai/types/inference/api_key_list_params.py b/src/gradientai/types/inference/api_key_list_params.py similarity index 100% rename from src/do_gradientai/types/inference/api_key_list_params.py rename to src/gradientai/types/inference/api_key_list_params.py diff --git a/src/do_gradientai/types/inference/api_key_list_response.py b/src/gradientai/types/inference/api_key_list_response.py similarity index 100% rename from src/do_gradientai/types/inference/api_key_list_response.py rename to src/gradientai/types/inference/api_key_list_response.py diff --git a/src/do_gradientai/types/inference/api_key_update_params.py b/src/gradientai/types/inference/api_key_update_params.py similarity index 100% rename from src/do_gradientai/types/inference/api_key_update_params.py rename to src/gradientai/types/inference/api_key_update_params.py diff --git a/src/do_gradientai/types/inference/api_key_update_regenerate_response.py b/src/gradientai/types/inference/api_key_update_regenerate_response.py similarity index 100% rename from src/do_gradientai/types/inference/api_key_update_regenerate_response.py rename to src/gradientai/types/inference/api_key_update_regenerate_response.py diff --git a/src/do_gradientai/types/inference/api_key_update_response.py b/src/gradientai/types/inference/api_key_update_response.py similarity index 100% rename from src/do_gradientai/types/inference/api_key_update_response.py rename to src/gradientai/types/inference/api_key_update_response.py diff --git a/src/do_gradientai/types/inference/api_model_api_key_info.py b/src/gradientai/types/inference/api_model_api_key_info.py similarity index 100% rename from src/do_gradientai/types/inference/api_model_api_key_info.py rename to src/gradientai/types/inference/api_model_api_key_info.py diff --git a/src/do_gradientai/types/knowledge_base_create_params.py b/src/gradientai/types/knowledge_base_create_params.py similarity index 100% rename from src/do_gradientai/types/knowledge_base_create_params.py rename to src/gradientai/types/knowledge_base_create_params.py diff --git a/src/do_gradientai/types/knowledge_base_create_response.py b/src/gradientai/types/knowledge_base_create_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_base_create_response.py rename to src/gradientai/types/knowledge_base_create_response.py diff --git a/src/do_gradientai/types/knowledge_base_delete_response.py b/src/gradientai/types/knowledge_base_delete_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_base_delete_response.py rename to src/gradientai/types/knowledge_base_delete_response.py diff --git a/src/do_gradientai/types/knowledge_base_list_params.py b/src/gradientai/types/knowledge_base_list_params.py similarity index 100% rename from src/do_gradientai/types/knowledge_base_list_params.py rename to src/gradientai/types/knowledge_base_list_params.py diff --git a/src/do_gradientai/types/knowledge_base_list_response.py b/src/gradientai/types/knowledge_base_list_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_base_list_response.py rename to src/gradientai/types/knowledge_base_list_response.py diff --git a/src/do_gradientai/types/knowledge_base_retrieve_response.py b/src/gradientai/types/knowledge_base_retrieve_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_base_retrieve_response.py rename to src/gradientai/types/knowledge_base_retrieve_response.py diff --git a/src/do_gradientai/types/knowledge_base_update_params.py b/src/gradientai/types/knowledge_base_update_params.py similarity index 100% rename from src/do_gradientai/types/knowledge_base_update_params.py rename to src/gradientai/types/knowledge_base_update_params.py diff --git a/src/do_gradientai/types/knowledge_base_update_response.py b/src/gradientai/types/knowledge_base_update_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_base_update_response.py rename to src/gradientai/types/knowledge_base_update_response.py diff --git a/src/do_gradientai/types/knowledge_bases/__init__.py b/src/gradientai/types/knowledge_bases/__init__.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/__init__.py rename to src/gradientai/types/knowledge_bases/__init__.py diff --git a/src/do_gradientai/types/knowledge_bases/api_file_upload_data_source.py b/src/gradientai/types/knowledge_bases/api_file_upload_data_source.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/api_file_upload_data_source.py rename to src/gradientai/types/knowledge_bases/api_file_upload_data_source.py diff --git a/src/do_gradientai/types/knowledge_bases/api_file_upload_data_source_param.py b/src/gradientai/types/knowledge_bases/api_file_upload_data_source_param.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/api_file_upload_data_source_param.py rename to src/gradientai/types/knowledge_bases/api_file_upload_data_source_param.py diff --git a/src/do_gradientai/types/knowledge_bases/api_indexed_data_source.py b/src/gradientai/types/knowledge_bases/api_indexed_data_source.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/api_indexed_data_source.py rename to src/gradientai/types/knowledge_bases/api_indexed_data_source.py diff --git a/src/do_gradientai/types/knowledge_bases/api_indexing_job.py b/src/gradientai/types/knowledge_bases/api_indexing_job.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/api_indexing_job.py rename to src/gradientai/types/knowledge_bases/api_indexing_job.py diff --git a/src/do_gradientai/types/knowledge_bases/api_knowledge_base_data_source.py b/src/gradientai/types/knowledge_bases/api_knowledge_base_data_source.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/api_knowledge_base_data_source.py rename to src/gradientai/types/knowledge_bases/api_knowledge_base_data_source.py diff --git a/src/do_gradientai/types/knowledge_bases/api_spaces_data_source.py b/src/gradientai/types/knowledge_bases/api_spaces_data_source.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/api_spaces_data_source.py rename to src/gradientai/types/knowledge_bases/api_spaces_data_source.py diff --git a/src/do_gradientai/types/knowledge_bases/api_spaces_data_source_param.py b/src/gradientai/types/knowledge_bases/api_spaces_data_source_param.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/api_spaces_data_source_param.py rename to src/gradientai/types/knowledge_bases/api_spaces_data_source_param.py diff --git a/src/do_gradientai/types/knowledge_bases/api_web_crawler_data_source.py b/src/gradientai/types/knowledge_bases/api_web_crawler_data_source.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/api_web_crawler_data_source.py rename to src/gradientai/types/knowledge_bases/api_web_crawler_data_source.py diff --git a/src/do_gradientai/types/knowledge_bases/api_web_crawler_data_source_param.py b/src/gradientai/types/knowledge_bases/api_web_crawler_data_source_param.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/api_web_crawler_data_source_param.py rename to src/gradientai/types/knowledge_bases/api_web_crawler_data_source_param.py diff --git a/src/do_gradientai/types/knowledge_bases/aws_data_source_param.py b/src/gradientai/types/knowledge_bases/aws_data_source_param.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/aws_data_source_param.py rename to src/gradientai/types/knowledge_bases/aws_data_source_param.py diff --git a/src/do_gradientai/types/knowledge_bases/data_source_create_params.py b/src/gradientai/types/knowledge_bases/data_source_create_params.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/data_source_create_params.py rename to src/gradientai/types/knowledge_bases/data_source_create_params.py diff --git a/src/do_gradientai/types/knowledge_bases/data_source_create_response.py b/src/gradientai/types/knowledge_bases/data_source_create_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/data_source_create_response.py rename to src/gradientai/types/knowledge_bases/data_source_create_response.py diff --git a/src/do_gradientai/types/knowledge_bases/data_source_delete_response.py b/src/gradientai/types/knowledge_bases/data_source_delete_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/data_source_delete_response.py rename to src/gradientai/types/knowledge_bases/data_source_delete_response.py diff --git a/src/do_gradientai/types/knowledge_bases/data_source_list_params.py b/src/gradientai/types/knowledge_bases/data_source_list_params.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/data_source_list_params.py rename to src/gradientai/types/knowledge_bases/data_source_list_params.py diff --git a/src/do_gradientai/types/knowledge_bases/data_source_list_response.py b/src/gradientai/types/knowledge_bases/data_source_list_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/data_source_list_response.py rename to src/gradientai/types/knowledge_bases/data_source_list_response.py diff --git a/src/do_gradientai/types/knowledge_bases/indexing_job_create_params.py b/src/gradientai/types/knowledge_bases/indexing_job_create_params.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/indexing_job_create_params.py rename to src/gradientai/types/knowledge_bases/indexing_job_create_params.py diff --git a/src/do_gradientai/types/knowledge_bases/indexing_job_create_response.py b/src/gradientai/types/knowledge_bases/indexing_job_create_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/indexing_job_create_response.py rename to src/gradientai/types/knowledge_bases/indexing_job_create_response.py diff --git a/src/do_gradientai/types/knowledge_bases/indexing_job_list_params.py b/src/gradientai/types/knowledge_bases/indexing_job_list_params.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/indexing_job_list_params.py rename to src/gradientai/types/knowledge_bases/indexing_job_list_params.py diff --git a/src/do_gradientai/types/knowledge_bases/indexing_job_list_response.py b/src/gradientai/types/knowledge_bases/indexing_job_list_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/indexing_job_list_response.py rename to src/gradientai/types/knowledge_bases/indexing_job_list_response.py diff --git a/src/do_gradientai/types/knowledge_bases/indexing_job_retrieve_data_sources_response.py b/src/gradientai/types/knowledge_bases/indexing_job_retrieve_data_sources_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/indexing_job_retrieve_data_sources_response.py rename to src/gradientai/types/knowledge_bases/indexing_job_retrieve_data_sources_response.py diff --git a/src/do_gradientai/types/knowledge_bases/indexing_job_retrieve_response.py b/src/gradientai/types/knowledge_bases/indexing_job_retrieve_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/indexing_job_retrieve_response.py rename to src/gradientai/types/knowledge_bases/indexing_job_retrieve_response.py diff --git a/src/do_gradientai/types/knowledge_bases/indexing_job_update_cancel_params.py b/src/gradientai/types/knowledge_bases/indexing_job_update_cancel_params.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/indexing_job_update_cancel_params.py rename to src/gradientai/types/knowledge_bases/indexing_job_update_cancel_params.py diff --git a/src/do_gradientai/types/knowledge_bases/indexing_job_update_cancel_response.py b/src/gradientai/types/knowledge_bases/indexing_job_update_cancel_response.py similarity index 100% rename from src/do_gradientai/types/knowledge_bases/indexing_job_update_cancel_response.py rename to src/gradientai/types/knowledge_bases/indexing_job_update_cancel_response.py diff --git a/src/do_gradientai/types/model.py b/src/gradientai/types/model.py similarity index 100% rename from src/do_gradientai/types/model.py rename to src/gradientai/types/model.py diff --git a/src/do_gradientai/types/model_list_response.py b/src/gradientai/types/model_list_response.py similarity index 100% rename from src/do_gradientai/types/model_list_response.py rename to src/gradientai/types/model_list_response.py diff --git a/src/do_gradientai/types/model_providers/__init__.py b/src/gradientai/types/model_providers/__init__.py similarity index 100% rename from src/do_gradientai/types/model_providers/__init__.py rename to src/gradientai/types/model_providers/__init__.py diff --git a/src/do_gradientai/types/model_providers/anthropic/__init__.py b/src/gradientai/types/model_providers/anthropic/__init__.py similarity index 100% rename from src/do_gradientai/types/model_providers/anthropic/__init__.py rename to src/gradientai/types/model_providers/anthropic/__init__.py diff --git a/src/do_gradientai/types/model_providers/anthropic/key_create_params.py b/src/gradientai/types/model_providers/anthropic/key_create_params.py similarity index 100% rename from src/do_gradientai/types/model_providers/anthropic/key_create_params.py rename to src/gradientai/types/model_providers/anthropic/key_create_params.py diff --git a/src/do_gradientai/types/model_providers/anthropic/key_create_response.py b/src/gradientai/types/model_providers/anthropic/key_create_response.py similarity index 100% rename from src/do_gradientai/types/model_providers/anthropic/key_create_response.py rename to src/gradientai/types/model_providers/anthropic/key_create_response.py diff --git a/src/do_gradientai/types/model_providers/anthropic/key_delete_response.py b/src/gradientai/types/model_providers/anthropic/key_delete_response.py similarity index 100% rename from src/do_gradientai/types/model_providers/anthropic/key_delete_response.py rename to src/gradientai/types/model_providers/anthropic/key_delete_response.py diff --git a/src/do_gradientai/types/model_providers/anthropic/key_list_agents_params.py b/src/gradientai/types/model_providers/anthropic/key_list_agents_params.py similarity index 100% rename from src/do_gradientai/types/model_providers/anthropic/key_list_agents_params.py rename to src/gradientai/types/model_providers/anthropic/key_list_agents_params.py diff --git a/src/do_gradientai/types/model_providers/anthropic/key_list_agents_response.py b/src/gradientai/types/model_providers/anthropic/key_list_agents_response.py similarity index 100% rename from src/do_gradientai/types/model_providers/anthropic/key_list_agents_response.py rename to src/gradientai/types/model_providers/anthropic/key_list_agents_response.py diff --git a/src/do_gradientai/types/model_providers/anthropic/key_list_params.py b/src/gradientai/types/model_providers/anthropic/key_list_params.py similarity index 100% rename from src/do_gradientai/types/model_providers/anthropic/key_list_params.py rename to src/gradientai/types/model_providers/anthropic/key_list_params.py diff --git a/src/do_gradientai/types/model_providers/anthropic/key_list_response.py b/src/gradientai/types/model_providers/anthropic/key_list_response.py similarity index 100% rename from src/do_gradientai/types/model_providers/anthropic/key_list_response.py rename to src/gradientai/types/model_providers/anthropic/key_list_response.py diff --git a/src/do_gradientai/types/model_providers/anthropic/key_retrieve_response.py b/src/gradientai/types/model_providers/anthropic/key_retrieve_response.py similarity index 100% rename from src/do_gradientai/types/model_providers/anthropic/key_retrieve_response.py rename to src/gradientai/types/model_providers/anthropic/key_retrieve_response.py diff --git a/src/do_gradientai/types/model_providers/anthropic/key_update_params.py b/src/gradientai/types/model_providers/anthropic/key_update_params.py similarity index 100% rename from src/do_gradientai/types/model_providers/anthropic/key_update_params.py rename to src/gradientai/types/model_providers/anthropic/key_update_params.py diff --git a/src/do_gradientai/types/model_providers/anthropic/key_update_response.py b/src/gradientai/types/model_providers/anthropic/key_update_response.py similarity index 100% rename from src/do_gradientai/types/model_providers/anthropic/key_update_response.py rename to src/gradientai/types/model_providers/anthropic/key_update_response.py diff --git a/src/do_gradientai/types/model_providers/openai/__init__.py b/src/gradientai/types/model_providers/openai/__init__.py similarity index 100% rename from src/do_gradientai/types/model_providers/openai/__init__.py rename to src/gradientai/types/model_providers/openai/__init__.py diff --git a/src/do_gradientai/types/model_providers/openai/key_create_params.py b/src/gradientai/types/model_providers/openai/key_create_params.py similarity index 100% rename from src/do_gradientai/types/model_providers/openai/key_create_params.py rename to src/gradientai/types/model_providers/openai/key_create_params.py diff --git a/src/do_gradientai/types/model_providers/openai/key_create_response.py b/src/gradientai/types/model_providers/openai/key_create_response.py similarity index 100% rename from src/do_gradientai/types/model_providers/openai/key_create_response.py rename to src/gradientai/types/model_providers/openai/key_create_response.py diff --git a/src/do_gradientai/types/model_providers/openai/key_delete_response.py b/src/gradientai/types/model_providers/openai/key_delete_response.py similarity index 100% rename from src/do_gradientai/types/model_providers/openai/key_delete_response.py rename to src/gradientai/types/model_providers/openai/key_delete_response.py diff --git a/src/do_gradientai/types/model_providers/openai/key_list_params.py b/src/gradientai/types/model_providers/openai/key_list_params.py similarity index 100% rename from src/do_gradientai/types/model_providers/openai/key_list_params.py rename to src/gradientai/types/model_providers/openai/key_list_params.py diff --git a/src/do_gradientai/types/model_providers/openai/key_list_response.py b/src/gradientai/types/model_providers/openai/key_list_response.py similarity index 100% rename from src/do_gradientai/types/model_providers/openai/key_list_response.py rename to src/gradientai/types/model_providers/openai/key_list_response.py diff --git a/src/do_gradientai/types/model_providers/openai/key_retrieve_agents_params.py b/src/gradientai/types/model_providers/openai/key_retrieve_agents_params.py similarity index 100% rename from src/do_gradientai/types/model_providers/openai/key_retrieve_agents_params.py rename to src/gradientai/types/model_providers/openai/key_retrieve_agents_params.py diff --git a/src/do_gradientai/types/model_providers/openai/key_retrieve_agents_response.py b/src/gradientai/types/model_providers/openai/key_retrieve_agents_response.py similarity index 100% rename from src/do_gradientai/types/model_providers/openai/key_retrieve_agents_response.py rename to src/gradientai/types/model_providers/openai/key_retrieve_agents_response.py diff --git a/src/do_gradientai/types/model_providers/openai/key_retrieve_response.py b/src/gradientai/types/model_providers/openai/key_retrieve_response.py similarity index 100% rename from src/do_gradientai/types/model_providers/openai/key_retrieve_response.py rename to src/gradientai/types/model_providers/openai/key_retrieve_response.py diff --git a/src/do_gradientai/types/model_providers/openai/key_update_params.py b/src/gradientai/types/model_providers/openai/key_update_params.py similarity index 100% rename from src/do_gradientai/types/model_providers/openai/key_update_params.py rename to src/gradientai/types/model_providers/openai/key_update_params.py diff --git a/src/do_gradientai/types/model_providers/openai/key_update_response.py b/src/gradientai/types/model_providers/openai/key_update_response.py similarity index 100% rename from src/do_gradientai/types/model_providers/openai/key_update_response.py rename to src/gradientai/types/model_providers/openai/key_update_response.py diff --git a/src/do_gradientai/types/region_list_params.py b/src/gradientai/types/region_list_params.py similarity index 100% rename from src/do_gradientai/types/region_list_params.py rename to src/gradientai/types/region_list_params.py diff --git a/src/do_gradientai/types/region_list_response.py b/src/gradientai/types/region_list_response.py similarity index 100% rename from src/do_gradientai/types/region_list_response.py rename to src/gradientai/types/region_list_response.py diff --git a/src/do_gradientai/types/shared/__init__.py b/src/gradientai/types/shared/__init__.py similarity index 100% rename from src/do_gradientai/types/shared/__init__.py rename to src/gradientai/types/shared/__init__.py diff --git a/src/do_gradientai/types/shared/api_links.py b/src/gradientai/types/shared/api_links.py similarity index 100% rename from src/do_gradientai/types/shared/api_links.py rename to src/gradientai/types/shared/api_links.py diff --git a/src/do_gradientai/types/shared/api_meta.py b/src/gradientai/types/shared/api_meta.py similarity index 100% rename from src/do_gradientai/types/shared/api_meta.py rename to src/gradientai/types/shared/api_meta.py diff --git a/tests/api_resources/agents/evaluation_metrics/test_workspaces.py b/tests/api_resources/agents/evaluation_metrics/test_workspaces.py index 42bfa79f..afeaa8f1 100644 --- a/tests/api_resources/agents/evaluation_metrics/test_workspaces.py +++ b/tests/api_resources/agents/evaluation_metrics/test_workspaces.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.agents.evaluation_metrics import ( +from gradientai.types.agents.evaluation_metrics import ( WorkspaceListResponse, WorkspaceCreateResponse, WorkspaceDeleteResponse, diff --git a/tests/api_resources/agents/evaluation_metrics/workspaces/test_agents.py b/tests/api_resources/agents/evaluation_metrics/workspaces/test_agents.py index e772d668..764e13e0 100644 --- a/tests/api_resources/agents/evaluation_metrics/workspaces/test_agents.py +++ b/tests/api_resources/agents/evaluation_metrics/workspaces/test_agents.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.agents.evaluation_metrics.workspaces import ( +from gradientai.types.agents.evaluation_metrics.workspaces import ( AgentListResponse, AgentMoveResponse, ) diff --git a/tests/api_resources/agents/test_api_keys.py b/tests/api_resources/agents/test_api_keys.py index 65351922..beb9666a 100644 --- a/tests/api_resources/agents/test_api_keys.py +++ b/tests/api_resources/agents/test_api_keys.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( APIKeyListResponse, APIKeyCreateResponse, APIKeyDeleteResponse, diff --git a/tests/api_resources/agents/test_evaluation_datasets.py b/tests/api_resources/agents/test_evaluation_datasets.py index 9e6dad52..e6ca2644 100644 --- a/tests/api_resources/agents/test_evaluation_datasets.py +++ b/tests/api_resources/agents/test_evaluation_datasets.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( EvaluationDatasetCreateResponse, EvaluationDatasetCreateFileUploadPresignedURLsResponse, ) diff --git a/tests/api_resources/agents/test_evaluation_metrics.py b/tests/api_resources/agents/test_evaluation_metrics.py index 82084f61..be83e330 100644 --- a/tests/api_resources/agents/test_evaluation_metrics.py +++ b/tests/api_resources/agents/test_evaluation_metrics.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.agents import EvaluationMetricListResponse +from gradientai.types.agents import EvaluationMetricListResponse base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010") diff --git a/tests/api_resources/agents/test_evaluation_runs.py b/tests/api_resources/agents/test_evaluation_runs.py index 6bd3cfa5..b2fce320 100644 --- a/tests/api_resources/agents/test_evaluation_runs.py +++ b/tests/api_resources/agents/test_evaluation_runs.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( EvaluationRunCreateResponse, EvaluationRunRetrieveResponse, EvaluationRunListResultsResponse, diff --git a/tests/api_resources/agents/test_evaluation_test_cases.py b/tests/api_resources/agents/test_evaluation_test_cases.py index 87f66b24..a0b5ee77 100644 --- a/tests/api_resources/agents/test_evaluation_test_cases.py +++ b/tests/api_resources/agents/test_evaluation_test_cases.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( EvaluationTestCaseListResponse, EvaluationTestCaseCreateResponse, EvaluationTestCaseUpdateResponse, diff --git a/tests/api_resources/agents/test_functions.py b/tests/api_resources/agents/test_functions.py index 2c5ceaf7..5a3693cb 100644 --- a/tests/api_resources/agents/test_functions.py +++ b/tests/api_resources/agents/test_functions.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( FunctionCreateResponse, FunctionDeleteResponse, FunctionUpdateResponse, diff --git a/tests/api_resources/agents/test_knowledge_bases.py b/tests/api_resources/agents/test_knowledge_bases.py index 0a007840..e62c05ff 100644 --- a/tests/api_resources/agents/test_knowledge_bases.py +++ b/tests/api_resources/agents/test_knowledge_bases.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.agents import APILinkKnowledgeBaseOutput, KnowledgeBaseDetachResponse +from gradientai.types.agents import APILinkKnowledgeBaseOutput, KnowledgeBaseDetachResponse base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010") diff --git a/tests/api_resources/agents/test_routes.py b/tests/api_resources/agents/test_routes.py index e2e85ab8..2e6dfd7b 100644 --- a/tests/api_resources/agents/test_routes.py +++ b/tests/api_resources/agents/test_routes.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( RouteAddResponse, RouteViewResponse, RouteDeleteResponse, diff --git a/tests/api_resources/agents/test_versions.py b/tests/api_resources/agents/test_versions.py index 314cd2e2..79f73672 100644 --- a/tests/api_resources/agents/test_versions.py +++ b/tests/api_resources/agents/test_versions.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.agents import ( +from gradientai.types.agents import ( VersionListResponse, VersionUpdateResponse, ) diff --git a/tests/api_resources/chat/test_completions.py b/tests/api_resources/chat/test_completions.py index 62f24534..b4c09579 100644 --- a/tests/api_resources/chat/test_completions.py +++ b/tests/api_resources/chat/test_completions.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.chat import CompletionCreateResponse +from gradientai.types.chat import CompletionCreateResponse base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010") diff --git a/tests/api_resources/inference/test_api_keys.py b/tests/api_resources/inference/test_api_keys.py index c48a5420..90bf95b9 100644 --- a/tests/api_resources/inference/test_api_keys.py +++ b/tests/api_resources/inference/test_api_keys.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.inference import ( +from gradientai.types.inference import ( APIKeyListResponse, APIKeyCreateResponse, APIKeyDeleteResponse, diff --git a/tests/api_resources/knowledge_bases/test_data_sources.py b/tests/api_resources/knowledge_bases/test_data_sources.py index 15665a84..9c466e2f 100644 --- a/tests/api_resources/knowledge_bases/test_data_sources.py +++ b/tests/api_resources/knowledge_bases/test_data_sources.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.knowledge_bases import ( +from gradientai.types.knowledge_bases import ( DataSourceListResponse, DataSourceCreateResponse, DataSourceDeleteResponse, diff --git a/tests/api_resources/knowledge_bases/test_indexing_jobs.py b/tests/api_resources/knowledge_bases/test_indexing_jobs.py index 206339e0..8bf1829f 100644 --- a/tests/api_resources/knowledge_bases/test_indexing_jobs.py +++ b/tests/api_resources/knowledge_bases/test_indexing_jobs.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.knowledge_bases import ( +from gradientai.types.knowledge_bases import ( IndexingJobListResponse, IndexingJobCreateResponse, IndexingJobRetrieveResponse, diff --git a/tests/api_resources/model_providers/anthropic/test_keys.py b/tests/api_resources/model_providers/anthropic/test_keys.py index b6ba0e9a..fd4ffb0f 100644 --- a/tests/api_resources/model_providers/anthropic/test_keys.py +++ b/tests/api_resources/model_providers/anthropic/test_keys.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.model_providers.anthropic import ( +from gradientai.types.model_providers.anthropic import ( KeyListResponse, KeyCreateResponse, KeyDeleteResponse, diff --git a/tests/api_resources/model_providers/openai/test_keys.py b/tests/api_resources/model_providers/openai/test_keys.py index b398f5cc..f0f1eda0 100644 --- a/tests/api_resources/model_providers/openai/test_keys.py +++ b/tests/api_resources/model_providers/openai/test_keys.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types.model_providers.openai import ( +from gradientai.types.model_providers.openai import ( KeyListResponse, KeyCreateResponse, KeyDeleteResponse, diff --git a/tests/api_resources/test_agents.py b/tests/api_resources/test_agents.py index 74c8cdab..2cc0e080 100644 --- a/tests/api_resources/test_agents.py +++ b/tests/api_resources/test_agents.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types import ( +from gradientai.types import ( AgentListResponse, AgentCreateResponse, AgentDeleteResponse, diff --git a/tests/api_resources/test_knowledge_bases.py b/tests/api_resources/test_knowledge_bases.py index 2132cd50..508820ce 100644 --- a/tests/api_resources/test_knowledge_bases.py +++ b/tests/api_resources/test_knowledge_bases.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types import ( +from gradientai.types import ( KnowledgeBaseListResponse, KnowledgeBaseCreateResponse, KnowledgeBaseDeleteResponse, diff --git a/tests/api_resources/test_models.py b/tests/api_resources/test_models.py index e1f3457b..afee0c1f 100644 --- a/tests/api_resources/test_models.py +++ b/tests/api_resources/test_models.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types import Model, ModelListResponse +from gradientai.types import Model, ModelListResponse base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010") diff --git a/tests/api_resources/test_regions.py b/tests/api_resources/test_regions.py index 4ed5bb27..8e25617f 100644 --- a/tests/api_resources/test_regions.py +++ b/tests/api_resources/test_regions.py @@ -7,9 +7,9 @@ import pytest +from gradientai import GradientAI, AsyncGradientAI from tests.utils import assert_matches_type -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai.types import RegionListResponse +from gradientai.types import RegionListResponse base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010") diff --git a/tests/conftest.py b/tests/conftest.py index 6048de1a..39547c5d 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -10,15 +10,15 @@ import pytest from pytest_asyncio import is_async_test -from do_gradientai import GradientAI, AsyncGradientAI, DefaultAioHttpClient -from do_gradientai._utils import is_dict +from gradientai import GradientAI, AsyncGradientAI, DefaultAioHttpClient +from gradientai._utils import is_dict if TYPE_CHECKING: from _pytest.fixtures import FixtureRequest # pyright: ignore[reportPrivateImportUsage] pytest.register_assert_rewrite("tests.utils") -logging.getLogger("do_gradientai").setLevel(logging.DEBUG) +logging.getLogger("gradientai").setLevel(logging.DEBUG) # automatically add `pytest.mark.asyncio()` to all of our async tests diff --git a/tests/test_client.py b/tests/test_client.py index 44dbc938..fc2c1325 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -21,11 +21,11 @@ from respx import MockRouter from pydantic import ValidationError -from do_gradientai import GradientAI, AsyncGradientAI, APIResponseValidationError -from do_gradientai._types import Omit -from do_gradientai._models import BaseModel, FinalRequestOptions -from do_gradientai._exceptions import APIStatusError, APITimeoutError, APIResponseValidationError -from do_gradientai._base_client import ( +from gradientai import GradientAI, AsyncGradientAI, APIResponseValidationError +from gradientai._types import Omit +from gradientai._models import BaseModel, FinalRequestOptions +from gradientai._exceptions import APIStatusError, APITimeoutError, APIResponseValidationError +from gradientai._base_client import ( DEFAULT_TIMEOUT, HTTPX_DEFAULT_TIMEOUT, BaseClient, @@ -247,10 +247,10 @@ def add_leak(leaks: list[tracemalloc.StatisticDiff], diff: tracemalloc.Statistic # to_raw_response_wrapper leaks through the @functools.wraps() decorator. # # removing the decorator fixes the leak for reasons we don't understand. - "do_gradientai/_legacy_response.py", - "do_gradientai/_response.py", + "gradientai/_legacy_response.py", + "gradientai/_response.py", # pydantic.BaseModel.model_dump || pydantic.BaseModel.dict leak memory for some reason. - "do_gradientai/_compat.py", + "gradientai/_compat.py", # Standard library leaks we don't care about. "/logging/__init__.py", ] @@ -804,7 +804,7 @@ def test_parse_retry_after_header(self, remaining_retries: int, retry_after: str calculated = client._calculate_retry_timeout(remaining_retries, options, headers) assert calculated == pytest.approx(timeout, 0.5 * 0.875) # pyright: ignore[reportUnknownMemberType] - @mock.patch("do_gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter, client: GradientAI) -> None: respx_mock.get("/v2/gen-ai/agents/uuid/versions").mock(side_effect=httpx.TimeoutException("Test timeout error")) @@ -814,7 +814,7 @@ def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter, clien assert _get_open_connections(self.client) == 0 - @mock.patch("do_gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter, client: GradientAI) -> None: respx_mock.get("/v2/gen-ai/agents/uuid/versions").mock(return_value=httpx.Response(500)) @@ -824,7 +824,7 @@ def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter, client assert _get_open_connections(self.client) == 0 @pytest.mark.parametrize("failures_before_success", [0, 2, 4]) - @mock.patch("do_gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) @pytest.mark.parametrize("failure_mode", ["status", "exception"]) def test_retries_taken( @@ -855,7 +855,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response: assert int(response.http_request.headers.get("x-stainless-retry-count")) == failures_before_success @pytest.mark.parametrize("failures_before_success", [0, 2, 4]) - @mock.patch("do_gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) def test_omit_retry_count_header( self, client: GradientAI, failures_before_success: int, respx_mock: MockRouter @@ -880,7 +880,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response: assert len(response.http_request.headers.get_list("x-stainless-retry-count")) == 0 @pytest.mark.parametrize("failures_before_success", [0, 2, 4]) - @mock.patch("do_gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) def test_overwrite_retry_count_header( self, client: GradientAI, failures_before_success: int, respx_mock: MockRouter @@ -1144,10 +1144,10 @@ def add_leak(leaks: list[tracemalloc.StatisticDiff], diff: tracemalloc.Statistic # to_raw_response_wrapper leaks through the @functools.wraps() decorator. # # removing the decorator fixes the leak for reasons we don't understand. - "do_gradientai/_legacy_response.py", - "do_gradientai/_response.py", + "gradientai/_legacy_response.py", + "gradientai/_response.py", # pydantic.BaseModel.model_dump || pydantic.BaseModel.dict leak memory for some reason. - "do_gradientai/_compat.py", + "gradientai/_compat.py", # Standard library leaks we don't care about. "/logging/__init__.py", ] @@ -1705,7 +1705,7 @@ async def test_parse_retry_after_header(self, remaining_retries: int, retry_afte calculated = client._calculate_retry_timeout(remaining_retries, options, headers) assert calculated == pytest.approx(timeout, 0.5 * 0.875) # pyright: ignore[reportUnknownMemberType] - @mock.patch("do_gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) async def test_retrying_timeout_errors_doesnt_leak( self, respx_mock: MockRouter, async_client: AsyncGradientAI @@ -1717,7 +1717,7 @@ async def test_retrying_timeout_errors_doesnt_leak( assert _get_open_connections(self.client) == 0 - @mock.patch("do_gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) async def test_retrying_status_errors_doesnt_leak( self, respx_mock: MockRouter, async_client: AsyncGradientAI @@ -1729,7 +1729,7 @@ async def test_retrying_status_errors_doesnt_leak( assert _get_open_connections(self.client) == 0 @pytest.mark.parametrize("failures_before_success", [0, 2, 4]) - @mock.patch("do_gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) @pytest.mark.asyncio @pytest.mark.parametrize("failure_mode", ["status", "exception"]) @@ -1761,7 +1761,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response: assert int(response.http_request.headers.get("x-stainless-retry-count")) == failures_before_success @pytest.mark.parametrize("failures_before_success", [0, 2, 4]) - @mock.patch("do_gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) @pytest.mark.asyncio async def test_omit_retry_count_header( @@ -1787,7 +1787,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response: assert len(response.http_request.headers.get_list("x-stainless-retry-count")) == 0 @pytest.mark.parametrize("failures_before_success", [0, 2, 4]) - @mock.patch("do_gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("gradientai._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) @pytest.mark.asyncio async def test_overwrite_retry_count_header( @@ -1823,8 +1823,8 @@ def test_get_platform(self) -> None: import nest_asyncio import threading - from do_gradientai._utils import asyncify - from do_gradientai._base_client import get_platform + from gradientai._utils import asyncify + from gradientai._base_client import get_platform async def test_main() -> None: result = await asyncify(get_platform)() diff --git a/tests/test_deepcopy.py b/tests/test_deepcopy.py index 5a98ce1b..9d1579a8 100644 --- a/tests/test_deepcopy.py +++ b/tests/test_deepcopy.py @@ -1,4 +1,4 @@ -from do_gradientai._utils import deepcopy_minimal +from gradientai._utils import deepcopy_minimal def assert_different_identities(obj1: object, obj2: object) -> None: diff --git a/tests/test_extract_files.py b/tests/test_extract_files.py index 341e65ae..2905d59c 100644 --- a/tests/test_extract_files.py +++ b/tests/test_extract_files.py @@ -4,8 +4,8 @@ import pytest -from do_gradientai._types import FileTypes -from do_gradientai._utils import extract_files +from gradientai._types import FileTypes +from gradientai._utils import extract_files def test_removes_files_from_input() -> None: diff --git a/tests/test_files.py b/tests/test_files.py index ff7914bb..4a723313 100644 --- a/tests/test_files.py +++ b/tests/test_files.py @@ -4,7 +4,7 @@ import pytest from dirty_equals import IsDict, IsList, IsBytes, IsTuple -from do_gradientai._files import to_httpx_files, async_to_httpx_files +from gradientai._files import to_httpx_files, async_to_httpx_files readme_path = Path(__file__).parent.parent.joinpath("README.md") diff --git a/tests/test_models.py b/tests/test_models.py index 575dc3af..28aff1f3 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -7,9 +7,9 @@ import pydantic from pydantic import Field -from do_gradientai._utils import PropertyInfo -from do_gradientai._compat import PYDANTIC_V2, parse_obj, model_dump, model_json -from do_gradientai._models import BaseModel, construct_type +from gradientai._utils import PropertyInfo +from gradientai._compat import PYDANTIC_V2, parse_obj, model_dump, model_json +from gradientai._models import BaseModel, construct_type class BasicModel(BaseModel): diff --git a/tests/test_qs.py b/tests/test_qs.py index c9213571..9080377b 100644 --- a/tests/test_qs.py +++ b/tests/test_qs.py @@ -4,7 +4,7 @@ import pytest -from do_gradientai._qs import Querystring, stringify +from gradientai._qs import Querystring, stringify def test_empty() -> None: diff --git a/tests/test_required_args.py b/tests/test_required_args.py index 434e9491..c4e6b9d8 100644 --- a/tests/test_required_args.py +++ b/tests/test_required_args.py @@ -2,7 +2,7 @@ import pytest -from do_gradientai._utils import required_args +from gradientai._utils import required_args def test_too_many_positional_params() -> None: diff --git a/tests/test_response.py b/tests/test_response.py index 001ce776..1a8f241e 100644 --- a/tests/test_response.py +++ b/tests/test_response.py @@ -6,8 +6,8 @@ import pytest import pydantic -from do_gradientai import BaseModel, GradientAI, AsyncGradientAI -from do_gradientai._response import ( +from gradientai import BaseModel, GradientAI, AsyncGradientAI +from gradientai._response import ( APIResponse, BaseAPIResponse, AsyncAPIResponse, @@ -15,8 +15,8 @@ AsyncBinaryAPIResponse, extract_response_type, ) -from do_gradientai._streaming import Stream -from do_gradientai._base_client import FinalRequestOptions +from gradientai._streaming import Stream +from gradientai._base_client import FinalRequestOptions class ConcreteBaseAPIResponse(APIResponse[bytes]): ... @@ -37,7 +37,7 @@ def test_extract_response_type_direct_classes() -> None: def test_extract_response_type_direct_class_missing_type_arg() -> None: with pytest.raises( RuntimeError, - match="Expected type to have a type argument at index 0 but it did not", + match="Expected type to have a type argument at index 0 but it did not", ): extract_response_type(AsyncAPIResponse) @@ -68,7 +68,7 @@ def test_response_parse_mismatched_basemodel(client: GradientAI) -> None: with pytest.raises( TypeError, - match="Pydantic models must subclass our base model type, e.g. `from do_gradientai import BaseModel`", + match="Pydantic models must subclass our base model type, e.g. `from gradientai import BaseModel`", ): response.parse(to=PydanticModel) @@ -86,7 +86,7 @@ async def test_async_response_parse_mismatched_basemodel(async_client: AsyncGrad with pytest.raises( TypeError, - match="Pydantic models must subclass our base model type, e.g. `from do_gradientai import BaseModel`", + match="Pydantic models must subclass our base model type, e.g. `from gradientai import BaseModel`", ): await response.parse(to=PydanticModel) diff --git a/tests/test_streaming.py b/tests/test_streaming.py index c1ce8e85..cdb41a77 100644 --- a/tests/test_streaming.py +++ b/tests/test_streaming.py @@ -5,8 +5,8 @@ import httpx import pytest -from do_gradientai import GradientAI, AsyncGradientAI -from do_gradientai._streaming import Stream, AsyncStream, ServerSentEvent +from gradientai import GradientAI, AsyncGradientAI +from gradientai._streaming import Stream, AsyncStream, ServerSentEvent @pytest.mark.asyncio diff --git a/tests/test_transform.py b/tests/test_transform.py index 30c06d6a..825fe048 100644 --- a/tests/test_transform.py +++ b/tests/test_transform.py @@ -8,15 +8,15 @@ import pytest -from do_gradientai._types import NOT_GIVEN, Base64FileInput -from do_gradientai._utils import ( +from gradientai._types import NOT_GIVEN, Base64FileInput +from gradientai._utils import ( PropertyInfo, transform as _transform, parse_datetime, async_transform as _async_transform, ) -from do_gradientai._compat import PYDANTIC_V2 -from do_gradientai._models import BaseModel +from gradientai._compat import PYDANTIC_V2 +from gradientai._models import BaseModel _T = TypeVar("_T") diff --git a/tests/test_utils/test_proxy.py b/tests/test_utils/test_proxy.py index 9ce2e0d3..3856b2c9 100644 --- a/tests/test_utils/test_proxy.py +++ b/tests/test_utils/test_proxy.py @@ -2,7 +2,7 @@ from typing import Any from typing_extensions import override -from do_gradientai._utils import LazyProxy +from gradientai._utils import LazyProxy class RecursiveLazyProxy(LazyProxy[Any]): diff --git a/tests/test_utils/test_typing.py b/tests/test_utils/test_typing.py index c9129fdc..66ad064f 100644 --- a/tests/test_utils/test_typing.py +++ b/tests/test_utils/test_typing.py @@ -2,7 +2,7 @@ from typing import Generic, TypeVar, cast -from do_gradientai._utils import extract_type_var_from_base +from gradientai._utils import extract_type_var_from_base _T = TypeVar("_T") _T2 = TypeVar("_T2") diff --git a/tests/utils.py b/tests/utils.py index 9def7c60..b539ed2c 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -8,8 +8,8 @@ from datetime import date, datetime from typing_extensions import Literal, get_args, get_origin, assert_type -from do_gradientai._types import Omit, NoneType -from do_gradientai._utils import ( +from gradientai._types import Omit, NoneType +from gradientai._utils import ( is_dict, is_list, is_list_type, @@ -18,8 +18,8 @@ is_annotated_type, is_type_alias_type, ) -from do_gradientai._compat import PYDANTIC_V2, field_outer_type, get_model_fields -from do_gradientai._models import BaseModel +from gradientai._compat import PYDANTIC_V2, field_outer_type, get_model_fields +from gradientai._models import BaseModel BaseModelT = TypeVar("BaseModelT", bound=BaseModel)