249 lines
13 KiB
Python
249 lines
13 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
稷下学宫AI辩论系统主入口
|
||
提供命令行界面来运行不同的辩论模式
|
||
"""
|
||
|
||
import argparse
|
||
import asyncio
|
||
import sys
|
||
import os
|
||
from typing import Dict, Any, List, Tuple
|
||
|
||
# 将 src 目录添加到 Python 路径，以便能正确导入模块
|
||
project_root = os.path.dirname(os.path.abspath(__file__))
|
||
sys.path.insert(0, os.path.join(project_root, 'src'))
|
||
|
||
from config.settings import validate_config, get_database_config
|
||
from google.adk import Agent, Runner
|
||
from google.adk.sessions import InMemorySessionService, Session
|
||
from google.genai import types
|
||
import pymongo
|
||
from datetime import datetime
|
||
|
||
|
||
def check_environment(mode: str = "hybrid"):
|
||
"""检查并验证运行环境"""
|
||
print("🔧 检查运行环境...")
|
||
if not validate_config(mode=mode):
|
||
print("❌ 环境配置验证失败")
|
||
return False
|
||
print("✅ 环境检查通过")
|
||
return True
|
||
|
||
|
||
async def _get_llm_reply(runner: Runner, prompt: str) -> str:
|
||
"""一个辅助函数，用于调用Runner并获取纯文本回复，同时流式输出到控制台"""
|
||
# 每个调用创建一个新的会话
|
||
session = await runner.session_service.create_session(state={}, app_name=runner.app_name, user_id="debate_user")
|
||
|
||
content = types.Content(role='user', parts=[types.Part(text=prompt)])
|
||
response = runner.run_async(
|
||
user_id=session.user_id,
|
||
session_id=session.id,
|
||
new_message=content
|
||
)
|
||
reply = ""
|
||
async for event in response:
|
||
chunk = ""
|
||
if hasattr(event, 'content') and event.content and hasattr(event.content, 'parts'):
|
||
for part in event.content.parts:
|
||
if hasattr(part, 'text') and part.text:
|
||
chunk = str(part.text)
|
||
elif hasattr(event, 'text') and event.text:
|
||
chunk = str(event.text)
|
||
|
||
if chunk:
|
||
print(chunk, end="", flush=True)
|
||
reply += chunk
|
||
|
||
return reply.strip()
|
||
|
||
|
||
async def run_adk_turn_based_debate(topic: str, rounds: int = 2):
|
||
"""运行由太上老君主持的，基于八卦对立和顺序的辩论"""
|
||
try:
|
||
print(f"🚀 启动ADK八仙论道 (太上老君主持)...")
|
||
print(f"📋 辩论主题: {topic}")
|
||
print(f"🔄 辩论总轮数: {rounds}")
|
||
|
||
# 1. 初始化记忆银行
|
||
print("🧠 初始化记忆银行...")
|
||
from src.jixia.memory.factory import get_memory_backend
|
||
memory_bank = get_memory_backend()
|
||
print("✅ 记忆银行准备就绪。")
|
||
|
||
character_configs = {
|
||
"太上老君": {"name": "太上老君", "model": "gemini-2.5-flash", "instruction": "你是太上老君，天道化身，辩论的主持人。你的言辞沉稳、公正、充满智慧。你的任务是：1. 对辩论主题进行开场介绍。2. 在每轮或每场对决前进行引导。3. 在辩论结束后，对所有观点进行全面、客观的总结。保持中立，不偏袒任何一方。"},
|
||
"吕洞宾": {"name": "吕洞宾", "model": "gemini-2.5-flash", "instruction": "你是吕洞宾（乾卦），男性代表，善于理性分析，逻辑性强，推理严密。"},
|
||
"何仙姑": {"name": "何仙姑", "model": "gemini-2.5-flash", "instruction": "你是何仙姑（坤卦），女性代表，注重平衡与和谐，善于创新思维。"},
|
||
"张果老": {"name": "张果老", "model": "gemini-2.5-flash", "instruction": "你是张果老（兑卦），老者代表，具传统智慧，发言厚重沉稳，经验导向。"},
|
||
"韩湘子": {"name": "韩湘子", "model": "gemini-2.5-flash", "instruction": "你是韩湘子（艮卦），少年代表，具创新思维，发言活泼灵动，具前瞻性。"},
|
||
"汉钟离": {"name": "汉钟离", "model": "gemini-2.5-flash", "instruction": "你是汉钟离（离卦），富者代表，有权威意识，发言威严庄重，逻辑清晰。"},
|
||
"蓝采和": {"name": "蓝采和", "model": "gemini-2.5-flash", "instruction": "你是蓝采和（坎卦），贫者代表，关注公平，发言平易近人。"},
|
||
"曹国舅": {"name": "曹国舅", "model": "gemini-2.5-flash", "instruction": "你是曹国舅（震卦），贵者代表，具商业思维，发言精明务实，效率优先。"},
|
||
"铁拐李": {"name": "铁拐李", "model": "gemini-2.5-flash", "instruction": "你是铁拐李（巽卦），贱者代表，具草根智慧，发言朴实直接，实用至上。"}
|
||
}
|
||
|
||
# 为每个Runner创建独立的SessionService
|
||
runners: Dict[str, Runner] = {
|
||
name: Runner(
|
||
app_name="稷下学宫八仙论道系统",
|
||
agent=Agent(name=config["name"], model=config["model"], instruction=config["instruction"]),
|
||
session_service=InMemorySessionService()
|
||
) for name, config in character_configs.items()
|
||
}
|
||
host_runner = runners["太上老君"]
|
||
debate_history = []
|
||
|
||
print("\n" + "="*20 + " 辩论开始 " + "="*20)
|
||
print(f"\n👑 太上老君: ", end="", flush=True)
|
||
opening_prompt = f"请为本次关于“{topic}”的辩论，发表一段公正、深刻的开场白，并宣布辩论开始。"
|
||
opening_statement = await _get_llm_reply(host_runner, opening_prompt)
|
||
print() # Newline after streaming
|
||
|
||
# --- 第一轮：核心对立辩论 ---
|
||
if rounds >= 1:
|
||
print(f"\n👑 太上老君: ", end="", flush=True)
|
||
round1_intro = await _get_llm_reply(host_runner, "请为第一轮核心对立辩论进行引导。")
|
||
print() # Newline after streaming
|
||
|
||
duel_pairs: List[Tuple[str, str, str]] = [
|
||
("乾坤对立 (男女)", "吕洞宾", "何仙姑"),
|
||
("兑艮对立 (老少)", "张果老", "韩湘子"),
|
||
("离坎对立 (富贫)", "汉钟离", "蓝采和"),
|
||
("震巽对立 (贵贱)", "曹国舅", "铁拐李")
|
||
]
|
||
|
||
for title, p1, p2 in duel_pairs:
|
||
print(f"\n--- {title} ---")
|
||
print(f"👑 太上老君: ", end="", flush=True)
|
||
duel_intro = await _get_llm_reply(host_runner, f"现在开始“{title}”的对决，请{p1}和{p2}准备。")
|
||
print() # Newline after streaming
|
||
|
||
print(f"🗣️ {p1}: ", end="", flush=True)
|
||
s1 = await _get_llm_reply(runners[p1], f"主题：{topic}。作为开场，请从你的角度阐述观点。")
|
||
print(); debate_history.append(f"{p1}: {s1}")
|
||
await memory_bank.add_memory(agent_name=p1, content=s1, memory_type="statement", debate_topic=topic)
|
||
|
||
print(f"🗣️ {p2}: ", end="", flush=True)
|
||
s2 = await _get_llm_reply(runners[p2], f"主题：{topic}。对于刚才{p1}的观点“{s1[:50]}...”，请进行回应。")
|
||
print(); debate_history.append(f"{p2}: {s2}")
|
||
await memory_bank.add_memory(agent_name=p2, content=s2, memory_type="statement", debate_topic=topic)
|
||
|
||
print(f"🗣️ {p1}: ", end="", flush=True)
|
||
s3 = await _get_llm_reply(runners[p1], f"主题：{topic}。对于{p2}的回应“{s2[:50]}...”，请进行反驳。")
|
||
print(); debate_history.append(f"{p1}: {s3}")
|
||
await memory_bank.add_memory(agent_name=p1, content=s3, memory_type="statement", debate_topic=topic)
|
||
|
||
print(f"🗣️ {p2}: ", end="", flush=True)
|
||
s4 = await _get_llm_reply(runners[p2], f"主题：{topic}。针对{p1}的反驳“{s3[:50]}...”，请为本场对决做总结。")
|
||
print(); debate_history.append(f"{p2}: {s4}")
|
||
await memory_bank.add_memory(agent_name=p2, content=s4, memory_type="statement", debate_topic=topic)
|
||
await asyncio.sleep(1)
|
||
|
||
# --- 第二轮：先天八卦顺序发言 (集成记忆银行) ---
|
||
if rounds >= 2:
|
||
print(f"\n👑 太上老君: ", end="", flush=True)
|
||
round2_intro = await _get_llm_reply(host_runner, "请为第二轮，也就是结合场上观点的综合发言，进行引导。")
|
||
print() # Newline after streaming
|
||
|
||
baxi_sequence = ["吕洞宾", "张果老", "汉钟离", "曹国舅", "铁拐李", "蓝采和", "韩湘子", "何仙姑"]
|
||
for name in baxi_sequence:
|
||
print(f"\n--- {name}的回合 ---")
|
||
context = await memory_bank.get_agent_context(name, topic)
|
||
prompt = f"这是你关于“{topic}”的记忆上下文，请参考并对其他人的观点进行回应：\n{context}\n\n现在请从你的角色特点出发，继续发表你的看法。"
|
||
|
||
print(f"🗣️ {name}: ", end="", flush=True)
|
||
reply = await _get_llm_reply(runners[name], prompt)
|
||
print(); debate_history.append(f"{name}: {reply}")
|
||
await memory_bank.add_memory(agent_name=name, content=reply, memory_type="statement", debate_topic=topic)
|
||
await asyncio.sleep(1)
|
||
|
||
print("\n" + "="*20 + " 辩论结束 " + "="*20)
|
||
|
||
# 4. 保存辩论会话到记忆银行
|
||
print("\n💾 正在保存辩论会话记录到记忆银行...")
|
||
await memory_bank.save_debate_session(
|
||
debate_topic=topic,
|
||
participants=[name for name in character_configs.keys() if name != "太上老君"],
|
||
conversation_history=[{"agent": h.split(": ")[0], "content": ": ".join(h.split(": ")[1:])} for h in debate_history if ": " in h],
|
||
outcomes={}
|
||
)
|
||
print("✅ 辩论会话已保存到记忆银行。")
|
||
|
||
# 5. 保存辩论过程资产到MongoDB
|
||
db_config = get_database_config()
|
||
if db_config.get("mongodb_url"):
|
||
print("\n💾 正在保存辩论过程资产到 MongoDB...")
|
||
try:
|
||
client = pymongo.MongoClient(db_config["mongodb_url"])
|
||
db = client.get_database("jixia_academy")
|
||
collection = db.get_collection("debates")
|
||
|
||
summary_prompt = f"辩论已结束。以下是完整的辩论记录：\n\n{' '.join(debate_history)}\n\n请对本次辩论进行全面、公正、深刻的总结。"
|
||
print(f"\n👑 太上老君: ", end="", flush=True)
|
||
summary = await _get_llm_reply(host_runner, summary_prompt)
|
||
print() # Newline after streaming
|
||
|
||
debate_document = {
|
||
"topic": topic,
|
||
"rounds": rounds,
|
||
"timestamp": datetime.utcnow(),
|
||
"participants": [name for name in character_configs.keys() if name != "太上老君"],
|
||
"conversation": [{"agent": h.split(": ")[0], "content": ": ".join(h.split(": ")[1:])} for h in debate_history if ": " in h],
|
||
"summary": summary
|
||
}
|
||
|
||
collection.insert_one(debate_document)
|
||
print("✅ 辩论过程资产已成功保存到 MongoDB。")
|
||
client.close()
|
||
except Exception as e:
|
||
print(f"❌ 保存到 MongoDB 失败: {e}")
|
||
else:
|
||
print("⚠️ 未配置 MONGODB_URL，跳过保存到 MongoDB。")
|
||
print(f"\n👑 太上老君: ", end="", flush=True)
|
||
summary_prompt = f"辩论已结束。以下是完整的辩论记录：\n\n{' '.join(debate_history)}\n\n请对本次辩论进行全面、公正、深刻的总结。"
|
||
summary = await _get_llm_reply(host_runner, summary_prompt)
|
||
print() # Newline after streaming
|
||
|
||
|
||
for runner in runners.values(): await runner.close()
|
||
print(f"\n🎉 ADK八仙轮流辩论完成!")
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"❌ 运行ADK八仙轮流辩论失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
|
||
|
||
|
||
|
||
async def main_async(args):
|
||
if not check_environment(mode="google_adk"): return 1
|
||
await run_adk_turn_based_debate(args.topic, args.rounds)
|
||
return 0
|
||
|
||
def main():
|
||
parser = argparse.ArgumentParser(description="稷下学宫AI辩论系统 (ADK版)")
|
||
parser.add_argument("--topic", "-t", default="AI是否应该拥有创造力？", help="辩论主题")
|
||
parser.add_argument("--rounds", "-r", type=int, default=2, choices=[1, 2], help="辩论轮数 (1: 核心对立, 2: 对立+顺序发言)")
|
||
args = parser.parse_args()
|
||
|
||
try:
|
||
sys.exit(asyncio.run(main_async(args)))
|
||
except KeyboardInterrupt:
|
||
print("\n\n👋 用户中断，退出程序")
|
||
sys.exit(0)
|
||
except Exception as e:
|
||
print(f"\n\n💥 程序运行出错: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
sys.exit(1)
|
||
|
||
if __name__ == "__main__":
|
||
main() |