diff --git a/metagpt/context.py b/metagpt/context.py index 74f7b133d..4083a1696 100644 --- a/metagpt/context.py +++ b/metagpt/context.py @@ -144,7 +144,7 @@ class ContextMixin(BaseModel): """Role context: role context > context""" if self._context: return self._context - return context + return CONTEXT @context.setter def context(self, context: Context): @@ -153,4 +153,4 @@ class ContextMixin(BaseModel): # Global context, not in Env -context = Context() +CONTEXT = Context() diff --git a/metagpt/llm.py b/metagpt/llm.py index aff72d3c5..d393738bb 100644 --- a/metagpt/llm.py +++ b/metagpt/llm.py @@ -9,11 +9,11 @@ from typing import Optional from metagpt.configs.llm_config import LLMType -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.provider.base_llm import BaseLLM def LLM(name: Optional[str] = None, provider: LLMType = LLMType.OPENAI) -> BaseLLM: """get the default llm provider if name is None""" # context.use_llm(name=name, provider=provider) - return context.llm(name=name, provider=provider) + return CONTEXT.llm(name=name, provider=provider) diff --git a/metagpt/roles/assistant.py b/metagpt/roles/assistant.py index 90a33ad6a..8939094ed 100644 --- a/metagpt/roles/assistant.py +++ b/metagpt/roles/assistant.py @@ -22,7 +22,7 @@ from pydantic import Field from metagpt.actions.skill_action import ArgumentsParingAction, SkillAction from metagpt.actions.talk_action import TalkAction -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.learn.skill_loader import SkillsDeclaration from metagpt.logs import logger from metagpt.memory.brain_memory import BrainMemory @@ -48,7 +48,7 @@ class Assistant(Role): def __init__(self, **kwargs): super().__init__(**kwargs) - self.constraints = self.constraints.format(language=kwargs.get("language") or context.kwargs.language) + self.constraints = self.constraints.format(language=kwargs.get("language") or CONTEXT.kwargs.language) async def think(self) -> bool: """Everything will be done part by part.""" @@ -56,7 +56,7 @@ class Assistant(Role): if not last_talk: return False if not self.skills: - skill_path = Path(context.kwargs.SKILL_PATH) if context.kwargs.SKILL_PATH else None + skill_path = Path(CONTEXT.kwargs.SKILL_PATH) if CONTEXT.kwargs.SKILL_PATH else None self.skills = await SkillsDeclaration.load(skill_yaml_file_name=skill_path) prompt = "" diff --git a/tests/conftest.py b/tests/conftest.py index fab1fa198..faa2d92e9 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -16,7 +16,7 @@ import uuid import pytest from metagpt.const import DEFAULT_WORKSPACE_ROOT, TEST_DATA_PATH -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.llm import LLM from metagpt.logs import logger from metagpt.utils.git_repository import GitRepository @@ -141,12 +141,12 @@ def loguru_caplog(caplog): # init & dispose git repo @pytest.fixture(scope="function", autouse=True) def setup_and_teardown_git_repo(request): - context.git_repo = GitRepository(local_path=DEFAULT_WORKSPACE_ROOT / f"unittest/{uuid.uuid4().hex}") - context.config.git_reinit = True + CONTEXT.git_repo = GitRepository(local_path=DEFAULT_WORKSPACE_ROOT / f"unittest/{uuid.uuid4().hex}") + CONTEXT.config.git_reinit = True # Destroy git repo at the end of the test session. def fin(): - context.git_repo.delete_repository() + CONTEXT.git_repo.delete_repository() # Register the function for destroying the environment. request.addfinalizer(fin) diff --git a/tests/metagpt/actions/test_debug_error.py b/tests/metagpt/actions/test_debug_error.py index ff9e9cd81..922aa8613 100644 --- a/tests/metagpt/actions/test_debug_error.py +++ b/tests/metagpt/actions/test_debug_error.py @@ -12,7 +12,7 @@ import pytest from metagpt.actions.debug_error import DebugError from metagpt.const import TEST_CODES_FILE_REPO, TEST_OUTPUTS_FILE_REPO -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.schema import RunCodeContext, RunCodeResult CODE_CONTENT = ''' @@ -117,7 +117,7 @@ if __name__ == '__main__': @pytest.mark.asyncio async def test_debug_error(): - context.src_workspace = context.git_repo.workdir / uuid.uuid4().hex + CONTEXT.src_workspace = CONTEXT.git_repo.workdir / uuid.uuid4().hex ctx = RunCodeContext( code_filename="player.py", test_filename="test_player.py", @@ -125,8 +125,8 @@ async def test_debug_error(): output_filename="output.log", ) - repo = context.file_repo - await repo.save_file(filename=ctx.code_filename, content=CODE_CONTENT, relative_path=context.src_workspace) + repo = CONTEXT.file_repo + await repo.save_file(filename=ctx.code_filename, content=CODE_CONTENT, relative_path=CONTEXT.src_workspace) await repo.save_file(filename=ctx.test_filename, content=TEST_CONTENT, relative_path=TEST_CODES_FILE_REPO) output_data = RunCodeResult( stdout=";", diff --git a/tests/metagpt/actions/test_design_api.py b/tests/metagpt/actions/test_design_api.py index 88cb612fc..027f7ca20 100644 --- a/tests/metagpt/actions/test_design_api.py +++ b/tests/metagpt/actions/test_design_api.py @@ -10,7 +10,7 @@ import pytest from metagpt.actions.design_api import WriteDesign from metagpt.const import PRDS_FILE_REPO -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.logs import logger from metagpt.schema import Message @@ -18,7 +18,7 @@ from metagpt.schema import Message @pytest.mark.asyncio async def test_design_api(): inputs = ["我们需要一个音乐播放器,它应该有播放、暂停、上一曲、下一曲等功能。"] # PRD_SAMPLE - repo = context.file_repo + repo = CONTEXT.file_repo for prd in inputs: await repo.save_file("new_prd.txt", content=prd, relative_path=PRDS_FILE_REPO) diff --git a/tests/metagpt/actions/test_prepare_documents.py b/tests/metagpt/actions/test_prepare_documents.py index a67f89874..fde971f3c 100644 --- a/tests/metagpt/actions/test_prepare_documents.py +++ b/tests/metagpt/actions/test_prepare_documents.py @@ -10,7 +10,7 @@ import pytest from metagpt.actions.prepare_documents import PrepareDocuments from metagpt.const import DOCS_FILE_REPO, REQUIREMENT_FILENAME -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.schema import Message @@ -18,12 +18,12 @@ from metagpt.schema import Message async def test_prepare_documents(): msg = Message(content="New user requirements balabala...") - if context.git_repo: - context.git_repo.delete_repository() - context.git_repo = None + if CONTEXT.git_repo: + CONTEXT.git_repo.delete_repository() + CONTEXT.git_repo = None - await PrepareDocuments(g_context=context).run(with_messages=[msg]) - assert context.git_repo - doc = await context.file_repo.get_file(filename=REQUIREMENT_FILENAME, relative_path=DOCS_FILE_REPO) + await PrepareDocuments(g_context=CONTEXT).run(with_messages=[msg]) + assert CONTEXT.git_repo + doc = await CONTEXT.file_repo.get_file(filename=REQUIREMENT_FILENAME, relative_path=DOCS_FILE_REPO) assert doc assert doc.content == msg.content diff --git a/tests/metagpt/actions/test_project_management.py b/tests/metagpt/actions/test_project_management.py index a462319b8..1eadb49fb 100644 --- a/tests/metagpt/actions/test_project_management.py +++ b/tests/metagpt/actions/test_project_management.py @@ -10,7 +10,7 @@ import pytest from metagpt.actions.project_management import WriteTasks from metagpt.const import PRDS_FILE_REPO, SYSTEM_DESIGN_FILE_REPO -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.logs import logger from metagpt.schema import Message from tests.metagpt.actions.mock_json import DESIGN, PRD @@ -18,9 +18,9 @@ from tests.metagpt.actions.mock_json import DESIGN, PRD @pytest.mark.asyncio async def test_design_api(): - await context.file_repo.save_file("1.txt", content=str(PRD), relative_path=PRDS_FILE_REPO) - await context.file_repo.save_file("1.txt", content=str(DESIGN), relative_path=SYSTEM_DESIGN_FILE_REPO) - logger.info(context.git_repo) + await CONTEXT.file_repo.save_file("1.txt", content=str(PRD), relative_path=PRDS_FILE_REPO) + await CONTEXT.file_repo.save_file("1.txt", content=str(DESIGN), relative_path=SYSTEM_DESIGN_FILE_REPO) + logger.info(CONTEXT.git_repo) action = WriteTasks() diff --git a/tests/metagpt/actions/test_summarize_code.py b/tests/metagpt/actions/test_summarize_code.py index 1c14d256d..2f7b5c61d 100644 --- a/tests/metagpt/actions/test_summarize_code.py +++ b/tests/metagpt/actions/test_summarize_code.py @@ -11,7 +11,7 @@ import pytest from metagpt.actions.summarize_code import SummarizeCode from metagpt.config import CONFIG from metagpt.const import SYSTEM_DESIGN_FILE_REPO, TASK_FILE_REPO -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.logs import logger from metagpt.schema import CodeSummarizeContext @@ -178,15 +178,15 @@ class Snake: @pytest.mark.asyncio async def test_summarize_code(): - context.src_workspace = context.git_repo.workdir / "src" - await context.file_repo.save_file(filename="1.json", relative_path=SYSTEM_DESIGN_FILE_REPO, content=DESIGN_CONTENT) - await context.file_repo.save_file(filename="1.json", relative_path=TASK_FILE_REPO, content=TASK_CONTENT) - await context.file_repo.save_file(filename="food.py", relative_path=CONFIG.src_workspace, content=FOOD_PY) - await context.file_repo.save_file(filename="game.py", relative_path=CONFIG.src_workspace, content=GAME_PY) - await context.file_repo.save_file(filename="main.py", relative_path=CONFIG.src_workspace, content=MAIN_PY) - await context.file_repo.save_file(filename="snake.py", relative_path=CONFIG.src_workspace, content=SNAKE_PY) + CONTEXT.src_workspace = CONTEXT.git_repo.workdir / "src" + await CONTEXT.file_repo.save_file(filename="1.json", relative_path=SYSTEM_DESIGN_FILE_REPO, content=DESIGN_CONTENT) + await CONTEXT.file_repo.save_file(filename="1.json", relative_path=TASK_FILE_REPO, content=TASK_CONTENT) + await CONTEXT.file_repo.save_file(filename="food.py", relative_path=CONFIG.src_workspace, content=FOOD_PY) + await CONTEXT.file_repo.save_file(filename="game.py", relative_path=CONFIG.src_workspace, content=GAME_PY) + await CONTEXT.file_repo.save_file(filename="main.py", relative_path=CONFIG.src_workspace, content=MAIN_PY) + await CONTEXT.file_repo.save_file(filename="snake.py", relative_path=CONFIG.src_workspace, content=SNAKE_PY) - src_file_repo = context.git_repo.new_file_repository(relative_path=CONFIG.src_workspace) + src_file_repo = CONTEXT.git_repo.new_file_repository(relative_path=CONFIG.src_workspace) all_files = src_file_repo.all_files ctx = CodeSummarizeContext(design_filename="1.json", task_filename="1.json", codes_filenames=all_files) action = SummarizeCode(context=ctx) diff --git a/tests/metagpt/actions/test_write_code.py b/tests/metagpt/actions/test_write_code.py index 2a7b8e696..cfc5863f4 100644 --- a/tests/metagpt/actions/test_write_code.py +++ b/tests/metagpt/actions/test_write_code.py @@ -18,7 +18,7 @@ from metagpt.const import ( TASK_FILE_REPO, TEST_OUTPUTS_FILE_REPO, ) -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.logs import logger from metagpt.provider.openai_api import OpenAILLM as LLM from metagpt.schema import CodingContext, Document @@ -53,35 +53,35 @@ async def test_write_code_directly(): @pytest.mark.asyncio async def test_write_code_deps(): # Prerequisites - context.src_workspace = context.git_repo.workdir / "snake1/snake1" + CONTEXT.src_workspace = CONTEXT.git_repo.workdir / "snake1/snake1" demo_path = Path(__file__).parent / "../../data/demo_project" - await context.file_repo.save_file( + await CONTEXT.file_repo.save_file( filename="test_game.py.json", content=await aread(str(demo_path / "test_game.py.json")), relative_path=TEST_OUTPUTS_FILE_REPO, ) - await context.file_repo.save_file( + await CONTEXT.file_repo.save_file( filename="20231221155954.json", content=await aread(str(demo_path / "code_summaries.json")), relative_path=CODE_SUMMARIES_FILE_REPO, ) - await context.file_repo.save_file( + await CONTEXT.file_repo.save_file( filename="20231221155954.json", content=await aread(str(demo_path / "system_design.json")), relative_path=SYSTEM_DESIGN_FILE_REPO, ) - await context.file_repo.save_file( + await CONTEXT.file_repo.save_file( filename="20231221155954.json", content=await aread(str(demo_path / "tasks.json")), relative_path=TASK_FILE_REPO ) - await context.file_repo.save_file( - filename="main.py", content='if __name__ == "__main__":\nmain()', relative_path=context.src_workspace + await CONTEXT.file_repo.save_file( + filename="main.py", content='if __name__ == "__main__":\nmain()', relative_path=CONTEXT.src_workspace ) ccontext = CodingContext( filename="game.py", - design_doc=await context.file_repo.get_file( + design_doc=await CONTEXT.file_repo.get_file( filename="20231221155954.json", relative_path=SYSTEM_DESIGN_FILE_REPO ), - task_doc=await context.file_repo.get_file(filename="20231221155954.json", relative_path=TASK_FILE_REPO), + task_doc=await CONTEXT.file_repo.get_file(filename="20231221155954.json", relative_path=TASK_FILE_REPO), code_doc=Document(filename="game.py", content="", root_path="snake1"), ) coding_doc = Document(root_path="snake1", filename="game.py", content=ccontext.json()) diff --git a/tests/metagpt/actions/test_write_prd.py b/tests/metagpt/actions/test_write_prd.py index 1f92c079b..faa5b77a4 100644 --- a/tests/metagpt/actions/test_write_prd.py +++ b/tests/metagpt/actions/test_write_prd.py @@ -10,7 +10,7 @@ import pytest from metagpt.actions import UserRequirement, WritePRD from metagpt.const import DOCS_FILE_REPO, PRDS_FILE_REPO, REQUIREMENT_FILENAME -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.logs import logger from metagpt.roles.product_manager import ProductManager from metagpt.roles.role import RoleReactMode @@ -33,7 +33,7 @@ async def test_write_prd(new_filename): # Assert the prd is not None or empty assert prd is not None assert prd.content != "" - assert context.git_repo.new_file_repository(relative_path=PRDS_FILE_REPO).changed_files + assert CONTEXT.git_repo.new_file_repository(relative_path=PRDS_FILE_REPO).changed_files if __name__ == "__main__": diff --git a/tests/metagpt/roles/test_architect.py b/tests/metagpt/roles/test_architect.py index 69afbcfe1..f9d6606ac 100644 --- a/tests/metagpt/roles/test_architect.py +++ b/tests/metagpt/roles/test_architect.py @@ -13,7 +13,7 @@ import pytest from metagpt.actions import WriteDesign, WritePRD from metagpt.const import PRDS_FILE_REPO -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.logs import logger from metagpt.roles import Architect from metagpt.schema import Message @@ -25,7 +25,7 @@ from tests.metagpt.roles.mock import MockMessages async def test_architect(): # Prerequisites filename = uuid.uuid4().hex + ".json" - await awrite(context.git_repo.workdir / PRDS_FILE_REPO / filename, data=MockMessages.prd.content) + await awrite(CONTEXT.git_repo.workdir / PRDS_FILE_REPO / filename, data=MockMessages.prd.content) role = Architect() rsp = await role.run(with_message=Message(content="", cause_by=WritePRD)) diff --git a/tests/metagpt/roles/test_assistant.py b/tests/metagpt/roles/test_assistant.py index 8797ba7f1..4ef44d77a 100644 --- a/tests/metagpt/roles/test_assistant.py +++ b/tests/metagpt/roles/test_assistant.py @@ -12,7 +12,7 @@ from pydantic import BaseModel from metagpt.actions.skill_action import SkillAction from metagpt.actions.talk_action import TalkAction -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.memory.brain_memory import BrainMemory from metagpt.roles.assistant import Assistant from metagpt.schema import Message @@ -21,7 +21,7 @@ from metagpt.utils.common import any_to_str @pytest.mark.asyncio async def test_run(): - context.kwargs.language = "Chinese" + CONTEXT.kwargs.language = "Chinese" class Input(BaseModel): memory: BrainMemory @@ -65,7 +65,7 @@ async def test_run(): "cause_by": any_to_str(SkillAction), }, ] - context.kwargs.agent_skills = [ + CONTEXT.kwargs.agent_skills = [ {"id": 1, "name": "text_to_speech", "type": "builtin", "config": {}, "enabled": True}, {"id": 2, "name": "text_to_image", "type": "builtin", "config": {}, "enabled": True}, {"id": 3, "name": "ai_call", "type": "builtin", "config": {}, "enabled": True}, @@ -77,8 +77,8 @@ async def test_run(): for i in inputs: seed = Input(**i) - context.kwargs.language = seed.language - context.kwargs.agent_description = seed.agent_description + CONTEXT.kwargs.language = seed.language + CONTEXT.kwargs.agent_description = seed.agent_description role = Assistant(language="Chinese") role.memory = seed.memory # Restore historical conversation content. while True: diff --git a/tests/metagpt/roles/test_engineer.py b/tests/metagpt/roles/test_engineer.py index b35321a1b..710e74b8f 100644 --- a/tests/metagpt/roles/test_engineer.py +++ b/tests/metagpt/roles/test_engineer.py @@ -19,7 +19,7 @@ from metagpt.const import ( SYSTEM_DESIGN_FILE_REPO, TASK_FILE_REPO, ) -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.logs import logger from metagpt.roles.engineer import Engineer from metagpt.schema import CodingContext, Message @@ -32,19 +32,19 @@ from tests.metagpt.roles.mock import STRS_FOR_PARSING, TASKS, MockMessages async def test_engineer(): # Prerequisites rqno = "20231221155954.json" - await context.file_repo.save_file(REQUIREMENT_FILENAME, content=MockMessages.req.content) - await context.file_repo.save_file(rqno, relative_path=PRDS_FILE_REPO, content=MockMessages.prd.content) - await context.file_repo.save_file( + await CONTEXT.file_repo.save_file(REQUIREMENT_FILENAME, content=MockMessages.req.content) + await CONTEXT.file_repo.save_file(rqno, relative_path=PRDS_FILE_REPO, content=MockMessages.prd.content) + await CONTEXT.file_repo.save_file( rqno, relative_path=SYSTEM_DESIGN_FILE_REPO, content=MockMessages.system_design.content ) - await context.file_repo.save_file(rqno, relative_path=TASK_FILE_REPO, content=MockMessages.json_tasks.content) + await CONTEXT.file_repo.save_file(rqno, relative_path=TASK_FILE_REPO, content=MockMessages.json_tasks.content) engineer = Engineer() rsp = await engineer.run(Message(content="", cause_by=WriteTasks)) logger.info(rsp) assert rsp.cause_by == any_to_str(WriteCode) - src_file_repo = context.git_repo.new_file_repository(context.src_workspace) + src_file_repo = CONTEXT.git_repo.new_file_repository(CONTEXT.src_workspace) assert src_file_repo.changed_files @@ -116,19 +116,19 @@ async def test_new_coding_context(): # Prerequisites demo_path = Path(__file__).parent / "../../data/demo_project" deps = json.loads(await aread(demo_path / "dependencies.json")) - dependency = await context.git_repo.get_dependency() + dependency = await CONTEXT.git_repo.get_dependency() for k, v in deps.items(): await dependency.update(k, set(v)) data = await aread(demo_path / "system_design.json") rqno = "20231221155954.json" - await awrite(context.git_repo.workdir / SYSTEM_DESIGN_FILE_REPO / rqno, data) + await awrite(CONTEXT.git_repo.workdir / SYSTEM_DESIGN_FILE_REPO / rqno, data) data = await aread(demo_path / "tasks.json") - await awrite(context.git_repo.workdir / TASK_FILE_REPO / rqno, data) + await awrite(CONTEXT.git_repo.workdir / TASK_FILE_REPO / rqno, data) - context.src_workspace = Path(context.git_repo.workdir) / "game_2048" - src_file_repo = context.git_repo.new_file_repository(relative_path=context.src_workspace) - task_file_repo = context.git_repo.new_file_repository(relative_path=TASK_FILE_REPO) - design_file_repo = context.git_repo.new_file_repository(relative_path=SYSTEM_DESIGN_FILE_REPO) + CONTEXT.src_workspace = Path(CONTEXT.git_repo.workdir) / "game_2048" + src_file_repo = CONTEXT.git_repo.new_file_repository(relative_path=CONTEXT.src_workspace) + task_file_repo = CONTEXT.git_repo.new_file_repository(relative_path=TASK_FILE_REPO) + design_file_repo = CONTEXT.git_repo.new_file_repository(relative_path=SYSTEM_DESIGN_FILE_REPO) filename = "game.py" ctx_doc = await Engineer._new_coding_doc( @@ -149,8 +149,8 @@ async def test_new_coding_context(): assert ctx.task_doc.content assert ctx.code_doc - context.git_repo.add_change({f"{TASK_FILE_REPO}/{rqno}": ChangeType.UNTRACTED}) - context.git_repo.commit("mock env") + CONTEXT.git_repo.add_change({f"{TASK_FILE_REPO}/{rqno}": ChangeType.UNTRACTED}) + CONTEXT.git_repo.commit("mock env") await src_file_repo.save(filename=filename, content="content") role = Engineer() assert not role.code_todos diff --git a/tests/metagpt/roles/test_qa_engineer.py b/tests/metagpt/roles/test_qa_engineer.py index 825fe58a3..c51642e6a 100644 --- a/tests/metagpt/roles/test_qa_engineer.py +++ b/tests/metagpt/roles/test_qa_engineer.py @@ -13,7 +13,7 @@ from pydantic import Field from metagpt.actions import DebugError, RunCode, WriteTest from metagpt.actions.summarize_code import SummarizeCode -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.environment import Environment from metagpt.roles import QaEngineer from metagpt.schema import Message @@ -23,10 +23,10 @@ from metagpt.utils.common import any_to_str, aread, awrite async def test_qa(): # Prerequisites demo_path = Path(__file__).parent / "../../data/demo_project" - context.src_workspace = Path(context.git_repo.workdir) / "qa/game_2048" + CONTEXT.src_workspace = Path(CONTEXT.git_repo.workdir) / "qa/game_2048" data = await aread(filename=demo_path / "game.py", encoding="utf-8") - await awrite(filename=context.src_workspace / "game.py", data=data, encoding="utf-8") - await awrite(filename=Path(context.git_repo.workdir) / "requirements.txt", data="") + await awrite(filename=CONTEXT.src_workspace / "game.py", data=data, encoding="utf-8") + await awrite(filename=Path(CONTEXT.git_repo.workdir) / "requirements.txt", data="") class MockEnv(Environment): msgs: List[Message] = Field(default_factory=list) diff --git a/tests/metagpt/roles/test_teacher.py b/tests/metagpt/roles/test_teacher.py index ff2139929..8bd37f482 100644 --- a/tests/metagpt/roles/test_teacher.py +++ b/tests/metagpt/roles/test_teacher.py @@ -10,7 +10,7 @@ from typing import Dict, Optional import pytest from pydantic import BaseModel -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.roles.teacher import Teacher from metagpt.schema import Message @@ -97,8 +97,8 @@ async def test_new_file_name(): @pytest.mark.asyncio async def test_run(): - context.kwargs.language = "Chinese" - context.kwargs.teaching_language = "English" + CONTEXT.kwargs.language = "Chinese" + CONTEXT.kwargs.teaching_language = "English" lesson = """ UNIT 1 Making New Friends TOPIC 1 Welcome to China! diff --git a/tests/metagpt/test_context.py b/tests/metagpt/test_context.py index 2d52325bc..f1c9da4e7 100644 --- a/tests/metagpt/test_context.py +++ b/tests/metagpt/test_context.py @@ -6,7 +6,7 @@ @File : test_context.py """ from metagpt.configs.llm_config import LLMType -from metagpt.context import AttrDict, Context, context +from metagpt.context import CONTEXT, AttrDict, Context def test_attr_dict_1(): @@ -52,11 +52,11 @@ def test_context_1(): def test_context_2(): - llm = context.config.get_openai_llm() + llm = CONTEXT.config.get_openai_llm() assert llm is not None assert llm.api_type == LLMType.OPENAI - kwargs = context.kwargs + kwargs = CONTEXT.kwargs assert kwargs is not None kwargs.test_key = "test_value" diff --git a/tests/metagpt/test_environment.py b/tests/metagpt/test_environment.py index d7d8d990a..49fd8a5fc 100644 --- a/tests/metagpt/test_environment.py +++ b/tests/metagpt/test_environment.py @@ -13,7 +13,7 @@ from pathlib import Path import pytest from metagpt.actions import UserRequirement -from metagpt.context import context +from metagpt.context import CONTEXT from metagpt.environment import Environment from metagpt.logs import logger from metagpt.roles import Architect, ProductManager, Role @@ -46,9 +46,9 @@ def test_get_roles(env: Environment): @pytest.mark.asyncio async def test_publish_and_process_message(env: Environment): - if context.git_repo: - context.git_repo.delete_repository() - context.git_repo = None + if CONTEXT.git_repo: + CONTEXT.git_repo.delete_repository() + CONTEXT.git_repo = None product_manager = ProductManager(name="Alice", profile="Product Manager", goal="做AI Native产品", constraints="资源有限") architect = Architect(