Spaces:
Running
Running
Update kig_core/llm_interface.py
Browse files
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
|
61 |
-
print("Error during .invoke
|
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)
|