Upload check_mogo.py

check_mogo.py

@@ -10,6 +10,89 @@ def get_mongo_col(col="config_beta"):
     col = db[col]  # data_package(7234543961263854561)  goods_feature("7234800834201014260")
     return col
 
-# 拿到 collection 对象
-collection = get_mongo_col()
-print(collection)
+def set_txt2txt_price_config():
+    # 拿到 collection 对象
+    collection = get_mongo_col()  # 默认 col="config_beta"
+
+    doc = collection.find_one(
+        {"name": "txt2txt_price_config"},
+        {"_id": 0, "data": 1}
+    )
+    if not doc:
+        raise RuntimeError("找不到配置 document")
+
+    data = doc["data"]
+
+    data["us.anthropic.claude-sonnet-4-5-20250929-v1:0"] = {
+        "completion_price": 0.015,
+        "prompt_price": 0.003,
+        "currency": "$"
+    }
+
+    # 3. 一次性把整个 data 写回去
+    result = collection.update_one(
+        {"name": "txt2txt_price_config"},
+        {"$set": {"data": data}}
+    )
+    print(f"matched={result.matched_count}, modified={result.modified_count}")
+
+
+    query = {"name": "txt2txt_price_config"}
+    projection = {"_id": 0, "data": 1}
+
+    doc = collection.find_one(query, projection)
+
+
+    if doc:
+        data_value = doc.get("data")
+        print(data_value)
+    else:
+        print("未找到 name 等于 txt2txt_price_config 的记录")               
+
+def set_txt2txt_config():
+    # 拿到 collection 对象
+    collection = get_mongo_col()  # 默认 col="config_beta"
+
+    doc = collection.find_one(
+        {"name": "txt2txt_config"},
+        {"_id": 0, "data": 1}
+    )
+    if not doc:
+        raise RuntimeError("找不到配置 document")
+
+    data = doc["data"]
+
+    data.append({
+        "stream_selected": 0,
+        "label": "us.anthropic.claude-sonnet-4-5-20250929-v1:0",
+        "value": "us.anthropic.claude-sonnet-4-5-20250929-v1:0",
+        "wechat_selected": 0,
+        "wechat_label": 0,
+        "selected": 0,
+        "max_tokens": 4000,
+        "is_think": 0,
+        "web_label": 0,
+        "stream_label": 0,
+        "manufacturer": "亚马逊"
+    })
+
+    # 3. 一次性把整个 data 写回去
+    result = collection.update_one(
+        {"name": "txt2txt_config"},
+        {"$set": {"data": data}}
+    )
+    print(f"matched={result.matched_count}, modified={result.modified_count}")
+
+    query = {"name": "txt2txt_config"}
+    projection = {"_id": 0, "data": 1}
+
+    doc = collection.find_one(query, projection)
+
+    if doc:
+        data_value = doc.get("data")
+        print(data_value)
+    else:
+        print("未找到 name 等于 txt2txt_config 的记录")       
+
+set_txt2txt_price_config()
+set_txt2txt_config()