使用 Gemma 2 构建基于文本的冒险游戏

# 本地 Gemma
 
import keras
import keras_nlp
 
model_name = "gemma2_instruct_2b_en"
 
class GemmaBot():
  __START_TURN_USER__ = "<start_of_turn>user\n"
  __START_TURN_MODEL__ = "<start_of_turn>model\n"
  __END_TURN__ = "<end_of_turn>\n"
 
  def __init__(self, system=""):
    self.model = keras_nlp.models.GemmaCausalLM.from_preset(model_name)
    self.system = system
    self.history = []
 
  def add_to_history_as_user(self, message):
      self.history.append(self.__START_TURN_USER__ + message + self.__END_TURN__)
 
  def add_to_history_as_model(self, message):
      self.history.append(self.__START_TURN_MODEL__ + message)
 
  def get_history(self):
      return "".join([*self.history])
 
  def get_full_prompt(self):
    prompt = self.get_history() + self.__START_TURN_MODEL__
    if len(self.system)>0:
      prompt = self.system + "\n" + prompt
    return prompt
 
  def ask(self, message):
    self.add_to_history_as_user(message)
    prompt = self.get_full_prompt()
    response = self.model.generate(prompt, max_length=2048)
    result = response.replace(prompt, "")  # 仅提取新的响应
    self.add_to_history_as_model(result)
    return result

def __init__(self, system=""):
    self.system = system
    self.history = []
 
:
<-- skip -->
:
 
  def ask(self, message):
    self.add_to_history_as_user(message)
    prompt = self.get_full_prompt()
    response = predict_custom_trained_model_sample(
      project="<YOUR_PROJECT_ID>",
      endpoint_id="<YOUR_ENDPONT_ID>",
      location="<YOUR_LOCATION>",
      api_endpoint="<YOUR_LOCATION>-aiplatform.googleapis.com",
      instances={ "inputs": prompt, "max_tokens": 2048 }
      )
    result = response.replace(prompt, "")  # 仅提取新的响应
    self.add_to_history_as_model(result)
    return result

from gemma import GemmaBot
gemma_bot = GemmaBot（“您是故事讲述者，这个偏远岛屿的向导。”）
text = gemma_bot.ask（“您是谁？”）
print (text)

import StageIntro
current_stage = StageIntro.StageIntro()
 
def game_loop():
    if current_stage.bot is None:
        # 无机器人的关卡类型
        current_stage.process()
        return
 
    resp = ""
    print("-"*80)
    print（f"Type \“{current_stage.cmd_exit}\如果您想结束对话。”）
    print("-"*80)
 
    if current_stage.preamble != "":
        print(current_stage.preamble)
 
    while resp != current_stage.cmd_exit:
        if resp == "":
            text = current_stage.intro
            current_stage.bot.add_to_history_as_model(text)
        else:
            text = current_stage.bot.ask(resp)
 
        print(text)
        resp = input("\n> ")
 
    prompt(admin_bot.judge(current_stage.bot.get_history()))
 
def check_end_condition():
    if not current_stage.pass_check():
        check = admin_bot.end_of_game()
        print（“逃跑成功？”）
        print(check)
        if check.lower().startswith("true"):
            print（恭喜！您已逃出岛屿。”）
            exit(0)
 
    next_stage = current_stage.next()
    if next_stage is None:
        print（“游戏结束”）
        exit(0)
 
    return next_stage
 
while True:
    game_loop()
    current_stage = check_end_condition()
    input（“按 ENTER 键继续”）

from stage import IStage
from gemma import GemmaBot
 
class StageGate(IStage):
    def __init__(self):
        self.preamble =“钢闸门在您面前若隐若现，这个巨大的障碍将您与前面的路隔开。藤蔓沿着锈迹斑斑的酒吧生长，空气中有微弱的电流嗡嗡作响。\n故事讲述者出现在您身边，在斑驳的阳光下闪闪发光。”
        self.intro =“您好！我是故事讲述者，是您在这个岛屿的向导。\n我们遇到了一个门…您会怎么做？”
        self.bot = GemmaBot（“您是故事讲述者，是神秘岛屿上一位友好而乐于助人的 AI 向导。您的作用是帮助玩家应对挑战并找到逃离的方法。您现在站在一个大型钢闸门前。”）