package controller

import (
	"context"
	"model-gateway/model/dto"
	"model-gateway/service"

	"gitea.redpowerfuture.com/red-future/common/beans"
)

// ModelManage 模型配置控制器
var ModelManage = new(modelManage)

type modelManage struct{}

// CreateModel 添加配置
func (c *modelManage) CreateModel(ctx context.Context, req *dto.CreateModelManageReq) (res *dto.CreateModelManageRes, err error) {
	return service.ModelManage.Create(ctx, req)
}

// ReferenceModel 引用系统模型
func (c *modelManage) ReferenceModel(ctx context.Context, req *dto.ReferenceSystemModelReq) (res *dto.ReferenceSystemModelRes, err error) {
	return service.ModelManage.ReferenceSystemModel(ctx, req)
}

// UpdateModel 更改配置
func (c *modelManage) UpdateModel(ctx context.Context, req *dto.UpdateModelManageReq) (res *dto.GetModelManageRes, err error) {
	return service.ModelManage.Update(ctx, req)
}

// DeleteModel 删除配置
func (c *modelManage) DeleteModel(ctx context.Context, req *dto.DeleteModelManageReq) (res *beans.ResponseEmpty, err error) {
	err = service.ModelManage.Delete(ctx, req)
	return
}

// GetModel 获取配置
func (c *modelManage) GetModel(ctx context.Context, req *dto.GetModelManageReq) (res *dto.GetModelManageRes, err error) {
	return service.ModelManage.Get(ctx, req)
}

// GetChatModel 获取聊天模型
func (c *modelManage) GetChatModel(ctx context.Context, req *dto.GetChatModelReq) (res *dto.GetChatModelRes, err error) {
	return service.ModelManage.GetChatModel(ctx, req)
}

// ListModel 配置列表
func (c *modelManage) ListModel(ctx context.Context, req *dto.ListModelManageReq) (res *dto.ListModelManageRes, err error) {
	return service.ModelManage.List(ctx, req)
}

// CheckChatModel 检查是否为聊天模型
func (c *modelManage) CheckChatModel(ctx context.Context, req *dto.CheckChatModelReq) (res *dto.CheckChatModelRes, err error) {
	return service.ModelManage.CheckChatModel(ctx, req)
}

// ListType 模型类型列表
func (c *modelManage) ListType(ctx context.Context, req *dto.ModelTypeReq) (res *dto.ModelTypeRes, err error) {
	return service.ModelManage.GetModelType(ctx, req)
}

// ListOperator 运营商列表
func (c *modelManage) ListOperator(ctx context.Context, req *dto.ModelSupplierReq) (res *dto.ModelSupplierRes, err error) {
	return service.ModelManage.GetModelSupplier(ctx, req)
}

// BuildSchemaMapping 自动构建 Schema 映射
func (c *modelManage) BuildSchemaMapping(ctx context.Context, req *dto.BuildSchemaMappingReq) (res *dto.BuildSchemaMappingRes, err error) {
	return service.SchemaMapping.BuildSchemaMapping(ctx, req)
}