injectlib/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2025-11-26 03:44:58 +08: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

2
pkg/pipeline/preproc/preproc.py
View File

@@ -33,7 +33,7 @@ class PreProcessor(stage.PipelineStage):
"""
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.session = session

4
pkg/provider/session/sessionmgr.py
View File

@@ -42,7 +42,7 @@ class SessionManager:
self.session_list.append(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:
@@ -51,7 +51,7 @@ class SessionManager:
# set prompt
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 = provider_entities.Prompt(