diff --git a/model/provider/openai_official.py b/model/provider/openai_official.py index c428af81f..981f33a6b 100644 --- a/model/provider/openai_official.py +++ b/model/provider/openai_official.py @@ -296,7 +296,7 @@ class ProviderOpenAIOfficial(Provider): extra_conf: Dict = None, **kwargs ) -> str: - if os.environ.get("TEST_LLM", "off") != "on": + if os.environ.get("TEST_LLM", "off") != "on" and os.environ.get("TEST_MODE", "off") == "on": return "这是一个测试消息。" super().accu_model_stat()