瀏覽代碼

fix: better error message for url add external knowledge (#11537)

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
yihong 4 月之前
父節點
當前提交
50b76dd5a2
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      api/services/external_knowledge_service.py

+ 4 - 1
api/services/external_knowledge_service.py

@@ -69,7 +69,10 @@ class ExternalDatasetService:
         endpoint = f"{settings['endpoint']}/retrieval"
         api_key = settings["api_key"]
         if not validators.url(endpoint, simple_host=True):
-            raise ValueError(f"invalid endpoint: {endpoint}")
+            if not endpoint.startswith("http://") and not endpoint.startswith("https://"):
+                raise ValueError(f"invalid endpoint: {endpoint} must start with http:// or https://")
+            else:
+                raise ValueError(f"invalid endpoint: {endpoint}")
         try:
             response = httpx.post(endpoint, headers={"Authorization": f"Bearer {api_key}"})
         except Exception as e: