Update TextGen/router.py

TextGen/router.py

@@ -64,6 +64,7 @@ def generate_text(messages: List[str], npc:str):
     # Initialize the LLM
     llm = ChatGoogleGenerativeAI(
         model="gemini-pro",
+        max_output_tokens=100,
         safety_settings={
             HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
         },