Fix/runner bugs (#1388)

* fix: bugs in runners

* fix: model entity of exists conversation is None after changed runner
This commit is contained in:
Junyan Qin (Chin)
2025-05-12 18:21:08 +08:00
committed by GitHub
parent 13d36412dd
commit cd68760c75
3 changed files with 14 additions and 16 deletions
+3 -3
View File
@@ -121,7 +121,7 @@ class DifyServiceAPIRunner(runner.RequestRunner):
user=f'{query.session.launcher_type.value}_{query.session.launcher_id}',
conversation_id=cov_id,
files=files,
timeout=self.pipeline_config['ai']['dify-service-api']['timeout'],
timeout=120,
):
self.ap.logger.debug('dify-chat-chunk: ' + str(chunk))
@@ -184,7 +184,7 @@ class DifyServiceAPIRunner(runner.RequestRunner):
response_mode='streaming',
conversation_id=cov_id,
files=files,
timeout=self.pipeline_config['ai']['dify-service-api']['timeout'],
timeout=120,
):
self.ap.logger.debug('dify-agent-chunk: ' + str(chunk))
@@ -276,7 +276,7 @@ class DifyServiceAPIRunner(runner.RequestRunner):
inputs=inputs,
user=f'{query.session.launcher_type.value}_{query.session.launcher_id}',
files=files,
timeout=self.pipeline_config['ai']['dify-service-api']['timeout'],
timeout=120,
):
self.ap.logger.debug('dify-workflow-chunk: ' + str(chunk))
if chunk['event'] in ignored_events:
-3
View File
@@ -4,7 +4,6 @@ import asyncio
from ...core import app, entities as core_entities
from ...provider import entities as provider_entities
from ...provider.modelmgr import entities as model_entities
class SessionManager:
@@ -42,7 +41,6 @@ class SessionManager:
query: core_entities.Query,
session: core_entities.Session,
prompt_config: list[dict],
llm_model: model_entities.LLMModelInfo,
) -> core_entities.Conversation:
"""获取对话或创建对话"""
@@ -64,7 +62,6 @@ class SessionManager:
conversation = core_entities.Conversation(
prompt=prompt,
messages=[],
use_llm_model=llm_model,
use_funcs=await self.ap.tool_mgr.get_all_functions(
plugin_enabled=True,
),