Merge pull request #126 from PrefectHQ/core-version-query

Add `core_version` to server api query

changes/pr126.yaml

@@ -0,0 +1,2 @@
+enhancement:
+  - "Add the Prefect Core version to the api query - [#126](https://github.com/PrefectHQ/server/pull/126)"

src/prefect_server/graphql/query.py

@@ -1,6 +1,7 @@
 import os
 from typing import Any, Dict, List
 
+import prefect
 import prefect_server
 from prefect.engine import state
 from prefect_server.utilities import graphql
@@ -44,5 +45,6 @@ def resolve_reference(parent: Any, info):
         "backend": "SERVER",
         "mode": "normal",
         "version": os.getenv("PREFECT_SERVER_VERSION", prefect_server.__version__),
+        "core_version": os.getenv("PREFECT_CORE_VERSION", prefect.__version__),
         "release_timestamp": os.getenv("RELEASE_TIMESTAMP"),
     }

src/prefect_server/graphql/schema/schema.graphql

@@ -119,6 +119,7 @@ type api_info {
   backend: String
   mode: String
   version: String
+  core_version: String
   release_timestamp: String
 }
 

tests/graphql/queries/test_api.py

@@ -9,6 +9,7 @@
             backend
             mode
             version
+            core_version
             release_timestamp
         }
     }
@@ -25,6 +26,20 @@ async def test_mode(run_query):
     assert result.data.api.mode == "normal"
 
 
+async def test_core_version_uses_env_var(run_query, monkeypatch):
+    monkeypatch.setenv("PREFECT_CORE_VERSION", "core-version")
+    result = await run_query(query=QUERY)
+    assert result.data.api.core_version == "core-version"
+
+
+async def test_core_version_uses_dunder_if_no_env_var(run_query, monkeypatch):
+    if "PREFECT_CORE_VERSION" in os.environ:
+        monkeypatch.delenv("PREFECT_CORE_VERSION")
+
+    result = await run_query(query=QUERY)
+    assert result.data.api.core_version == prefect.__version__
+
+
 async def test_server_version_picks_up_dunder_version_if_no_server_env_var(
     run_query, monkeypatch
 ):