gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

针对memory修改

Browse Source
This commit is contained in:
tcmofashi
2025-03-02 16:06:11 +08:00
parent 97834e7a93
commit c0cb28e10d
8 changed files with 23 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/plugins/chat/llm_generator.py
View File

@@ -110,7 +110,7 @@ class LLMResponseGenerator:
"model": model_name,
"messages": [{"role": "user", "content": prompt}],
"stream": False,
"max_tokens": 1024,
"max_tokens": 2048,
"temperature": 0.7
}
@@ -118,7 +118,7 @@ class LLMResponseGenerator:
"model": "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
"messages": [{"role": "user", "content": prompt_check}],
"stream": False,
"max_tokens": 1024,
"max_tokens": 2048,
"temperature": 0.7
}
@@ -210,13 +210,13 @@ class LLMResponseGenerator:
return await self._generate_base_response(
message,
"deepseek-reasoner",
{"temperature": 0.7, "max_tokens": 1024}
{"temperature": 0.7, "max_tokens": 2048}
)
else:
return await self._generate_base_response(
message,
"Pro/deepseek-ai/DeepSeek-R1",
{"temperature": 0.7, "max_tokens": 1024}
{"temperature": 0.7, "max_tokens": 2048}
)
async def _generate_v3_response(self, message: Message) -> Optional[str]:
@@ -225,13 +225,13 @@ class LLMResponseGenerator:
return await self._generate_base_response(
message,
"deepseek-chat",
{"temperature": 0.8, "max_tokens": 1024}
{"temperature": 0.8, "max_tokens": 2048}
)
else:
return await self._generate_base_response(
message,
"Pro/deepseek-ai/DeepSeek-V3",
{"temperature": 0.8, "max_tokens": 1024}
{"temperature": 0.8, "max_tokens": 2048}
)
async def _generate_r1_distill_response(self, message: Message) -> Optional[str]:
@@ -239,7 +239,7 @@ class LLMResponseGenerator:
return await self._generate_base_response(
message,
"deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
{"temperature": 0.7, "max_tokens": 1024}
{"temperature": 0.7, "max_tokens": 2048}
)
async def _get_group_chat_context(self, message: Message) -> str: