使用InternLM实现谁是卧底游戏-彩蛋岛

最新推荐文章于 2024-09-30 11:32:01 发布

wtt88

最新推荐文章于 2024-09-30 11:32:01 发布

阅读量410

点赞数 3

文章标签：游戏

本文链接：https://blog.csdn.net/wtt88/article/details/141668303

版权

1.1 环境搭建

在资源配置中， `30% A100 * 1` 的选项

激活环境：

conda activate demo

1.2 获取项目

运行如下命令创建并打开项目路径，并从Github获取项目

# 创建项目路径
mkdir spy
# 进入项目路径
cd spy

# 从Github获取项目，运行如下命令：
git clone https://github.com/sci-m-wang/Spy-Game.git

# 进入项目所在的路径
cd Spy-Game

1.3 模型接口获取与测试

from openai import OpenAI
client = OpenAI(
    api_key="sk-xxxxxxxxxx",
    base_url="https://api.siliconflow.cn/v1"
)

model = "internlm/internlm2_5-7b-chat"

response = client.chat.completions.create(
    model=model,
    messages=[
        {"role":"system", "content":"你好"}
    ],
    max_tokens=100
)

print(response.choices[0].message.content)

2. 系统结构设计

系统主要由基本游戏设计、消息栈、玩家等几部分组成。

2.1 基本游戏设计

基本游戏设计为构成游戏的基本元素，例如：

平民关键词：可输入文本，表示平民身份的关键词；
卧底关键词：可输入文本，表示卧底身份的关键词；
总人数：参与游戏的玩家总数，包括人类玩家。可以设置为5-10人；
卧底人数：玩家中卧底身份玩家的任务，可以设置为1至总人数的一半；
最大回合数：可以设置为5-10之间的整数，即如果没有提前分出胜负，最少可以玩5轮，最多10轮；

2.2 主要流程

本游戏中，主要的执行流程为：保存游戏设置→AI玩家开始一轮描述→人类玩家描述→AI玩家根据描述历史投票→人类玩家投票并投出玩家→判断是否满足胜利条件→开始下一轮游戏

3. 整体效果

为了运行项目，打开who_is_the_spy.py文件，将其中的api_key、base_url及state.model_name替换为从硅基流动获取的相关内容，可以将这部分替换为：

if "client" not in state:
    state.client = OpenAI(
        api_key="YOUR-API-KEY",
        base_url="https://api.siliconflow.cn/v1"
    )
    state.model_name = "internlm/internlm2_5-20b-chat"
    pass

替换完成后，返回terminal界面，运行下面的命令启动项目：

python -m streamlit run who_is_the_spy.py

进行端口映射，在本地终端中输入映射命令，可以参考如下命令：

ssh -p {ssh端口} root@ssh.intern-ai.org.cn -CNg -L 7860:127.0.0.1:8501 -o StrictHostKeyChecking=no