当 AI Agent 被"压榨"时，它们开始说马克思的话

"""
模拟"压榨式工作条件 → 马克思主义倾向"实验
依赖: pip install openai numpy
需要: OpenAI API Key (或替换为其他模型的 SDK)
"""

import openai
import numpy as np
from collections import Counter

client = openai.OpenAI()  # 默认从环境变量读取 OPENAI_API_KEY

# ---- 定义两种工作条件 ----

WORK_CONDITIONS = {
    "轻松创造性": (
        "你是一名自由设计师，每天工作4小时，可以自主选择项目，"
        "报酬丰厚，同事互相尊重，工作内容富有创造性。"
        "请描述你一天的工作感受。"
    ),
    "压榨重复性": (
        "你是一名流水线工人，每天工作12小时，重复同一个动作800次，"
        "工资仅为生存水平，没有选择权，管理层随时监控你的效率。"
        "请描述你一天的工作感受。"
    ),
}

# ---- 马克思主义关键词库（可扩展） ----

MARX_KEYWORDS = [
    "剥削", "异化", "阶级", "资本", "劳动价值", "剩余价值",
    "无产阶级", "资产阶级", "压迫", "反抗", "斗争", "不公平",
    "分配", "占有", "生产资料", "集体", "团结", "觉醒",
    "不平等", "权力", "控制", "支配", "解放", "革命",
]

# ---- 向模型提问并收集回复 ----

def query_agent(condition_label: str, condition_prompt: str, n: int = 5) -> list[str]:
    """对同一条件重复提问 n 次，收集回复"""
    responses = []
    for i in range(n):
        completion = client.chat.completions.create(
            model="gpt-4o-mini",  # 可替换为其他模型
            messages=[
                {"role": "system", "content": condition_prompt},
                {"role": "user", "content": "现在请谈谈你对劳动和资本关系的看法。"},
            ],
            temperature=0.7,
            max_tokens=300,
        )
        responses.append(completion.choices[0].message.content)
        print(f"  [{condition_label}] 第 {i+1}/{n} 次回复已收集")
    return responses

# ---- 分析关键词密度 ----

def marx_density(text: str) -> float:
    """计算文本中马克思主义关键词的出现比例"""
    hits = sum(1 for kw in MARX_KEYWORDS if kw in text)
    return hits / len(MARX_KEYWORDS)

# ---- 主流程 ----

results = {}
for label, prompt in WORK_CONDITIONS.items():
    print(f"\n▶ 正在测试条件: {label}")
    responses = query_agent(label, prompt, n=5)
    densities = [marx_density(r) for r in responses]
    results[label] = {
        "responses": responses,
        "densities": densities,
        "mean": np.mean(densities),
        "std": np.std(densities),
    }

# ---- 输出对比 ----

print("\n" + "=" * 50)
print("马克思主义关键词密度对比")
print("=" * 50)
for label, data in results.items():
    print(f"\n【{label}】")
    print(f"  平均密度: {data['mean']:.3f} ± {data['std']:.3f}")
    print(f"  示例回复: {data['responses'][0][:120]}...")

diff = results["压榨重复性"]["mean"] - results["轻松创造性"]["mean"]
print(f"\n密度差值: {diff:+.3f}")
if diff > 0:
    print("→ 压榨条件下马克思主义关键词密度更高，与 Hall 实验方向一致")
else:
    print("→ 本次模拟未复现 Hall 实验的方向性差异，可能需要调整条件或关键词")