[improvement][chat]Support reviewing query memory based on direct user feedback.

This commit is contained in:
jerryjzhang
2024-12-26 09:47:13 +08:00
parent 68963b9ec9
commit d04a086c88
15 changed files with 49 additions and 20 deletions

View File

@@ -190,7 +190,7 @@ public abstract class S2BaseDemo implements CommandLineRunner {
return relateDimension;
}
protected void updateQueryScore(Integer queryId) {
protected void updateQueryScore(Long queryId) {
chatManageService.updateFeedback(queryId, 5, "");
}

View File

@@ -115,8 +115,8 @@ public class S2VisitsDemo extends S2BaseDemo {
// create agent
Integer agentId = addAgent(s2DataSet.getId());
addSampleChats(agentId);
updateQueryScore(1);
updateQueryScore(4);
updateQueryScore(1L);
updateQueryScore(4L);
} catch (Exception e) {
log.error("Failed to add S2Visits demo data", e);
}

View File

@@ -402,3 +402,6 @@ alter table s2_agent add column `viewer` varchar(1000) COLLATE utf8_unicode_ci D
--20241201
ALTER TABLE s2_query_stat_info RENAME COLUMN `user` TO `query_user`;
ALTER TABLE s2_chat_context RENAME COLUMN `user` TO `query_user`;
--20241226
alter table s2_chat_memory add column `query_id` BIGINT DEFAULT NULL;

View File

@@ -86,6 +86,7 @@ COMMENT ON TABLE s2_chat_config IS 'chat config information table ';
CREATE TABLE IF NOT EXISTS `s2_chat_memory` (
`id` INT NOT NULL AUTO_INCREMENT,
`question` varchar(655) ,
`query_id` BIGINT ,
`agent_id` INT ,
`db_schema` TEXT ,
`s2_sql` TEXT ,

View File

@@ -77,6 +77,7 @@ CREATE TABLE IF NOT EXISTS `s2_chat_memory` (
`id` INT NOT NULL AUTO_INCREMENT,
`question` varchar(655) COMMENT '用户问题' ,
`side_info` TEXT COMMENT '辅助信息' ,
`query_id` BIGINT COMMENT '问答ID' ,
`agent_id` INT COMMENT '助理ID' ,
`db_schema` TEXT COMMENT 'Schema映射' ,
`s2_sql` TEXT COMMENT '大模型解析SQL' ,

View File

@@ -70,6 +70,7 @@ CREATE TABLE IF NOT EXISTS s2_chat_memory (
id SERIAL PRIMARY KEY,
question varchar(655),
side_info TEXT,
query_id bigint,
agent_id INTEGER,
db_schema TEXT,
s2_sql TEXT,