chore: disable introspection and add additional fields (#366)

Paperspace · Sep 19, 2021 · 1e54b5c · 1e54b5c
1 parent 1517ac6
commit 1e54b5c
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 1 deletion.
diff --git a/gradient/api_sdk/graphql.py b/gradient/api_sdk/graphql.py
@@ -10,4 +10,4 @@ def graphql_client(api_key=config.PAPERSPACE_API_KEY):
 	}
 	transport = RequestsHTTPTransport(headers=headers, url=config.API_HOST, verify=True, retries=3)
 
-	return Client(transport=transport, fetch_schema_from_transport=True)
+	return Client(transport=transport)
diff --git a/gradient/api_sdk/repositories/gradient_deployments.py b/gradient/api_sdk/repositories/gradient_deployments.py
@@ -69,6 +69,14 @@ def get_deployment(id, first=100):
 				deploymentSpecs(first: $first) {
 					nodes {
 						id
+						data {
+							image
+							port
+							resources {
+								instanceType
+								replicas
+							}
+						}
 						endpointUrl
 						actor {
 							avatarUrl
@@ -134,6 +142,19 @@ def list_deployments(first=100):
 					deploymentSpecs(first: $first) {
 						nodes {
 							id
+							data {
+								image
+								port
+								resources {
+									instanceType
+									replicas
+								}
+							}
+							endpointUrl
+							actor {
+								avatarUrl
+								fullName
+							}
 							deploymentRuns(first: $first) {
 								nodes {
 									id

diff --git a/gradient/cli/gradient_deployments.py b/gradient/cli/gradient_deployments.py
@@ -11,6 +11,7 @@
 from gradient.cli.cli import cli
 from gradient.cli.common import  api_key_option, ClickGroup
 from gradient.commands.helpers import print_table, formatted_graphql
+from gradient.exceptions import ApplicationError
 
 
 logger = logging.getLogger(__name__)