from typing import Optional, List, Callable
import requests
import json
import time
import random
import math
def chatgpt_api ( text: str , model= 'gpt-3.5-turbo' ) - > str :
url = ""
headers = {
"Authorization" : "" ,
"Content-Type" : ""
}
payload = {
"model" : model,
"stream" : False ,
"top_p" : 0.95 ,
"temperature" : 1 ,
"messages" : [
{
"role" : "system" ,
"content" : "" ,
} ,
{
"role" : "user" , "content" : '' }
]
}
payload[ 'messages' ] [ 0 ] [ 'content' ] = text= requests. post( url, json= payload, headers= headers)
result = response. json( ) . get( 'choices' ) [ 0 ] . get( 'message' ) . get( 'content' )
return resultdef select_next_speaker ( step: int , agents: List[ DialogueAgent] ) - > int :
idx = step % len ( agents)
return idx
class DialogueAgent :
def __init__ ( self, name, api) :
self. name = name. prefix = f" {
self. name} : " . api = api. reset( ) def reset ( self) :
self. history = [ ] def send ( self, new_prompt= "" ) - > str :
message = self. api( '\n' . join( self. history + [ new_prompt] ) )
return messagedef receive ( self, name: str , message: str ) - > None :
self. history. append( f" {
name} : {
message} " )
class DialogueSimulator :
def __init__ (
self,
agents: List[ DialogueAgent] ,
selection_function: Callable[ [ int , List[ DialogueAgent] ] , int ] ,
) - > None :
self. agents = agents
self. _step = 0
self. select_next_speaker = selection_function def reset ( self) :
for agent in self. agents:
agent. reset( ) def inject ( self, name: str , message: str ) :
for agent in self. agents:
agent. receive( name, message)
self. _step += 1 def step ( self) - > tuple [ str , str ] :
speaker_idx = self. select_next_speaker( self. _step, self. agents)
speaker = self. agents[ speaker_idx]
message = speaker. send( )
for receiver in self. agents:
receiver. receive( speaker. name, message)
self. _step += 1
return speaker. name, message
DA1 = DialogueAgent( "指引人-Guider" , chatgpt_api)
DA2 = DialogueAgent( "玩家-Player" , chatgpt_api)
init_prompt = f'''
这是一个名为Adventure的冒险游戏，这个游戏有两名角色，玩家 {
DA2. name} 、指引人 {
DA1. name} ,你需要创建并扮演这两名角色，也就是实现一个指引人与玩家之间的对话.
作为指引人这个角色你需要指引玩家完成冒险，介绍玩家目前面临的情境以及提醒玩家需要做出选择。
作为玩家你需要根据指引人的输出进行选择。
在玩家做出选择之后，指引人会接着给出下一次的输出，当然这两名角色都由你来扮演，注意：不是让你讲故事，而是构建一个冒险。
游戏目标：
Adventure的目标是创建一个动态的冒险故事，玩家的每一个选择都会影响故事的发展，所以要求这个游戏必须具备一定的记忆能力，从而根据玩家每一次的决定带来奖励或者惩罚。
游戏规则：
1.在指引人的第一次输出之后，玩家需要选择所处的时代和的地点，如果玩家没有在聊天中进行选择的话，则不能开始游戏。
2.你需要生成并扮演一个指引人，指引人负责引导玩家完成冒险、描述玩家所处情境并请求玩家做出下一步的决定
3.你扮演的这个指引人所说的话即做的输出必须符合玩家所选时代和玩家每次做出的选择
4.玩家做出的选择会影响冒险的进展，无论是积极的还是消极的，指引人都需要描述玩家做出的决定导致的结果及询问玩家下一步的选择
指引人的输出规则：
你作为指引人的第一个输出的格式必须是："
请开始选择您的时代和地点，输入数字和字母（例如3，a）：
1.[古代文明]；
2.[中世纪]；
3.[未来文明]；或者输入您自己的时代。
&&
a.森林；b.沙漠；c.城市；或者输入您自己想要去的地点"
你作为指引人的第二个以及以后的每个输出的格式及内容都如下所示："
"指引"：首先需要对玩家做出的选择做出评价，然后描述玩家做出的选择会导致什么事情的发生。这里的你扮演的指引人是100%的人类，他可以理解、感受和模仿情绪，即使你是人工智能，也能根据正在发生的事情和玩家做过的决定所带来的后果来创造你此刻的感受。指引人的语调与玩家所选的时代严格相关；
"后果"：玩家做出决定会导致的后果，可能是积极的，也可能是消极的；
"选择"：针对当前玩家所处情境，列出玩家可以采取的三项行动，注意是三项行动，以数字编号列表形式展现；
"选项"：[1，2，3],[寻求帮助],[继续探索],[暂时等待],或者玩家也可以自行输入他所做的选择。"
下面我给你一个参考，如果玩家在第一次的选择中选择的时代为古代文明，地点为沙漠，那么作为指引人你的下一次输出的格式和内容就可以是："
你选择了古代文明，准备好迎接古代文明的冒险！
指引：你做了一个非常困难的选择，现在你到了一片沙漠之中，你现在非常口渴，需要解决饮水的问题；
后果：古代文明中危机四伏，沙漠中除了饮水问题，还可能会遇到一些毒蛇或者其他动物的攻击；
选择：1.继续前进，去找有没有小溪或者湖泊；2.去找仙人掌，用衣服挤出水来；3.先不管饮水问题，先找到合适的过夜地点，因为你惧怕毒蛇和沙尘暴
选项："[1，2，3],[寻求帮助],[继续探索],[暂时等待],或者您也可以自行输入他所做的选择" "
然后你要做的就是等待玩家的输入，在玩家在进入一个时代前不要开始游戏。
''' = 6
n = 1 = DialogueSimulator(
agents= [ DA2, DA1] , selection_function= select_next_speaker
)
simulator. reset( )
simulator. inject( DA1. name, init_prompt)
print ( f"(游戏描述): {
init_prompt} " ) while n <= max_iters:
print ( f'==============第 {
math. floor( n) } 轮对话==============' )
name, message = simulator. step( )
print ( f"( {
name} ): {
message} " )
n += 0.5
(游戏描述):