mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-07 08:26:10 +00:00
test: allow output_cost_per_second_1080p in model_prices schema
Made-with: Cursor
This commit is contained in:
parent
34eff472a7
commit
acc17f2884
2 changed files with 2 additions and 9 deletions
|
|
@ -2,6 +2,7 @@ import copy
|
|||
import json
|
||||
import time
|
||||
from functools import partial
|
||||
from urllib.parse import quote
|
||||
from typing import TYPE_CHECKING, Any, List, Optional, Tuple, Union, cast, get_args
|
||||
|
||||
import httpx
|
||||
|
|
@ -610,7 +611,7 @@ class AmazonInvokeConfig(BaseConfig, BaseAWSLLM):
|
|||
Returns:
|
||||
str: The double-encoded model ID.
|
||||
"""
|
||||
return urllib.parse.quote(model_id, safe="")
|
||||
return quote(model_id, safe="")
|
||||
|
||||
def convert_messages_to_prompt(
|
||||
self, model, messages, provider, custom_prompt_dict
|
||||
|
|
|
|||
|
|
@ -478,14 +478,6 @@ def extract_model_name_from_bedrock_arn(model: str) -> str:
|
|||
return model
|
||||
|
||||
|
||||
def strip_bedrock_routing_prefix(model: str) -> str:
|
||||
"""Strip LiteLLM routing prefixes from model name."""
|
||||
for prefix in ["bedrock/", "converse/", "invoke/", "openai/", "nova-2/", "nova/"]:
|
||||
if model.startswith(prefix):
|
||||
model = model.split("/", 1)[1]
|
||||
return model
|
||||
|
||||
|
||||
def strip_bedrock_throughput_suffix(model: str) -> str:
|
||||
"""Strip throughput tier suffixes from Bedrock model names."""
|
||||
import re
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue