package com.smtaiserver.smtaiserver.javaai.qwen.agent;
|
|
import java.util.Map.Entry;
|
|
import org.apache.logging.log4j.LogManager;
|
import org.apache.logging.log4j.Logger;
|
import org.dom4j.Document;
|
import org.dom4j.Element;
|
import org.mozilla.javascript.Context;
|
|
import com.smtaiserver.smtaiserver.core.SMTAIServerRequest;
|
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecord;
|
import com.smtaiserver.smtaiserver.javaai.SMTJavaAIError;
|
import com.smtaiserver.smtaiserver.javaai.ast.ASTDBMap;
|
import com.smtaiserver.smtaiserver.javaai.llm.core.SMTLLMConnect;
|
import com.smtaiserver.smtaiserver.javaai.qwen.agent.script.SMTQwenAgentScriptJet;
|
import com.smtservlet.util.Json;
|
import com.smtservlet.util.SMTStatic;
|
|
public class SMTQwenAgentJavascript extends SMTQwenAgent
|
{
|
private static Logger _logger = LogManager.getLogger(SMTQwenAgentJavascript.class);
|
|
|
protected String _prompt;
|
|
@Override
|
public void initInstance(DBRecord rec) throws Exception
|
{
|
super.initInstance(rec);
|
|
try
|
{
|
Document doc = SMTStatic.convStrToXmlDoc("<ROOT>" + rec.getString("clz_arguments") + "</ROOT>");
|
Element xmlPrompt = (Element)doc.selectSingleNode("ROOT/PROMPT");
|
|
_prompt = xmlPrompt.getText();
|
|
if(this._mapName2Argument.size() > 0)
|
{
|
StringBuilder sbArgs = new StringBuilder();
|
sbArgs.append("其中系统提供的参数是:\n");
|
for(Entry<String, AgentArgument> entry : this._mapName2Argument.entrySet())
|
{
|
sbArgs.append(entry.getKey() + " : " + entry.getValue()._prompt + "\n");
|
}
|
|
_prompt += sbArgs.toString();
|
}
|
|
}
|
catch(Exception ex)
|
{
|
throw new Exception("init mertic agent error : " + this._agentId, ex);
|
}
|
}
|
|
@Override
|
public SMTJavaAIError callAgents(String jsonPath, Json jsonArgs, SMTLLMConnect llm, String question, SMTAIServerRequest tranReq) throws Exception
|
{
|
int maxCount = 1;
|
|
// 失败最多尝试3次
|
for(int j = 1; j <= maxCount; j ++)
|
{
|
// 分析生成js代码
|
String jsCode = llm.callWithMessage(new String[] {_prompt}, question, tranReq).replace("\r", "");
|
tranReq.traceLLMDebug(jsCode);
|
jsCode = SMTStatic.convLLMAnswerToJavascript(jsCode);
|
jsCode = "function ffffff(){" + jsCode + "};ffffff();";
|
//String jsCode = "var arr = testRS(); var l = arr[0]; for(var i in arr){}";
|
|
try(ASTDBMap dbMap = new ASTDBMap())
|
{
|
// 执行脚本
|
SMTQwenAgentScriptJet scriptJet = new SMTQwenAgentScriptJet(jsonArgs, dbMap, tranReq);
|
try
|
{
|
Context cx = scriptJet.entryContext();
|
try
|
{
|
scriptJet.executeScript(cx, jsCode);
|
break;
|
}
|
finally
|
{
|
Context.exit();
|
}
|
}
|
catch(Exception ex)
|
{
|
StringBuilder sbCode = new StringBuilder();
|
String[] lines = jsCode.replace("\r", "").split("\n");
|
for(int i = 0; i < lines.length; i ++)
|
{
|
sbCode.append(String.format("%04d %s", i + 1, lines[i]) + "\n");
|
_logger.fatal(String.format("%04d %s", i + 1, lines[i]));
|
}
|
|
tranReq.sendChunkedBlock("begin", "解析失败(" + ex.getMessage() + "),重新解析");
|
|
if(j < maxCount)
|
_logger.fatal("execute js agent error : " + this.getAgentId() + "\n" + sbCode, ex);
|
else
|
throw new Exception("execute js agent error : " + this.getAgentId() + "\n" + sbCode + "\n" + SMTStatic.toString(ex));
|
}
|
}
|
}
|
return null;
|
}
|
|
}
|
