接入新模型

2023-10-28 19:23:43 +08:00
parent cf085565a7
commit 127385b846
18 changed files with 253 additions and 40 deletions
--- a/tests/test_llms.py
+++ b/tests/test_llms.py
@@ -17,7 +17,8 @@ if __name__ == "__main__":
    # from request_llms.bridge_claude import predict_no_ui_long_connection
    # from request_llms.bridge_internlm import predict_no_ui_long_connection
    # from request_llms.bridge_qwen import predict_no_ui_long_connection
-    from request_llms.bridge_spark import predict_no_ui_long_connection
+    # from request_llms.bridge_spark import predict_no_ui_long_connection
+    from request_llms.bridge_zhipu import predict_no_ui_long_connection

    llm_kwargs = {
        'max_length': 4096,