Jofthomas HF staff commited on
Commit
2733ad7
·
1 Parent(s): ee84e6b

Update TextGen/router.py

Browse files
Files changed (1) hide show
  1. TextGen/router.py +1 -0
TextGen/router.py CHANGED
@@ -200,6 +200,7 @@ def check(input: Logs):
200
  print(input.logs)
201
  system_prompt="You are a game master in a roguelike. You previously decided of an objective for the player. You have to answer with YES or NO on wether the objective as been sucessfully completed. Be kind and flexible as the content of the game is AI generated and might not be feasible."
202
  user_message=f"The objective was : {input.objective} and the player did the following actions : {input.logs}. Do you grant acess ? ONLY answer YES or NO"
 
203
  answer=inference_model(system_prompt,user_message)
204
  return answer
205
 
 
200
  print(input.logs)
201
  system_prompt="You are a game master in a roguelike. You previously decided of an objective for the player. You have to answer with YES or NO on wether the objective as been sucessfully completed. Be kind and flexible as the content of the game is AI generated and might not be feasible."
202
  user_message=f"The objective was : {input.objective} and the player did the following actions : {input.logs}. Do you grant acess ? ONLY answer YES or NO"
203
+ print(system_prompt,user_message,answer)
204
  answer=inference_model(system_prompt,user_message)
205
  return answer
206