mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-06-29 23:41:12 +08:00
e290a0d23e
## Summary - Align Langfuse API key set/get/delete behavior with the Python implementation. - Improve DAO handling for Langfuse credential save/delete flows. - Add tests for Langfuse service error handling and API key lifecycle behavior.
123 lines
3.7 KiB
Go
123 lines
3.7 KiB
Go
//
|
|
// Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
package handler
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"ragflow/internal/common"
|
|
"ragflow/internal/entity"
|
|
"ragflow/internal/service"
|
|
)
|
|
|
|
// LangfuseService is the behaviour the handler depends on (interface enables
|
|
// mocking in tests).
|
|
type LangfuseService interface {
|
|
SetAPIKey(tenantID, secretKey, publicKey, host string) (*entity.TenantLangfuse, common.ErrorCode, error)
|
|
GetAPIKey(tenantID string) (*entity.LangfuseInfoResponse, common.ErrorCode, string, error)
|
|
DeleteAPIKey(tenantID string) (bool, common.ErrorCode, string, error)
|
|
}
|
|
|
|
// LangfuseHandler handles /langfuse/api-key HTTP requests.
|
|
type LangfuseHandler struct {
|
|
langfuseService LangfuseService
|
|
}
|
|
|
|
// NewLangfuseHandler creates a new Langfuse handler.
|
|
func NewLangfuseHandler(langfuseService LangfuseService) *LangfuseHandler {
|
|
return &LangfuseHandler{langfuseService: langfuseService}
|
|
}
|
|
|
|
// NewLangfuse keeps a zero-arg constructor consistent with other handlers.
|
|
func NewLangfuse() *LangfuseHandler {
|
|
return NewLangfuseHandler(service.NewLangfuseService())
|
|
}
|
|
|
|
// SetLangfuseRequest is the POST/PUT body. Empty-value validation happens in
|
|
// the service layer to reproduce the Python "Missing required fields" message.
|
|
type SetLangfuseRequest struct {
|
|
SecretKey string `json:"secret_key"`
|
|
PublicKey string `json:"public_key"`
|
|
Host string `json:"host"`
|
|
}
|
|
|
|
// SetAPIKey handles POST/PUT /langfuse/api-key.
|
|
func (h *LangfuseHandler) SetAPIKey(c *gin.Context) {
|
|
user, errorCode, errorMessage := GetUser(c)
|
|
if errorCode != common.CodeSuccess {
|
|
jsonError(c, errorCode, errorMessage)
|
|
return
|
|
}
|
|
|
|
var req SetLangfuseRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
jsonError(c, common.CodeDataError, "Invalid request: "+err.Error())
|
|
return
|
|
}
|
|
|
|
row, code, err := h.langfuseService.SetAPIKey(user.ID, req.SecretKey, req.PublicKey, req.Host)
|
|
if err != nil {
|
|
jsonError(c, code, err.Error())
|
|
return
|
|
}
|
|
|
|
// Echo back the stored keys, matching the Python langfuse_keys payload.
|
|
jsonResponse(c, common.CodeSuccess, gin.H{
|
|
"tenant_id": row.TenantID,
|
|
"secret_key": row.SecretKey,
|
|
"public_key": row.PublicKey,
|
|
"host": row.Host,
|
|
}, "success")
|
|
}
|
|
|
|
// GetAPIKey handles GET /langfuse/api-key.
|
|
func (h *LangfuseHandler) GetAPIKey(c *gin.Context) {
|
|
user, errorCode, errorMessage := GetUser(c)
|
|
if errorCode != common.CodeSuccess {
|
|
jsonError(c, errorCode, errorMessage)
|
|
return
|
|
}
|
|
|
|
data, code, message, err := h.langfuseService.GetAPIKey(user.ID)
|
|
if err != nil {
|
|
jsonError(c, code, message)
|
|
return
|
|
}
|
|
jsonResponse(c, code, data, message)
|
|
}
|
|
|
|
// DeleteAPIKey handles DELETE /langfuse/api-key.
|
|
func (h *LangfuseHandler) DeleteAPIKey(c *gin.Context) {
|
|
user, errorCode, errorMessage := GetUser(c)
|
|
if errorCode != common.CodeSuccess {
|
|
jsonError(c, errorCode, errorMessage)
|
|
return
|
|
}
|
|
|
|
ok, code, message, err := h.langfuseService.DeleteAPIKey(user.ID)
|
|
if err != nil {
|
|
jsonError(c, code, message)
|
|
return
|
|
}
|
|
// No record: mirror get_json_result(message=...) with data=nil.
|
|
if message != "" {
|
|
jsonResponse(c, common.CodeSuccess, nil, message)
|
|
return
|
|
}
|
|
jsonResponse(c, common.CodeSuccess, ok, "success")
|
|
}
|