Update kig_core/llm_interface.py

kig_core/llm_interface.py

@@ -56,11 +56,12 @@ def get_llm(model_name: str) -> BaseChatModel:
 
 def invoke_llm(var,parameters):
     try:
-        var.invoke(parameters)
+        return var.invoke(parameters)
     except error:
         print("Error during .invoke, waiting 60 seconds")
         time.sleep(60)
         print("Waiting is finished")
+        return var.invoke(parameters)
         
 # Example usage (could be called from other modules)
 # main_llm = get_llm(settings.main_llm_model)