Ver código fonte

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

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
tags/0.14.0
yihong 10 meses atrás
pai
commit
50b76dd5a2
Nenhuma conta vinculada ao e-mail do autor do commit
1 arquivos alterados com 4 adições e 1 exclusões
  1. 4
    1
      api/services/external_knowledge_service.py

+ 4
- 1
api/services/external_knowledge_service.py Ver arquivo

@@ -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:

Carregando…
Cancelar
Salvar