add CoT

modules/chat_func.py

@@ -213,50 +213,52 @@ def predict_all(
     else:
         all_token_counts.append(count_token(construct_user(inputs)))
     try:
-        logging.info(f"system_prompt：{system_prompt}")
-        logging.info(f"history：{history}")
-        # 如果能传入index，则此处里获得初筛后的店铺和菜名
-        response = get_response(
-            openai_api_key,
-            system_prompt,
-            history,
-            temperature,
-            top_p,
-            False,
-            selected_model,
-        )
-        response = json.loads(response.text)
-        logging.info(f"初次响应推荐店铺：{response}")
-        response = response["choices"][0]["message"]["content"]
-
-        logging.info(f"初次响应推荐店铺：{response}")
-        # 将response中的店铺和菜名提取出来
-        import re
-
-        # text = """
-        #         好的，针对您想吃韩式烤肉的需求，我向您推荐以下店铺和菜品：
+        # logging.info(f"system_prompt：{system_prompt}")
+        # logging.info(f"history：{history}")
+        # # 如果能传入index，则此处里获得初筛后的店铺和菜名
+        # response = get_response(
+        #     openai_api_key,
+        #     system_prompt,
+        #     history,
+        #     temperature,
+        #     top_p,
+        #     False,
+        #     selected_model,
+        # )
+        # response = json.loads(response.text)
+        # logging.info(f"初次响应推荐店铺：{response}")
+        # response = response["choices"][0]["message"]["content"]
         #
-        #         店铺名称：“青年烤肉店” 推荐菜品：烤牛肉、烤猪肉、烤羊肉
+        # logging.info(f"初次响应推荐店铺：{response}")
+        # # 将response中的店铺和菜名提取出来
+        # import re
         #
-        #         店铺名称：“西西里烤肉店” 推荐菜品：烤牛肉串、烤排骨、烤鸡肉
+        # # text = """
+        # #         好的，针对您想吃韩式烤肉的需求，我向您推荐以下店铺和菜品：
+        # #
+        # #         店铺名称：“青年烤肉店” 推荐菜品：烤牛肉、烤猪肉、烤羊肉
+        # #
+        # #         店铺名称：“西西里烤肉店” 推荐菜品：烤牛肉串、烤排骨、烤鸡肉
+        # #
+        # #         店铺名称：“韩式烤肉店” 推荐菜品：石锅拌饭、铁板烧、烤牛舌"""
         #
-        #         店铺名称：“韩式烤肉店” 推荐菜品：石锅拌饭、铁板烧、烤牛舌"""
-
-        pattern = r'店铺名称：(.+?) 推荐菜品：(.+)。'
-
-        results = re.findall(pattern, response)
-
-        dicts = {}
-        import string
-        for result in results:
-            dicts[result[0]] = result[1].split('、')
-
-        logging.info(f"初筛后的店铺和菜品：{dicts}")
-        dishes = []
-        for restaurant, dish in dicts.items():
-            dishes.extend(dish)
+        # pattern = r'店铺名称：(.+?) 推荐菜品：(.+)。'
+        #
+        # results = re.findall(pattern, response)
+        #
+        # dicts = {}
+        # import string
+        # for result in results:
+        #     dicts[result[0]] = result[1].split('、')
+        #
+        # logging.info(f"初筛后的店铺和菜品：{dicts}")
+        # dishes = []
+        # for restaurant, dish in dicts.items():
+        #     dishes.extend(dish)
+        #
+        # dishes = '、'.join(dishes)
 
-        dishes = '、'.join(dishes)
+        dishes = "半熟芝士拿铁、白桃半熟芝士拿铁、招牌烤全羊、羊肉串"
 
         # 将初筛后的店铺和菜品送入构建好的CoT
         prompt_with_ingredient = f"""