package com.smtaiserver.smtaiserver.javaai.jsonflow.node;

import java.util.HashSet;
import java.util.List;
import java.util.Set;

import com.smtaiserver.smtaiserver.core.SMTAIServerApp;
import com.smtaiserver.smtaiserver.core.SMTAIServerRequest;
import com.smtaiserver.smtaiserver.javaai.SMTJavaAIError;
import com.smtaiserver.smtaiserver.javaai.jsonflow.core.SMTJsonFlowExecArg;
import com.smtaiserver.smtaiserver.javaai.jsonflow.core.SMTJsonFlowManager;
import com.smtaiserver.smtaiserver.javaai.jsonflow.core.SMTJsonFlowNodeOnlyOutput;
import com.smtaiserver.smtaiserver.javaai.llm.core.SMTLLMConnect;
import com.smtservlet.util.Json;
import com.smtservlet.util.SMTJsonWriter;
import com.smtservlet.util.SMTStatic;

public class SMTJsonFlowNodeProcedure extends SMTJsonFlowNodeOnlyOutput 
{
	private String		_procId;
	
	@Override
	public void initInstane(SMTJsonFlowManager manager, Json jsonNode) throws Exception
	{
		super.initInstane(manager, jsonNode);
		
		Json jsonValues = jsonNode.getJsonPath("data|group_params|0|params|0|value", false);
		if(jsonValues == null || !jsonValues.isString())
			throw new Exception("flow json agenet not set call agent id");
		_procId = jsonValues.asString();
	}
	
	@Override
	public SMTJavaAIError executeFlowNode(SMTJsonFlowExecArg execArg) throws Exception 
	{
		SMTJavaAIError error;
		if("assessment".equals(_procId))
		{
			execArg._tranReq.sendChunkedBlock("begin", "开始评估结果集");
			error = callProc_assessment(execArg);
		}
		else
		{
			return new SMTJavaAIError("未知工作流操作:" + _procId);
		}
		if(error != null)
			return error;
		
		return super.executeFlowNode(execArg);
	}
	
	private SMTJavaAIError callProc_assessment(SMTJsonFlowExecArg execArg) throws Exception
	{
		// 获取要评估的结果集
		List<SMTJsonWriter> listJsonWrResult = execArg._tranReq.getResultJsonWrList();
		if(listJsonWrResult == null || listJsonWrResult.size() == 0)
		{
			execArg._tranReq.sendChunkedBlock("begin", "无任何结果集需要评估");
			return null;
		}
		
		// 对结果进行评估
		for(SMTJsonWriter jsonWrResult : listJsonWrResult)
		{
			Json jsonResult = jsonWrResult.getRootJson();
			Set<String> setExistAgentKey = new HashSet<>();
			SMTJsonWriter jsonWrConclusion = new SMTJsonWriter(true);
			for(Json jsonSummary : jsonResult.getJson("summary").asJsonList())
			{
				String agentKey = jsonSummary.safeGetStr("agent_key", "");
				if(setExistAgentKey.contains(agentKey))
					continue;
				
				if(!"recordset".equals(jsonSummary.safeGetStr("type", "")))
					continue;
				
				setExistAgentKey.add(agentKey);
				
				SMTLLMConnect llm = SMTAIServerApp.getApp().allocLLMConnect(null);
				try
				{
					parseRecordsetTypeConclusionWithLLM(agentKey, llm, jsonSummary, execArg._tranReq, jsonWrConclusion);
				}
				finally
				{
					llm.close();
				}
			}			
		}
		
		
		return null;
	}

	private SMTJavaAIError parseRecordsetTypeConclusionWithLLM(String agentKey, SMTLLMConnect llm, Json jsonSummary, SMTAIServerRequest tranReq, SMTJsonWriter jsonWrConclusion) throws Exception
	{
		tranReq.sendChunkedBlock("begin", "分析[" + jsonSummary.safeGetStr("title", "") + "]的数据");
		
		StringBuilder sbQuestion = new StringBuilder();
		sbQuestion.append("以下是[" + jsonSummary.safeGetStr("title", "") + "]的数据，请对其进行分析，并提炼出最精简的结果。\n");
		
		String unitName = jsonSummary.safeGetStr("unit", null);

		
		boolean isFirst = true;
		int maxCol = 0;
		for(Json jsonCol : jsonSummary.getJson("cols").asJsonList())
		{
			if(isFirst)
				isFirst = false;
			else
				sbQuestion.append(",");
			
			String colName = jsonCol.safeGetStr("title", "值");
			if(!SMTStatic.isNullOrEmpty(unitName) && "value".equals(jsonCol.safeGetStr("type", "")))
				sbQuestion.append(colName + "(" + unitName + ")");
			else
				sbQuestion.append(colName);
			maxCol ++;
		}
		sbQuestion.append("\n");
		
		for(Json jsonRow : jsonSummary.getJson("values").asJsonList())
		{
			List<Json> listJsonRow = jsonRow.asJsonList();
			for(int i = 0; i < maxCol; i ++)
			{
				if(i > 0)
					sbQuestion.append(",");
				Json jsonValue = listJsonRow.get(i);
				if(jsonValue.isNull())
				{
					sbQuestion.append("");
				}
				else
				{
					sbQuestion.append(listJsonRow.get(i).asString());
				}
			}
			sbQuestion.append("\n");
			
			if(sbQuestion.length() > 4096)
				break;
		}
		
		String answer = llm.callWithMessage(null, sbQuestion.toString(), tranReq).replace("\r", "");
		
		jsonWrConclusion.beginMap(null);
		{
			jsonWrConclusion.addKeyValue("agent_key", agentKey);
			jsonWrConclusion.addKeyValue("report", answer);
		}
		jsonWrConclusion.endMap();
		
		tranReq.sendChunkedBlock("conclusion", jsonWrConclusion.getRootJson());
		
		return null;
	}
}