alihan/tokencost
1
0
Fork 0
mirror of https://github.com/AgentOps-AI/tokencost.git synced 2024-06-22 04:30:40 +03:00
Code Issues Packages Projects Releases Wiki Activity

removed unused file

Browse Source
This commit is contained in:
reibs
2023-12-25 23:04:27 -08:00
parent df1fa6fff6
commit 16f77b2f05
1 changed files with 0 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
tests/cost_fetch.py
View File

@@ -1,43 +0,0 @@
import openai
from tokencost import TOKEN_COSTS
models = (
"gpt-3.5-turbo",
"gpt-3.5-turbo-0301",
"gpt-3.5-turbo-0613",
"gpt-3.5-turbo-16k",
"gpt-3.5-turbo-16k-0613",
"gpt-3.5-turbo-1106",
"gpt-3.5-turbo-instruct",
"gpt-4-0314",
"gpt-4",
"gpt-4-32k",
"gpt-4-32k-0314",
"gpt-4-0613",
"gpt-4-1106-preview",
"gpt-4-vision-preview",
"text-embedding-ada-002",
)
STRING = "Hello, world!"
MESSAGES = [
{"role": "user", "content": "Hello"},
{"role": "assistant", "content": "Hi there!"},
]
client = openai.Client()
# for model in models:
# print(model)
# d = {"model": model}
# completion = client.chat.completions.create(messages=MESSAGES, model=model)
# d["prompt_tokens"] = completion.usage.prompt_tokens
# d["cost"] = d["prompt_tokens"] * TOKEN_COSTS[model]["prompt"]
# print(d)
# print("-" * 100)
for model in models:
print(model, TOKEN_COSTS[model]["completion"] * 4)
print("-" * 100)
# print(client.embeddings.create(input=STRING, model="text-embedding-ada-002"))