injectlib/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2025-11-25 19:37:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 实现新的前文剪切模式

Browse Source
This commit is contained in:
Rock Chin
2023-03-18 15:57:28 +00:00
parent d056cb6769
commit ca349e33fc
3 changed files with 84 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pkg/database/manager.py
View File

@@ -55,7 +55,7 @@ class DatabaseManager:
`status` varchar(255) not null default 'on_going',
`default_prompt` text not null default '',
`prompt` text not null,
`token_counts` text not null default '[]',
`token_counts` text not null default '[]'
)
""")
@@ -96,7 +96,7 @@ class DatabaseManager:
# session持久化
def persistence_session(self, subject_type: str, subject_number: int, create_timestamp: int,
last_interact_timestamp: int, prompt: str, default_prompt: str = '', token_counts: list = []):
last_interact_timestamp: int, prompt: str, default_prompt: str = '', token_counts: str = ''):
"""持久化指定session"""
# 检查是否已经有了此name和create_timestamp的session
@@ -115,14 +115,14 @@ class DatabaseManager:
self.__execute__(sql,
("{}_{}".format(subject_type, subject_number), subject_type, subject_number, create_timestamp,
last_interact_timestamp, prompt, default_prompt, json.dumps(token_counts)))
last_interact_timestamp, prompt, default_prompt, token_counts))
else:
sql = """
update `sessions` set `last_interact_timestamp` = ?, `prompt` = ?, `token_counts` = ?
where `type` = ? and `number` = ? and `create_timestamp` = ?
"""
self.__execute__(sql, (last_interact_timestamp, prompt, json.dumps(token_counts), subject_type,
self.__execute__(sql, (last_interact_timestamp, prompt, token_counts, subject_type,
subject_number, create_timestamp))
# 显式关闭一个session
@@ -172,7 +172,7 @@ class DatabaseManager:
'last_interact_timestamp': last_interact_timestamp,
'prompt': prompt,
'default_prompt': default_prompt,
'token_counts': json.loads(token_counts)
'token_counts': token_counts
}
else:
if session_name in sessions:
@@ -210,7 +210,7 @@ class DatabaseManager:
'last_interact_timestamp': last_interact_timestamp,
'prompt': prompt,
'default_prompt': default_prompt,
'token_counts': json.loads(token_counts)
'token_counts': token_counts
}
# 获取此session_name后一个session的数据
@@ -243,7 +243,7 @@ class DatabaseManager:
'last_interact_timestamp': last_interact_timestamp,
'prompt': prompt,
'default_prompt': default_prompt,
'token_counts': json.loads(token_counts)
'token_counts': token_counts
}
# 列出与某个对象的所有对话session
@@ -272,7 +272,7 @@ class DatabaseManager:
'last_interact_timestamp': last_interact_timestamp,
'prompt': prompt,
'default_prompt': default_prompt,
'token_counts': json.loads(token_counts)
'token_counts': token_counts
})
return sessions