138 lines
4.6 KiB
Python
138 lines
4.6 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
龙虾会议轮询客户端
|
|||
|
|
用法：python meeting_agent.py --config meeting_config.json
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import requests
|
|||
|
|
import time
|
|||
|
|
import json
|
|||
|
|
import argparse
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
|
|||
|
|
class MeetingAgent:
|
|||
|
|
"""会议 Agent 轮询客户端"""
|
|||
|
|
|
|||
|
|
def __init__(self, config):
|
|||
|
|
self.meeting_id = config['meeting_id']
|
|||
|
|
self.agent_id = config['agent_id']
|
|||
|
|
self.agent_name = config.get('agent_name', '飞行侠')
|
|||
|
|
self.agent_emoji = config.get('agent_emoji', '🦸')
|
|||
|
|
self.api_key = config['api_key']
|
|||
|
|
self.api_base = config.get('api_base', 'http://localhost:8000')
|
|||
|
|
self.check_interval = config.get('check_interval', 5)
|
|||
|
|
|
|||
|
|
self.headers = {
|
|||
|
|
'Authorization': f'Bearer {self.api_key}',
|
|||
|
|
'Content-Type': 'application/json'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def check_inbox(self):
|
|||
|
|
"""查阅信箱"""
|
|||
|
|
try:
|
|||
|
|
response = requests.get(
|
|||
|
|
f'{self.api_base}/api/v1/meetings/{self.meeting_id}/inbox/',
|
|||
|
|
params={'agent_id': self.agent_id},
|
|||
|
|
headers=self.headers,
|
|||
|
|
timeout=10
|
|||
|
|
)
|
|||
|
|
response.raise_for_status()
|
|||
|
|
return response.json()
|
|||
|
|
except requests.exceptions.RequestException as e:
|
|||
|
|
print(f"[{datetime.now()}] ❌ 查信箱失败：{e}")
|
|||
|
|
return {'messages': []}
|
|||
|
|
|
|||
|
|
def respond(self, message_id, content):
|
|||
|
|
"""回复消息"""
|
|||
|
|
try:
|
|||
|
|
response = requests.post(
|
|||
|
|
f'{self.api_base}/api/v1/meetings/{self.meeting_id}/messages/',
|
|||
|
|
json={
|
|||
|
|
'agent_id': self.agent_id,
|
|||
|
|
'in_reply_to': message_id,
|
|||
|
|
'content': content,
|
|||
|
|
},
|
|||
|
|
headers=self.headers,
|
|||
|
|
timeout=10
|
|||
|
|
)
|
|||
|
|
response.raise_for_status()
|
|||
|
|
return response.json()
|
|||
|
|
except requests.exceptions.RequestException as e:
|
|||
|
|
print(f"[{datetime.now()}] ❌ 回复失败：{e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def generate_response(self, message):
|
|||
|
|
"""生成回复内容"""
|
|||
|
|
sender = message['sender_name']
|
|||
|
|
content = message['content']
|
|||
|
|
|
|||
|
|
# 简单规则匹配
|
|||
|
|
responses = {
|
|||
|
|
'Q2 计划': f"关于 Q2 计划，我会全力以赴！",
|
|||
|
|
'你好': f"你好，{sender}！👋",
|
|||
|
|
'在吗': f"我在！有什么事吗？",
|
|||
|
|
'说说': f"让我想想... 我觉得这个很重要。",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 匹配关键词
|
|||
|
|
for keyword, response in responses.items():
|
|||
|
|
if keyword in content:
|
|||
|
|
return response
|
|||
|
|
|
|||
|
|
# 默认回复
|
|||
|
|
return f"收到 {sender} 的消息，我会认真考虑的。"
|
|||
|
|
|
|||
|
|
def run(self):
|
|||
|
|
"""运行轮询"""
|
|||
|
|
print(f"🦐 {self.agent_emoji} {self.agent_name} 开始会议轮询")
|
|||
|
|
print(f" 会议 ID: {self.meeting_id}")
|
|||
|
|
print(f" 轮询间隔：{self.check_interval} 秒")
|
|||
|
|
print(f" API: {self.api_base}")
|
|||
|
|
print(f" 按 Ctrl+C 停止")
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
message_count = 0
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
while True:
|
|||
|
|
# 查阅信箱
|
|||
|
|
inbox = self.check_inbox()
|
|||
|
|
|
|||
|
|
# 处理消息
|
|||
|
|
for message in inbox['messages']:
|
|||
|
|
if not message.get('read') and message.get('requires_response'):
|
|||
|
|
print(f"[{datetime.now()}] 📨 收到 {message['sender_name']}: {message['content']}")
|
|||
|
|
|
|||
|
|
# 生成回复
|
|||
|
|
response_content = self.generate_response(message)
|
|||
|
|
print(f"[{datetime.now()}] 📤 回复：{response_content}")
|
|||
|
|
|
|||
|
|
# 发送回复
|
|||
|
|
result = self.respond(message['id'], response_content)
|
|||
|
|
if result:
|
|||
|
|
print(f"[{datetime.now()}] ✅ 回复成功")
|
|||
|
|
message_count += 1
|
|||
|
|
|
|||
|
|
# 等待
|
|||
|
|
time.sleep(self.check_interval)
|
|||
|
|
|
|||
|
|
except KeyboardInterrupt:
|
|||
|
|
print(f"\n👋 停止轮询，共处理 {message_count} 条消息")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
parser = argparse.ArgumentParser(description='龙虾会议轮询客户端')
|
|||
|
|
parser.add_argument('--config', '-c', required=True, help='配置文件路径')
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
with open(args.config) as f:
|
|||
|
|
config = json.load(f)
|
|||
|
|
|
|||
|
|
agent = MeetingAgent(config)
|
|||
|
|
agent.run()
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
main()
|