Skip if less than 3.8

Author: aravind-segu

tests/test_open_ai_mixin.py

@@ -1,4 +1,7 @@
 import sys
+
+import pytest
+
 from databricks.sdk.core import Config
 
 
@@ -14,19 +17,14 @@ def test_open_ai_client(monkeypatch):
     assert client.api_key == "test_token"
 
 
+@pytest.mark.skipif(sys.version_info < (3, 8), reason="Requires Python > 3.7")
 def test_langchain_open_ai_client(monkeypatch):
     from databricks.sdk import WorkspaceClient
-    print(sys.version_info)
-    print(sys.version_info <= (3,7))
-    if sys.version_info <= (3, 7):
-        with pytest.raises(ImportError):
-            w = WorkspaceClient(config=Config())
-            client = w.serving_endpoints.get_langchain_chat_open_ai_client("databricks-meta-llama-3-1-70b-instruct")
-    else:
-        monkeypatch.setenv('DATABRICKS_HOST', 'test_host')
-        monkeypatch.setenv('DATABRICKS_TOKEN', 'test_token')
-        w = WorkspaceClient(config=Config())
-        client = w.serving_endpoints.get_langchain_chat_open_ai_client("databricks-meta-llama-3-1-70b-instruct")
-
-        assert client.openai_api_base == "https://test_host/serving-endpoints"
-        assert client.model_name == "databricks-meta-llama-3-1-70b-instruct"
+
+    monkeypatch.setenv('DATABRICKS_HOST', 'test_host')
+    monkeypatch.setenv('DATABRICKS_TOKEN', 'test_token')
+    w = WorkspaceClient(config=Config())
+    client = w.serving_endpoints.get_langchain_chat_open_ai_client("databricks-meta-llama-3-1-70b-instruct")
+
+    assert client.openai_api_base == "https://test_host/serving-endpoints"
+    assert client.model_name == "databricks-meta-llama-3-1-70b-instruct"