Upload mongo.py

mongo.py

@@ -23,14 +23,49 @@ if not doc:
 
 data = doc["data"]
 
-# 2. 在本地 dict 里插入/修改
-data["us.anthropic.claude-3-7-sonnet-20250219-v1:0"] = {
-    "completion_price": 0.015,
-    "prompt_price": 0.003,
+
+data["gpt-4.1"] = {
+    "completion_price": 0.008,
+    "prompt_price": 0.002,
+    "currency": "$"
+}
+
+data["gpt-4.1-mini"] = {
+    "completion_price": 0.0016,
+    "prompt_price": 0.0004,
+    "currency": "$"
+}
+
+data["gpt-4.1-nano"] = {
+    "completion_price": 0.0004,
+    "prompt_price": 0.0001,
+    "currency": "$"
+}
+
+data["gpt-5"] = {
+    "completion_price": 0.01,
+    "prompt_price": 0.00125,
+    "currency": "$"
+}
+
+data["gpt-5-chat"] = {
+    "completion_price": 0.01,
+    "prompt_price": 0.00125,
+    "currency": "$"
+}
+
+data["gpt-5-nano"] = {
+    "completion_price": 0.0004,
+    "prompt_price": 0.00005,
+    "currency": "$"
+}
+
+data["gpt-5-mini"] = {
+    "completion_price": 0.002,
+    "prompt_price": 0.00025,
     "currency": "$"
 }
 
-del data["us"]
 
 # 3. 一次性把整个 data 写回去
 result = collection.update_one(
@@ -39,20 +74,6 @@ result = collection.update_one(
 )
 print(f"matched={result.matched_count}, modified={result.modified_count}")
 
-# service_name = "us.anthropic.claude-3-7-sonnet-20250219-v1:0"
-# service_data = {
-#     "completion_price": 0.015,
-#     "prompt_price": 0.003,
-#     "currency": "$"
-# }
-
-# result = collection.update_one(
-#     {"name": "txt2txt_price_config"},
-#     {"$set": {f"data.{service_name}": service_data}}
-# )
-
-# print(f"matched={result.matched_count}, modified={result.modified_count}")
-
 
 query = {"name": "txt2txt_price_config"}
 projection = {"_id": 0, "data": 1}
@@ -66,14 +87,3 @@ if doc:
 else:
     print("未找到 name 等于 txt2txt_price_config 的记录")
 
-
-# 查询单条文档
-# doc = collection.find_one(query, projection)
-
-# db = col.database  # 或者直接用 client["dashboard"]
-
-# # 列出所有集合名
-# collections = db.list_collection_names()
-# print("当前 database 下的集合：", collections)
-
-# ['mp_collection', 'system.profile', 'mtxx_odz_aimr_im_chat_order_label_aggregate', 'xx_expose_history', 'aibeauty_trigger', 'mtxx_odz_aimr_im_chat_order', 'my_music_history', 'aibeauty_data_beta', 'my_music_history_count14', 'xx_area_mapping', 'aibeauty_trigger_beta', 'aibeauty_feedback', 'counters', 'xx_music_history', 'aibeauty_directive_beta', 'gd_history', 'message_spark_list_beta', 'my_expose', 'aibeauty_directive', 'xx_material_history', 'message_tongyi_list', 'aibeauty_im_msg', 'xx_formula_save_top100', 'aibeauty_title_stat', 'aibeauty_data', 'config', 'message_wenxin_list_beta', 'im_chat_msg_type', 'mp_neg_feedback', 'my_click', 'my_material_use_info', 'message_list_beta', 'message_pangu_list', 'aibeauty_data_history', 'my_search_history', 'mp_live', 'aibeauty_feedback_beta', 'aibeauty_top_data', 'message_wenxin_list', 'mp_expose', 'aibeauty_title_beta', 'aibeauty_data_history_beta', 'mtxx_odz_aimr_im_chat_order_label', 'related_history', 'xx_new_user', 'mht_rec_path', 'xx_feed_count', 'sdk_adz_aigc_chatgpt_info', 'aibeauty_top_data_gid_room_id', 'video_4k_transcoder', 'my_material_info', 'xx_music_material_category', 'mp_click', 'my_banner_history', 'webui_task_beta', 'config_beta', 'mht_dynamic_info', 'aibeauty_top_data_prompt_roomlist', 'my_search_history_count14', 'my_material_info_30', 'webui_record_beta', 'my_material_history', 'message_list', 'my_tmpl_history', 'aibeauty_title_stat_beta', 'message_pangu_list_beta', 'xx_dislike', 'my_music_history_count30', 'my_recommend_view', 'nlp_timeliness', 'blackflow_history', 'aibeauty_title', 'aibeauty_top_data_beta', 'message_spark_list', 'message_tongyi_list_beta', 'click_history']