From c0e93263383d04d6b5a05eeb5348d9f72736a14a Mon Sep 17 00:00:00 2001
From: seeker <liushaojie@fuzhi.ai>
Date: Fri, 19 Jul 2024 14:29:23 +0800
Subject: [PATCH] update: update token_counter, add models:
 deepseek-coder,deepseek-chat

---
 metagpt/utils/token_counter.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/metagpt/utils/token_counter.py b/metagpt/utils/token_counter.py
index 83b4826db..eb9e7a051 100644
--- a/metagpt/utils/token_counter.py
+++ b/metagpt/utils/token_counter.py
@@ -70,6 +70,8 @@ TOKEN_COSTS = {
     "google/gemini-flash-1.5": {"prompt": 0.00025, "completion": 0.00075},
     "deepseek/deepseek-coder": {"prompt": 0.00014, "completion": 0.00028},
     "deepseek/deepseek-chat": {"prompt": 0.00014, "completion": 0.00028},  # end, for openrouter
+    "deepseek-chat": {"prompt": 0.00014, "completion": 0.00028},
+    "deepseek-coder": {"prompt": 0.00014, "completion": 0.00028},
 }
 
 
@@ -215,6 +217,8 @@ TOKEN_MAX = {
     "google/gemini-flash-1.5": 2800000,
     "deepseek/deepseek-coder": 128000,
     "deepseek/deepseek-chat": 128000,  # end, for openrouter
+    "deepseek-chat": 128000,
+    "deepseek-coder": 128000,
 }