package util

import (
	"context"
	"strings"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/util/gconv"
)

// GetServerPort 从配置获取服务端口
func GetServerPort(ctx context.Context) string {
	address := g.Cfg().MustGet(ctx, "server.address", ":8080").String()
	// address 格式如 ":3009"，去掉冒号
	if strings.HasPrefix(address, ":") {
		return address[1:]
	}
	return "8080"
}

// GetModelPrompt 获取请求模型的提示词
func GetModelPrompt(ctx context.Context, modelType int) string {
	key := "modelPrompts.types." + gconv.String(modelType)
	return g.Cfg().MustGet(ctx, key, "").String()
}

// GetBuildPrompt 获取节点构建提示词
func GetBuildPrompt(ctx context.Context) string {
	return g.Cfg().MustGet(ctx, "nodePrompts", "").String()
}