Upload mongo.py

mongo.py

@@ -13,20 +13,46 @@ def get_mongo_col(col="config_beta"):
 # 拿到 collection 对象
 collection = get_mongo_col()  # 默认 col="config_beta"
 
-service_name = "us.anthropic.claude-3-7-sonnet-20250219-v1:0"
-service_data = {
+
+doc = collection.find_one(
+    {"name": "txt2txt_price_config"},
+    {"_id": 0, "data": 1}
+)
+if not doc:
+    raise RuntimeError("找不到配置 document")
+
+data = doc["data"]
+
+# 2. 在本地 dict 里插入/修改
+data["us.anthropic.claude-3-7-sonnet-20250219-v1:0"] = {
     "completion_price": 0.015,
     "prompt_price": 0.003,
     "currency": "$"
 }
 
+del data["us"]
+
+# 3. 一次性把整个 data 写回去
 result = collection.update_one(
     {"name": "txt2txt_price_config"},
-    {"$set": {f"data.{service_name}": service_data}}
+    {"$set": {"data": data}}
 )
-
 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}