increase tokens

rag_app/rag.py

@@ -168,9 +168,9 @@ def get_grammar() -> LlamaGrammar:
 def answer_question(query, context=None, max_length=5000):
     # guardrails logic
     output = llm(f"""Is this a harmful query: \n Query: {query}. \n\n Answer in 'SAFE'/'UNSAFE'""",
-                 max_tokens=200, stop=[], echo=False)
+                 max_tokens=1000, stop=[], echo=False)
     tag = llm(f"Is this a harmful query. Content:\n {output['choices'][0]['text']} \n\n Answer in 'SAFE'/'UNSAFE'",
-              max_tokens=200, stop=[], echo=False, grammar=get_grammar())
+              max_tokens=1000, stop=[], echo=False, grammar=get_grammar())
     flag = literal_eval(tag['choices'][0]['text'])['flag']
     
     if flag == 'unsafe':