AskaEth
d15acf587c
- docs: 17个文件重命名为 YYYY-MM-DD.HH-mm-SS-内容.md 格式 - config: 管理员凭据移至 backend/.env (ADMIN_USERNAME/PASSWORD) - gateway: 新增 SessionState 会话追踪 + GET /api/v1/admin/sessions - devtools: 新增7个代理端点 (dashboard/sessions/memory) - devtools: WebUI重构为侧边栏 + 5面板 (仪表盘/记忆/会话/服务/性能)
130 lines
3.0 KiB
Go
130 lines
3.0 KiB
Go
package handler
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/yourname/cyrene-ai/gateway/internal/middleware"
|
|
)
|
|
|
|
// MemoryHandler 记忆查询处理器 — 代理到 AI-Core
|
|
type MemoryHandler struct {
|
|
aiCoreURL string
|
|
client *http.Client
|
|
}
|
|
|
|
// NewMemoryHandler 创建记忆处理器
|
|
func NewMemoryHandler(aiCoreURL string) *MemoryHandler {
|
|
return &MemoryHandler{
|
|
aiCoreURL: aiCoreURL,
|
|
client: &http.Client{
|
|
Timeout: 15 * time.Second,
|
|
},
|
|
}
|
|
}
|
|
|
|
// Query 搜索用户记忆 — 代理 GET /api/v1/memory/search?user_id=...&q=...
|
|
func (h *MemoryHandler) Query(c *gin.Context) {
|
|
userID := middleware.GetUserID(c)
|
|
query := c.Query("q")
|
|
if query == "" {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "查询参数q不能为空"})
|
|
return
|
|
}
|
|
|
|
url := fmt.Sprintf("%s/api/v1/memory/search?user_id=%s&q=%s",
|
|
h.aiCoreURL, userID, query)
|
|
|
|
resp, err := h.client.Get(url)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadGateway, gin.H{"error": fmt.Sprintf("AI-Core 不可达: %v", err)})
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
body, _ := io.ReadAll(resp.Body)
|
|
|
|
var result interface{}
|
|
json.Unmarshal(body, &result)
|
|
c.JSON(resp.StatusCode, result)
|
|
}
|
|
|
|
// List 列出用户所有记忆 — 代理 GET /api/v1/memory?user_id=...
|
|
func (h *MemoryHandler) List(c *gin.Context) {
|
|
userID := middleware.GetUserID(c)
|
|
|
|
url := fmt.Sprintf("%s/api/v1/memory?user_id=%s", h.aiCoreURL, userID)
|
|
|
|
resp, err := h.client.Get(url)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadGateway, gin.H{"error": fmt.Sprintf("AI-Core 不可达: %v", err)})
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
body, _ := io.ReadAll(resp.Body)
|
|
|
|
var result interface{}
|
|
json.Unmarshal(body, &result)
|
|
c.JSON(resp.StatusCode, result)
|
|
}
|
|
|
|
// Add 手动添加记忆 — 代理 POST /api/v1/memory
|
|
func (h *MemoryHandler) Add(c *gin.Context) {
|
|
userID := middleware.GetUserID(c)
|
|
|
|
var req struct {
|
|
Content string `json:"content" binding:"required"`
|
|
Category string `json:"category"`
|
|
Priority int `json:"priority"`
|
|
}
|
|
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "请求参数无效"})
|
|
return
|
|
}
|
|
|
|
if req.Category == "" {
|
|
req.Category = "other"
|
|
}
|
|
if req.Priority <= 0 {
|
|
req.Priority = 1
|
|
}
|
|
|
|
// 转发到 AI-Core
|
|
aiReq := map[string]interface{}{
|
|
"user_id": userID,
|
|
"content": req.Content,
|
|
"category": req.Category,
|
|
"priority": req.Priority,
|
|
}
|
|
reqBody, _ := json.Marshal(aiReq)
|
|
|
|
url := fmt.Sprintf("%s/api/v1/memory", h.aiCoreURL)
|
|
httpReq, err := http.NewRequest("POST", url, bytes.NewReader(reqBody))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "构建请求失败"})
|
|
return
|
|
}
|
|
httpReq.Header.Set("Content-Type", "application/json")
|
|
|
|
resp, err := h.client.Do(httpReq)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadGateway, gin.H{"error": fmt.Sprintf("AI-Core 不可达: %v", err)})
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
body, _ := io.ReadAll(resp.Body)
|
|
|
|
var result interface{}
|
|
json.Unmarshal(body, &result)
|
|
c.JSON(resp.StatusCode, result)
|
|
}
|