update

app.py

@@ -204,6 +204,9 @@ import torch
 from tqdm import tqdm
 import numpy as np
 import time
+import requests
+from requests.adapters import HTTPAdapter
+from requests.packages.urllib3.util.retry import Retry
 
 # 设置缓存目录
 os.environ['MPLCONFIGDIR'] = '/tmp/matplotlib'
@@ -212,10 +215,32 @@ os.makedirs(os.environ['MPLCONFIGDIR'], exist_ok=True)
 os.makedirs(os.environ['TRANSFORMERS_CACHE'], exist_ok=True)
 
 # Weaviate 连接配置
+retry_strategy = Retry(
+    total=3,  # 总共重试次数
+    status_forcelist=[429, 500, 502, 503, 504],  # 需要重试的状态码
+    method_whitelist=["HEAD", "GET", "OPTIONS", "POST"],  # 需要重试的方法
+    backoff_factor=1  # 重试间隔时间的倍数
+)
+adapter = HTTPAdapter(max_retries=retry_strategy)
+
+http = requests.Session()
+http.mount("https://", adapter)
+http.mount("http://", adapter)
+
+timeout =10 # 超时时间（秒）
+
 WEAVIATE_API_KEY = "Y7c8DRmcxZ4nP5IJLwkznIsK84l6EdwfXwcH"
 WEAVIATE_URL = "https://39nlafviqvard82k6y8btq.c0.asia-southeast1.gcp.weaviate.cloud"
+# weaviate_auth_config = weaviate.AuthApiKey(api_key=WEAVIATE_API_KEY)
+# weaviate_client = weaviate.Client(url=WEAVIATE_URL, auth_client_secret=weaviate_auth_config)
+# 创建 Weaviate 客户端
 weaviate_auth_config = weaviate.AuthApiKey(api_key=WEAVIATE_API_KEY)
-weaviate_client = weaviate.Client(url=WEAVIATE_URL, auth_client_secret=weaviate_auth_config)
+weaviate_client = weaviate.Client(
+    url=WEAVIATE_URL,
+    auth_client_secret=weaviate_auth_config,
+    timeout_config=(timeout, timeout)  # 连接超时和读取超时
+)
+
 
 # 预训练模型配置
 MODEL_NAME = "bert-base-chinese"
@@ -421,13 +446,13 @@ def chatbot_response(message, history, window_size, threshold, score_threshold,u
             else:
                 group_scores[group_id] = score
 
+    distance=1000
     if group_scores:
         max_group_id = max(group_scores, key=group_scores.get)
         max_score = group_scores[max_group_id]
         if(max_score>=score_threshold):
             distance,ad_summary,ad_keywords=[(candidate['distance'],candidate['summary'],candidate['keyword_list']) for candidate in candidates if candidate['group_id']==max_group_id][0]
-        else:
-            distance=1000
+
 
     # if(candidates):
     #     # distance, ad_summary, ad_keywords=keyword_match(keywords_dict,candidates)  
@@ -435,7 +460,7 @@ def chatbot_response(message, history, window_size, threshold, score_threshold,u
     # else:
     #     distance=1000
 
-    if distance and distance < 1000:
+    if distance < 1000:
         brands=['腾讯','阿里巴巴','百度','京东','华为','小米','苹果','微软','谷歌','亚马逊']
         brand=random.choice(brands)
         ad_message = f"{message} <sep>品牌{brand}<sep>{ad_summary}"