fix(utils.py): fix types

This commit is contained in:
Krrish Dholakia 2024-08-06 12:23:12 -07:00
parent 621b5d5fba
commit 834b437eb4
2 changed files with 9 additions and 3 deletions

View file

@ -2,3 +2,9 @@ model_list:
- model_name: "*"
litellm_params:
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

View file

@ -2,7 +2,7 @@ import json
import time
import uuid
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 pydantic import ConfigDict, Field, PrivateAttr
@ -219,7 +219,7 @@ class ChatCompletionDeltaToolCall(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
api_base: Optional[str] = None # returns api base used for making completion call