forked from phoenix/litellm-mirror
fix(utils.py): fix types
This commit is contained in:
parent
621b5d5fba
commit
834b437eb4
2 changed files with 9 additions and 3 deletions
|
@ -1,4 +1,10 @@
|
||||||
model_list:
|
model_list:
|
||||||
- model_name: "*"
|
- model_name: "*"
|
||||||
litellm_params:
|
litellm_params:
|
||||||
model: "*"
|
model: "*"
|
||||||
|
|
||||||
|
litellm_settings:
|
||||||
|
guardrails:
|
||||||
|
- prompt_injection: # your custom name for guardrail
|
||||||
|
callbacks: [lakera_prompt_injection] # litellm callbacks to use
|
||||||
|
default_on: true # will run on all llm requests when true
|
|
@ -2,7 +2,7 @@ import json
|
||||||
import time
|
import time
|
||||||
import uuid
|
import uuid
|
||||||
from enum import Enum
|
from enum import Enum
|
||||||
from typing import Dict, List, Literal, Optional, Tuple, Union
|
from typing import Any, Dict, List, Literal, Optional, Tuple, Union
|
||||||
|
|
||||||
from openai._models import BaseModel as OpenAIObject
|
from openai._models import BaseModel as OpenAIObject
|
||||||
from pydantic import ConfigDict, Field, PrivateAttr
|
from pydantic import ConfigDict, Field, PrivateAttr
|
||||||
|
@ -219,7 +219,7 @@ class ChatCompletionDeltaToolCall(OpenAIObject):
|
||||||
|
|
||||||
|
|
||||||
class HiddenParams(OpenAIObject):
|
class HiddenParams(OpenAIObject):
|
||||||
original_response: Optional[str] = None
|
original_response: Optional[Union[str, Any]] = None
|
||||||
model_id: Optional[str] = None # used in Router for individual deployments
|
model_id: Optional[str] = None # used in Router for individual deployments
|
||||||
api_base: Optional[str] = None # returns api base used for making completion call
|
api_base: Optional[str] = None # returns api base used for making completion call
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue