Spaces:
Running
Running
Update kig_core/llm_interface.py
Browse files
kig_core/llm_interface.py
CHANGED
@@ -54,6 +54,14 @@ def get_llm(model_name: str) -> BaseChatModel:
|
|
54 |
logger.error(f"Unsupported model provider for model name: {model_name}")
|
55 |
raise ValueError(f"Model '{model_name}' is not supported or configuration is missing.")
|
56 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
57 |
# Example usage (could be called from other modules)
|
58 |
# main_llm = get_llm(settings.main_llm_model)
|
59 |
# eval_llm = get_llm(settings.eval_llm_model)
|
|
|
54 |
logger.error(f"Unsupported model provider for model name: {model_name}")
|
55 |
raise ValueError(f"Model '{model_name}' is not supported or configuration is missing.")
|
56 |
|
57 |
+
def invoke_llm(var,parameters):
|
58 |
+
try:
|
59 |
+
var.invoke(parameters)
|
60 |
+
except error:
|
61 |
+
print("Error during .invoke, waiting 60 seconds")
|
62 |
+
time.sleep(60)
|
63 |
+
print("Waiting is finished")
|
64 |
+
|
65 |
# Example usage (could be called from other modules)
|
66 |
# main_llm = get_llm(settings.main_llm_model)
|
67 |
# eval_llm = get_llm(settings.eval_llm_model)
|