Update TextGen/router.py

TextGen/router.py

@@ -49,9 +49,12 @@ class Generate(BaseModel):
 
 def generate_text(messages: List[str], npc:str):
     print(npc)
-    system_prompt=main_npc_system_prompts[npc]
+    if npc in main_npcs:
+        system_prompt=main_npc_system_prompts[npc]
+    else:
+        system_prompt="you're a character in a video game. Play along."
     print(system_prompt)    
-    new_messages=[{"role": "system", "content": system_prompt}]
+    new_messages=[{"role": "user", "content": system_prompt}]
     for index, message in enumerate(messages):
       if index%2==0:
         new_messages.append({"role": "user", "content": message})