package prompt

import (
	"data_ai/ai"
)

// 获取外围抽取字段
func AcquireTitleBuyerInfo(detail string) map[string]interface{} {
	content := PromptBuyerText(detail)
	res := map[string]interface{}{}
	res = ai.PostZhiPuInfo(content)
	return res
}

// 获取外围抽取字段
func AcquireBuyerInfo(detail string) map[string]interface{} {
	content := PromptBuyerText(detail)
	res := map[string]interface{}{}
	res = ai.PostZhiPuInfo(content)
	return res
}

// 获取外围抽取字段
func AcquireDeepSeekBuyerInfo(detail string) map[string]interface{} {
	content := PromptBuyerText(detail)
	res := map[string]interface{}{}
	res = ai.PostDeepSeekInfo(content)
	return res
}

// 提示词优选 - 提问词补偿不需要限制
func PromptBuyerText(detail string) string {
	content := `请根据我提供的正文进行"实体单位"的抽取；
你在识别"实体单位"的时候，只能返回一个最全的实体单位，请务必不要省略分公司，全称的中小学校等局级单位等，如果是多个实体单位，请填写"无"
请将上述的识别结果、信息分类结果，按照JSON格式输出，
严格按照json格式
{
"实体单位":"实体单位",
}
请回答我的问题，不要联想，不要无中生有，不要生成解释，对于尚未确定或未明确的信息请在JSON对应的值填写：无
正文内容：` + "\n" + detail + "\n结果JSON："
	return content
}