model rebuild for openapi tests

microsoft · Dec 19, 2024 · b6f216e · b6f216e
1 parent e1e9e74
commit b6f216e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/python/tests/unit/connectors/openapi_plugin/test_sk_openapi.py b/python/tests/unit/connectors/openapi_plugin/test_sk_openapi.py
@@ -733,6 +733,10 @@ async def dummy_auth_callback(**kwargs):
 
 @pytest.fixture
 def openapi_runner_with_predicate_callback():
+    from semantic_kernel.connectors.openapi_plugin import OperationSelectionPredicateContext  # noqa: F401
+
+    OpenAPIFunctionExecutionParameters.model_rebuild()
+
     # Define a dummy predicate callback
     def predicate_callback(context):
         # Skip operations with DELETE method or containing 'internal' in the path
@@ -809,6 +813,10 @@ async def test_run_operation_with_error(mock_request, openapi_runner):
 
 
 def test_invalid_server_url_override():
+    from semantic_kernel.connectors.openapi_plugin import OperationSelectionPredicateContext  # noqa: F401
+
+    OpenAPIFunctionExecutionParameters.model_rebuild()
+
     with pytest.raises(ValueError, match="Invalid server_url_override: invalid_url"):
         params = OpenAPIFunctionExecutionParameters(server_url_override="invalid_url")
         params.model_post_init(None)