添加rag graph的agent

TangCheng

2025-04-10 19a6c672450cd7d580b071b66ab80f62f080f6f1

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
package com.smtaiserver.smtaiserver.javaai.qwen.agent; 
 
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; 
 
public class SMTQwenAgentRAGGraph extends SMTQwenAgent  
{ 
 
    @Override 
    public void initInstance(DBRecord rec) throws Exception 
    { 
        super.initInstance(rec); 
         
    } 
     
    @Override 
    public SMTJavaAIError callAgents(String jsonPath, Json jsonArgs, SMTLLMConnect llm, String question, SMTAIServerRequest tranReq) throws Exception  
    { 
        SMTJsonWriter jsonWrResult = tranReq.getResultJsonWr(); 
         
        StringBuilder sbAnswer = new StringBuilder(); 
        SMTJavaAIError error = callRAGServer(question, tranReq, sbAnswer); 
        if(error != null) 
            return error; 
        jsonWrResult.addKeyValue("json_ok", true); 
        jsonWrResult.addKeyValue("answer_type", "knowledge"); 
        jsonWrResult.beginArray("knowledge"); 
        { 
            jsonWrResult.beginMap(null); 
            jsonWrResult.addKeyValue("answer", sbAnswer.toString()); 
            jsonWrResult.endMap(); 
        } 
        jsonWrResult.endArray(); 
        return null; 
    } 
     
    private SMTJavaAIError callRAGServer(String question, SMTAIServerRequest tranReq, StringBuilder sbAnswer) throws Exception 
    { 
        String answer = "ABCDEFGHIJKLM"; 
         
        tranReq.sendChunkedBlock("begin_stream", ""); 
        try 
        { 
            for(int i = 0; i < answer.length(); i ++) 
            { 
                char ch = answer.charAt(i); 
                tranReq.sendChunkedStreamBlock(com.smtservlet.util.SMTStatic.toString(ch)); 
                sbAnswer.append(ch); 
                 
                Thread.sleep(1000); 
            } 
        } 
        finally 
        { 
            tranReq.sendChunkedBlock("end_stream", ""); 
        } 
         
        return null; 
    } 
 
}