mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-06-05 14:55:18 +02:00
add use_system_prompt to judge if need to add system_prompt part
This commit is contained in:
parent
66f27ca2d5
commit
8e201384bf
4 changed files with 10 additions and 3 deletions
|
|
@ -31,6 +31,9 @@ RPM: 10
|
|||
#DEPLOYMENT_NAME: "YOUR_DEPLOYMENT_NAME"
|
||||
#DEPLOYMENT_ID: "YOUR_DEPLOYMENT_ID"
|
||||
|
||||
#### if zhipuai from `https://open.bigmodel.cn`. You can set here or export API_KEY="YOUR_API_KEY"
|
||||
ZHIPUAI_API_KEY: "YOUR_API_KEY"
|
||||
|
||||
#### for Search
|
||||
|
||||
## Supported values: serpapi/google/serper/ddg
|
||||
|
|
|
|||
|
|
@ -13,6 +13,7 @@ from dataclasses import dataclass
|
|||
class BaseChatbot(ABC):
|
||||
"""Abstract GPT class"""
|
||||
mode: str = "API"
|
||||
use_system_prompt: bool = True
|
||||
|
||||
@abstractmethod
|
||||
def ask(self, msg: str) -> str:
|
||||
|
|
|
|||
|
|
@ -32,15 +32,17 @@ class BaseGPTAPI(BaseChatbot):
|
|||
return self._system_msg(self.system_prompt)
|
||||
|
||||
def ask(self, msg: str) -> str:
|
||||
message = [self._default_system_msg(), self._user_msg(msg)]
|
||||
message = [self._default_system_msg(), self._user_msg(msg)] if self.use_system_prompt else [self._user_msg(msg)]
|
||||
rsp = self.completion(message)
|
||||
return self.get_choice_text(rsp)
|
||||
|
||||
async def aask(self, msg: str, system_msgs: Optional[list[str]] = None) -> str:
|
||||
if system_msgs:
|
||||
message = self._system_msgs(system_msgs) + [self._user_msg(msg)]
|
||||
message = self._system_msgs(system_msgs) + [self._user_msg(msg)] if self.use_system_prompt \
|
||||
else [self._user_msg(msg)]
|
||||
else:
|
||||
message = [self._default_system_msg(), self._user_msg(msg)]
|
||||
message = [self._default_system_msg(), self._user_msg(msg)] if self.use_system_prompt \
|
||||
else [self._user_msg(msg)]
|
||||
rsp = await self.acompletion_text(message, stream=True)
|
||||
logger.debug(message)
|
||||
# logger.debug(rsp)
|
||||
|
|
|
|||
|
|
@ -44,3 +44,4 @@ ta==0.10.2
|
|||
semantic-kernel==0.3.13.dev0
|
||||
wrapt==1.15.0
|
||||
websocket-client==0.58.0
|
||||
zhipuai==1.0.7
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue