youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 09:28:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix stream output

Browse Source
This commit is contained in:
takatost
2024-03-19 20:34:43 +08:00
parent 45017f3f35
commit 0183651cd5
2 changed files with 13 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/core/workflow/nodes/knowledge_retrieval/structed_multi_dataset_router_agent.py
View File

@ -1,13 +1,14 @@
from collections.abc import Sequence
from typing import Optional, Union, Generator
from collections.abc import Generator, Sequence
from typing import Optional, Union
from langchain import PromptTemplate
from langchain.agents.structured_chat.base import HUMAN_MESSAGE_TEMPLATE
from langchain.agents.structured_chat.prompt import PREFIX, SUFFIX
from core.app.entities.app_invoke_entities import ModelConfigWithCredentialsEntity
from core.model_manager import ModelInstance
from core.model_runtime.entities.llm_entities import LLMUsage
from core.model_runtime.entities.message_entities import PromptMessageTool, PromptMessageRole, PromptMessage
from core.model_runtime.entities.message_entities import PromptMessage, PromptMessageRole, PromptMessageTool
from core.prompt.advanced_prompt_transform import AdvancedPromptTransform
from core.prompt.entities.advanced_prompt_entities import ChatModelMessage
from core.workflow.nodes.knowledge_retrieval.entities import KnowledgeRetrievalNodeData