apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-07-02 16:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge branch 'dev' of https://github.com/geekan/MetaGPT into geekan/dev

Browse source
This commit is contained in:
莘权 马 2024-01-09 09:57:57 +08:00
commit 2bac5535f7
11 changed files with 59 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/metagpt/memory/test_brain_memory.py
View file

@ -46,7 +46,7 @@ def test_extract_info(input, tag, val):
@pytest.mark.asyncio
@pytest.mark.parametrize("llm", [LLM(provider=LLMType.OPENAI), LLM(provider=LLMType.METAGPT)])
@pytest.mark.parametrize("llm", [LLM(provider=LLMType.OPENAI)]) # , LLM(provider=LLMType.METAGPT)
async def test_memory_llm(llm):
memory = BrainMemory()
for i in range(500):

8
tests/metagpt/memory/test_longterm_memory.py
View file

@ -10,17 +10,15 @@ import os
import pytest
from metagpt.actions import UserRequirement
from metagpt.config import CONFIG
from metagpt.config2 import config
from metagpt.memory.longterm_memory import LongTermMemory
from metagpt.roles.role import RoleContext
from metagpt.schema import Message
os.environ.setdefault("OPENAI_API_KEY", config.get_openai_llm().api_key)
def test_ltm_search():
assert hasattr(CONFIG, "long_term_memory") is True
os.environ.setdefault("OPENAI_API_KEY", CONFIG.openai_api_key)
assert len(CONFIG.openai_api_key) > 20
role_id = "UTUserLtm(Product Manager)"
from metagpt.environment import Environment

2
tests/metagpt/memory/test_memory.py
View file

@ -32,7 +32,7 @@ def test_memory():
messages = memory.get_by_action(UserRequirement)
assert len(messages) == 2
messages = memory.get_by_actions([UserRequirement])
messages = memory.get_by_actions({UserRequirement})
assert len(messages) == 2
messages = memory.try_remember("test message")

4
tests/metagpt/memory/test_memory_storage.py
View file

@ -11,12 +11,12 @@ from typing import List
from metagpt.actions import UserRequirement, WritePRD
from metagpt.actions.action_node import ActionNode
from metagpt.config import CONFIG
from metagpt.config2 import config
from metagpt.const import DATA_PATH
from metagpt.memory.memory_storage import MemoryStorage
from metagpt.schema import Message
os.environ.setdefault("OPENAI_API_KEY", CONFIG.openai_api_key)
os.environ.setdefault("OPENAI_API_KEY", config.get_openai_llm().api_key)
def test_idea_message():

4
tests/metagpt/provider/test_ollama_api.py
View file

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

4
tests/metagpt/provider/test_open_llm_api.py
View file

@ -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",