phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #4279 from tomusher/feature/return-output-vector-size-in-modelinfo

Browse source 
Return output_vector_size in get_model_info
This commit is contained in:
Krish Dholakia 2024-07-04 17:03:39 -07:00 • committed by GitHub
commit 1807c49f18
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/types/utils.py
View file

@ -61,6 +61,7 @@ class ModelInfo(TypedDict, total=False):
float float
] # only for vertex ai models ] # only for vertex ai models
output_cost_per_image: Optional[float] output_cost_per_image: Optional[float]
output_vector_size: Optional[int]
output_cost_per_video_per_second: Optional[float] # only for vertex ai models output_cost_per_video_per_second: Optional[float] # only for vertex ai models
output_cost_per_audio_per_second: Optional[float] # only for vertex ai models output_cost_per_audio_per_second: Optional[float] # only for vertex ai models
litellm_provider: Required[str] litellm_provider: Required[str]

1
litellm/utils.py
View file

@ -4685,6 +4685,7 @@ def get_model_info(model: str, custom_llm_provider: Optional[str] = None) -> Mod
output_cost_per_character_above_128k_tokens=_model_info.get( output_cost_per_character_above_128k_tokens=_model_info.get(
"output_cost_per_character_above_128k_tokens", None "output_cost_per_character_above_128k_tokens", None
), ),
output_vector_size=_model_info.get("output_vector_size", None),
litellm_provider=_model_info.get( litellm_provider=_model_info.get(
"litellm_provider", custom_llm_provider "litellm_provider", custom_llm_provider
), ),