1

秦芳睿

9 天以前 e70a362606b78a822e93d5117a9013e8f9086faf

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package com.smtaiserver.smtaiserver.javaai.qwen.agent; 
 
import java.util.Map.Entry; 
 
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 
{ 
    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  
    { 
 
        // 分析生成js代码 
        String jsCode = llm.callWithMessage(new String[] {_prompt}, question, tranReq).replace("\r", ""); 
        tranReq.traceLLMDebug(jsCode); 
        jsCode = SMTStatic.convLLMAnswerToJavascript(jsCode); 
        //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); 
                } 
                finally 
                { 
                    Context.exit(); 
                } 
            } 
            catch(Exception ex) 
            { 
                throw new Exception("execute js agent error : " + this.getAgentId(), ex); 
            } 
        } 
        return null; 
    } 
 
}