sanguo_vnpy/test/backtest/court_discuss_client.py

#!/usr/bin/env python3
"""
朝堂议政引擎 - 手动调用脚本
用于测试和使用朝堂议政功能
"""

import requests
import json
import time

# edict API基础地址
BASE_URL = "http://localhost:7891/api"

# 官员ID列表（可以选择参与议政的官员）
ALL_OFFICIALS = [
    'taizi',      # 太子
    'zhongshu',   # 中书省
    'menxia',     # 门下省
    'shangshu',   # 尚书省
    'libu',       # 礼部
    'hubu',       # 户部
    'bingbu',     # 兵部
    'xingbu',     # 刑部
    'gongbu',     # 工部
    'libu_hr',    # 吏部
]

def print_separator(title=""):
    """打印分隔线"""
    print("\n" + "="*80)
    if title:
        print(f"  {title}")
        print("="*80)

def list_officials():
    """列出所有可用官员"""
    print_separator("👥 可用官员列表")
    try:
        response = requests.get(f"{BASE_URL}/court-discuss/officials")
        result = response.json()
        if result.get('ok'):
            officials = result.get('officials', {})
            for oid, profile in officials.items():
                print(f"\n  {profile['emoji']} {profile['name']} ({oid})")
                print(f"     角色: {profile['role']}")
                print(f"     职责: {profile['duty'][:80]}...")
        else:
            print(f"  ❌ 获取官员列表失败: {result.get('error')}")
    except Exception as e:
        print(f"  ❌ 请求失败: {e}")

def create_session(topic, official_ids, task_id=""):
    """创建朝堂议政会话"""
    print_separator("🏛️ 创建议政会话")
    print(f"  议题: {topic}")
    print(f"  参与官员: {', '.join(official_ids)}")
    if task_id:
        print(f"  关联任务: {task_id}")

    try:
        response = requests.post(
            f"{BASE_URL}/court-discuss/start",
            json={
                'topic': topic,
                'officials': official_ids,
                'taskId': task_id
            }
        )
        result = response.json()
        if result.get('ok'):
            session_id = result.get('taskId') or result.get('sessionId')
            print(f"\n  ✅ 议政会话创建成功！")
            print(f"  会话ID: {session_id}")
            return session_id
        else:
            print(f"\n  ❌ 创建失败: {result.get('error')}")
            return None
    except Exception as e:
        print(f"\n  ❌ 请求失败: {e}")
        return None

def advance_discussion(session_id, user_message=None, decree=None):
    """推进一轮讨论"""
    print_separator("💬 推进讨论")
    print(f"  会话ID: {session_id}")
    if user_message:
        print(f"  皇帝发言: {user_message}")
    if decree:
        print(f"  天命降临: {decree}")

    try:
        response = requests.post(
            f"{BASE_URL}/court-discuss/advance",
            json={
                'sessionId': session_id,
                'userMessage': user_message,
                'decree': decree
            }
        )
        result = response.json()
        if result.get('ok'):
            print(f"\n  ✅ 讨论推进成功！")
            print(f"  第 {result.get('round')} 轮讨论")
            print(f"  新增 {len(result.get('new_messages', []))} 条消息")

            # 打印官员发言
            new_messages = result.get('new_messages', [])
            if new_messages:
                print("\n  📜 官员发言:")
                for msg in new_messages:
                    name = msg.get('name', '?')
                    content = msg.get('content', '')
                    emotion = msg.get('emotion', 'neutral')
                    if len(content) > 100:
                        print(f"\n     {name}: {content[:100]}...")
                    else:
                        print(f"\n     {name}: {content}")
                    if emotion != 'neutral':
                        print(f"     (情绪: {emotion})")

            scene_note = result.get('scene_note')
            if scene_note:
                print(f"\n  🎭 场景: {scene_note}")

            return True
        else:
            print(f"\n  ❌ 推进失败: {result.get('error')}")
            return False
    except Exception as e:
        print(f"\n  ❌ 请求失败: {e}")
        return False

def get_session(session_id):
    """获取会话详情"""
    print_separator("📋 会话详情")
    try:
        response = requests.get(f"{BASE_URL}/court-discuss/session/{session_id}")
        result = response.json()
        if result.get('ok') or result.get('session_id'):
            session = result
            print(f"\n  会话ID: {session.get('session_id')}")
            print(f"  议题: {session.get('topic')}")
            print(f"  轮数: {session.get('round')}")
            print(f"  状态: {session.get('phase')}")
            print(f"  参与官员: {len(session.get('officials', []))} 位")
            print(f"  消息总数: {len(session.get('messages', []))}")

            # 打印所有消息
            messages = session.get('messages', [])
            if messages:
                print("\n  📜 完整消息历史:")
                for msg in messages:
                    msg_type = msg.get('type')
                    timestamp = time.strftime('%H:%M:%S', time.localtime(msg.get('timestamp', 0)))

                    if msg_type == 'system':
                        print(f"\n  [{timestamp}] 系统: {msg.get('content')}")
                    elif msg_type == 'emperor':
                        print(f"\n  [{timestamp}] 👑 皇帝: {msg.get('content')}")
                    elif msg_type == 'decree':
                        print(f"\n  [{timestamp}] ⚡ 天命降临: {msg.get('content')}")
                    elif msg_type == 'official':
                        name = msg.get('official_name', '?')
                        content = msg.get('content', '')
                        print(f"\n  [{timestamp}] {name}: {content}")
                    elif msg_type == 'scene_note':
                        print(f"\n  [{timestamp}] 🎭 场景: {msg.get('content')}")

            return session
        else:
            print(f"\n  ❌ 获取失败: {result.get('error')}")
            return None
    except Exception as e:
        print(f"\n  ❌ 请求失败: {e}")
        return None

def list_sessions():
    """列出所有活跃会话"""
    print_separator("📋 活跃会话列表")
    try:
        response = requests.get(f"{BASE_URL}/court-discuss/list")
        result = response.json()
        if result.get('ok'):
            sessions = result.get('sessions', [])
            if sessions:
                for s in sessions:
                    print(f"\n  会话ID: {s.get('session_id')}")
                    print(f"  议题: {s.get('topic')}")
                    print(f"  轮数: {s.get('round')}")
                    print(f"  状态: {s.get('phase')}")
                    print(f"  官员数: {s.get('official_count')}")
                    print(f"  消息数: {s.get('message_count')}")
            else:
                print("  暂无活跃会话")
            return sessions
        else:
            print(f"  ❌ 获取失败: {result.get('error')}")
            return []
    except Exception as e:
        print(f"  ❌ 请求失败: {e}")
        return []

def conclude_session(session_id):
    """结束议政会话"""
    print_separator("🏁 结束议政")
    try:
        response = requests.post(
            f"{BASE_URL}/court-discuss/conclude",
            json={'sessionId': session_id}
        )
        result = response.json()
        if result.get('ok'):
            print(f"\n  ✅ 议政结束！")
            print(f"  总结: {result.get('summary')}")
            return True
        else:
            print(f"\n  ❌ 结束失败: {result.get('error')}")
            return False
    except Exception as e:
        print(f"\n  ❌ 请求失败: {e}")
        return False

def get_fate_event():
    """获取随机命运事件"""
    print_separator("🎲 命运骰子")
    try:
        response = requests.get(f"{BASE_URL}/court-discuss/fate")
        result = response.json()
        if result.get('ok'):
            event = result.get('event')
            print(f"\n  🎭 {event}")
            return event
        else:
            print(f"  ❌ 获取失败: {result.get('error')}")
            return None
    except Exception as e:
        print(f"  ❌ 请求失败: {e}")
        return None

def demo_full_flow():
    """演示完整流程"""
    print_separator("🚀 朝堂议政完整流程演示")

    # 1. 列出官员
    list_officials()

    # 2. 选择官员参与议政
    selected_officials = ['zhongshu', 'menxia', 'shangshu', 'hubu']
    print(f"\n\n  选择参与议政的官员: {', '.join(selected_officials)}")

    # 3. 创建议政会话
    topic = "如何建立一个稳定高效的量化交易系统？"
    session_id = create_session(topic, selected_officials)

    if not session_id:
        return

    # 等待一下
    time.sleep(2)

    # 4. 推进几轮讨论
    for i in range(3):
        print_separator(f"🔄 第 {i+1} 轮讨论")
        advance_discussion(session_id)
        time.sleep(3)

    # 5. 皇帝发言
    print_separator("👑 皇帝发言")
    advance_discussion(
        session_id,
        user_message="诸位爱卿的讨论很有见地，朕想听听关于风险控制方面的具体建议。"
    )

    time.sleep(3)

    # 6. 再来一轮
    advance_discussion(session_id)

    time.sleep(3)

    # 7. 获取会话详情
    get_session(session_id)

    # 8. 结束议政
    conclude_session(session_id)

def interactive_mode():
    """交互式模式"""
    print_separator("🎮 朝堂议政 - 交互式模式")
    print("\n  欢迎使用朝堂议政引擎！")
    print("  输入 'help' 查看可用命令")

    current_session_id = None

    while True:
        try:
            cmd = input("\n  > ").strip()

            if cmd == 'help' or cmd == '?':
                print("\n  可用命令:")
                print("    officials        - 列出所有官员")
                print("    create           - 创建议政会话")
                print("    advance          - 推进讨论")
                print("    emperor <msg>    - 皇帝发言")
                print("    decree <msg>     - 天命降临（下旨）")
                print("    fate             - 命运骰子")
                print("    session          - 查看当前会话")
                print("    list             - 列出所有会话")
                print("    conclude         - 结束议政")
                print("    demo             - 运行完整演示")
                print("    clear            - 清屏")
                print("    quit / exit      - 退出")

            elif cmd == 'officials':
                list_officials()

            elif cmd == 'create':
                topic = input("  请输入议题: ").strip()
                if not topic:
                    print("  ❌ 议题不能为空！")
                    continue

                print("\n  可用官员:")
                for i, oid in enumerate(ALL_OFFICIALS, 1):
                    print(f"    {i}. {oid}")

                selection = input("\n  请选择参与官员（用逗号分隔，如: 1,2,3,4）: ").strip()
                try:
                    indices = [int(x.strip()) - 1 for x in selection.split(',')]
                    selected = [ALL_OFFICIALS[i] for i in indices if 0 <= i < len(ALL_OFFICIALS)]
                except:
                    print("  ❌ 选择无效！")
                    continue

                if len(selected) < 2:
                    print("  ❌ 至少需要选择2位官员！")
                    continue

                task_id = input("  关联任务ID（可选，直接回车跳过）: ").strip()

                current_session_id = create_session(topic, selected, task_id)

            elif cmd == 'advance':
                if not current_session_id:
                    print("  ❌ 没有活跃会话，请先创建！")
                    continue
                advance_discussion(current_session_id)

            elif cmd.startswith('emperor '):
                if not current_session_id:
                    print("  ❌ 没有活跃会话，请先创建！")
                    continue
                msg = cmd[len('emperor '):].strip()
                advance_discussion(current_session_id, user_message=msg)

            elif cmd.startswith('decree '):
                if not current_session_id:
                    print("  ❌ 没有活跃会话，请先创建！")
                    continue
                msg = cmd[len('decree '):].strip()
                advance_discussion(current_session_id, decree=msg)

            elif cmd == 'fate':
                event = get_fate_event()
                if event and current_session_id:
                    if input("  是否将此事件加入讨论？(y/n): ").strip().lower() == 'y':
                        advance_discussion(current_session_id, decree=f"【命运骰子】{event}")

            elif cmd == 'session':
                if not current_session_id:
                    print("  ❌ 没有活跃会话！")
                    continue
                get_session(current_session_id)

            elif cmd == 'list':
                sessions = list_sessions()
                if sessions and not current_session_id:
                    choice = input("  选择会话（输入ID）: ").strip()
                    if choice:
                        current_session_id = choice

            elif cmd == 'conclude':
                if not current_session_id:
                    print("  ❌ 没有活跃会话！")
                    continue
                if conclude_session(current_session_id):
                    current_session_id = None

            elif cmd == 'demo':
                demo_full_flow()

            elif cmd == 'clear':
                import os
                os.system('clear' if os.name == 'posix' else 'cls')

            elif cmd == 'quit' or cmd == 'exit':
                print("\n  👋 再见！")
                break

            elif cmd:
                print(f"  ❌ 未知命令: {cmd}")

        except KeyboardInterrupt:
            print("\n\n  👋 再见！")
            break
        except Exception as e:
            print(f"  ❌ 错误: {e}")

if __name__ == '__main__':
    import sys

    if len(sys.argv) > 1 and sys.argv[1] == 'demo':
        demo_full_flow()
    else:
        interactive_mode()