kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-06-02 00:57:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: support skills creator and gemini models

Browse Source
This commit is contained in:
saboteur7
2026-01-30 18:00:10 +08:00
parent 49fb4034c6
commit dd6a9c26bd
30 changed files with 3562 additions and 833 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
agent/protocol/agent.py
View File

@@ -162,9 +162,16 @@ class Agent:
# DeepSeek
elif 'deepseek' in model_name:
return 64000
# Gemini models
elif 'gemini' in model_name:
if '2.0' in model_name or 'exp' in model_name:
return 2000000 # Gemini 2.0: 2M tokens
else:
return 1000000 # Gemini 1.5: 1M tokens
# Default conservative value
return 10000
return 128000
def _get_context_reserve_tokens(self) -> int:
"""
@@ -176,9 +183,10 @@ class Agent:
if self.context_reserve_tokens is not None:
return self.context_reserve_tokens
# Reserve ~20% of context window for new requests
# Reserve ~10% of context window, with min 10K and max 200K
context_window = self._get_model_context_window()
return max(4000, int(context_window * 0.2))
reserve = int(context_window * 0.1)
return max(10000, min(200000, reserve))
def _estimate_message_tokens(self, message: dict) -> int:
"""