|
|
@@ -12,6 +12,7 @@ import com.cyksj.common.util.StringUtil;
|
|
|
import com.cyksj.mapper.mini.*;
|
|
|
import com.cyksj.model.entity.*;
|
|
|
import com.cyksj.model.request.MiniAiPicturesReq;
|
|
|
+import com.cyksj.model.response.MiniPopularizeResp;
|
|
|
import com.cyksj.redis.RedisService;
|
|
|
import com.cyksj.server.listener.ConsoleStreamListener;
|
|
|
import com.cyksj.service.mini.MiniPopularizeCommonService;
|
|
|
@@ -69,7 +70,7 @@ public class MiniPopularizeFrontServiceImpl implements MiniPopularizeFrontServic
|
|
|
|
|
|
@Override
|
|
|
@Transactional(rollbackFor = Throwable.class)
|
|
|
- public Long createPictures(MiniAiPicturesReq miniAiPicturesReq) throws IOException {
|
|
|
+ public MiniPopularizeResp createPictures(MiniAiPicturesReq miniAiPicturesReq) throws IOException {
|
|
|
String conversationId = miniAiPicturesReq.getConversationId();
|
|
|
AtomicBoolean recoverNum = new AtomicBoolean(true);
|
|
|
Long userId = miniAiPicturesReq.getUserId();
|
|
|
@@ -117,7 +118,12 @@ public class MiniPopularizeFrontServiceImpl implements MiniPopularizeFrontServic
|
|
|
.recoverNum(recoverNum.get())
|
|
|
.build();
|
|
|
streamChatCompletion(miniAiPicturesReq, listener);
|
|
|
- return imageRecord.getId();
|
|
|
+
|
|
|
+ MiniPopularizeResp resp = MiniPopularizeResp.builder()
|
|
|
+ .conversationId(conversationId)
|
|
|
+ .chatId(imageRecord.getId())
|
|
|
+ .build();
|
|
|
+ return resp;
|
|
|
}
|
|
|
|
|
|
@Override
|