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

add: openai moderation

Browse source
This commit is contained in:
zhanglei 2023-09-14 15:21:34 +08:00
parent f43208204c
commit 9bf1e51d02
1 changed files with 15 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

16
metagpt/provider/openai_api.py
View file

@ -6,7 +6,7 @@
"""
import asyncio
import time
from typing import NamedTuple
from typing import NamedTuple, Union, List, Optional
import openai
from openai.error import APIConnectionError
@ -286,3 +286,17 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter):
if not self.auto_max_tokens:
return CONFIG.max_tokens_rsp
return get_max_completion_tokens(messages, self.model, CONFIG.max_tokens_rsp)
def moderation(self,content: Union[str, List[str]],model: Optional[str] = None,api_key: Optional[str] = None,):
try:
if content is None or content == "" or len(content) == 0:
logger.error("content cannot be empty!")
else:
rsp = self._moderation(content=content,model=model,api_key=api_key)
return rsp
except Exception as e:
logger.error("moderating failed!", e)
def _moderation(self,content: Union[str, List[str]],model: Optional[str] = None,api_key: Optional[str] = None):
rsp = self.llm.Moderation.create(input=content,model=model,api_key=api_key)
return rsp