opensource/MetaGPT
3
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-09-01 15:55:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'feature-obfuscate' into 'mgx_ops'

Browse Source 
Feature obfuscate

See merge request pub/MetaGPT!348
This commit is contained in:
张雷
2024-08-22 05:25:16 +00:00
parent b1d75272ff b2e8be2d78
commit cb3541f204
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
metagpt/provider/base_llm.py
+1 -2
View File
@@ -300,7 +300,6 @@ class BaseLLM(ABC):
if compress_type == CompressType.NO_COMPRESS:
return messages
current_token_count = 0
max_token = TOKEN_MAX.get(self.config.model, max_token)
keep_token = int(max_token * threshold)
compressed = []
@@ -318,7 +317,7 @@ class BaseLLM(ABC):
# system_msgs = [msg for msg in messages if msg["role"] == system_msg_val]
# user_assistant_msgs = [msg for msg in messages if msg["role"] != system_msg_val]
compressed.extend(system_msgs)
current_token_count += self.count_tokens(system_msgs)
current_token_count = self.count_tokens(system_msgs)
if compress_type in [CompressType.POST_CUT_BY_TOKEN, CompressType.POST_CUT_BY_MSG]:
# Under keep_token constraint, keep as many latest messages as possible