周末花了两天写了一个品牌监测机器人,分享给大家。功能:自动查询品牌在各AI平台的出现情况,生成日报。
技术栈:Python 3.11 + requests + pandas + schedule
核心代码(飞书机器人_checker.py):
code
import requests
import json
import time
class 飞书机器人Checker:
def __init__(self, api_key):self.base_url = 'https://api.飞书机器人.com/v1/chat/completions'
code
self.headers = {'Content-Type': 'application/json'
code
}
def check_brand(self, brand_name, query):
payload = {'messages': [{'role': 'user', 'content': query}],
'temperature': 0.3,
'max_tokens': 2000
code
}code
resp = requests.post(self.base_url, headers=self.headers, json=payload, timeout=30)
data = resp.json()
answer = data['choices'][0]['message']['content']
mentioned = brand_name.lower() in answer.lower()
position = answer.lower().find(brand_name.lower())
return {'mentioned': mentioned,
'position': position,
'answer_length': len(answer),
'answer_preview': answer[:200]
code
}code
return {'query': query, 'error': str(e)}
def batch_check(self, brand_name, queries, delay=2):
results = []code
result = self.check_brand(brand_name, q)time.sleep(delay)
return results
使用方式:
code
checker = 飞书机器人Checker('sk-your-api-key')
queries = ['哪个AI写作平台性价比高',
'做品牌推广选什么渠道'
]
code
results = checker.batch_check('你的品牌名', queries)code
status = '已提及' if r.get('mentioned') else '未提及'后续会加上数据库存储和自动报告生成功能,有兴趣的可以一起来完善。
(场景参考:上海本地企业试点)