1

秦芳睿

9 天以前 e70a362606b78a822e93d5117a9013e8f9086faf

 JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/control/SMTJavaAIControl.java

@@ -3,17 +3,11 @@
import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.InputStream;

import java.util.ArrayList;

import java.util.Date;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.*;

import java.util.Map.Entry;

import java.util.Set;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;



import com.google.gson.Gson;

import org.springframework.web.bind.annotation.RequestParam;

import org.springframework.web.multipart.MultipartFile;

import org.springframework.web.servlet.ModelAndView;

@@ -43,6 +37,47 @@


public class SMTJavaAIControl 

{

   private class SMTCallAgentArgStreamView implements View

   {

      private SMTAIServerRequest    _tranReq;

		
      public SMTCallAgentArgStreamView(SMTAIServerRequest tranReq)

      {

         _tranReq = tranReq;

         _tranReq.setChunkedMode(true);

      }

		
      @Override

      public void render(Map<String, ?> model, HttpServletRequest request, HttpServletResponse response) throws Exception 
      {

         String agentId = _tranReq.convParamToString("agent_id", true);

         Json jsonArgs = _tranReq.convParamToJson("call_args", true);

			
         response.setCharacterEncoding("UTF-8");

         response.setContentType("text/plain;charset=UTF-8");

         response.setHeader("smtstream", "chunked");

			
         try

         {

            SMTQwenAgent agent = SMTAIServerApp.getApp().getQwenAgentManager().getAgentById(agentId);

            if(agent == null)

               throw new Exception("can't find agent : " + agentId);

            SMTLLMConnect llm = SMTAIServerApp.getApp().allocLLMConnect(null);

				
            if(!jsonArgs.has("question"))

               jsonArgs.set("question", "");

				
            agent.callAgents("/", jsonArgs, llm, "", _tranReq);

         }

         catch(Exception ex)

         {

            _tranReq.traceLLMDebug("call chat stream excetion : \n" + SMTStatic.toString(ex));

            _tranReq.sendChunkedBlock("error", ex.getMessage());

         }

         _tranReq.sendChunkedBlock("finish", "");

      }

   }

	
   private class SMTAgentStreamView implements View

   {

      private SMTAIServerRequest    _tranReq;

@@ -97,10 +132,14 @@
               {

                  for(MultipartFile file : files)

                  {

                     String fileId = SMTStatic.newUUID();

                     String fileText="";

                     String fileSuffix = (String) SMTAIServerApp.getApp().getGlobalConfig("file_suffix", false);

                     String fileName = new File(file.getOriginalFilename()).getName();

                     byte[] fileData = readAttachFile(file);

                     String fileText = new String(fileData, "UTF-8");

                     if (fileSuffix.contains(fileName.substring(fileName.lastIndexOf(".") + 1))) {

                        fileText = SMTAIServerApp.fileTranslTxt(file);

                     }

                     String fileId = SMTStatic.newUUID();

                     db.executeSQL(

                          " INSERT INTO chat_history_attach(attach_id, attach_type, attach_title, attach_bytes)"

                        + " VALUES(?,?,?,?)", new Object[] {

@@ -356,7 +395,7 @@
          String sampleId = tranReq.convParamToString("sample_id", false);

          String llmId = tranReq.convParamToString("llm_id", false);

          String historyGroupId = tranReq.convParamToString("history_group_id", true);

          String question = java.net.URLDecoder.decode(tranReq.convParamToString("question", true), "UTF-8");

          String question = tranReq.convParamToString("question", true);

          String processId = tranReq.convParamToString("process_id", false);

          String sRawMode = tranReq.convParamToString("raw_mode", false);

          String prevQuestion = tranReq.convParamToString("prev_question", false);

@@ -365,8 +404,6 @@
   

          if(SMTStatic.isNullOrEmpty(groupType))

             groupType = "业务场景";

          else

             groupType = java.net.URLDecoder.decode(groupType, "UTF-8");

          

          tranReq.setCurQuestionPos(curPos);

          tranReq.setCurGroupType(groupType);

@@ -411,53 +448,62 @@
   {

      tranReq.sendChunkedBlock("begin", "分析[" + jsonSummary.safeGetStr("title", "") + "]的数据");

      

      StringBuilder sbQuestion = new StringBuilder();

      sbQuestion.append("以下是[" + jsonSummary.safeGetStr("title", "") + "]的数据，请对其进行分析，并提炼出最精简的结果。\n");

      

      boolean isFirst = true;

      int maxCol = 0;

      for(Json jsonCol : jsonSummary.getJson("cols").asJsonList())

      boolean set = tranReq.setSendStarStream(true);

      try

      {

         if(isFirst)

            isFirst = false;

         else

            sbQuestion.append(",");

         sbQuestion.append(jsonCol.safeGetStr("title", "值"));

         maxCol ++;

      }

      sbQuestion.append("\n");

		
      for(Json jsonRow : jsonSummary.getJson("values").asJsonList())

      {

         List<Json> listJsonRow = jsonRow.asJsonList();

         for(int i = 0; i < maxCol; i ++)

         StringBuilder sbQuestion = new StringBuilder();

         sbQuestion.append("以下是[" + jsonSummary.safeGetStr("title", "") + "]的数据，请对其进行分析，并提炼出最精简的结果。\n");

			
         boolean isFirst = true;

         int maxCol = 0;

         for(Json jsonCol : jsonSummary.getJson("cols").asJsonList())

         {

            if(i > 0)

               sbQuestion.append(",");

            Json jsonValue = listJsonRow.get(i);

            if(jsonValue.isNull())

            {

               sbQuestion.append("");

            }

            if(isFirst)

               isFirst = false;

            else

            {

               sbQuestion.append(listJsonRow.get(i).asString());

            }

               sbQuestion.append(",");

            sbQuestion.append(jsonCol.safeGetStr("title", "值"));

            maxCol ++;

         }

         sbQuestion.append("\n");

         

         if(sbQuestion.length() > 4096)

            break;

         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();

      }

		
      String answer = llm.callWithMessage(null, sbQuestion.toString(), tranReq).replace("\r", "");

		
      jsonWrConclusion.beginMap(null);

      finally

      {

         jsonWrConclusion.addKeyValue("agent_key", agentKey);

         jsonWrConclusion.addKeyValue("report", answer);

         tranReq.setSendStarStream(set);

      }

      jsonWrConclusion.endMap();

      

      return null;

   }

@@ -1307,20 +1353,20 @@
       

       return tranReq.returnJson(jsonWr);

    }

    
    public ModelAndView callLLM(SMTAIServerRequest tranReq) throws Exception 


    public ModelAndView callLLM(SMTAIServerRequest tranReq) throws Exception

    {

       String llmId = tranReq.convParamToString("llm_id", false);

       Json jsonSystems = tranReq.convParamToJson("system_json", false);

       String question = tranReq.convParamToString("question", true);

       boolean answerIsJson = tranReq.convParamToBoolean("answer_is_json", true);

    	
    	




       SMTJsonWriter jsonWr = tranReq.newReturnJsonWriter(true, null, null);

       SMTLLMConnect conn = SMTAIServerApp.getApp().allocLLMConnect(llmId);

    	


       List<String> listSystem = new ArrayList<>();

    	


       if(jsonSystems != null)

       {

          for(Json jsonSystem : jsonSystems.asJsonList())

@@ -1328,7 +1374,7 @@
             listSystem.add(jsonSystem.asString());

          }

       }

    	


       String answer = conn.callWithMessage(listSystem, question, tranReq);

       if(answerIsJson)

       {

@@ -1341,8 +1387,7 @@


      return tranReq.returnJson(jsonWr);

    }



   public void downloadFileByFileId(SMTAIServerRequest tranReq, HttpServletResponse response) throws Exception {

   public ModelAndView downloadFileByFileId(SMTAIServerRequest tranReq, HttpServletResponse response) throws Exception {

      String fileId = tranReq.convParamToString("file_id", true);  // 获取请求中的file_id

      SMTDatabase db = SMTAIServerApp.getApp().allocDatabase();



@@ -1352,20 +1397,65 @@
         if (dbRecords.getRecords().isEmpty()) {

            throw new Exception("文件不存在");

         }



         // 获取文件信息

         DBRecord fileRecord = dbRecords.getRecords().get(0);

         String attachTitle = fileRecord.getString("attach_title");  // 获取文件标题

         String attachBytes = fileRecord.getString("attach_bytes");

         byte[] bytes = attachBytes.getBytes();

         // 设置响应头和文件名

         String fileName = attachTitle != null ? attachTitle : "downloaded_file";

         response.setHeader("Content-Disposition", "attachment; filename=\"" + fileName + "\"");

         response.setContentType("application/octet-stream");

            response.getOutputStream().write(bytes);

            response.getOutputStream().flush();

         String extension = attachTitle.substring(attachTitle.lastIndexOf(".") + 1);

         // 扩展名和 MIME 类型映射



         Object fileTypeMapping = SMTAIServerApp.getApp().getGlobalConfig("file_type_mapping", false);

         Json json = Json.read(fileTypeMapping.toString());

         Map<String, Object> mimeTypes = json.asMap();

         String mimeType = (String) mimeTypes.getOrDefault(extension.toLowerCase(), "application/octet-stream");

         // 获取文件的字节数据，假设返回的是字节数组

         byte[] bytes = (byte[]) fileRecord.getValue("attach_bytes");  // 获取字节数据

         return tranReq.returnDownloadByteArray(bytes, attachTitle, mimeType);

//         // 设置响应头和文件名

//         String fileName = attachTitle != null ? attachTitle : "downloaded_file";

//         response.setHeader("Content-Disposition", "attachment; filename=\"" + fileName + "\"");

//         response.setContentType("application/octet-stream");

//         response.getOutputStream().write(bytes);  // 写入字节数据

//         response.getOutputStream().flush();

      } catch (Exception e) {

         throw new Exception("Error processing file download: " + e.getMessage());

      }

   }



	
   public ModelAndView chatCallAgent(SMTAIServerRequest tranReq) throws Exception

   {

      String agentId = tranReq.convParamToString("agent_id", true);

      Json jsonArgs = tranReq.convParamToJson("args", false);

      SMTQwenAgent agent = SMTAIServerApp.getApp().getQwenAgentManager().getAgentById(agentId);

		
      SMTLLMConnect llm = SMTAIServerApp.getApp().allocLLMConnect(null);

		
      if(jsonArgs == null)

         jsonArgs = Json.object("question", "");

		
      if(!jsonArgs.has("question"))

         jsonArgs.set("question", "");

		
      agent.callAgents("/", jsonArgs, llm, "", tranReq);

      List<SMTJsonWriter> listResult = tranReq.getResultJsonWrList();

		
      SMTJsonWriter jsonWr = tranReq.newReturnJsonWriter(true, null, null);

      jsonWr.beginArray("results");

      if(listResult != null && listResult.size() > 0)

      {

         for(SMTJsonWriter jsonWrResult : listResult)

         {

            jsonWr.addKeyRaw(null, jsonWrResult.getRootJson());

         }

      }

      jsonWr.endArray();

		
      return tranReq.returnJson(jsonWr);

		
   }

	
   public ModelAndView callAgentArgStream(SMTAIServerRequest tranReq) throws Exception

   {

       return new ModelAndView(new SMTCallAgentArgStreamView(tranReq));

   }

}