增加followup信息

app/api/v2/zhipu_controller_v2.py

@@ -281,9 +281,43 @@ def analysis_stream():
                 chunk_out = format_chunk(chunk, None, None) 
                 yield json.dumps(chunk_out) + '\n'
 
-        yield json.dumps(format_chunk("", "如有新的销售进展，请补充相关信息，我会为您做进一步分析", "")) + '\n'
+        followup_info = get_analysis_followup_info(message,knowledge_id)
+        followup_info += "如有新的销售进展，请补充相关信息，我会为您做进一步分析"
+        yield json.dumps(format_chunk("", followup_info, "")) + '\n'
         logger.info(f'accumulated_result: {accumulated_result}')
+        logger.info(f'followup_info: {followup_info}')
 
     return Response(event_stream(), mimetype='text/event-stream', headers=response_headers)
 
 
+def get_analysis_followup_info(message, knowledge_id):
+    followup_info = ""
+    # 1. Get business contact info
+    if '汇理' in message:
+        followup_info += "客户联系人: 陈明宇 (技术总监), 王雪梅 (项目经理)\n" 
+        followup_info += "客户联系电话: 186-2155-7823, 135-8867-4932"
+    elif '芝麻地网科' in message:
+        followup_info += "客户联系人: 刘建华 (产品总监), 张婷婷 (商务经理)\n" 
+        followup_info += "客户联系电话: 138-9876-5432, 159-2468-1357"
+    elif '联特科技' in message:
+        followup_info += "客户联系人: 郭志强 (研发经理), 林小华 (项目主管)\n" 
+        followup_info += "客户联系电话: 177-3344-5566, 136-9988-7766"
+    elif '视睿电子' in message:
+        followup_info += "客户联系人: 黄伟东 (技术经理), 赵敏 (销售总监)\n" 
+        followup_info += "客户联系电话: 158-1122-3344, 139-5544-3322"
+    elif '深圳麦克韦尔' in message:
+        followup_info += "客户联系人: 吴晓峰 (部门主管), 孙丽华 (商务代表)\n" 
+        followup_info += "客户联系电话: 182-8899-7766, 137-6655-4433"
+    else:
+        followup_info = ""
+        return followup_info
+
+    action_guide_url = "https://fiq58om72ph.feishu.cn/wiki/GbV3wXWu2igPoRkXSP8cMDzvnkd?from=from_copylink"
+    followup_info += f"\n行动指南: {action_guide_url}"
+
+    sales_stage_definition_url = "https://fiq58om72ph.feishu.cn/wiki/I8Raw0MVKizZTvkR4UWcRmy4n7f?from=from_copylink"
+    followup_info += f"\n销售阶段手册: {sales_stage_definition_url}"
+
+    followup_info += "\n\n"
+
+    return followup_info