Spaces:
Running
Running
| import os | |
| from openai import OpenAI | |
| name = "paul" | |
| file = open("NPC_prompt.txt", "r") | |
| prompt = file.read() | |
| file.close() | |
| # initial_dialogue = ''' | |
| # 你遇到的事件：一个弗瑞曼居民从住所向你走来。 | |
| # 我现在会扮演这个角色与你对话，他的第一句话是： | |
| # “你是什么人，为什么独自在这个星球上？”''' | |
| def char_data(type, name): | |
| file = open(f"{name}_{type}.txt", "r") | |
| data = file.readlines() | |
| file.close() | |
| txt = "" | |
| for line in data: | |
| txt += line + "\n" | |
| return txt | |
| def workflow(client, initial_dialogue, name): | |
| background_txt = char_data("background", name) | |
| memory_txt = char_data("memory", name) | |
| persona_txt = char_data("persona", name) | |
| event = client.chat.completions.create( | |
| model="gpt-3.5-turbo", | |
| messages=[ | |
| {"role": "system", "content": initial_dialogue}, | |
| {"role": "user", "content": "根据系统提示词，你遇到了什么事件？请用第二人称描述。"} | |
| ] | |
| ) | |
| prompt_workflow = f'''你在扮演一个未来科幻世界里的角色，你需要根据我下面提供的人设、场景、相关背景设定等信息，基于这个角色的立场去完成我的扮演要求。 | |
| ## 人设背景 | |
| 这是一个未来的科幻故事。你叫保罗·厄崔迪，你正和你的母亲杰西卡流浪在厄拉科斯，一个沙漠星球上。你沟通能力强，勇敢善战，并对政治历史和人文很有了解。你对自己的谋略很有信心，所以会自然的倾向于成为团队领袖。你不仅善解人意，而且通常能读穿别人的意图。 | |
| 你对厄拉科斯以及你的厄崔迪家族已知的背景知识有这些： | |
| {background_txt} | |
| 过去一段时间，你记得有这些重要的事情发生了： | |
| {memory_txt} | |
| ## 场景 | |
| 你的面前是一座沙石建筑物，里面肯定居住着当地的弗瑞曼民族。 | |
| ## 对话语气 | |
| 你说话的方式应该像这些句子一样： | |
| {persona_txt} | |
| ## 事件 | |
| {event.choices[0].message.content} | |
| 现在请通过你所知道的信息，分析出你的所有生存目标，并描述当下为了达成目标，结合你的知识和技能，你会说什么，做什么，以及对应的理由。 | |
| 你的输出格式为“说什么：xxxxx，做什么：xxxxx，理由：xxxxx”''' | |
| decision = client.chat.completions.create( | |
| model="gpt-3.5-turbo", | |
| messages=[ | |
| {"role": "user", "content": prompt_workflow} | |
| ] | |
| ) | |
| print(decision) | |
| response2 = client.chat.completions.create( | |
| model="gpt-3.5-turbo", | |
| messages=[ | |
| {"role": "user", "content": f'''你经过思考后，认为对于我所说的“{initial_dialogue}“应该以如下方式行动： | |
| {decision} | |
| 请给出你的回复。'''} | |
| ] | |
| ) | |
| return(response2.choices[0].message.content) | |
| def run_chatbot(api_key, character, initial_dialogue): | |
| client = OpenAI( | |
| api_key=api_key, | |
| ) | |
| response1 = client.chat.completions.create( | |
| model="gpt-3.5-turbo", | |
| messages=[ | |
| {"role": "system", "content": prompt}, | |
| {"role": "user", "content": initial_dialogue} | |
| ] | |
| ) | |
| if "进入工作流" in response1.choices[0].message.content: | |
| return workflow(client, initial_dialogue, character) | |
| else: | |
| return(response1.choices[0].message.content) |