package com.smtaiserver.smtaiserver.javaai.qwen.agent;

import java.util.ArrayList;
import java.util.List;

import com.smtaiserver.smtaiserver.core.SMTAIServerRequest;
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecord;
import com.smtaiserver.smtaiserver.javaai.SMTJavaAIError;
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 SMTQwenAgentKnowlgFile extends SMTQwenAgent 
{
	protected int	_vectorScope;
	
	@Override
	public void initInstance(DBRecord rec) throws Exception
	{
		_inSupervisor = !("Y".equals(rec.getString("inner_call")));
		_agentTitle = rec.getString("knowlg_title");
		
		// 不要调用基类的initInstance，因为它是从knowlg_name_list过来的表
		_agentId = rec.getString("knowlg_id");
		_agentType = "knowledge";
		_agentPrompt = 
		      _agentId + "\n"
			+ "    功能:\n"
			+ "        " + SMTStatic.indentStrLines(rec.getString("knowlg_prompt"), "        ") + "\n"
			+ "    参数:\n"
			+ "        question:提出的问题\n"
			;
		
		_agentGroup = rec.getString("agent_group");
		if(rec.getFieldMap().containsKey("GROUP_TYPE"))
			_agentGroupType = rec.getString("group_type");
		else
			_agentGroupType = "";

		Integer vectorScope = rec.getInteger("vector_scope");
		if(vectorScope != null)
			_vectorScope = vectorScope;
		else
			_vectorScope = 90;
	}
	
	@Override
	public SMTJavaAIError callAgents(String jsonPath, Json jsonArgs, SMTLLMConnect llm, String question, SMTAIServerRequest tranReq) throws Exception
	{
		SMTJsonWriter jsonWrResult = tranReq.getResultJsonWr();
		
		String vectorSQL = 
			   "SELECT file_name as vector_title, vector_text as vector_message, ?::vector <=> vector_value AS ratio  \r\n"
			+ "FROM ai_doc.doc_file_vector A \r\n"
			+ "INNER JOIN ai_doc.doc_file_list B \r\n"
			+ "ON A.file_id = B.file_id\r\n"
			+ "INNER JOIN ai_doc.knowlg_file_list c\r\n"
			+ "ON C.knowlg_file_id=B.file_id\r\n"
			+ "AND C.knowlg_id='" + _agentId + "'\r\n"
			+ "ORDER BY ratio LIMIT 4"
			;
		
		jsonWrResult.addKeyValue("answer_type", "knowledge");
		jsonWrResult.beginArray("knowledge");
		{
			jsonWrResult.beginMap(null);
			{
				long tick = System.currentTimeMillis();
				List<String[]> contexts = new ArrayList<>();
				String result = llm.callWithVector(question, tranReq.getRawQuestion(), vectorSQL, _vectorScope, tranReq, contexts);
				tranReq.traceLLMDebug("executeCmd_QUERY回答问题：[" + ((double)(System.currentTimeMillis() - tick) / 1000) + "秒] [\n" + result + "\n]");
				jsonWrResult.addKeyValue("answer", result);
				jsonWrResult.beginArray("contexts");
				for(String[] context : contexts)
				{
					jsonWrResult.beginMap(null);
					{
						jsonWrResult.addKeyValue("page_content", context[1]);
						jsonWrResult.beginMap("metadata");
						{
							jsonWrResult.addKeyValue("Title", context[0]);
						}
						jsonWrResult.endMap();
					}
					jsonWrResult.endMap();
				}
				jsonWrResult.endArray();
			}
			jsonWrResult.endMap();
		}
		jsonWrResult.endArray();

		return null;
	}

}