feat(model): Support yi proxy LLM (eosphoros-ai#1303)

FOkvj · Mar 15, 2024 · 130ffb0 · 130ffb0
1 parent 6974329
commit 130ffb0
Show file tree

Hide file tree

Showing 144 changed files with 1,286 additions and 1,088 deletions.
diff --git a/.env.template b/.env.template
@@ -208,6 +208,11 @@ TONGYI_PROXY_API_KEY={your-tongyi-sk}
 #XUNFEI_SPARK_API_KEY={your_api_key}
 #XUNFEI_SPARK_API_SECRET={your_api_secret}
 
+## Yi Proxyllm, https://platform.lingyiwanwu.com/docs
+#YI_MODEL_VERSION=yi-34b-chat-0205
+#YI_API_BASE=https://api.lingyiwanwu.com/v1
+#YI_API_KEY={your-yi-api-key}
+
 
 
 #*******************************************************************#

diff --git a/.flake8 b/.flake8
@@ -14,6 +14,7 @@ ignore =
   E126
   E203
   E226
+  E231
   E24
   E704
   W503

diff --git a/.mypy.ini b/.mypy.ini
@@ -71,3 +71,6 @@ ignore_missing_imports = True
 
 [mypy-pymilvus.*]
 ignore_missing_imports = True
+
+[mypy-cryptography.*]
+ignore_missing_imports = True
diff --git a/dbgpt/_private/config.py b/dbgpt/_private/config.py
@@ -106,6 +106,17 @@ def __init__(self) -> None:
                 "GEMINI_MODEL_VERSION", "gemini-pro"
             )
 
+        # Yi proxy
+        self.yi_proxy_api_key = os.getenv("YI_API_KEY")
+        if self.yi_proxy_api_key:
+            os.environ["yi_proxyllm_proxy_api_key"] = self.yi_proxy_api_key
+            os.environ["yi_proxyllm_proxyllm_backend"] = os.getenv(
+                "YI_MODEL_VERSION", "yi-34b-chat-0205"
+            )
+            os.environ["yi_proxyllm_proxy_api_base"] = os.getenv(
+                "YI_API_BASE", "https://api.lingyiwanwu.com/v1"
+            )
+
         self.proxy_server_url = os.getenv("PROXY_SERVER_URL")
 
         self.elevenlabs_api_key = os.getenv("ELEVENLABS_API_KEY")

diff --git a/dbgpt/app/static/404.html b/dbgpt/app/static/404.html
diff --git a/dbgpt/app/static/404/index.html b/dbgpt/app/static/404/index.html
diff --git a/dbgpt/app/static/_next/static/HlyvA5dZ3qEZyPQ5_f33x/_buildManifest.js b/dbgpt/app/static/_next/static/HlyvA5dZ3qEZyPQ5_f33x/_buildManifest.js
diff --git a/dbgpt/app/static/_next/static/KdUNA5m3s-yKZ2dNV_cuN/_buildManifest.js b/dbgpt/app/static/_next/static/KdUNA5m3s-yKZ2dNV_cuN/_buildManifest.js
diff --git a/...tic/HlyvA5dZ3qEZyPQ5_f33x/_ssgManifest.js → ...tic/KdUNA5m3s-yKZ2dNV_cuN/_ssgManifest.js b/...tic/HlyvA5dZ3qEZyPQ5_f33x/_ssgManifest.js → ...tic/KdUNA5m3s-yKZ2dNV_cuN/_ssgManifest.js
diff --git a/dbgpt/app/static/_next/static/chunks/1009-4b2af86bde623424.js b/dbgpt/app/static/_next/static/chunks/1009-4b2af86bde623424.js
diff --git a/dbgpt/app/static/_next/static/chunks/1193-0c6475e280185569.js b/dbgpt/app/static/_next/static/chunks/1193-0c6475e280185569.js
-Original file line number
+Diff line change
@@ Expand Up / @@ -14,6 +14,7 @@ ignore = @@
       E126
       E203
       E226
+      E231
       E24
       E704
       W503
@@ Expand Down @@