Merge pull request #4 from Preemo-Inc/atmiguel-minor-fixes

Upgrading gradientai to 1.0.0.dev4

Author: atmiguel

.gitignore

@@ -158,3 +158,5 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+
+.tool-versions

README.md

@@ -7,18 +7,17 @@ Quick example of how to use Gradient API with Python.
 The code relies on environment variables. They can be injected through `dotenv` by creating `.env` file in the current directory:
 
 ```
-GRADIENT_API_KEY=...
+GRADIENT_ACCESS_TOKEN=...
 GRADIENT_WORKSPACE_ID=...
-GRADIENT_API_URL=https://api-staging2.gradient.ai/api
 ```
 
 ## Usage
 
 1. Clone this repo
-2. Run `poetry shell`
-3. Run `python gradient_sdk_python_example/list.py`
+2. Run `poetry install`
+3. Run `poetry shell`
+4. Run `python gradient_sdk_python_example/list.py`
 
 ## Example
 
 ![Alt Text](example.gif)
-

gradient_sdk_python_example/list.py

@@ -5,11 +5,13 @@
 load_dotenv()
 
 configuration = gradientai.Configuration(
-    host=os.getenv("GRADIENT_API_URL"),
     access_token=os.getenv("GRADIENT_ACCESS_TOKEN")
 )
 
-models_api = gradientai.ModelsApi(gradientai.ApiClient(configuration))
-model_list = models_api.list_models(only_base="false", x_preemo_workspace_id=os.getenv("GRADIENT_WORKSPACE_ID"))
+with gradientai.ApiClient(configuration) as api_client:
+    models_api = gradientai.ModelsApi(api_client)
+    model_list = models_api.list_models(
+        x_gradient_workspace_id=os.getenv("GRADIENT_WORKSPACE_ID")
+    )
 
-print(model_list.models[0].actual_instance)
+print(model_list.models[0].actual_instance)

poetry.lock

@@ -7,11 +7,10 @@ readme = "README.md"
 packages = [{include = "gradient_sdk_python_example"}]
 
 [tool.poetry.dependencies]
-python = "^3.11"
-gradientai = "^1.0.0.dev2"
+python = "^3.10"
+gradientai = "^1.0.0.dev4"
 python-dotenv = "^1.0.0"
 
-
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"