apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-24 14:15:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix feature

Browse source
This commit is contained in:
rainyrfeng 2023-09-05 11:26:16 +08:00
parent ce0c7b62a0
commit a0be280879
1 changed files with 15 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

32
metagpt/provider/openai_api.py
View file

@ -176,25 +176,23 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter):
return full_reply_content
def _cons_kwargs(self, messages: list[dict]) -> dict:
kwargs = {
"messages": messages,
"max_tokens": self.get_max_tokens(messages),
"n": 1,
"stop": None,
"temperature": 0.3,
}
if CONFIG.openai_api_type == "azure":
kwargs = {
"engine": CONFIG.openai_api_engine,
"deployment_id": CONFIG.deployment_id,
"messages": messages,
"max_tokens": self.get_max_tokens(messages),
"n": 1,
"stop": None,
"temperature": 0.3,
}
if CONFIG.openai_api_engine and CONFIG.deployment_id:
raise ValueError("You can only use one of the `deployment_id` or `engine` model")
elif not CONFIG.openai_api_engine and not CONFIG.deployment_id:
raise ValueError("You must specify `OPENAI_API_ENGINE` or `DEPLOYMENT_ID` parameter")
kwargs_mode = {"engine": CONFIG.openai_api_engine} if CONFIG.openai_api_engine \
else {"deployment_id": CONFIG.deployment_id}
else:
kwargs = {
"model": self.model,
"messages": messages,
"max_tokens": self.get_max_tokens(messages),
"n": 1,
"stop": None,
"temperature": 0.3,
}
kwargs_mode = {"model": self.model}
kwargs.update(kwargs_mode)
kwargs["timeout"] = 3
return kwargs