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

(fix) cost calculator for FT: gpt-3.5

Browse source
This commit is contained in:
ishaan-jaff 2023-11-23 18:28:21 -08:00
parent b6f8dbbc0e
commit 695eaac542
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/utils.py
View file

@ -1587,12 +1587,12 @@ def token_counter(model="", text=None, messages: Optional[List] = None):
return num_tokens return num_tokens
def cost_per_token(model="gpt-3.5-turbo", prompt_tokens=0, completion_tokens=0): def cost_per_token(model="", prompt_tokens=0, completion_tokens=0):
""" """
Calculates the cost per token for a given model, prompt tokens, and completion tokens. Calculates the cost per token for a given model, prompt tokens, and completion tokens.
Parameters: Parameters:
model (str): The name of the model to use. Default is "gpt-3.5-turbo". model (str): The name of the model to use. Default is ""
prompt_tokens (int): The number of tokens in the prompt. prompt_tokens (int): The number of tokens in the prompt.
completion_tokens (int): The number of tokens in the completion. completion_tokens (int): The number of tokens in the completion.
@ -1637,7 +1637,7 @@ def cost_per_token(model="gpt-3.5-turbo", prompt_tokens=0, completion_tokens=0):
def completion_cost( def completion_cost(
completion_response=None, completion_response=None,
model="gpt-3.5-turbo", model=None,
prompt="", prompt="",
messages: List = [], messages: List = [],
completion="", completion="",