heymenn commited on
Commit
5c79574
·
verified ·
1 Parent(s): f646915

Update kig_core/llm_interface.py

Browse files
Files changed (1) hide show
  1. kig_core/llm_interface.py +2 -2
kig_core/llm_interface.py CHANGED
@@ -57,8 +57,8 @@ def get_llm(model_name: str) -> BaseChatModel:
57
  def invoke_llm(var,parameters):
58
  try:
59
  return var.invoke(parameters)
60
- except error:
61
- print("Error during .invoke, waiting 60 seconds")
62
  time.sleep(60)
63
  print("Waiting is finished")
64
  return var.invoke(parameters)
 
57
  def invoke_llm(var,parameters):
58
  try:
59
  return var.invoke(parameters)
60
+ except Exception as e:
61
+ print(f"Error during .invoke : {e} \nwaiting 60 seconds")
62
  time.sleep(60)
63
  print("Waiting is finished")
64
  return var.invoke(parameters)