Upload 5 files

app.py

@@ -28,6 +28,8 @@ COMPUTE_POINTS_URL = "https://apps.abacus.ai/api/_getOrganizationComputePoints"
 COMPUTE_POINTS_LOG_URL = "https://abacus.ai/api/v0/_getOrganizationComputePointLog"
 CREATE_CONVERSATION_URL = "https://apps.abacus.ai/api/createDeploymentConversation"
 DELETE_CONVERSATION_URL = "https://apps.abacus.ai/api/deleteDeploymentConversation"
+GET_CONVERSATION_URL = "https://apps.abacus.ai/api/getDeploymentConversation"
+COMPUTE_POINT_TOGGLE_URL = "https://abacus.ai/api/v0/_updateOrganizationComputePointToggle"
 
 
 USER_AGENTS = [
@@ -375,6 +377,42 @@ def init_session():
             model_map, models_set = get_model_map(session, cookies, session_token)
             all_models.update(models_set)
             USER_DATA.append((session, cookies, session_token, conversation_id, model_map, i))
+            
+            # 对第一个成功配置的用户，初始化计算点数记录功能
+            if i == 0:
+                try:
+                    headers = {
+                        "accept": "application/json, text/plain, */*",
+                        "accept-language": "zh-CN,zh;q=0.9",
+                        "content-type": "application/json",
+                        "reai-ui": "1",
+                        "sec-ch-ua": "\"Chromium\";v=\"116\", \"Not)A;Brand\";v=\"24\", \"Google Chrome\";v=\"116\"",
+                        "sec-ch-ua-mobile": "?0",
+                        "sec-ch-ua-platform": "\"Windows\"",
+                        "sec-fetch-dest": "empty",
+                        "sec-fetch-mode": "cors",
+                        "sec-fetch-site": "same-site",
+                        "x-abacus-org-host": "apps",
+                        "session-token": session_token
+                    }
+                    
+                    response = session.post(
+                        COMPUTE_POINT_TOGGLE_URL,
+                        headers=headers,
+                        json={"alwaysDisplay": True},
+                        cookies=None
+                    )
+                    
+                    if response.status_code == 200:
+                        result = response.json()
+                        if result.get("success"):
+                            print("成功初始化计算点数记录功能为开启状态")
+                        else:
+                            print(f"初始化计算点数记录功能失败: {result.get('error', '未知错误')}")
+                    else:
+                        print(f"初始化计算点数记录功能失败，状态码: {response.status_code}")
+                except Exception as e:
+                    print(f"初始化计算点数记录功能时出错: {e}")
         except Exception as e:
             print(f"配置用户 {i+1} 失败: {e}")
             continue
@@ -696,7 +734,8 @@ def send_message(message, model, think=False):
         "sec-fetch-mode": "cors",
         "sec-fetch-site": "same-origin",
         "sentry-trace": sentry_trace,
-        "user-agent": random.choice(USER_AGENTS)
+        "user-agent": random.choice(USER_AGENTS),
+        "x-abacus-org-host": "apps"
     }
     
     if session_token:
@@ -712,9 +751,7 @@ def send_message(message, model, think=False):
             "language": "zh-CN"
         },
         "llmName": model_map[model][1],
-        "externalApplicationId": model_map[model][0],
-        "regenerate": True,
-        "editPrompt": True
+        "externalApplicationId": model_map[model][0]
     }
     
     if think:
@@ -725,7 +762,8 @@ def send_message(message, model, think=False):
             CHAT_URL,
             headers=headers,
             data=json.dumps(payload),
-            stream=True
+            stream=True,
+            cookies=None
         )
         
         response.raise_for_status()
@@ -789,7 +827,12 @@ def send_message(message, model, think=False):
             
             # 在流式传输完成后计算token并更新统计
             completion_result, _ = num_tokens_from_string(completion_buffer.getvalue(), model)
-            update_model_stats(model, prompt_tokens, completion_result, calculation_method)
+            
+            # 保存对话历史并获取计算点数
+            _, compute_points = save_conversation_history(session, cookies, session_token, conversation_id)
+            
+            # 更新统计信息
+            update_model_stats(model, prompt_tokens, completion_result, calculation_method, compute_points)
             
             # 如果需要删除上一个对话且上一个对话ID不为空且与当前不同
             if DELETE_CHAT and last_conversation_id and last_conversation_id != conversation_id:
@@ -834,7 +877,8 @@ def send_message_non_stream(message, model, think=False):
         "sec-fetch-mode": "cors",
         "sec-fetch-site": "same-origin",
         "sentry-trace": sentry_trace,
-        "user-agent": random.choice(USER_AGENTS)
+        "user-agent": random.choice(USER_AGENTS),
+        "x-abacus-org-host": "apps"
     }
     
     if session_token:
@@ -850,9 +894,7 @@ def send_message_non_stream(message, model, think=False):
             "language": "zh-CN"
         },
         "llmName": model_map[model][1],
-        "externalApplicationId": model_map[model][0],
-        "regenerate": True,
-        "editPrompt": True
+        "externalApplicationId": model_map[model][0]
     }
     
     if think:
@@ -863,7 +905,8 @@ def send_message_non_stream(message, model, think=False):
             CHAT_URL,
             headers=headers,
             data=json.dumps(payload),
-            stream=True
+            stream=True,
+            cookies=None
         )
         
         response.raise_for_status()
@@ -918,7 +961,12 @@ def send_message_non_stream(message, model, think=False):
             
             # 计算输出token并更新统计信息
             completion_result, _ = num_tokens_from_string(think_content + response_content, model)
-            update_model_stats(model, prompt_tokens, completion_result, calculation_method)
+            
+            # 保存对话历史并获取计算点数
+            _, compute_points = save_conversation_history(session, cookies, session_token, conversation_id)
+            
+            # 更新统计信息
+            update_model_stats(model, prompt_tokens, completion_result, calculation_method, compute_points)
             
             # 如果需要删除上一个对话且上一个对话ID不为空且与当前不同
             if DELETE_CHAT and last_conversation_id and last_conversation_id != conversation_id:
@@ -955,7 +1003,12 @@ def send_message_non_stream(message, model, think=False):
             
             # 计算输出token并更新统计信息
             completion_result, _ = num_tokens_from_string(response_content, model)
-            update_model_stats(model, prompt_tokens, completion_result, calculation_method)
+            
+            # 保存对话历史并获取计算点数
+            _, compute_points = save_conversation_history(session, cookies, session_token, conversation_id)
+            
+            # 更新统计信息
+            update_model_stats(model, prompt_tokens, completion_result, calculation_method, compute_points)
             
             # 如果需要删除上一个对话且上一个对话ID不为空且与当前不同
             if DELETE_CHAT and last_conversation_id and last_conversation_id != conversation_id:
@@ -1111,7 +1164,7 @@ def num_tokens_from_string(string, model=""):
 
 
 # 更新模型使用统计
-def update_model_stats(model, prompt_tokens, completion_tokens, calculation_method="estimate"):
+def update_model_stats(model, prompt_tokens, completion_tokens, calculation_method="estimate", compute_points=None):
     global model_usage_stats, total_tokens, model_usage_records
     
     # 添加调用记录
@@ -1126,7 +1179,8 @@ def update_model_stats(model, prompt_tokens, completion_tokens, calculation_meth
         "call_time": call_time,
         "prompt_tokens": prompt_tokens,
         "completion_tokens": completion_tokens,
-        "calculation_method": calculation_method
+        "calculation_method": calculation_method,
+        "compute_points": compute_points
     }
     model_usage_records.append(record)
     
@@ -1382,6 +1436,52 @@ def update_delete_chat_setting():
         return jsonify({"success": False, "error": str(e)})
 
 
+# 添加更新计算点数记录设置的路由
+@app.route("/update_compute_point_toggle", methods=["POST"])
+@require_auth
+def update_compute_point_toggle():
+    try:
+        (session, cookies, session_token, conversation_id, model_map, user_index) = get_user_data()
+        data = request.get_json()
+        if data and "always_display" in data:
+            headers = {
+                "accept": "application/json, text/plain, */*",
+                "accept-language": "zh-CN,zh;q=0.9",
+                "content-type": "application/json",
+                "reai-ui": "1",
+                "sec-ch-ua": "\"Chromium\";v=\"116\", \"Not)A;Brand\";v=\"24\", \"Google Chrome\";v=\"116\"",
+                "sec-ch-ua-mobile": "?0",
+                "sec-ch-ua-platform": "\"Windows\"",
+                "sec-fetch-dest": "empty",
+                "sec-fetch-mode": "cors",
+                "sec-fetch-site": "same-site",
+                "x-abacus-org-host": "apps"
+            }
+            
+            if session_token:
+                headers["session-token"] = session_token
+            
+            response = session.post(
+                COMPUTE_POINT_TOGGLE_URL,
+                headers=headers,
+                json={"alwaysDisplay": data["always_display"]},
+                cookies=None
+            )
+            
+            if response.status_code == 200:
+                result = response.json()
+                if result.get("success"):
+                    print(f"更新计算点数记录设置为: {data['always_display']}")
+                    return jsonify({"success": True})
+            
+            return jsonify({"success": False, "error": "API调用失败"})
+        else:
+            return jsonify({"success": False, "error": "缺少always_display参数"})
+    except Exception as e:
+        print(f"更新计算点数记录设置失败: {e}")
+        return jsonify({"success": False, "error": str(e)})
+
+
 # 获取Hugging Face Space URL
 def get_space_url():
     # 尝试从环境变量获取
@@ -1418,6 +1518,67 @@ if SPACE_URL:
     print("注意：Hugging Face生成的URL会自动将空间名称中的下划线(_)替换为连字符(-)")
 
 
+def save_conversation_history(session, cookies, session_token, conversation_id, deployment_id="14b2a314cc"):
+    """保存对话历史，返回使用的计算点数"""
+    if not conversation_id:
+        return False, None
+        
+    headers = {
+        "accept": "application/json, text/plain, */*",
+        "accept-language": "zh-CN,zh;q=0.9",
+        "baggage": f"sentry-environment=production,sentry-release=946244517de08b08598b94f18098411f5a5352d5,sentry-public_key=3476ea6df1585dd10e92cdae3a66ff49,sentry-trace_id={TRACE_ID}",
+        "reai-ui": "1",
+        "sec-ch-ua": "\"Chromium\";v=\"116\", \"Not)A;Brand\";v=\"24\", \"Google Chrome\";v=\"116\"",
+        "sec-ch-ua-mobile": "?0",
+        "sec-ch-ua-platform": "\"Windows\"",
+        "sec-fetch-dest": "empty",
+        "sec-fetch-mode": "cors",
+        "sec-fetch-site": "same-origin",
+        "sentry-trace": f"{TRACE_ID}-800cb7f4613dec52",
+        "x-abacus-org-host": "apps"
+    }
+    
+    if session_token:
+        headers["session-token"] = session_token
+        
+    params = {
+        "deploymentId": deployment_id,
+        "deploymentConversationId": conversation_id,
+        "skipDocumentBoundingBoxes": "true",
+        "filterIntermediateConversationEvents": "false",
+        "getUnusedDocumentUploads": "false"
+    }
+    
+    try:
+        response = session.get(
+            GET_CONVERSATION_URL,
+            headers=headers,
+            params=params,
+            cookies=None
+        )
+        
+        if response.status_code == 200:
+            data = response.json()
+            if data.get("success"):
+                # 从最后一条BOT消息中获取计算点数
+                history = data.get("result", {}).get("history", [])
+                compute_points = None
+                for msg in reversed(history):
+                    if msg.get("role") == "BOT":
+                        compute_points = msg.get("computePointsUsed")
+                        break
+                print(f"成功保存对话历史: {conversation_id}, 使用计算点: {compute_points}")
+                return True, compute_points
+            else:
+                print(f"保存对话历史失败: {data.get('error', '未知错误')}")
+        else:
+            print(f"保存对话历史失败，状态码: {response.status_code}")
+        return False, None
+    except Exception as e:
+        print(f"保存对话历史时出错: {e}")
+        return False, None
+
+
 if __name__ == "__main__":
     # 启动保活线程
     threading.Thread(target=keep_alive, daemon=True).start()

templates/dashboard.html

@@ -695,6 +695,20 @@
                 <span class="status-label">设置说明</span>
                 <span class="status-value info-text">开启后，系统将在每次对话完成后自动删除上一次对话，只保留最新对话</span>
             </div>
+            <div class="status-item">
+                <span class="status-label">是否记录计算点数</span>
+                <div class="toggle-switch-container">
+                    <label class="toggle-switch">
+                        <input type="checkbox" id="compute-point-toggle" checked>
+                        <span class="toggle-slider"></span>
+                    </label>
+                    <span class="toggle-status" id="compute-point-status">开启</span>
+                </div>
+            </div>
+            <div class="status-item">
+                <span class="status-label">设置说明</span>
+                <span class="status-value info-text">开启后，系统将记录每次对话使用的计算点数，用于统计和分析</span>
+            </div>
         </div>
         
         <div class="grid">
@@ -868,7 +882,9 @@
                             <th>模型</th>
                             <th>输入Token</th>
                             <th>输出Token</th>
+                            <th>总Token</th>
                             <th>计算方式</th>
+                            <th>计算点数</th>
                         </tr>
                     </thead>
                     <tbody>
@@ -876,15 +892,11 @@
                         <tr class="model-row {% if loop.index > 10 %}hidden-model{% endif %}">
                             <td class="datetime">{{ record.call_time }}</td>
                             <td>{{ record.model }}</td>
-                            <td class="token-count">{{ record.prompt_tokens|int }}</td>
-                            <td class="token-count">{{ record.completion_tokens|int }}</td>
-                            <td>
-                                {% if record.calculation_method in ["api"] %}
-                                <span class="token-method api">精确</span>
-                                {% else %}
-                                <span class="token-method estimate">估算</span>
-                                {% endif %}
-                            </td>
+                            <td class="token-count">{{ record.prompt_tokens }}</td>
+                            <td class="token-count">{{ record.completion_tokens }}</td>
+                            <td>{{ record.prompt_tokens + record.completion_tokens }}</td>
+                            <td>{{ record.calculation_method }}</td>
+                            <td>{{ record.compute_points if record.compute_points is not none else 'null' }}</td>
                         </tr>
                         {% endfor %}
                     </tbody>
@@ -1025,6 +1037,46 @@
                     });
                 });
             }
+
+            // 处理计算点数记录开关
+            const computePointToggle = document.getElementById('compute-point-toggle');
+            const computePointStatus = document.getElementById('compute-point-status');
+            
+            if (computePointToggle && computePointStatus) {
+                computePointToggle.addEventListener('change', function() {
+                    const isChecked = this.checked;
+                    computePointStatus.textContent = isChecked ? '开启' : '关闭';
+                    
+                    // 发送更新请求到后端
+                    fetch('/update_compute_point_toggle', {
+                        method: 'POST',
+                        headers: {
+                            'Content-Type': 'application/json',
+                        },
+                        body: JSON.stringify({ always_display: isChecked })
+                    })
+                    .then(response => response.json())
+                    .then(data => {
+                        if (data.success) {
+                            // 显示成功提示
+                            showNotification(isChecked ? '已开启计算点数记录功能' : '已关闭计算点数记录功能', 'success');
+                        } else {
+                            // 显示错误提示
+                            showNotification('设置更新失败: ' + data.error, 'error');
+                            // 回滚UI状态
+                            computePointToggle.checked = !isChecked;
+                            computePointStatus.textContent = !isChecked ? '开启' : '关闭';
+                        }
+                    })
+                    .catch(error => {
+                        console.error('更新设置出错:', error);
+                        showNotification('更新设置失败，请重试', 'error');
+                        // 回滚UI状态
+                        computePointToggle.checked = !isChecked;
+                        computePointStatus.textContent = !isChecked ? '开启' : '关闭';
+                    });
+                });
+            }
         });
         
         // 通知函数