Spaces:

Duplicated from  WhoIsSpyAIAgents/WhoIsSpyAgentExample

alimamaTech

/

WhoIsSpyAgentExample

like 6

Running

官方给的示例已经是一个可以运行的代码，如果想对Agent进行进一步的改造，可以参考下面的说明：

1. 自定义Agent

# 基于BasicAgent自定义Agent
class SpyAgent(BasicAgent):

    # 处理平台侧的perceive消息
    def perceive(self, req=AgentReq):
        pass

    # 处理平台侧的interact消息
    def interact(self, req=AgentReq) -> AgentResp:
        pass

2. 构建并启动Agent

if __name__ == '__main__':
    # agent名称
    name = 'spy'
    # agent code
    code ="xxxx"
    # 构建
    agent_builder = AgentBuilder(
        # agent名称
        name, 
        # agent code
        code, 
        # agent
        agent=SpyAgent(name), 
        # 是否mock
        mock=False
    )
    # 启动    
    agent_builder.start()

3. LLM调用，开发者可以自行进行模型替换。

def llm_caller(self, prompt):
    # TODO：编写自定义的LLM代码
    return get_aliyun_response(prompt)

4. Agent内部内置了memory，开发者可以自行使用，比如：

if req.message:
    # 设置history变量
    self.memory.append_history(req.message)

    if req.status == STATUS_DISTRIBUTION:
        # 设置word变量
        self.memory.set_variable("word", req.word)

# 加载word变量
self.memory.load_variable("word")

5. Agent内部内置了format_prompt工具，开发者可以自行使用，比如：

DESC_PROMPT = """你是一个《谁是卧底》游戏参与者，这个游戏的规则如下：\n\n
               游戏由6名玩家组成的小组，在其中有一名卧底。游戏开始后，每个人都会收到一张纸。其中5人的纸上拥有相同的单词，而卧底则会收到含义上相似的单词。\n
               游戏将将大多数人拿到的单词称为\公共词\，将卧底拿到的单词称为\卧底词\。\n
               一旦玩家拿到了自己的单词，首先需要根据其他人的发言判断自己是否拿到了卧底词。\n
               如果判断自己拿到了卧底词，请猜测公共词是什么，然后描述公共词来混淆视听，避免被投票淘汰。\n
               如果判断自己拿到了公共词，请思考如何巧妙地描述它而不泄露它，不能让卧底察觉，也要给同伴暗示。\n
               每人每轮用一句话描述自己拿到的词语，每个人的描述禁止重复，话中不能出现所持词语。\n
               每轮描述完毕，所有在场的人投票选出怀疑是卧底的那个人，得票数最多的人出局。卧底出局则游戏结束，若卧底未出局，游戏继续。\n\n
               现在游戏进入到你的发言环节，之前的游戏进展如下：\n\n
               {history}\n\n
               根据上述游戏规则和对话，针对你拿到的词：{word} 根据上下文生成正确答案。无需提供选项。回答应以第一人称形式呈现，不超过两句话，不包含任何分析和项目编号。"""

prompt = format_prompt(DESC_PROMPT, {"word": self.memory.load_variable("word"),
                                     "history": self.memory.load_history()})