Update app.py

app.py

@@ -18,7 +18,7 @@ os.environ['TZ'] = 'Asia/Shanghai'
 time.tzset()
 
 logging.basicConfig(level=logging.INFO,
-                    format='%(asctime)s - %(levelname)s - %(message)s')
+                    format='%(asctime)s - %(levelname)s - %(client_ip)s - %(message)s')
 
 API_ENDPOINT = "https://api.siliconflow.cn/v1/user/info"
 TEST_MODEL_ENDPOINT = "https://api.siliconflow.cn/v1/chat/completions"
@@ -46,6 +46,15 @@ request_timestamps = []
 token_counts = []
 data_lock = threading.Lock()
 
+def get_client_ip():
+    """
+    获取客户端IP地址。
+    """
+    if request.headers.getlist("X-Forwarded-For"):
+        return request.headers.getlist("X-Forwarded-For")[0]
+    else:
+        return request.remote_addr
+
 def get_credit_summary(api_key):
     """
     使用 API 密钥获取额度信息。
@@ -469,7 +478,6 @@ def index():
     one_minute_ago = current_time - 60
 
     with data_lock:
-        # Clean up old data
         while request_timestamps and request_timestamps[0] < one_minute_ago:
             request_timestamps.pop(0)
             token_counts.pop(0)
@@ -544,7 +552,8 @@ def check_tokens():
                     )
             except Exception as exc:
                 logging.error(
-                    f"处理 Token {token} 生成异常: {exc}"
+                    f"处理 Token {token} 生成异常: {exc}", 
+                    extra={'client_ip': get_client_ip()}
                 )
 
     return jsonify(results)
@@ -774,7 +783,8 @@ def handsome_embeddings():
         except (KeyError, ValueError, IndexError) as e:
             logging.error(
                 f"解析响应 JSON 失败: {e}, "
-                f"完整内容: {response_json}"
+                f"完整内容: {response_json}", 
+                extra={'client_ip': get_client_ip()}
             )
             prompt_tokens = 0
             embedding_data = []
@@ -783,7 +793,8 @@ def handsome_embeddings():
             f"使用的key: {api_key}, "
             f"提示token: {prompt_tokens}, "
             f"总共用时: {total_time:.4f}秒, "
-            f"使用的模型: {model_name}"
+            f"使用的模型: {model_name}", 
+            extra={'client_ip': get_client_ip()}
         )
 
         with data_lock:
@@ -937,12 +948,12 @@ def handsome_images_generations():
                                 img_str = base64.b64encode(buffered.getvalue()).decode()
                                 openai_images.append({"b64_json": img_str})
                            except Exception as e:
-                                logging.error(f"图片转base64失败: {e}")
+                                logging.error(f"图片转base64失败: {e}", extra={'client_ip': get_client_ip()})
                                 openai_images.append({"url": image_url})
                         else:
                             openai_images.append({"url": image_url})
                     else:
-                        logging.error(f"无效的图片数据: {item}")
+                        logging.error(f"无效的图片数据: {item}", extra={'client_ip': get_client_ip()})
                         openai_images.append({"url": item})
 
 
@@ -953,7 +964,8 @@ def handsome_images_generations():
             except (KeyError, ValueError, IndexError) as e:
                 logging.error(
                     f"解析响应 JSON 失败: {e}, "
-                    f"完整内容: {response_json}"
+                    f"完整内容: {response_json}", 
+                    extra={'client_ip': get_client_ip()}
                 )
                 response_data = {
                     "created": int(time.time()),
@@ -963,7 +975,8 @@ def handsome_images_generations():
             logging.info(
                 f"使用的key: {api_key}, "
                 f"总共用时: {total_time:.4f}秒, "
-                f"使用的模型: {model_name}"
+                f"使用的模型: {model_name}", 
+                extra={'client_ip': get_client_ip()}
             )
 
             with data_lock:
@@ -973,7 +986,7 @@ def handsome_images_generations():
             return jsonify(response_data)
 
         except requests.exceptions.RequestException as e:
-            logging.error(f"请求转发异常: {e}")
+            logging.error(f"请求转发异常: {e}", extra={'client_ip': get_client_ip()})
             return jsonify({"error": str(e)}), 500
     else:
         return jsonify({"error": "Unsupported model"}), 400
@@ -1131,10 +1144,10 @@ def handsome_chat_completions():
                         image_url = ""
                         if images and isinstance(images[0], dict) and "url" in images[0]:
                             image_url = images[0]["url"]
-                            logging.info(f"Extracted image URL: {image_url}")
+                            logging.info(f"Extracted image URL: {image_url}", extra={'client_ip': get_client_ip()})
                         elif images and isinstance(images[0], str):
                             image_url = images[0]
-                            logging.info(f"Extracted image URL: {image_url}")
+                            logging.info(f"Extracted image URL: {image_url}", extra={'client_ip': get_client_ip()})
                         
                         markdown_image_link = f"![image]({image_url})"
                         if image_url:
@@ -1194,7 +1207,7 @@ def handsome_chat_completions():
                             request_timestamps.append(time.time())
                             token_counts.append(0)
                     except requests.exceptions.RequestException as e:
-                        logging.error(f"请求转发异常: {e}")
+                        logging.error(f"请求转发异常: {e}", extra={'client_ip': get_client_ip()})
                         error_chunk_data = {
                             "id": f"chatcmpl-{uuid.uuid4()}",
                             "object": "chat.completion.chunk",
@@ -1228,7 +1241,8 @@ def handsome_chat_completions():
                         yield f"data: {json.dumps(end_chunk_data)}\n\n".encode('utf-8')
                     logging.info(
                         f"使用的key: {api_key}, "
-                        f"使用的模型: {model_name}"
+                        f"使用的模型: {model_name}", 
+                        extra={'client_ip': get_client_ip()}
                     )
                     yield "data: [DONE]\n\n".encode('utf-8')
                 return Response(stream_with_context(generate()), content_type='text/event-stream')
@@ -1245,10 +1259,10 @@ def handsome_chat_completions():
                     image_url = ""
                     if images and isinstance(images[0], dict) and "url" in images[0]:
                         image_url = images[0]["url"]
-                        logging.info(f"Extracted image URL: {image_url}")
+                        logging.info(f"Extracted image URL: {image_url}", extra={'client_ip': get_client_ip()})
                     elif images and isinstance(images[0], str):
                         image_url = images[0]
-                        logging.info(f"Extracted image URL: {image_url}")
+                        logging.info(f"Extracted image URL: {image_url}", extra={'client_ip': get_client_ip()})
                     
                     markdown_image_link = f"![image]({image_url})"
                     response_data = {
@@ -1270,7 +1284,8 @@ def handsome_chat_completions():
                 except (KeyError, ValueError, IndexError) as e:
                     logging.error(
                         f"解析响应 JSON 失败: {e}, "
-                        f"完整内容: {response_json}"
+                        f"完整内容: {response_json}", 
+                        extra={'client_ip': get_client_ip()}
                     )
                     response_data = {
                         "id": f"chatcmpl-{uuid.uuid4()}",
@@ -1292,7 +1307,8 @@ def handsome_chat_completions():
                 logging.info(
                     f"使用的key: {api_key}, "
                     f"总共用时: {total_time:.4f}秒, "
-                    f"使用的模型: {model_name}"
+                    f"使用的模型: {model_name}", 
+                    extra={'client_ip': get_client_ip()}
                 )
                 with data_lock:
                     request_timestamps.append(time.time())
@@ -1300,7 +1316,7 @@ def handsome_chat_completions():
                 return jsonify(response_data)
 
         except requests.exceptions.RequestException as e:
-            logging.error(f"请求转发异常: {e}")
+            logging.error(f"请求转发异常: {e}", extra={'client_ip': get_client_ip()})
             return jsonify({"error": str(e)}), 500
     else:
         try:
@@ -1325,7 +1341,7 @@ def handsome_chat_completions():
                                 chunk = chunk.decode('utf-8')
                                 yield f"{chunk}\n\n".encode('utf-8')
                     except requests.exceptions.RequestException as e:
-                        logging.error(f"请求转发异常: {e}")
+                        logging.error(f"请求转发异常: {e}", extra={'client_ip': get_client_ip()})
                         error_chunk_data = {
                             "id": f"chatcmpl-{uuid.uuid4()}",
                             "object": "chat.completion.chunk",
@@ -1392,7 +1408,8 @@ def handsome_chat_completions():
                 except (KeyError, ValueError, IndexError) as e:
                     logging.error(
                         f"解析响应 JSON 失败: {e}, "
-                        f"完整内容: {response_json}"
+                        f"完整内容: {response_json}", 
+                        extra={'client_ip': get_client_ip()}
                     )
                     response_data = {
                         "id": f"chatcmpl-{uuid.uuid4()}",
@@ -1414,19 +1431,20 @@ def handsome_chat_completions():
                 logging.info(
                    f"使用的key: {api_key}, "
                    f"总共用时: {total_time:.4f}秒, "
-                   f"使用的模型: {model_name}"
+                   f"使用的模型: {model_name}", 
+                    extra={'client_ip': get_client_ip()}
                 )
                 with data_lock:
                     request_timestamps.append(time.time())
                     token_counts.append(0)
                 return jsonify(response_data)
         except requests.exceptions.RequestException as e:
-            logging.error(f"请求转发异常: {e}")
+            logging.error(f"请求转发异常: {e}", extra={'client_ip': get_client_ip()})
             return jsonify({"error": str(e)}), 500
 
 if __name__ == '__main__':
     import json
-    logging.info(f"环境变量：{os.environ}")
+    logging.info(f"环境变量：{os.environ}", extra={'client_ip': get_client_ip()})
 
     invalid_keys_global = []
     free_keys_global = []
@@ -1434,14 +1452,14 @@ if __name__ == '__main__':
     valid_keys_global = []
 
     load_keys()
-    logging.info("程序启动时首次加载 keys 已执行")
+    logging.info("程序启动时首次加载 keys 已执行", extra={'client_ip': get_client_ip()})
 
     scheduler.start()
 
-    logging.info("首次加载 keys 已手动触发执行")
+    logging.info("首次加载 keys 已手动触发执行", extra={'client_ip': get_client_ip()})
 
     refresh_models()
-    logging.info("首次刷新模型列表已手动触发执行")
+    logging.info("首次刷新模型列表已手动触发执行", extra={'client_ip': get_client_ip()})
 
     app.run(
         debug=False,