parent
9bfbd3674d
commit
f0337024f1
@ -0,0 +1,23 @@
|
||||
package xyz.wbsite.achat;
|
||||
|
||||
import org.springframework.web.bind.annotation.GetMapping;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
|
||||
/**
|
||||
* 门户入口
|
||||
*
|
||||
* @author wangbing
|
||||
* @version 0.0.1
|
||||
* @since 1.8
|
||||
*/
|
||||
@RestController
|
||||
public class HelloController {
|
||||
|
||||
/**
|
||||
* 门户
|
||||
*/
|
||||
@GetMapping("/")
|
||||
public String chat() {
|
||||
return "AChat is running!";
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,86 @@
|
||||
package xyz.wbsite.achat;
|
||||
|
||||
import org.springframework.web.bind.annotation.DeleteMapping;
|
||||
import org.springframework.web.bind.annotation.GetMapping;
|
||||
import org.springframework.web.bind.annotation.PathVariable;
|
||||
import org.springframework.web.bind.annotation.PostMapping;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
import xyz.wbsite.achat.core.session.Result;
|
||||
import xyz.wbsite.achat.core.session.Session;
|
||||
import xyz.wbsite.achat.core.session.SessionService;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 会话服务接口
|
||||
* 提供会话、消息的创建、删除、查询等功能
|
||||
*
|
||||
* @author wangbing
|
||||
* @version 0.0.1
|
||||
* @since 1.8
|
||||
*/
|
||||
@RestController
|
||||
public class SessionController {
|
||||
|
||||
@Resource
|
||||
private SessionService sessionService;
|
||||
|
||||
/**
|
||||
* 创建会话
|
||||
*
|
||||
* @param uid 用户标识
|
||||
* @return 创建会话响应
|
||||
*/
|
||||
@PostMapping("/{uid}/session")
|
||||
public Result<Session> createSession(@PathVariable("uid") String uid) {
|
||||
return sessionService.createSession(uid);
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除会话
|
||||
*
|
||||
* @param uid 用户标识
|
||||
* @param sid 会话ID
|
||||
* @return 删除会话响应
|
||||
*/
|
||||
@DeleteMapping("/{uid}/session/{sid}")
|
||||
public Result<Void> deleteSession(@PathVariable("uid") String uid, @PathVariable("sid") String sid) {
|
||||
return sessionService.deleteSession(uid, sid);
|
||||
}
|
||||
|
||||
/**
|
||||
* 会话详情
|
||||
*
|
||||
* @param uid 用户标识
|
||||
* @param sid 会话ID
|
||||
* @return 删除会话响应
|
||||
*/
|
||||
@GetMapping("/{uid}/session/{sid}")
|
||||
public Result<Session> getSession(@PathVariable("uid") String uid, @PathVariable("sid") String sid) {
|
||||
return sessionService.getSession(uid, sid);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取会话列表
|
||||
*
|
||||
* @param uid 用户标识
|
||||
* @return 会话列表响应
|
||||
*/
|
||||
@GetMapping("/{uid}/session/list")
|
||||
public Result<List<Session>> listSession(@PathVariable("uid") String uid) {
|
||||
return sessionService.listSessions(uid);
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除会话历史消息
|
||||
*
|
||||
* @param uid 用户标识
|
||||
* @param sid 会话ID
|
||||
* @return 删除会话响应
|
||||
*/
|
||||
@DeleteMapping("/{uid}/session/{sid}/messages/{mid}")
|
||||
public Result<Void> deleteMessage(@PathVariable("uid") String uid, @PathVariable("sid") String sid, @PathVariable("mid") String mid) {
|
||||
return sessionService.deleteMessage(uid, sid, mid);
|
||||
}
|
||||
}
|
||||
@ -1,30 +0,0 @@
|
||||
//package xyz.wbsite.achat.core;
|
||||
//
|
||||
///**
|
||||
// * 附件
|
||||
// *
|
||||
// * @author wangbing
|
||||
// * @version 0.0.1
|
||||
// * @since 1.8
|
||||
// */
|
||||
//public class Attachment {
|
||||
// private String filename;
|
||||
//
|
||||
// private String fid;
|
||||
//
|
||||
// public String getFilename() {
|
||||
// return filename;
|
||||
// }
|
||||
//
|
||||
// public void setFilename(String filename) {
|
||||
// this.filename = filename;
|
||||
// }
|
||||
//
|
||||
// public String getFid() {
|
||||
// return fid;
|
||||
// }
|
||||
//
|
||||
// public void setFid(String fid) {
|
||||
// this.fid = fid;
|
||||
// }
|
||||
//}
|
||||
@ -1,75 +0,0 @@
|
||||
//package xyz.wbsite.achat.core;
|
||||
//
|
||||
///**
|
||||
// * 服务器推送事件
|
||||
// *
|
||||
// * @author wangbing
|
||||
// * @version 0.0.1
|
||||
// * @since 1.8
|
||||
// */
|
||||
//public class Event {
|
||||
//
|
||||
// private String id;
|
||||
//
|
||||
// private String object;
|
||||
//
|
||||
// private String model;
|
||||
//
|
||||
// private Long created;
|
||||
//
|
||||
// private String sid;
|
||||
//
|
||||
// private String uid;
|
||||
//
|
||||
// public Event() {
|
||||
// this.created = System.currentTimeMillis();
|
||||
// }
|
||||
//
|
||||
// public String getId() {
|
||||
// return id;
|
||||
// }
|
||||
//
|
||||
// public void setId(String id) {
|
||||
// this.id = id;
|
||||
// }
|
||||
//
|
||||
// public String getObject() {
|
||||
// return object;
|
||||
// }
|
||||
//
|
||||
// public void setObject(String object) {
|
||||
// this.object = object;
|
||||
// }
|
||||
//
|
||||
// public String getModel() {
|
||||
// return model;
|
||||
// }
|
||||
//
|
||||
// public void setModel(String model) {
|
||||
// this.model = model;
|
||||
// }
|
||||
//
|
||||
// public Long getCreated() {
|
||||
// return created;
|
||||
// }
|
||||
//
|
||||
// public void setCreated(Long created) {
|
||||
// this.created = created;
|
||||
// }
|
||||
//
|
||||
// public String getSid() {
|
||||
// return sid;
|
||||
// }
|
||||
//
|
||||
// public void setSid(String sid) {
|
||||
// this.sid = sid;
|
||||
// }
|
||||
//
|
||||
// public String getUid() {
|
||||
// return uid;
|
||||
// }
|
||||
//
|
||||
// public void setUid(String uid) {
|
||||
// this.uid = uid;
|
||||
// }
|
||||
//}
|
||||
@ -1,17 +0,0 @@
|
||||
//package xyz.wbsite.achat.core;
|
||||
//
|
||||
//public class Prompt {
|
||||
//
|
||||
// /**
|
||||
// * 提示词
|
||||
// */
|
||||
// private String prompt;
|
||||
//
|
||||
// public String getPrompt() {
|
||||
// return prompt;
|
||||
// }
|
||||
//
|
||||
// public void setPrompt(String prompt) {
|
||||
// this.prompt = prompt;
|
||||
// }
|
||||
//}
|
||||
@ -1,126 +0,0 @@
|
||||
//package xyz.wbsite.achat.core;
|
||||
//
|
||||
//
|
||||
///**
|
||||
// * 会话
|
||||
// *
|
||||
// * @author wangbing
|
||||
// * @version 0.0.1
|
||||
// * @since 1.8
|
||||
// */
|
||||
//public class Session {
|
||||
// /**
|
||||
// * 主键
|
||||
// */
|
||||
// private String id;
|
||||
// /**
|
||||
// * 用户ID
|
||||
// */
|
||||
// private String uid;
|
||||
// private String title;
|
||||
// private String model;
|
||||
// private String prompt;
|
||||
// private String temperature;
|
||||
// private String topP;
|
||||
// private String frequencyPenalty;
|
||||
// private String presencePenalty;
|
||||
// private String maxTokens;
|
||||
// private String lastTime;
|
||||
// private String lastMessage;
|
||||
//
|
||||
// public String getId() {
|
||||
// return id;
|
||||
// }
|
||||
//
|
||||
// public void setId(String id) {
|
||||
// this.id = id;
|
||||
// }
|
||||
//
|
||||
// public String getUid() {
|
||||
// return uid;
|
||||
// }
|
||||
//
|
||||
// public void setUid(String uid) {
|
||||
// this.uid = uid;
|
||||
// }
|
||||
//
|
||||
// public String getTitle() {
|
||||
// return title;
|
||||
// }
|
||||
//
|
||||
// public void setTitle(String title) {
|
||||
// this.title = title;
|
||||
// }
|
||||
//
|
||||
// public String getModel() {
|
||||
// return model;
|
||||
// }
|
||||
//
|
||||
// public void setModel(String model) {
|
||||
// this.model = model;
|
||||
// }
|
||||
//
|
||||
// public String getPrompt() {
|
||||
// return prompt;
|
||||
// }
|
||||
//
|
||||
// public void setPrompt(String prompt) {
|
||||
// this.prompt = prompt;
|
||||
// }
|
||||
//
|
||||
// public String getTemperature() {
|
||||
// return temperature;
|
||||
// }
|
||||
//
|
||||
// public void setTemperature(String temperature) {
|
||||
// this.temperature = temperature;
|
||||
// }
|
||||
//
|
||||
// public String getTopP() {
|
||||
// return topP;
|
||||
// }
|
||||
//
|
||||
// public void setTopP(String topP) {
|
||||
// this.topP = topP;
|
||||
// }
|
||||
//
|
||||
// public String getFrequencyPenalty() {
|
||||
// return frequencyPenalty;
|
||||
// }
|
||||
//
|
||||
// public void setFrequencyPenalty(String frequencyPenalty) {
|
||||
// this.frequencyPenalty = frequencyPenalty;
|
||||
// }
|
||||
//
|
||||
// public String getPresencePenalty() {
|
||||
// return presencePenalty;
|
||||
// }
|
||||
//
|
||||
// public void setPresencePenalty(String presencePenalty) {
|
||||
// this.presencePenalty = presencePenalty;
|
||||
// }
|
||||
//
|
||||
// public String getMaxTokens() {
|
||||
// return maxTokens;
|
||||
// }
|
||||
//
|
||||
// public void setMaxTokens(String maxTokens) {
|
||||
// this.maxTokens = maxTokens;
|
||||
// }
|
||||
//
|
||||
// public String getLastTime() {
|
||||
// return lastTime;
|
||||
// }
|
||||
//
|
||||
// public void setLastTime(String lastTime) {
|
||||
// this.lastTime = lastTime;
|
||||
// }
|
||||
//
|
||||
// public String getLastMessage() {
|
||||
// return lastMessage;
|
||||
// }
|
||||
//
|
||||
// public void setLastMessage(String lastMessage) {
|
||||
// this.lastMessage = lastMessage;
|
||||
// }
|
||||
//}
|
||||
@ -1,103 +0,0 @@
|
||||
//package xyz.wbsite.achat.core.event;
|
||||
//
|
||||
//import xyz.wbsite.achat.core.Event;
|
||||
//
|
||||
///**
|
||||
// * 完成事件
|
||||
// * 用于标识流式响应的结束
|
||||
// *
|
||||
// * @author wangbing
|
||||
// * @version 0.0.1
|
||||
// * @since 1.8
|
||||
// */
|
||||
//public class CompleteEvent extends Event {
|
||||
//
|
||||
// /**
|
||||
// * 完整响应内容
|
||||
// */
|
||||
// private String content;
|
||||
//
|
||||
// /**
|
||||
// * 构造函数
|
||||
// *
|
||||
// * @param sid 会话ID
|
||||
// */
|
||||
// public CompleteEvent(String sid) {
|
||||
// super();
|
||||
// this.setSid(sid);
|
||||
// this.setObject("chat.completion");
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 生成的令牌总数
|
||||
// */
|
||||
// private Integer completionTokens;
|
||||
//
|
||||
// /**
|
||||
// * 提示词的令牌数量
|
||||
// */
|
||||
// private Integer promptTokens;
|
||||
//
|
||||
// /**
|
||||
// * 总令牌数量
|
||||
// */
|
||||
// private Integer totalTokens;
|
||||
//
|
||||
// /**
|
||||
// * 完成状态
|
||||
// */
|
||||
// private String finishReason;
|
||||
//
|
||||
// /**
|
||||
// * 生成用时（毫秒）
|
||||
// */
|
||||
// private Long generationTime;
|
||||
//
|
||||
// public String getContent() {
|
||||
// return content;
|
||||
// }
|
||||
//
|
||||
// public void setContent(String content) {
|
||||
// this.content = content;
|
||||
// }
|
||||
//
|
||||
// public Integer getCompletionTokens() {
|
||||
// return completionTokens;
|
||||
// }
|
||||
//
|
||||
// public void setCompletionTokens(Integer completionTokens) {
|
||||
// this.completionTokens = completionTokens;
|
||||
// }
|
||||
//
|
||||
// public Integer getPromptTokens() {
|
||||
// return promptTokens;
|
||||
// }
|
||||
//
|
||||
// public void setPromptTokens(Integer promptTokens) {
|
||||
// this.promptTokens = promptTokens;
|
||||
// }
|
||||
//
|
||||
// public Integer getTotalTokens() {
|
||||
// return totalTokens;
|
||||
// }
|
||||
//
|
||||
// public void setTotalTokens(Integer totalTokens) {
|
||||
// this.totalTokens = totalTokens;
|
||||
// }
|
||||
//
|
||||
// public String getFinishReason() {
|
||||
// return finishReason;
|
||||
// }
|
||||
//
|
||||
// public void setFinishReason(String finishReason) {
|
||||
// this.finishReason = finishReason;
|
||||
// }
|
||||
//
|
||||
// public Long getGenerationTime() {
|
||||
// return generationTime;
|
||||
// }
|
||||
//
|
||||
// public void setGenerationTime(Long generationTime) {
|
||||
// this.generationTime = generationTime;
|
||||
// }
|
||||
//}
|
||||
@ -1,66 +0,0 @@
|
||||
//package xyz.wbsite.achat.core.event;
|
||||
//
|
||||
//import xyz.wbsite.achat.core.Event;
|
||||
//
|
||||
///**
|
||||
// * 开始推送事件
|
||||
// * 用于标识流式响应的开始
|
||||
// *
|
||||
// * @author wangbing
|
||||
// * @version 0.0.1
|
||||
// * @since 1.8
|
||||
// */
|
||||
//public class StartEvent extends Event {
|
||||
//
|
||||
// /**
|
||||
// * 提示词令牌数量
|
||||
// */
|
||||
// private Integer promptTokens;
|
||||
//
|
||||
// /**
|
||||
// * 最大令牌数量限制
|
||||
// */
|
||||
// private Integer maxTokens;
|
||||
//
|
||||
// /**
|
||||
// * 温度参数
|
||||
// */
|
||||
// private Double temperature;
|
||||
//
|
||||
// /**
|
||||
// * 随机种子
|
||||
// */
|
||||
// private Integer seed;
|
||||
//
|
||||
// public Integer getPromptTokens() {
|
||||
// return promptTokens;
|
||||
// }
|
||||
//
|
||||
// public void setPromptTokens(Integer promptTokens) {
|
||||
// this.promptTokens = promptTokens;
|
||||
// }
|
||||
//
|
||||
// public Integer getMaxTokens() {
|
||||
// return maxTokens;
|
||||
// }
|
||||
//
|
||||
// public void setMaxTokens(Integer maxTokens) {
|
||||
// this.maxTokens = maxTokens;
|
||||
// }
|
||||
//
|
||||
// public Double getTemperature() {
|
||||
// return temperature;
|
||||
// }
|
||||
//
|
||||
// public void setTemperature(Double temperature) {
|
||||
// this.temperature = temperature;
|
||||
// }
|
||||
//
|
||||
// public Integer getSeed() {
|
||||
// return seed;
|
||||
// }
|
||||
//
|
||||
// public void setSeed(Integer seed) {
|
||||
// this.seed = seed;
|
||||
// }
|
||||
//}
|
||||
@ -1,105 +0,0 @@
|
||||
//package xyz.wbsite.achat.core.prompt;
|
||||
//
|
||||
//import xyz.wbsite.achat.core.chat.Message;
|
||||
//import xyz.wbsite.achat.core.Prompt;
|
||||
//import xyz.wbsite.achat.core.message.UserMessage;
|
||||
//
|
||||
//import java.util.List;
|
||||
//import java.util.Map;
|
||||
//import java.util.stream.Collectors;
|
||||
//
|
||||
//public class MessagePrompt extends Prompt {
|
||||
//
|
||||
// /**
|
||||
// * 模型
|
||||
// */
|
||||
// private String model;
|
||||
//
|
||||
// /**
|
||||
// * 消息列表
|
||||
// */
|
||||
// private List<Message> messages;
|
||||
//
|
||||
// /**
|
||||
// * 是否流式返回
|
||||
// */
|
||||
// private Boolean stream;
|
||||
//
|
||||
// /**
|
||||
// * 最大token数
|
||||
// */
|
||||
// private Integer maxTokens;
|
||||
//
|
||||
// /**
|
||||
// * 温度参
|
||||
// */
|
||||
// private Double temperature;
|
||||
//
|
||||
// /**
|
||||
// * 额外参数
|
||||
// */
|
||||
// private Map<String, Object> extraParams;
|
||||
//
|
||||
// public String getModel() {
|
||||
// return model;
|
||||
// }
|
||||
//
|
||||
// public void setModel(String model) {
|
||||
// this.model = model;
|
||||
// }
|
||||
//
|
||||
// public List<Message> getMessages() {
|
||||
// return messages;
|
||||
// }
|
||||
//
|
||||
// public void setMessages(List<Message> messages) {
|
||||
// this.messages = messages;
|
||||
// }
|
||||
//
|
||||
// public Boolean getStream() {
|
||||
// return stream;
|
||||
// }
|
||||
//
|
||||
// public void setStream(Boolean stream) {
|
||||
// this.stream = stream;
|
||||
// }
|
||||
//
|
||||
// public Integer getMaxTokens() {
|
||||
// return maxTokens;
|
||||
// }
|
||||
//
|
||||
// public void setMaxTokens(Integer maxTokens) {
|
||||
// this.maxTokens = maxTokens;
|
||||
// }
|
||||
//
|
||||
// public Double getTemperature() {
|
||||
// return temperature;
|
||||
// }
|
||||
//
|
||||
// public void setTemperature(Double temperature) {
|
||||
// this.temperature = temperature;
|
||||
// }
|
||||
//
|
||||
// public Map<String, Object> getExtraParams() {
|
||||
// return extraParams;
|
||||
// }
|
||||
//
|
||||
// public void setExtraParams(Map<String, Object> extraParams) {
|
||||
// this.extraParams = extraParams;
|
||||
// }
|
||||
//
|
||||
// public UserMessage getLastUserMessage() {
|
||||
// List<Message> messageList = messages.stream().filter(message -> message instanceof UserMessage).collect(Collectors.toList());
|
||||
// UserMessage userMessage = (UserMessage)messageList.get(messageList.size() - 1);
|
||||
// return userMessage;
|
||||
// }
|
||||
//
|
||||
// public String getUid() {
|
||||
// return getLastUserMessage().getUid();
|
||||
// }
|
||||
//
|
||||
// public String getSid(){
|
||||
// return getLastUserMessage().getSid();
|
||||
// }
|
||||
//
|
||||
//}
|
||||
@ -1,18 +0,0 @@
|
||||
package xyz.wbsite.achat.core.service;
|
||||
|
||||
import xyz.wbsite.achat.core.chat.ChatCompletionRequest;
|
||||
import xyz.wbsite.achat.core.chat.StreamEmitter;
|
||||
|
||||
/**
|
||||
* 推理生成器
|
||||
* <p>
|
||||
* 抽象出来用于生成消息的实现层
|
||||
*
|
||||
* @author wangbing
|
||||
* @version 0.0.1
|
||||
* @since 1.8
|
||||
*/
|
||||
public interface ChatCompletionGenerator {
|
||||
|
||||
void on(StreamEmitter emitter, ChatCompletionRequest chatCompletionRequest);
|
||||
}
|
||||
@ -1,121 +0,0 @@
|
||||
//package xyz.wbsite.achat.core.service.impl;
|
||||
//
|
||||
//import org.springframework.stereotype.Service;
|
||||
//import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
|
||||
//import xyz.wbsite.achat.core.message.MessageSseEmitter;
|
||||
//import xyz.wbsite.achat.core.chat.Message;
|
||||
//import xyz.wbsite.achat.core.Result;
|
||||
//import xyz.wbsite.achat.core.Session;
|
||||
//import xyz.wbsite.achat.core.prompt.MessagePrompt;
|
||||
//import xyz.wbsite.achat.core.service.SessionService;
|
||||
//
|
||||
//import java.util.ArrayList;
|
||||
//import java.util.HashMap;
|
||||
//import java.util.List;
|
||||
//import java.util.Map;
|
||||
//import java.util.UUID;
|
||||
//import java.util.stream.Collectors;
|
||||
//
|
||||
///**
|
||||
// * 会话管理服务实现类
|
||||
// * 实现会话的创建、删除、查询、消息收发等功能
|
||||
// *
|
||||
// * @author wangbing
|
||||
// * @version 0.0.1
|
||||
// * @since 1.8
|
||||
// */
|
||||
//@Service
|
||||
//public class SessionServiceMemoryImpl implements SessionService {
|
||||
//
|
||||
// private final Map<String, Session> sessionStore = new HashMap<>();
|
||||
//
|
||||
// private final List<Message> messageStore = new ArrayList<>();
|
||||
//
|
||||
// /**
|
||||
// * 创建新会话
|
||||
// */
|
||||
// @Override
|
||||
// public Result<Session> createSession(String uid) {
|
||||
// Session session = new Session();
|
||||
// session.setId(String.valueOf(UUID.randomUUID().toString()));
|
||||
// session.setUid(uid);
|
||||
// session.setTitle("新对话");
|
||||
// sessionStore.put(session.getId(), session);
|
||||
// return Result.success(session);
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 删除会话
|
||||
// */
|
||||
// @Override
|
||||
// public Result<Void> deleteSession(String sid) {
|
||||
// sessionStore.remove(sid);
|
||||
// return Result.success();
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 查询会话列表
|
||||
// */
|
||||
// @Override
|
||||
// public Result<List<Session>> listSessions(String uid) {
|
||||
// List<Session> collect = sessionStore.values()
|
||||
// .stream()
|
||||
// .filter(item -> item.getUid().equals(uid))
|
||||
// .collect(Collectors.toList());
|
||||
// return Result.success(collect);
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 获取会话详情
|
||||
// */
|
||||
// @Override
|
||||
// public Result<Session> getSession(String sid) {
|
||||
// Session session = sessionStore.get(sid);
|
||||
// if (session == null) {
|
||||
// return Result.error("会话不存在");
|
||||
// }
|
||||
// return Result.success(session);
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 发送消息并获取流式响应
|
||||
// */
|
||||
// @Override
|
||||
// public SseEmitter sendMessage(MessagePrompt message) {
|
||||
// // 创建VChatSseEmitter来处理流式响应
|
||||
// return new MessageSseEmitter(message, (emitter, message1) -> {
|
||||
// // 这边模拟LLM复述一遍用户问题
|
||||
// String text = message1.getContent();
|
||||
// for (char c : text.toCharArray()) {
|
||||
// if (emitter.isComplete()) {
|
||||
// return;
|
||||
// }
|
||||
// emitter.onPartialResponse(String.valueOf(c));
|
||||
// try {
|
||||
// Thread.sleep(100);
|
||||
// } catch (InterruptedException e) {
|
||||
// throw new RuntimeException(e);
|
||||
// }
|
||||
// }
|
||||
// emitter.onCompleteResponse(null);
|
||||
// });
|
||||
// }
|
||||
//
|
||||
// @Override
|
||||
// public Result<Void> stopSession(String sid) {
|
||||
// // todo
|
||||
// return Result.success();
|
||||
// }
|
||||
//
|
||||
// @Override
|
||||
// public Result<List<Message>> listMessage(String sid) {
|
||||
// List<Message> messages = messageStore.stream().filter(item -> item.getSid().equals(sid)).collect(Collectors.toList());
|
||||
// return Result.success(messages);
|
||||
// }
|
||||
//
|
||||
// @Override
|
||||
// public Result<Void> deleteMessage(String sid) {
|
||||
// messageStore.forEach(item -> messageStore.remove(item));
|
||||
// return null;
|
||||
// }
|
||||
//}
|
||||
@ -0,0 +1,34 @@
|
||||
package xyz.wbsite.achat.core.session;
|
||||
|
||||
/**
|
||||
*
|
||||
*/
|
||||
public class Message extends xyz.wbsite.achat.core.chat.Message {
|
||||
private String id;
|
||||
private String uid;
|
||||
private String sid;
|
||||
|
||||
public String getId() {
|
||||
return id;
|
||||
}
|
||||
|
||||
public void setId(String id) {
|
||||
this.id = id;
|
||||
}
|
||||
|
||||
public String getUid() {
|
||||
return uid;
|
||||
}
|
||||
|
||||
public void setUid(String uid) {
|
||||
this.uid = uid;
|
||||
}
|
||||
|
||||
public String getSid() {
|
||||
return sid;
|
||||
}
|
||||
|
||||
public void setSid(String sid) {
|
||||
this.sid = sid;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,138 @@
|
||||
package xyz.wbsite.achat.core.session;
|
||||
|
||||
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 会话
|
||||
*
|
||||
* @author wangbing
|
||||
* @version 0.0.1
|
||||
* @since 1.8
|
||||
*/
|
||||
public class Session {
|
||||
/**
|
||||
* 主键
|
||||
*/
|
||||
private String id;
|
||||
/**
|
||||
* 用户ID
|
||||
*/
|
||||
private String uid;
|
||||
private String title;
|
||||
private String model;
|
||||
private String prompt;
|
||||
private String temperature;
|
||||
private String topP;
|
||||
private String frequencyPenalty;
|
||||
private String presencePenalty;
|
||||
private String maxTokens;
|
||||
private String lastTime;
|
||||
private String lastMessage;
|
||||
|
||||
private List<Message> messages;
|
||||
|
||||
public String getId() {
|
||||
return id;
|
||||
}
|
||||
|
||||
public void setId(String id) {
|
||||
this.id = id;
|
||||
}
|
||||
|
||||
public String getUid() {
|
||||
return uid;
|
||||
}
|
||||
|
||||
public void setUid(String uid) {
|
||||
this.uid = uid;
|
||||
}
|
||||
|
||||
public String getTitle() {
|
||||
return title;
|
||||
}
|
||||
|
||||
public void setTitle(String title) {
|
||||
this.title = title;
|
||||
}
|
||||
|
||||
public String getModel() {
|
||||
return model;
|
||||
}
|
||||
|
||||
public void setModel(String model) {
|
||||
this.model = model;
|
||||
}
|
||||
|
||||
public String getPrompt() {
|
||||
return prompt;
|
||||
}
|
||||
|
||||
public void setPrompt(String prompt) {
|
||||
this.prompt = prompt;
|
||||
}
|
||||
|
||||
public String getTemperature() {
|
||||
return temperature;
|
||||
}
|
||||
|
||||
public void setTemperature(String temperature) {
|
||||
this.temperature = temperature;
|
||||
}
|
||||
|
||||
public String getTopP() {
|
||||
return topP;
|
||||
}
|
||||
|
||||
public void setTopP(String topP) {
|
||||
this.topP = topP;
|
||||
}
|
||||
|
||||
public String getFrequencyPenalty() {
|
||||
return frequencyPenalty;
|
||||
}
|
||||
|
||||
public void setFrequencyPenalty(String frequencyPenalty) {
|
||||
this.frequencyPenalty = frequencyPenalty;
|
||||
}
|
||||
|
||||
public String getPresencePenalty() {
|
||||
return presencePenalty;
|
||||
}
|
||||
|
||||
public void setPresencePenalty(String presencePenalty) {
|
||||
this.presencePenalty = presencePenalty;
|
||||
}
|
||||
|
||||
public String getMaxTokens() {
|
||||
return maxTokens;
|
||||
}
|
||||
|
||||
public void setMaxTokens(String maxTokens) {
|
||||
this.maxTokens = maxTokens;
|
||||
}
|
||||
|
||||
public String getLastTime() {
|
||||
return lastTime;
|
||||
}
|
||||
|
||||
public void setLastTime(String lastTime) {
|
||||
this.lastTime = lastTime;
|
||||
}
|
||||
|
||||
public String getLastMessage() {
|
||||
return lastMessage;
|
||||
}
|
||||
|
||||
public void setLastMessage(String lastMessage) {
|
||||
this.lastMessage = lastMessage;
|
||||
}
|
||||
|
||||
public List<Message> getMessages() {
|
||||
return messages;
|
||||
}
|
||||
|
||||
public void setMessages(List<Message> messages) {
|
||||
this.messages = messages;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,56 @@
|
||||
package xyz.wbsite.achat.core.session;
|
||||
|
||||
import xyz.wbsite.achat.core.chat.Message;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 会话管理服务接口
|
||||
* 提供会话的创建、删除、查询、消息收发等功能
|
||||
*
|
||||
* @author wangbing
|
||||
* @version 0.0.1
|
||||
* @since 1.8
|
||||
*/
|
||||
public interface SessionService {
|
||||
|
||||
/**
|
||||
* 创建新会话
|
||||
*
|
||||
* @param uid 用户编号
|
||||
* @return 创建的会话对象
|
||||
*/
|
||||
Result<Session> createSession(String uid);
|
||||
|
||||
/**
|
||||
* 删除会话
|
||||
*
|
||||
* @param sid 会话ID
|
||||
* @return 删除是否成功
|
||||
*/
|
||||
Result<Void> deleteSession(String uid, String sid);
|
||||
|
||||
/**
|
||||
* 查询会话列表
|
||||
*
|
||||
* @param uid 用户编号
|
||||
* @return 会话列表
|
||||
*/
|
||||
Result<List<Session>> listSessions(String uid);
|
||||
|
||||
/**
|
||||
* 获取会话详情
|
||||
*
|
||||
* @param sid 会话ID
|
||||
* @return 会话对象
|
||||
*/
|
||||
Result<Session> getSession(String uid,String sid);
|
||||
|
||||
/**
|
||||
* 获取会话历史消息
|
||||
*
|
||||
* @param sid 会话ID
|
||||
* @return 消息列表
|
||||
*/
|
||||
Result<Void> deleteMessage(String uid, String sid, String mid);
|
||||
}
|
||||
@ -0,0 +1,176 @@
|
||||
package xyz.wbsite.achat.core.session;
|
||||
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.UUID;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
/**
|
||||
* 会话管理服务实现类
|
||||
* 实现会话的创建、删除、查询、消息收发等功能
|
||||
*
|
||||
* @author wangbing
|
||||
* @version 0.0.1
|
||||
* @since 1.8
|
||||
*/
|
||||
@Service
|
||||
public class SessionServiceMemoryImpl implements SessionService {
|
||||
|
||||
/**
|
||||
* 会话存储
|
||||
*/
|
||||
private final List<Session> sessionStore = new ArrayList<>();
|
||||
|
||||
/**
|
||||
* 消息存储
|
||||
*/
|
||||
private final List<Message> messageStore = new ArrayList<>();
|
||||
|
||||
/**
|
||||
* 创建新会话
|
||||
*
|
||||
* @param uid 用户ID
|
||||
* @return 结果
|
||||
*/
|
||||
@Override
|
||||
public Result<Session> createSession(String uid) {
|
||||
// 检查参数合法性
|
||||
if (uid == null || uid.isEmpty()) {
|
||||
return Result.error("用户ID不能为空");
|
||||
}
|
||||
|
||||
Session session = new Session();
|
||||
session.setId(String.valueOf(UUID.randomUUID().toString()));
|
||||
session.setUid(uid);
|
||||
session.setTitle("新对话");
|
||||
sessionStore.add(session);
|
||||
return Result.success(session);
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除会话
|
||||
*
|
||||
* @param uid 用户ID
|
||||
* @param sid 会话ID
|
||||
* @return 结果
|
||||
*/
|
||||
@Override
|
||||
public Result<Void> deleteSession(String uid, String sid) {
|
||||
// 检查参数合法性
|
||||
if (uid == null || uid.isEmpty() || sid == null || sid.isEmpty()) {
|
||||
return Result.error("用户ID或会话ID不能为空");
|
||||
}
|
||||
|
||||
// 从sessionStore查找比对uid和sid然后删除
|
||||
boolean sessionRemoved = sessionStore.removeIf(session ->
|
||||
session.getId().equals(sid) && session.getUid().equals(uid)
|
||||
);
|
||||
|
||||
// 从messageStore查找比对uid和sid然后删除
|
||||
messageStore.removeIf(message ->
|
||||
message.getUid().equals(uid) && message.getSid().equals(sid)
|
||||
);
|
||||
|
||||
if (!sessionRemoved) {
|
||||
return Result.error("会话不存在或无权限删除");
|
||||
}
|
||||
|
||||
return Result.success();
|
||||
}
|
||||
|
||||
/**
|
||||
* 查询会话列表
|
||||
*
|
||||
* @param uid 用户ID
|
||||
* @return
|
||||
*/
|
||||
/**
|
||||
* 查询会话列表
|
||||
*
|
||||
* @param uid 用户ID
|
||||
* @return 会话列表
|
||||
*/
|
||||
@Override
|
||||
public Result<List<Session>> listSessions(String uid) {
|
||||
// 检查参数合法性
|
||||
if (uid == null || uid.isEmpty()) {
|
||||
return Result.error("用户ID不能为空");
|
||||
}
|
||||
|
||||
// 从sessionStore查找比对uid返回所有会话
|
||||
List<Session> userSessions = sessionStore.stream()
|
||||
.filter(session -> session.getUid().equals(uid))
|
||||
.collect(Collectors.toList());
|
||||
|
||||
return Result.success(userSessions);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取会话详情
|
||||
*
|
||||
* @param uid 用户ID
|
||||
* @param sid 会话ID
|
||||
* @return 会话详情
|
||||
*/
|
||||
@Override
|
||||
public Result<Session> getSession(String uid, String sid) {
|
||||
// 检查参数合法性
|
||||
if (uid == null || uid.isEmpty() || sid == null || sid.isEmpty()) {
|
||||
return Result.error("用户ID或会话ID不能为空");
|
||||
}
|
||||
|
||||
// 从sessionStore查找比对uid和sid返回会话
|
||||
Session session = sessionStore.stream()
|
||||
.filter(s -> s.getId().equals(sid) && s.getUid().equals(uid))
|
||||
.findFirst()
|
||||
.orElse(null);
|
||||
|
||||
if (session == null) {
|
||||
return Result.error("会话不存在或无权限查看");
|
||||
}
|
||||
|
||||
// 创建会话副本
|
||||
Session sessionCopy = new Session();
|
||||
sessionCopy.setId(session.getId());
|
||||
sessionCopy.setUid(session.getUid());
|
||||
sessionCopy.setTitle(session.getTitle());
|
||||
|
||||
// 从messageStore检索出当前会话的聊天记录
|
||||
List<Message> sessionMessages = messageStore.stream()
|
||||
.filter(message -> message.getUid().equals(uid) && message.getSid().equals(sid))
|
||||
.collect(Collectors.toList());
|
||||
|
||||
sessionCopy.setMessages(sessionMessages);
|
||||
|
||||
return Result.success(sessionCopy);
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除会话中的消息
|
||||
*
|
||||
* @param uid 用户ID
|
||||
* @param sid 会话ID
|
||||
* @param mid 消息ID
|
||||
* @return 结果
|
||||
*/
|
||||
@Override
|
||||
public Result<Void> deleteMessage(String uid, String sid, String mid) {
|
||||
// 检查参数合法性
|
||||
if (uid == null || uid.isEmpty() || sid == null || sid.isEmpty() || mid == null || mid.isEmpty()) {
|
||||
return Result.error("用户ID、会话ID或消息ID不能为空");
|
||||
}
|
||||
|
||||
// 从messageStore查找比对uid、sid和mid然后删除
|
||||
boolean mr = messageStore.removeIf(message ->
|
||||
message.getId().equals(mid) && message.getUid().equals(uid) && message.getSid().equals(sid)
|
||||
);
|
||||
|
||||
if (!mr) {
|
||||
return Result.error("消息不存在或无权限删除");
|
||||
}
|
||||
|
||||
return Result.success();
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue