phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add completion configs

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-21 14:00:07 -07:00
parent 371e0428d3
commit 2f44191642
7 changed files with 102 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/main.py
View file

@ -1321,11 +1321,8 @@ def text_completion(*args, **kwargs):
return completion(*args, **kwargs)
##### Moderation #######################
def moderation(*args, **kwargs):
def moderation(input: str, api_key: Optional[str]=None):
# only supports open ai for now
api_key = None
if "api_key" in kwargs:
api_key = kwargs["api_key"]
api_key = (
api_key or
litellm.api_key or
@ -1336,7 +1333,7 @@ def moderation(*args, **kwargs):
openai.api_type = "open_ai"
openai.api_version = None
openai.api_base = "https://api.openai.com/v1"
response = openai.Moderation.create(*args, **kwargs)
response = openai.Moderation.create(input)
return response
####### HELPER FUNCTIONS ################