Typing test (#24651)

This commit is contained in:
Asuka Minato 2025-08-28 10:36:39 +09:00 • committed by GitHub
commit 7995ff1410
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 23 additions and 20 deletions

View file

@ -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)