feat: add data source

frontend/src/App.tsx

@@ -7,6 +7,7 @@ import { Settings } from "./pages/Settings";
 import { Users } from "./pages/Users";
 import { Login } from "./pages/Login";
 import { ModelConfigs } from "./pages/ModelConfigs";
+import { DataSources } from "./pages/DataSources";
 import { useAuthStore } from "./store/authStore";
 
 // Protected Route Component
@@ -91,6 +92,14 @@ function App() {
             </MainLayout>
           </ProtectedRoute>
         } />
+        
+        <Route path="/datasources" element={
+          <ProtectedRoute requireAdmin={true}>
+            <MainLayout>
+              <DataSources />
+            </MainLayout>
+          </ProtectedRoute>
+        } />
       </Routes>
     </BrowserRouter>
   );

frontend/src/components/ChatInterface.tsx

@@ -72,6 +72,12 @@ export function ChatInterface() {
   const [models, setModels] = useState<ModelConfig[]>([]);
   const [selectedModelId, setSelectedModelId] = useState<string>("");
   const [modelOpen, setModelOpen] = useState(false);
+  
+  // Data Source selection state
+  const [availableDataSources, setAvailableDataSources] = useState<{id: string, name: string}[]>([
+    { id: "postgres-main", name: "PostgreSQL" },
+    { id: "clickhouse-main", name: "ClickHouse" }
+  ]);
 
   // Try to parse active session from URL query
   const queryParams = new URLSearchParams(location.search);
@@ -86,8 +92,21 @@ export function ChatInterface() {
 
   useEffect(() => {
     fetchModels();
+    fetchDataSources();
   }, []);
 
+  const fetchDataSources = async () => {
+    try {
+      const data = await api.get<Array<{id: number, name: string}>>("/api/v1/datasources");
+      setAvailableDataSources(prev => [
+        ...prev.filter(d => !d.id.startsWith("ds:")),
+        ...data.map(d => ({ id: `ds:${d.id}`, name: d.name }))
+      ]);
+    } catch (e) {
+      console.error("Failed to fetch data sources", e);
+    }
+  };
+
   const syncSessionFileContext = async (file: DataFileContext | null) => {
     try {
       await api.put(`/nanobot/sessions/${encodeURIComponent(activeSessionKey)}/context-file`, {
@@ -506,8 +525,9 @@ export function ChatInterface() {
             onChange={(e) => setSelectedDataSource(e.target.value)}
             className="bg-transparent border-none outline-none text-sm font-medium"
           >
-            <option value="postgres-main">PostgreSQL</option>
-            <option value="clickhouse-main">ClickHouse</option>
+            {availableDataSources.map(ds => (
+              <option key={ds.id} value={ds.id}>{ds.name}</option>
+            ))}
             {activeDataFile?.url?.startsWith("local://") ? (
               <option value="upload-main">上传文件</option>
             ) : null}

frontend/src/components/DataSourceForm.tsx

@@ -0,0 +1,262 @@
+import { useState } from "react";
+import { Button } from "@/components/ui/button";
+import { Input } from "@/components/ui/input";
+import { Loader2, Check, AlertTriangle } from "lucide-react";
+
+export interface DataSourceConfig {
+  id?: number;
+  name: string;
+  type: string;
+  config: Record<string, any>;
+}
+
+interface DataSourceFormProps {
+  initialData?: DataSourceConfig | null;
+  onSubmit: (data: Omit<DataSourceConfig, "id">) => Promise<void>;
+  onTest: (type: string, config: Record<string, any>) => Promise<boolean>;
+  onCancel: () => void;
+}
+
+export function DataSourceForm({ initialData, onSubmit, onTest, onCancel }: DataSourceFormProps) {
+  const [name, setName] = useState(initialData?.name || "");
+  const [type, setType] = useState(initialData?.type || "postgres");
+  const [config, setConfig] = useState<Record<string, any>>(initialData?.config || {});
+  const [isTesting, setIsTesting] = useState(false);
+  const [testResult, setTestResult] = useState<{ success: boolean; message: string } | null>(null);
+  const [isSaving, setIsSaving] = useState(false);
+
+  const handleConfigChange = (key: string, value: any) => {
+    setConfig(prev => ({ ...prev, [key]: value }));
+  };
+
+  const handleTest = async () => {
+    setIsTesting(true);
+    setTestResult(null);
+    try {
+      const success = await onTest(type, config);
+      setTestResult({
+        success,
+        message: success ? "连接成功" : "连接失败",
+      });
+    } catch (e: any) {
+      setTestResult({
+        success: false,
+        message: e.message || "连接失败",
+      });
+    } finally {
+      setIsTesting(false);
+    }
+  };
+
+  const handleSubmit = async (e: React.FormEvent) => {
+    e.preventDefault();
+    setIsSaving(true);
+    try {
+      await onSubmit({ name, type, config });
+    } finally {
+      setIsSaving(false);
+    }
+  };
+
+  const renderConfigFields = () => {
+    switch (type) {
+      case "postgres":
+      case "supabase":
+        return (
+          <div className="space-y-4">
+             <div className="grid grid-cols-2 gap-4">
+              <div className="space-y-2">
+                <label className="text-sm font-medium">Host</label>
+                <Input 
+                  value={config.host || ""} 
+                  onChange={e => handleConfigChange("host", e.target.value)} 
+                  placeholder="localhost" 
+                />
+              </div>
+              <div className="space-y-2">
+                <label className="text-sm font-medium">Port</label>
+                <Input 
+                  type="number"
+                  value={config.port || 5432} 
+                  onChange={e => handleConfigChange("port", parseInt(e.target.value))} 
+                  placeholder="5432" 
+                />
+              </div>
+            </div>
+            <div className="space-y-2">
+              <label className="text-sm font-medium">Database</label>
+              <Input 
+                value={config.database || ""} 
+                onChange={e => handleConfigChange("database", e.target.value)} 
+                placeholder="postgres" 
+              />
+            </div>
+            <div className="space-y-2">
+              <label className="text-sm font-medium">Username</label>
+              <Input 
+                value={config.user || ""} 
+                onChange={e => handleConfigChange("user", e.target.value)} 
+                placeholder="postgres" 
+              />
+            </div>
+            <div className="space-y-2">
+              <label className="text-sm font-medium">Password</label>
+              <Input 
+                type="password"
+                value={config.password || ""} 
+                onChange={e => handleConfigChange("password", e.target.value)} 
+                placeholder="••••••" 
+              />
+            </div>
+            <div className="text-xs text-zinc-500 pt-2">
+              或者使用连接字符串 (覆盖上述设置):
+            </div>
+            <div className="space-y-2">
+              <label className="text-sm font-medium">Connection String</label>
+              <Input 
+                value={config.connection_string || ""} 
+                onChange={e => handleConfigChange("connection_string", e.target.value)} 
+                placeholder="postgresql://user:pass@host:5432/db" 
+              />
+            </div>
+          </div>
+        );
+      case "clickhouse":
+        return (
+          <div className="space-y-4">
+             <div className="grid grid-cols-2 gap-4">
+              <div className="space-y-2">
+                <label className="text-sm font-medium">Host</label>
+                <Input 
+                  value={config.host || ""} 
+                  onChange={e => handleConfigChange("host", e.target.value)} 
+                  placeholder="localhost" 
+                />
+              </div>
+              <div className="space-y-2">
+                <label className="text-sm font-medium">Port</label>
+                <Input 
+                  type="number"
+                  value={config.port || 9000} 
+                  onChange={e => handleConfigChange("port", parseInt(e.target.value))} 
+                  placeholder="9000" 
+                />
+              </div>
+            </div>
+            <div className="space-y-2">
+              <label className="text-sm font-medium">Database</label>
+              <Input 
+                value={config.database || ""} 
+                onChange={e => handleConfigChange("database", e.target.value)} 
+                placeholder="default" 
+              />
+            </div>
+            <div className="space-y-2">
+              <label className="text-sm font-medium">Username</label>
+              <Input 
+                value={config.user || ""} 
+                onChange={e => handleConfigChange("user", e.target.value)} 
+                placeholder="default" 
+              />
+            </div>
+            <div className="space-y-2">
+              <label className="text-sm font-medium">Password</label>
+              <Input 
+                type="password"
+                value={config.password || ""} 
+                onChange={e => handleConfigChange("password", e.target.value)} 
+                placeholder="••••••" 
+              />
+            </div>
+          </div>
+        );
+      case "sqlite":
+        return (
+          <div className="space-y-4">
+            <div className="space-y-2">
+              <label className="text-sm font-medium">File Path (Server Side)</label>
+              <Input 
+                value={config.file_path || ""} 
+                onChange={e => handleConfigChange("file_path", e.target.value)} 
+                placeholder="/path/to/database.db" 
+              />
+            </div>
+          </div>
+        );
+      case "parquet":
+        return (
+           <div className="space-y-4">
+            <div className="space-y-2">
+              <label className="text-sm font-medium">File Path (Server Side)</label>
+              <Input 
+                value={config.file_path || ""} 
+                onChange={e => handleConfigChange("file_path", e.target.value)} 
+                placeholder="/path/to/data.parquet" 
+              />
+            </div>
+          </div>
+        );
+      default:
+        return null;
+    }
+  };
+
+  return (
+    <form onSubmit={handleSubmit} className="space-y-6">
+      <div className="space-y-2">
+        <label className="text-sm font-medium">名称</label>
+        <Input 
+          value={name} 
+          onChange={e => setName(e.target.value)} 
+          placeholder="我的数据源" 
+          required 
+        />
+      </div>
+
+      <div className="space-y-2">
+        <label className="text-sm font-medium">类型</label>
+        <select
+          className="w-full h-10 px-3 rounded-md border border-zinc-200 bg-white text-sm focus:outline-none focus:ring-2 focus:ring-zinc-950 focus:border-transparent"
+          value={type}
+          onChange={e => setType(e.target.value)}
+        >
+          <option value="postgres">PostgreSQL</option>
+          <option value="clickhouse">ClickHouse</option>
+          <option value="sqlite">SQLite</option>
+          <option value="supabase">Supabase</option>
+          <option value="parquet">Parquet</option>
+        </select>
+      </div>
+
+      <div className="p-4 border border-zinc-200 rounded-lg bg-zinc-50/50">
+        {renderConfigFields()}
+      </div>
+
+      {testResult && (
+        <div className={`p-3 rounded-md flex items-center gap-2 text-sm ${testResult.success ? 'bg-green-50 text-green-700' : 'bg-red-50 text-red-700'}`}>
+          {testResult.success ? <Check className="h-4 w-4" /> : <AlertTriangle className="h-4 w-4" />}
+          {testResult.message}
+        </div>
+      )}
+
+      <div className="flex justify-end gap-3 pt-4">
+        <Button type="button" variant="outline" onClick={onCancel}>
+          取消
+        </Button>
+        <Button 
+          type="button" 
+          variant="secondary" 
+          onClick={handleTest}
+          disabled={isTesting}
+        >
+          {isTesting && <Loader2 className="mr-2 h-4 w-4 animate-spin" />}
+          测试连接
+        </Button>
+        <Button type="submit" disabled={isSaving}>
+          {isSaving && <Loader2 className="mr-2 h-4 w-4 animate-spin" />}
+          保存
+        </Button>
+      </div>
+    </form>
+  );
+}

frontend/src/components/Sidebar.tsx

@@ -1,7 +1,7 @@
 import { Sheet, SheetContent, SheetTrigger } from "@/components/ui/sheet";
 import { Button } from "@/components/ui/button";
 import { ScrollArea } from "@/components/ui/scroll-area";
-import { Menu, LayoutDashboard, Plus, MoreVertical, User, Search, Wrench, Settings, Brain, Trash2, Pencil, Pin, Archive } from "lucide-react";
+import { Menu, LayoutDashboard, Plus, MoreVertical, User, Search, Wrench, Settings, Brain, Trash2, Pencil, Pin, Archive, Database } from "lucide-react";
 import { useState, useRef, useEffect } from "react";
 import { Link, useNavigate, useLocation } from "react-router-dom";
 import { useAuthStore } from "@/store/authStore";
@@ -465,6 +465,17 @@ function SidebarBody() {
                   <Brain className="h-4 w-4 text-zinc-500" />
                   模型配置
                 </button>
+
+                <button
+                  className="w-full flex items-center gap-2 px-3 py-2 text-sm text-zinc-700 hover:bg-zinc-100 transition-colors"
+                  onClick={() => {
+                    navigate("/datasources");
+                    setShowUserMenu(false);
+                  }}
+                >
+                  <Database className="h-4 w-4 text-zinc-500" />
+                  数据源配置
+                </button>
                 
                 <button 
                   className="w-full flex items-center gap-2 px-3 py-2 text-sm text-indigo-600 hover:bg-indigo-50 transition-colors"

frontend/src/pages/DataSources.tsx

@@ -0,0 +1,171 @@
+import { useState, useEffect } from "react";
+import { api } from "@/lib/api";
+import { DataSourceForm, type DataSourceConfig } from "@/components/DataSourceForm";
+import { Button } from "@/components/ui/button";
+import { Plus, Database, Pencil, Trash2, Loader2 } from "lucide-react";
+import { Dialog, DialogContent, DialogHeader, DialogTitle } from "@/components/ui/dialog";
+import { useAuthStore } from "@/store/authStore";
+import { useNavigate } from "react-router-dom";
+
+export function DataSources() {
+  const [datasources, setDatasources] = useState<DataSourceConfig[]>([]);
+  const [isLoading, setIsLoading] = useState(false);
+  const [isOpen, setIsOpen] = useState(false);
+  const [editingDs, setEditingDs] = useState<DataSourceConfig | null>(null);
+  const { user } = useAuthStore();
+  const navigate = useNavigate();
+
+  useEffect(() => {
+    if (!user?.is_admin) {
+      navigate("/");
+      return;
+    }
+    fetchDataSources();
+  }, [user]);
+
+  const fetchDataSources = async () => {
+    setIsLoading(true);
+    try {
+      const data = await api.get<DataSourceConfig[]>("/api/v1/datasources");
+      setDatasources(data);
+    } catch (e) {
+      console.error("Failed to fetch data sources", e);
+    } finally {
+      setIsLoading(false);
+    }
+  };
+
+  const handleCreate = () => {
+    setEditingDs(null);
+    setIsOpen(true);
+  };
+
+  const handleEdit = (ds: DataSourceConfig) => {
+    setEditingDs(ds);
+    setIsOpen(true);
+  };
+
+  const handleDelete = async (id: number) => {
+    if (!window.confirm("确定要删除这个数据源吗？")) return;
+    try {
+      await api.delete(`/api/v1/datasources/${id}`);
+      fetchDataSources();
+    } catch (e) {
+      console.error("Failed to delete data source", e);
+    }
+  };
+
+  const handleSubmit = async (data: Omit<DataSourceConfig, "id">) => {
+    try {
+      if (editingDs?.id) {
+        await api.put(`/api/v1/datasources/${editingDs.id}`, data);
+      } else {
+        await api.post("/api/v1/datasources", data);
+      }
+      setIsOpen(false);
+      fetchDataSources();
+    } catch (e) {
+      console.error("Failed to save data source", e);
+      alert("保存失败: " + (e as any).message);
+    }
+  };
+
+  const handleTest = async (type: string, config: Record<string, any>) => {
+    try {
+      const res = await api.post<{ success: boolean; message: string }>("/api/v1/datasources/test", { type, config });
+      return res.success;
+    } catch (e) {
+      console.error("Test connection failed", e);
+      throw e;
+    }
+  };
+
+  return (
+    <div className="h-full flex flex-col bg-white">
+      <div className="border-b border-zinc-100 px-8 py-5 flex items-center justify-between">
+        <div>
+          <h1 className="text-2xl font-bold text-zinc-900">数据源配置</h1>
+          <p className="text-sm text-zinc-500 mt-1">管理可用于问答的数据源连接</p>
+        </div>
+        <Button onClick={handleCreate} className="bg-indigo-600 hover:bg-indigo-700 text-white gap-2">
+          <Plus className="h-4 w-4" />
+          新建数据源
+        </Button>
+      </div>
+
+      <div className="flex-1 overflow-auto p-8">
+        {isLoading ? (
+          <div className="flex justify-center items-center h-64">
+            <Loader2 className="h-8 w-8 animate-spin text-zinc-400" />
+          </div>
+        ) : datasources.length === 0 ? (
+          <div className="flex flex-col items-center justify-center h-64 border-2 border-dashed border-zinc-200 rounded-xl bg-zinc-50/50">
+            <Database className="h-10 w-10 text-zinc-300 mb-3" />
+            <p className="text-zinc-500 font-medium">暂无数据源</p>
+            <p className="text-zinc-400 text-sm mt-1">点击右上角按钮添加第一个数据源</p>
+          </div>
+        ) : (
+          <div className="grid grid-cols-1 md:grid-cols-2 lg:grid-cols-3 gap-6">
+            {datasources.map((ds) => (
+              <div 
+                key={ds.id} 
+                className="group relative bg-white border border-zinc-200 rounded-xl p-5 hover:shadow-md transition-all hover:border-zinc-300"
+              >
+                <div className="flex items-start justify-between mb-4">
+                  <div className="flex items-center gap-3">
+                    <div className="w-10 h-10 rounded-lg bg-indigo-50 flex items-center justify-center text-indigo-600">
+                      <Database className="h-5 w-5" />
+                    </div>
+                    <div>
+                      <h3 className="font-semibold text-zinc-900">{ds.name}</h3>
+                      <p className="text-xs text-zinc-500 font-mono mt-0.5 uppercase">{ds.type}</p>
+                    </div>
+                  </div>
+                  <div className="flex items-center gap-1 opacity-0 group-hover:opacity-100 transition-opacity">
+                    <Button variant="ghost" size="icon" className="h-8 w-8 text-zinc-400 hover:text-zinc-600" onClick={() => handleEdit(ds)}>
+                      <Pencil className="h-4 w-4" />
+                    </Button>
+                    <Button variant="ghost" size="icon" className="h-8 w-8 text-zinc-400 hover:text-red-600 hover:bg-red-50" onClick={() => handleDelete(ds.id!)}>
+                      <Trash2 className="h-4 w-4" />
+                    </Button>
+                  </div>
+                </div>
+                
+                <div className="space-y-2">
+                  <div className="flex items-center justify-between text-sm">
+                    <span className="text-zinc-500">Host</span>
+                    <span className="font-medium text-zinc-700 truncate max-w-[150px]">
+                      {ds.config.host || "Local / File"}
+                    </span>
+                  </div>
+                  <div className="flex items-center justify-between text-sm">
+                    <span className="text-zinc-500">Database</span>
+                    <span className="font-medium text-zinc-700 truncate max-w-[150px]">
+                      {ds.config.database || ds.config.file_path ? (ds.config.file_path?.split('/').pop()) : "-"}
+                    </span>
+                  </div>
+                </div>
+              </div>
+            ))}
+          </div>
+        )}
+      </div>
+
+      <Dialog open={isOpen} onOpenChange={setIsOpen}>
+        <DialogContent className="sm:max-w-[600px] max-h-[90vh] overflow-y-auto">
+          <DialogHeader>
+            <DialogTitle>{editingDs ? "编辑数据源" : "新建数据源"}</DialogTitle>
+          </DialogHeader>
+          <div className="py-4">
+            <DataSourceForm 
+              initialData={editingDs} 
+              onSubmit={handleSubmit} 
+              onTest={handleTest}
+              onCancel={() => setIsOpen(false)}
+            />
+          </div>
+        </DialogContent>
+      </Dialog>
+    </div>
+  );
+}