Update TextGen/router.py

TextGen/router.py

@@ -87,12 +87,13 @@ main_npc_system_prompts={
 class Generate(BaseModel):
     text:str
 
-class Invoke(BaseModel):
+class Rooms(BaseModel):
     rooms:List
     room_of_interest:List
     index_exit:int
+    possible_entities:List
 
-class Rooms(BaseModel):
+class Invoke(BaseModel):
     system_prompt:str
     message:str