mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-06 02:48:13 +00:00
test_router_amoderation
This commit is contained in:
parent
2d836dfb6d
commit
1acc321eb3
1 changed files with 2 additions and 2 deletions
|
|
@ -1700,7 +1700,7 @@ async def test_router_amoderation():
|
|||
{
|
||||
"model_name": "openai-moderations",
|
||||
"litellm_params": {
|
||||
"model": "text-moderation-stable",
|
||||
"model": "omni-moderation-latest",
|
||||
"api_key": os.getenv("OPENAI_API_KEY", None),
|
||||
},
|
||||
}
|
||||
|
|
@ -1709,7 +1709,7 @@ async def test_router_amoderation():
|
|||
router = Router(model_list=model_list)
|
||||
## Test 1: user facing function
|
||||
result = await router.amoderation(
|
||||
model="text-moderation-stable", input="this is valid good text"
|
||||
model="omni-moderation-latest", input="this is valid good text"
|
||||
)
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue