apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-28 18:36:22 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add yi-34b

Browse source
This commit is contained in:
better629 2024-03-14 21:34:23 +08:00
parent 7f39377a74
commit 1306971d2d
3 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
metagpt/utils/token_counter.py
View file

@ -48,6 +48,8 @@ TOKEN_COSTS = {
"claude-2.1": {"prompt": 0.008, "completion": 0.024},
"claude-3-sonnet-20240229": {"prompt": 0.003, "completion": 0.015},
"claude-3-opus-20240229": {"prompt": 0.015, "completion": 0.075},
"yi-34b-chat-0205": {"prompt": 0.0003, "completion": 0.0003},
"yi-34b-chat-200k": {"prompt": 0.0017, "completion": 0.0017},
}
@ -176,6 +178,8 @@ TOKEN_MAX = {
"claude-2.1": 200000,
"claude-3-sonnet-20240229": 200000,
"claude-3-opus-20240229": 200000,
"yi-34b-chat-0205": 4000,
"yi-34b-chat-200k": 200000,
}