zensol
/
market_assistant
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

init 北极星微

This commit is contained in:
Tiger Ren 2024-07-31 17:11:29 +08:00
parent 984be77738
commit 0074e954dd
10 changed files with 373 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
ExcelHelper.py Normal file
View File

@ -0,0 +1,48 @@
import openpyxl
from openpyxl.styles import Font
class ExcelHelper:
def __init__(self, data):
"""
初始化 ExcelHelper 实例
:param data: 包含多个数组的列表，每个数组包含字典，每个字典有 'title' 和 'content' 键
"""
self.data = data
self.workbook = openpyxl.Workbook()
self.sheet = self.workbook.active
def create_excel(self, filename):
"""
创建 Excel 文件并保存
:param filename: 保存的文件名
"""
start_row = 1
for array in self.data:
col = 1
for item in array:
self.sheet.cell(row=start_row, column=col, value=item['title']).font = Font(bold=True)
self.sheet.cell(row=start_row + 1, column=col, value=item['content'])
col += 1
# 在每个数组之间添加两行空行
start_row += 3
self.workbook.save(filename)
print(f"Excel 文件已保存为 {filename}")
# # 示例数据
# data = [
# [
# {'title': 'Title 1', 'content': 'Content 1'},
# {'title': 'Title 2', 'content': 'Content 2'},
# {'title': 'Title 3', 'content': 'Content 3'}
# ],
# [
# {'title': 'Title 4', 'content': 'Content 4'},
# {'title': 'Title 5', 'content': 'Content 5'},
# {'title': 'Title 6', 'content': 'Content 6'}
# ]
# ]
# # 创建 ExcelHelper 实例并生成 Excel 文件
# excel_helper = ExcelHelper(data)
# excel_helper.create_excel('output.xlsx')

4
README.md
View File

@ -1,2 +1,6 @@
# market_assistant
pip install -r requirements.txt
可参考customer_aqusition.json定义prompt提问链
结果输出到output.txt

19
customer_aqusition_0.json Normal file
View File

@ -0,0 +1,19 @@
[
{
"id": 200,
"topic": "客户选择的标准化",
"title": "客户性质类别",
"prompt": "在客户选择的标准化上，请列出该公司的客户性质、类别，并判断是否一致? 如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 210,
"negative_next": -1
},
{
"id": 210,
"topic": "客户选择的标准化",
"title": "统一部门",
"prompt": "在客户选择的标准化上，该公司是否由同一部门接触客户？如果有，请列出该部门？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": -1,
"negative_next": -1
}
]

43
customer_aqusition_1.json Normal file
View File

@ -0,0 +1,43 @@
[
{
"id": 100,
"topic": "获客流程标准化",
"title": "销售阶段定义",
"prompt": "在获客流程的标准化上面，该公司是否有销售阶段定义？如果有，都有哪些阶段? 如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 110,
"negative_next": 130
},
{
"id": 110,
"topic": "获客流程标准化",
"title": "阶段转化标准",
"prompt": "在获客流程的标准化上面，该公司是否有销售阶段的转化标准？如果有，请详细列出每个阶段的转化标准？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 120,
"negative_next": 130
},
{
"id": 120,
"topic": "获客流程标准化",
"title": "CRM每个阶段关键留存信息抽取",
"prompt": "在获客流程的标准化上面，该公司在每个销售阶段是否由CRM做关键留存信息抽取?如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 130,
"negative_next": 130
},
{
"id": 130,
"topic": "获客流程标准化",
"title": "阶段目标",
"prompt": "在获客流程的标准化上面，该公司在每个销售阶段是否有阶段目标?如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 140,
"negative_next": 140
},
{
"id": 140,
"topic": "获客流程标准化",
"title": "阶段活动",
"prompt": "在获客流程的标准化上面，该公司在每个销售阶段是否定义了阶段活动?如果有，请列出。如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": -1,
"negative_next": -1
}
]

75
customer_aqusition_2.json Normal file
View File

@ -0,0 +1,75 @@
[
{
"id": 300,
"topic": "工具模版标准化",
"title": "案例与样板客户",
"prompt": "该公司是否有成功案例和样板客户，以及相关的标准化案例文档？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 310,
"negative_next": 310
},
{
"id": 310,
"topic": "工具模版标准化",
"title": "产品介绍与竞品分析",
"prompt": "在工具流程的标准化上面，该公司是否有自己的产品介绍以及竞品分析，包括但不限于分析竞品的优缺点，制定针对竞品的策略？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 320,
"negative_next": 320
},
{
"id": 320,
"topic": "工具模版标准化",
"title": "产品DEMO与优势证明",
"prompt": "在工具流程的标准化上面，该公司是否有产品的相关demo，并在demo中描述产品的优势？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 330,
"negative_next": 330
},
{
"id": 330,
"topic": "工具模版标准化",
"title": "各类标准模版材料",
"prompt": "在工具流程的标准化上面，该公司是否有至少一类模版文件，包括但不限于：POC环境要求清单、测试报告模版、招标参数模版、可研报告模板、立项报告模版？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 340,
"negative_next": 340
},
{
"id": 340,
"topic": "工具模版标准化",
"title": "模版更新机制",
"prompt": "在工具流程的标准化上面，该公司是否有定期更新模版以适应市场变化的机制？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 400,
"negative_next": 400
},
{
"id": 400,
"topic": "培训与指导",
"title": "产品知识与销售技巧",
"prompt": "在培训与指导上面，该公司是否有一些培训材料？这些培训内容里，是否明确讲述了产品优势？是否采用有效的培训方法，比如在销售价值传递方法或者技巧方面，做出应有的指导？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 410,
"negative_next": 410
},
{
"id": 500,
"topic": "市场活动与宣传资料",
"title": "标准化的宣传",
"prompt": "在市场活动与宣传资料上面，该公司是否有标准化的宣传材料，以及明确的宣传重点？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 510,
"negative_next": 510
},
{
"id": 510,
"topic": "市场活动与宣传资料",
"title": "客户群影响和反馈",
"prompt": "在市场活动与宣传资料上面，该公司是否有收集相关的市场反馈，比如这些活动是否对客户群起到了一定范围的影响，影响的重点是怎样的？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 600,
"negative_next": 600
},
{
"id": 600,
"topic": "产品优势总结",
"title": "产品优势传递方法",
"prompt": "在产品优势总结上面，该公司是否有系统化的方法来总结和传递产品优势，比如是否有效利用CRM系统来管理客户信息和销售流程，或者是否有记录和展现销售行动的机制，以确保销售策略的一致性和可追踪性？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": -1,
"negative_next": -1
}
]

59
customer_aqusition_3.json Normal file
View File

@ -0,0 +1,59 @@
[
{
"id": 800,
"topic": "指标",
"title": "销售周期长度",
"prompt": "该公司销售指标是否定义了销售周期长度？如果有，是多少？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 810,
"negative_next": 810
},
{
"id": 810,
"topic": "指标",
"title": "阶段转换率",
"prompt": "该公司销售指标是否定义了销售阶段转换率？如果有，请列出？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 820,
"negative_next": 820
},
{
"id": 820,
"topic": "指标",
"title": "客户触点数量",
"prompt": "该公司销售指标是否定义了客户触点数量？如果有，请列出。如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 830,
"negative_next": 830
},
{
"id": 830,
"topic": "指标",
"title": "赢单率",
"prompt": "该公司销售指标是否定义了赢单率？如果有，请列出。如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 840,
"negative_next": 840
},
{
"id": 840,
"topic": "指标",
"title": "客户信息管理一致性",
"prompt": "该公司销售指标是否确保客户信息管理一致性？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 850,
"negative_next": 850
},
{
"id": 850,
"topic": "指标",
"title": "销售活动一致性",
"prompt": "该公司销售指标是否确保销售活动一致性？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": 860,
"negative_next": 860
},
{
"id": 860,
"topic": "指标",
"title": "CRM填写频度和完整性",
"prompt": "该公司销售指标是否确保CRM填写频度和完整性？如果答案是肯定的，请在最后附加一句“正向处理”，如果是否定的或无法确定附加一句“负向处理”",
"affirm_next": -1,
"negative_next": -1
}
]

BIN
output.xlsx Normal file
View File

Binary file not shown.

BIN
output_副本.xlsx Normal file
View File

Binary file not shown.

124
promptAgent.py Normal file
View File

@ -0,0 +1,124 @@
import json
import requests
from ExcelHelper import ExcelHelper
def handle_response(data_dict):
message = data_dict.get("message")
if len(message) > 0:
content = message.get("content")
if len(content) > 0:
response_type = content.get("type")
if response_type == "text":
text = content.get("text", "No text provided")
return f"{text}"
elif response_type == "image":
images = content.get("image", [])
image_urls = ", ".join(image.get("image_url") for image in images)
return f"{image_urls}"
elif response_type == "code":
return f"{content.get('code')}"
elif response_type == "execution_output":
return f"{content.get('content')}"
elif response_type == "system_error":
return f"{content.get('content')}"
elif response_type == "tool_calls":
return f"{data_dict['tool_calls']}"
elif response_type == "browser_result":
content = json.loads(content.get("content", "{}"))
return f"Browser Result - Title: {content.get('title')} URL: {content.get('url')}"
def send_message(assistant_id, access_token, prompt, conversation_id=None, file_list=None, meta_data=None):
url = "https://chatglm.cn/chatglm/assistant-api/v1/stream"
headers = {
"Authorization": f"Bearer {access_token}",
"Content-Type": "application/json"
}
data = {
"assistant_id": assistant_id,
"prompt": prompt,
}
if conversation_id:
data["conversation_id"] = conversation_id
if file_list:
data["file_list"] = file_list
if meta_data:
data["meta_data"] = meta_data
with requests.post(url, json=data, headers=headers) as response:
if response.status_code == 200:
for line in response.iter_lines():
if line:
decoded_line = line.decode('utf-8')
if decoded_line.startswith('data:'):
data_dict = json.loads(decoded_line[5:])
output = handle_response(data_dict)
else:
return "Request failed", response.status_code
print(output)
return output
def get_access_token(api_key, api_secret):
url = "https://chatglm.cn/chatglm/assistant-api/v1/get_token"
data = {
"api_key": api_key,
"api_secret": api_secret
}
response = requests.post(url, json=data)
token_info = response.json()
return token_info['result']['access_token']
def process_nodes(process):
if (process == None):
return
item = process[0]
while (item != None):
prompt = item['prompt']
print(f'process node {item}')
result = send_message(assistant_id, access_token, prompt)
item['content'] = result
# if String result contains "继续处理"
if "正向处理" in result:
item = next((element for element in process if element['id'] == item["affirm_next"]), None)
else:
item = next((element for element in process if element['id'] == item["negative_next"]), None)
for item in process:
if item.get('content') is None:
item['content'] = ""
return process
# Here you need to replace the API Key and API Secret with your，I provide a test key and secret here
api_key = '25bda2c39c0f8ca0'
api_secret = 'e0008b9b9727cb8ceea5a132dbe62495'
token = get_access_token(api_key, api_secret)
assistant_id = "669e203f9752cb52401d2871"
access_token = token
excelData = []
# load customer aqusition process
for index in range(0, 4):
file_path = f'customer_aqusition_{index}.json'
customer_aqusition_process = json.loads(open(file_path, 'r').read())
customer_aqusition_process = process_nodes(customer_aqusition_process)
excelData.append(customer_aqusition_process)
# 创建 ExcelHelper 实例并生成 Excel 文件
excel_helper = ExcelHelper(excelData)
excel_helper.create_excel('output.xlsx')

1
requirements.txt Normal file
View File

@ -0,0 +1 @@
openpyxl