fix(backend): data source name should not be blank

pingcap · Sep 26, 2024 · f1ea345 · f1ea345
1 parent ae9ccee
commit f1ea345
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/backend/app/api/admin_routes/data_source.py b/backend/app/api/admin_routes/data_source.py
@@ -1,4 +1,4 @@
-from pydantic import BaseModel
+from pydantic import BaseModel, field_validator
 from fastapi import APIRouter, Depends, HTTPException, status
 from fastapi_pagination import Params, Page
 
@@ -27,6 +27,12 @@ class DataSourceCreate(BaseModel):
     build_kg_index: bool = False
     llm_id: int | None = None
 
+    @field_validator("name")
+    def name_must_not_be_blank(cls, v: str) -> str:
+        if not v.strip():
+            raise ValueError("Please provide a name for the data source")
+        return v
+
 
 @router.post("/admin/datasources")
 def create_datasource(