Merge branch 'frontier' into production

2023-11-13 00:16:44 +08:00
parent aa7c47e821 b9b7bf38ab
commit 56f42397b1
6 changed files with 45 additions and 24 deletions
--- a/toolbox.py
+++ b/toolbox.py
@@ -1150,4 +1150,10 @@ def get_chat_default_kwargs():

 def get_max_token(llm_kwargs):
    from request_llms.bridge_all import model_info
-    return model_info[llm_kwargs['llm_model']]['max_token']
+    return model_info[llm_kwargs['llm_model']]['max_token']
+
+def check_packages(packages=[]):
+    import importlib.util
+    for p in packages:
+        spam_spec = importlib.util.find_spec(p)
+        if spam_spec is None: raise ModuleNotFoundError