Typing test (#24651)

api/tests/integration_tests/model_runtime/__mock/plugin_daemon.py

@@ -3,15 +3,12 @@ from collections.abc import Callable
 
 import pytest
 
-# import monkeypatch
-from _pytest.monkeypatch import MonkeyPatch
-
 from core.plugin.impl.model import PluginModelClient
 from tests.integration_tests.model_runtime.__mock.plugin_model import MockModelClass
 
 
 def mock_plugin_daemon(
-    monkeypatch: MonkeyPatch,
+    monkeypatch: pytest.MonkeyPatch,
 ) -> Callable[[], None]:
     """
     mock openai module
@@ -34,7 +31,7 @@ MOCK = os.getenv("MOCK_SWITCH", "false").lower() == "true"
 
 
 @pytest.fixture
-def setup_model_mock(monkeypatch):
+def setup_model_mock(monkeypatch: pytest.MonkeyPatch):
     if MOCK:
         unpatch = mock_plugin_daemon(monkeypatch)