mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-04 18:45:38 +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.
105 lines
3.0 KiB
Go
105 lines
3.0 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 dao
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
|
|
"ragflow/internal/entity"
|
|
)
|
|
|
|
// LangfuseDAO is the data access object for tenant Langfuse credentials.
|
|
type LangfuseDAO struct{}
|
|
|
|
// NewLangfuse creates a new Langfuse DAO.
|
|
func NewLangfuse() *LangfuseDAO {
|
|
return &LangfuseDAO{}
|
|
}
|
|
|
|
// GetByTenantID returns the Langfuse credentials row for a tenant.
|
|
// It returns (nil, nil) when no row exists, mirroring the Python
|
|
// TenantLangfuseService.filter_by_tenant behaviour (DoesNotExist -> None).
|
|
func (dao *LangfuseDAO) GetByTenantID(tenantID string) (*entity.TenantLangfuse, error) {
|
|
var row entity.TenantLangfuse
|
|
err := DB.Where("tenant_id = ?", tenantID).First(&row).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
return &row, nil
|
|
}
|
|
|
|
// Create inserts a new Langfuse credentials row (mirrors save).
|
|
func (dao *LangfuseDAO) Create(row *entity.TenantLangfuse) error {
|
|
return DB.Create(row).Error
|
|
}
|
|
|
|
// UpdateByTenantID updates the Langfuse credentials row for a tenant
|
|
func (dao *LangfuseDAO) UpdateByTenantID(tenantID string, updates map[string]any) error {
|
|
res := DB.Model(&entity.TenantLangfuse{}).Where("tenant_id = ?", tenantID).Updates(updates)
|
|
if res.Error != nil {
|
|
return res.Error
|
|
}
|
|
if res.RowsAffected == 0 {
|
|
return gorm.ErrRecordNotFound
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// DeleteByTenantID deletes the Langfuse credentials row for a tenant
|
|
// (mirrors delete_model / delete_ty_tenant_id).
|
|
func (dao *LangfuseDAO) DeleteByTenantID(tenantID string) error {
|
|
res := DB.Where("tenant_id = ?", tenantID).Delete(&entity.TenantLangfuse{})
|
|
if res.Error != nil {
|
|
return res.Error
|
|
}
|
|
if res.RowsAffected == 0 {
|
|
return gorm.ErrRecordNotFound
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (dao *LangfuseDAO) SaveByTenantID(row *entity.TenantLangfuse) error {
|
|
return DB.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "tenant_id"}},
|
|
DoUpdates: clause.Assignments(map[string]any{
|
|
"secret_key": row.SecretKey,
|
|
"public_key": row.PublicKey,
|
|
"host": row.Host,
|
|
}),
|
|
}).Create(row).Error
|
|
}
|
|
|
|
func (dao *LangfuseDAO) DeleteExistingByTenantID(tenantID string) error {
|
|
return DB.Transaction(func(tx *gorm.DB) error {
|
|
var row entity.TenantLangfuse
|
|
err := tx.Where("tenant_id = ?", tenantID).First(&row).Error
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return gorm.ErrRecordNotFound
|
|
}
|
|
return err
|
|
}
|
|
return tx.Delete(&row).Error
|
|
})
|
|
}
|