Skip to content

Restore and sync pricing table in readme #196

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
areibman wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

Restore and sync pricing table in readme #196

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 21 additions & 1 deletion README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -119,4 +119,24 @@ For Anthropic models above version 3 (i.e. Sonnet 3.5, Haiku 3.5, and Opus 3), w


## Cost table
Units denominated in USD. All prices can be located [here](pricing_table.md).
Units denominated in USD. All prices can be located [here](pricing_table.md).

<details>
<summary>📊 View Full Pricing Table</summary>

| Model Name | Prompt Cost (USD) per 1M tokens | Completion Cost (USD) per 1M tokens | Max Prompt Tokens | Max Output Tokens |
|:-----------------------|:----------------------------------|:--------------------------------------|:--------------------|--------------------:|
| gpt-4 | $30 | $60 | 8192 | 4096 |
| gpt-4o | $2.5 | $10 | 128,000 | 16384 |
| gpt-4o-mini | $0.15 | $0.6 | 128,000 | 16384 |
| o1-mini | $1.1 | $4.4 | 128,000 | 65536 |
| o1-preview | $15 | $60 | 128,000 | 32768 |
| gpt-3.5-turbo | $1.5 | $2 | 16,385 | 4096 |
| gpt-3.5-turbo-0125 | $0.5 | $1.5 | 16,385 | 4096 |
| text-embedding-3-large | $0.13 | $0 | 8,191 | nan |
| text-embedding-3-small | $0.02 | $0 | 8,191 | nan |
| text-embedding-ada-002 | $0.1 | $0 | 8,191 | nan |

*For the complete pricing table with all models, see [pricing_table.md](pricing_table.md)*

</details>
68 changes: 68 additions & 0 deletions update_prices.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -122,3 +122,71 @@ def format_cost(x):
f.write(table_md)

print("Pricing table updated in pricing_table.md")

# Update the README.md with the pricing table
def update_readme_pricing_table():
"""Update the pricing table in README.md with the latest data"""
# Read the current README
with open("README.md", "r") as f:
readme_content = f.read()

# Create a summary table with key models for the README
# Filter for the most important models to keep the README table manageable
important_models = [
'gpt-4', 'gpt-4o', 'gpt-4o-mini', 'gpt-3.5-turbo', 'gpt-3.5-turbo-0125',
'o1-mini', 'o1-preview', 'text-embedding-3-large', 'text-embedding-3-small',
'text-embedding-ada-002'
]

# Filter the dataframe for important models
summary_df = df[df.index.isin(important_models)]

# Generate the summary table
summary_table_md = summary_df[
[
"Model Name",
"Prompt Cost (USD) per 1M tokens",
"Completion Cost (USD) per 1M tokens",
"Max Prompt Tokens",
"Max Output Tokens",
]
].to_markdown(index=False)

# Create the new pricing table section
new_pricing_section = f"""## Cost table
Units denominated in USD. All prices can be located [here](pricing_table.md).

<details>
<summary>📊 View Full Pricing Table</summary>

{summary_table_md}

*For the complete pricing table with all models, see [pricing_table.md](pricing_table.md)*

</details>"""

# Find and replace the existing pricing table section
import re
pattern = r'## Cost table\nUnits denominated in USD\. All prices can be located \[here\]\(pricing_table\.md\)\.\n\n<details>\n<summary>📊 View Full Pricing Table</summary>\n\n.*?\n\n\*For the complete pricing table with all models, see \[pricing_table\.md\]\(pricing_table\.md\)\*\n\n</details>'

if re.search(pattern, readme_content, re.DOTALL):
# Replace existing section
updated_readme = re.sub(pattern, new_pricing_section, readme_content, flags=re.DOTALL)
else:
# If no existing section found, add it before the end of the file
# Find the last section and add before it
lines = readme_content.split('\n')
for i in range(len(lines) - 1, -1, -1):
if lines[i].startswith('## '):
lines.insert(i, new_pricing_section)
break
updated_readme = '\n'.join(lines)

# Write the updated README
with open("README.md", "w") as f:
f.write(updated_readme)

print("Pricing table updated in README.md")

# Update the README
update_readme_pricing_table()