Enterprise-Knowledge-Base/internal/handler/dto/model.go

package dto

import (
	"time"

	"github.com/Tencent/WeKnora/internal/types"
)

// ModelResponse mirrors types.Model for response bodies, with all secret
// fields (APIKey, AppSecret) removed by construction. Credential presence
// metadata lives behind the /credentials subresource, not inlined here.
//
// BaseURL is preserved for tenant-owned models (the frontend needs it to
// render which endpoint a custom model points at). For builtin models it is
// stripped along with every other field that could leak how a particular
// tenant configured the upstream provider.
type ModelResponse struct {
	ID          string                             `json:"id"`
	TenantID    uint64                             `json:"tenant_id"`
	Name        string                             `json:"name"`
	Type        types.ModelType                    `json:"type"`
	Source      types.ModelSource                  `json:"source"`
	Description string                             `json:"description"`
	Parameters  ModelParametersDTO                 `json:"parameters"`
	IsDefault   bool                               `json:"is_default"`
	IsBuiltin   bool                               `json:"is_builtin"`
	Status      types.ModelStatus                  `json:"status"`
	CreatedAt   time.Time                          `json:"created_at"`
	UpdatedAt   time.Time                          `json:"updated_at"`
	// Per-field "configured?" map. Omitted for builtin models (no
	// per-tenant credentials). See MCPServiceResponse.Credentials.
	Credentials map[string]CredentialFieldMetadata `json:"credentials,omitempty"`
}

// ModelParametersDTO carries every parameter field EXCEPT the two secret
// ones (APIKey, AppSecret). AppID is non-secret and stays — it's an account
// identifier the WeKnora Cloud frontend renders. CustomHeaders is also kept
// (structural metadata, not a credential).
type ModelParametersDTO struct {
	BaseURL             string                    `json:"base_url"`
	InterfaceType       string                    `json:"interface_type"`
	EmbeddingParameters types.EmbeddingParameters `json:"embedding_parameters"`
	ParameterSize       string                    `json:"parameter_size"`
	Provider            string                    `json:"provider"`
	ExtraConfig         map[string]string         `json:"extra_config,omitempty"`
	CustomHeaders       map[string]string         `json:"custom_headers,omitempty"`
	SupportsVision     bool                      `json:"supports_vision"`
	AppID               string                    `json:"app_id,omitempty"`
}

// NewModelResponse converts a stored Model into its response shape.
//
// Builtin models are shared across tenants — strip BaseURL (which can leak
// the tenant's private endpoint) and any non-shared parameters.
func NewModelResponse(m *types.Model) *ModelResponse {
	if m == nil {
		return nil
	}
	params := ModelParametersDTO{
		BaseURL:             m.Parameters.BaseURL,
		InterfaceType:       m.Parameters.InterfaceType,
		EmbeddingParameters: m.Parameters.EmbeddingParameters,
		ParameterSize:       m.Parameters.ParameterSize,
		Provider:            m.Parameters.Provider,
		ExtraConfig:         m.Parameters.ExtraConfig,
		CustomHeaders:       m.Parameters.CustomHeaders,
		SupportsVision:      m.Parameters.SupportsVision,
		AppID:               m.Parameters.AppID,
	}
	if m.IsBuiltin {
		// Builtin: strip everything that could reveal per-tenant config.
		// EmbeddingParameters and ParameterSize / Provider / InterfaceType /
		// SupportsVision are intentionally preserved (they describe the
		// capability surface, not the configured endpoint).
		params.BaseURL = ""
		params.ExtraConfig = nil
		params.CustomHeaders = nil
		params.AppID = ""
	}
	var creds map[string]CredentialFieldMetadata
	if !m.IsBuiltin {
		creds = map[string]CredentialFieldMetadata{
			"api_key":    {Configured: m.Parameters.APIKey != ""},
			"app_secret": {Configured: m.Parameters.AppSecret != ""},
		}
	}
	return &ModelResponse{
		ID:          m.ID,
		TenantID:    m.TenantID,
		Name:        m.Name,
		Type:        m.Type,
		Source:      m.Source,
		Description: m.Description,
		Parameters:  params,
		IsDefault:   m.IsDefault,
		IsBuiltin:   m.IsBuiltin,
		Status:      m.Status,
		CreatedAt:   m.CreatedAt,
		UpdatedAt:   m.UpdatedAt,
		Credentials: creds,
	}
}

// NewModelResponses is the slice convenience wrapper.
func NewModelResponses(models []*types.Model) []*ModelResponse {
	out := make([]*ModelResponse, 0, len(models))
	for _, m := range models {
		out = append(out, NewModelResponse(m))
	}
	return out
}