秦芳睿
2025-03-13 d9111b072c5bbf196435be9aa7f1f849cd07252b
JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/core/SMTAIServerRequest.java
@@ -50,6 +50,13 @@
      {
         return "附加文件 : " + _fileName + "\n" + _fileText + "\n";
      }
      public void addToJson(SMTJsonWriter jsonWr)
      {
         jsonWr.addKeyValue("file_id", _fileId);
         jsonWr.addKeyValue("file_name", _fileName);
         jsonWr.addKeyValue("file_content", _fileText);
      }
   }
   
   private static Logger          _logger = LogManager.getLogger(SMTAIServerRequest.class);
@@ -89,6 +96,9 @@
   private List<SMTJsonWriter>      _listJsonWrResult = null;
   private boolean               _disableConclusion = false;
   private List<AIAttachFile>      _listAttchFile = null;
   private Json               _jsonAttachTables = null;
   private Json               _jsonAttachMetrics = null;
   private boolean               _sendStarStream = false; // 将输出文字变成*
   @Override
@@ -114,6 +124,22 @@
      }
   }
   
   public boolean setSendStarStream(boolean isSetStartStream)
   {
      boolean org = _sendStarStream;
      _sendStarStream = isSetStartStream;
      return org;
   }
   public void setAttachTables(Json jsonTables)
   {
      _jsonAttachTables = jsonTables;
   }
   public void setJsonAttachMetrics(Json attachMetrics)
   {
      _jsonAttachMetrics = attachMetrics;
   }
   public String[] getAttachMessage()
   {
      List<String> list = new ArrayList<>();
@@ -126,7 +152,99 @@
         }
      }
      
      if(_jsonAttachTables != null)
      {
         for(Json jsonAttachTable : _jsonAttachTables.asJsonList())
         {
            StringBuilder sbText = new StringBuilder();
            sbText.append("现有表格:" + jsonAttachTable.getJson("title").asString() + "\n");
            List<Json> jsonColumns = jsonAttachTable.getJson("columns").asJsonList();
            for(int i = 0; i < jsonColumns.size(); i ++)
            {
               if(i > 0)
                  sbText.append(",");
               sbText.append(jsonColumns.get(i).asString());
            }
            sbText.append("\n");
            for(Json jsonRecord : jsonAttachTable.getJson("values").asJsonList())
            {
               List<Json> listRec = jsonRecord.asJsonList();
               for(int i = 0; i < listRec.size(); i ++)
               {
                  if(i > 0)
                     sbText.append(",");
                  if(listRec.get(i).isNull())
                     sbText.append("");
                  else
                     sbText.append(listRec.get(i).asString());
               }
               sbText.append("\n");
            }
            sbText.append("\n");
            list.add(sbText.toString());
         }
      }
      if(_jsonAttachMetrics != null)
      {
         for(Json jsonAttachTable : _jsonAttachMetrics.asJsonList())
         {
            StringBuilder sbText = new StringBuilder();
            sbText.append("现有表格:" + jsonAttachTable.getJson("title").asString() + "\n");
            List<Json> jsonColumns = jsonAttachTable.getJson("columns").asJsonList();
            for(int i = 0; i < jsonColumns.size(); i ++)
            {
               if(i > 0)
                  sbText.append(",");
               sbText.append(jsonColumns.get(i).asString());
            }
            sbText.append("\n");
            for(Json jsonRecord : jsonAttachTable.getJson("values").asJsonList())
            {
               List<Json> listRec = jsonRecord.asJsonList();
               for(int i = 0; i < listRec.size(); i ++)
               {
                  if(i > 0)
                     sbText.append(",");
                  if(listRec.get(i).isNull())
                     sbText.append("");
                  else
                     sbText.append(listRec.get(i).asString());
               }
               sbText.append("\n");
            }
            sbText.append("\n");
            list.add(sbText.toString());
         }
      }
      return list.size() == 0 ? null : list.toArray(new String[list.size()]);
   }
   public void addAttachMessageToJson(Json jsonResult)
   {
      if(_listAttchFile != null)
      {
         SMTJsonWriter jsonWr = new SMTJsonWriter(true);
         for(AIAttachFile file : _listAttchFile)
         {
            jsonWr.beginMap(null);
            file.addToJson(jsonWr);
            jsonWr.endMap();
         }
         jsonResult.set("attach_files", jsonWr.getRootJson());
      }
      if(_jsonAttachTables != null)
      {
         jsonResult.set("attach_tables", _jsonAttachTables);
      }
      if(_jsonAttachMetrics != null)
      {
         jsonResult.set("attach_metrics", _jsonAttachMetrics);
      }
   }
   
   public void addAttachFile(AIAttachFile attachFile)
@@ -440,6 +558,17 @@
      }
   }
   
   public void sendChunkedStreamBlock(String stream) throws Exception
   {
      if(_sendStarStream)
      {
         stream = "#";
         if(stream.indexOf("\n") >= 0)
            stream += "\n";
      }
      sendChunkedBlock("send_stream", stream);
   }
   public void sendChunkedBlock(String mode, Object value) throws Exception
   {
      if(!_isChunked)