xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-08-14 14:31:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

perf: param for get_conversation

Browse Source
This commit is contained in:
Junyan Qin
2025-04-03 17:19:27 +08:00
parent fb18278bdc
commit 472d472bc1
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/pipeline/preproc/preproc.py
+1 -1
View File
@@ -33,7 +33,7 @@ class PreProcessor(stage.PipelineStage):
""" """
session = await self.ap.sess_mgr.get_session(query) session = await self.ap.sess_mgr.get_session(query)
conversation = await self.ap.sess_mgr.get_conversation(query, session, query.pipeline_config) conversation = await self.ap.sess_mgr.get_conversation(query, session, query.pipeline_config['ai']['local-agent']['prompt'])
# 设置query # 设置query
query.session = session query.session = session
pkg/provider/session/sessionmgr.py
+2 -2
View File
@@ -42,7 +42,7 @@ class SessionManager:
self.session_list.append(session) self.session_list.append(session)
return session return session
async def get_conversation(self, query: core_entities.Query, session: core_entities.Session, pipeline_config: dict) -> core_entities.Conversation: async def get_conversation(self, query: core_entities.Query, session: core_entities.Session, prompt_config: list[dict]) -> core_entities.Conversation:
"""获取对话或创建对话""" """获取对话或创建对话"""
if not session.conversations: if not session.conversations:
@@ -51,7 +51,7 @@ class SessionManager:
# set prompt # set prompt
prompt_messages = [] prompt_messages = []
for prompt_message in pipeline_config['ai']['local-agent']['prompt']: for prompt_message in prompt_config:
prompt_messages.append(provider_entities.Message(**prompt_message)) prompt_messages.append(provider_entities.Message(**prompt_message))
prompt = provider_entities.Prompt( prompt = provider_entities.Prompt(