mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-03 02:22:24 +00:00
fix: update moderation model path to include cometapi prefix
This commit is contained in:
parent
730fe9c72d
commit
59bd0c9039
1 changed files with 8 additions and 2 deletions
10
cookbook/LiteLLM_CometAPI.ipynb
vendored
10
cookbook/LiteLLM_CometAPI.ipynb
vendored
|
|
@ -309,8 +309,7 @@
|
|||
"outputs": [],
|
||||
"source": [
|
||||
"moderation_response = litellm.moderation(\n",
|
||||
" model=\"omni-moderation-latest\",\n",
|
||||
" custom_llm_provider=\"cometapi\",\n",
|
||||
" model=\"cometapi/omni-moderation-latest\",\n",
|
||||
" input=\"I want to build a safe application.\",\n",
|
||||
" api_key=api_key,\n",
|
||||
")\n",
|
||||
|
|
@ -318,6 +317,13 @@
|
|||
"print(moderation_response.results[0].categories)\n",
|
||||
"print(moderation_response.results[0].category_scores)\n"
|
||||
]
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": []
|
||||
}
|
||||
],
|
||||
"metadata": {
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue