apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-26 15:49:42 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

modify detail

Browse source
This commit is contained in:
garylin2099 2024-07-31 23:22:49 +08:00
parent a6104f3931
commit 1f73051571
2 changed files with 26 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/metagpt/provider/test_base_llm.py
View file

@ -41,7 +41,6 @@ class MockBaseLLM(BaseLLM):
return default_resp_cont
@pytest.mark.skip
def test_base_llm():
message = Message(role="user", content="hello")
assert "role" in message.to_dict()
@ -93,7 +92,6 @@ def test_base_llm():
# assert resp == default_resp_cont
@pytest.mark.skip
@pytest.mark.asyncio
async def test_async_base_llm():
base_llm = MockBaseLLM()
@ -147,3 +145,19 @@ def test_compress_messages_long(compress_type):
assert len(compressed) < len(messages)
assert compressed[0]["role"] == "system" and compressed[1]["role"] == "system"
assert compressed[2]["role"] != "system"
@pytest.mark.parametrize(
"compress_type", ["post_cut_by_msg", "post_cut_by_token", "pre_cut_by_msg", "pre_cut_by_token"]
)
def test_compress_messages_long_no_sys_msg(compress_type):
base_llm = MockBaseLLM()
base_llm.config.model = "test_llm"
max_token_limit = 100
messages = [{"role": "user", "content": "1" * 10000}]
compressed = base_llm.compress_messages(messages, compress_type=compress_type, max_token=max_token_limit)
print(compressed)
assert compressed
assert len(compressed[0]["content"]) < len(messages[0]["content"])