Update kig_core/llm_interface.py

kig_core/llm_interface.py

@@ -54,6 +54,14 @@ def get_llm(model_name: str) -> BaseChatModel:
         logger.error(f"Unsupported model provider for model name: {model_name}")
         raise ValueError(f"Model '{model_name}' is not supported or configuration is missing.")
 
+def invoke_llm(var,parameters):
+    try:
+        var.invoke(parameters)
+    except error:
+        print("Error during .invoke, waiting 60 seconds")
+        time.sleep(60)
+        print("Waiting is finished")
+        
 # Example usage (could be called from other modules)
 # main_llm = get_llm(settings.main_llm_model)
 # eval_llm = get_llm(settings.eval_llm_model)