diff --git a/tests/metagpt/provider/test_ollama_api.py b/tests/metagpt/provider/test_ollama_api.py index 41f02bf2c..5d942598b 100644 --- a/tests/metagpt/provider/test_ollama_api.py +++ b/tests/metagpt/provider/test_ollama_api.py @@ -7,7 +7,6 @@ from typing import Any, Tuple import pytest -from metagpt.config import CONFIG from metagpt.provider.ollama_api import OllamaLLM from tests.metagpt.provider.mock_llm_config import mock_llm_config @@ -17,9 +16,6 @@ messages = [{"role": "user", "content": prompt_msg}] resp_content = "I'm ollama" default_resp = {"message": {"role": "assistant", "content": resp_content}} -CONFIG.ollama_api_base = "http://xxx" -CONFIG.max_budget = 10 - async def mock_ollama_arequest(self, stream: bool = False, **kwargs) -> Tuple[Any, Any, bool]: if stream: diff --git a/tests/metagpt/provider/test_open_llm_api.py b/tests/metagpt/provider/test_open_llm_api.py index f74bc9c49..fc7b510cc 100644 --- a/tests/metagpt/provider/test_open_llm_api.py +++ b/tests/metagpt/provider/test_open_llm_api.py @@ -13,14 +13,10 @@ from openai.types.chat.chat_completion_chunk import Choice as AChoice from openai.types.chat.chat_completion_chunk import ChoiceDelta from openai.types.completion_usage import CompletionUsage -from metagpt.config import CONFIG from metagpt.provider.open_llm_api import OpenLLM from metagpt.utils.cost_manager import Costs from tests.metagpt.provider.mock_llm_config import mock_llm_config -CONFIG.max_budget = 10 -CONFIG.calc_usage = True - resp_content = "I'm llama2" default_resp = ChatCompletion( id="cmpl-a6652c1bb181caae8dd19ad8",