fix linters

google · Aug 23, 2024 · 931beee · 931beee
1 parent 8b78496
commit 931beee
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/timesketch/api/v1/resources/nl2q.py b/timesketch/api/v1/resources/nl2q.py
@@ -209,15 +209,18 @@ def post(self, sketch_id):
             llm = manager.LLMManager().get_provider(llm_provider)()
         except Exception as e:  # pylint: disable=broad-except
             logger.error("Error LLM Provider: {}".format(e))
-            result_schema["error"] = "Error loading LLM Provider. Please try again later!"
+            result_schema["error"] = (
+                "Error loading LLM Provider. Please try again later!"
+            )
             return jsonify(result_schema)
 
         try:
             prediction = llm.generate(prompt)
         except Exception as e:  # pylint: disable=broad-except
             logger.error("Error NL2Q prompt: {}".format(e))
             result_schema["error"] = (
-                "An error occurred generating the query via the defined LLM. Please try again later!"
+                "An error occurred generating the query via the defined LLM. "
+                "Please try again later!"
             )
             return jsonify(result_schema)
         # The model sometimes output tripple backticks that needs to be removed.