修改prompt，增加需求数量

examples/demand/data_query_tools.py

@@ -97,13 +97,13 @@ where t1.缺量>= {count}
         for r in data:
             lack_count = r[9]
             if lack_count > 1000:
-                count = 30
+                count = 70
             elif 500 < lack_count <= 1000:
-                count = 20
+                count = 50
             elif 100 < lack_count <= 500:
-                count = 10
+                count = 30
             elif 50 < lack_count <= 100:
-                count = 5
+                count = 10
             else:
                 count = 0
             if count == 0:

examples/demand/demand.md

@@ -110,4 +110,4 @@ $user$
 2. 分类的共现组合，必须来自于`get_weight_score_topn`查询到的分类作为起点
 3. 最终结果的保留，必须要有权重分或者支持度进行支持
 4. 尽量保证最终产生的需求数量在「%count%」个左右
-
+5. 元素的需求更有意义，尽可能多的产生元素的需求，当元素的需求不足的时候，用有意义的其他类型需求补充

examples/demand/run.py

@@ -319,6 +319,9 @@ def write_demand_items_to_mysql(execution_id: int, merge_level2: str) -> int:
 
     affected = mysql_db.insert_many("demand_content", rows)
     log(f"[mysql] 写入 demand_content 完成，rows={len(rows)}, affected={affected}")
+    # with open(f'/Users/shimeng/Desktop/py/Agent/examples/demand/result/{merge_level2}.json', 'w',
+    #           encoding='utf-8') as f:
+    #     json.dump(rows, f, ensure_ascii=False, indent=4)
     return len(rows)