Skip to content

Commit 21972ba

Browse files
committed
use gpt-4o for lower price; update price calculation
1 parent c687360 commit 21972ba

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

configs/config.ini

+2-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@ run_openai = true
55
#model = gpt-3.5-turbo
66
#model = gpt-3.5-turbo-1106
77
#model = gpt-4
8-
model = gpt-4-turbo-preview
8+
#model = gpt-4-turbo-preview
9+
model = gpt-4o
910
# cost quality tradeoff - larger batches are cheaper but less accurate.
1011
batch_size = 5
1112

filter_papers.py

+6-4
Original file line numberDiff line numberDiff line change
@@ -52,12 +52,14 @@ def filter_papers_by_hindex(all_authors, papers, config):
5252

5353

5454
def calc_price(model, usage):
55+
if model == 'gpt-4o':
56+
return (5 * usage.prompt_tokens + 15 * usage.completion_tokens) / 1e6
5557
if model in ["gpt-4-1106-preview", "gpt-4-0125-preview", "gpt-4-turbo-preview"]:
56-
return (0.01 * usage.prompt_tokens + 0.03 * usage.completion_tokens) / 1000.0
58+
return (10 * usage.prompt_tokens + 30 * usage.completion_tokens) / 1e6
5759
if model == "gpt-4":
58-
return (0.03 * usage.prompt_tokens + 0.06 * usage.completion_tokens) / 1000.0
59-
if model.startswith("gpt-3.5-turbo"):
60-
return (0.0015 * usage.prompt_tokens + 0.002 * usage.completion_tokens) / 1000.0
60+
return (30 * usage.prompt_tokens + 60 * usage.completion_tokens) / 1e6
61+
if model.startswith("gpt-3.5"):
62+
return (0.5 * usage.prompt_tokens + 1.5 * usage.completion_tokens) / 1e6
6163

6264

6365
@retry.retry(tries=3, delay=2)

0 commit comments

Comments
 (0)