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

add graph engine test

Browse Source
This commit is contained in:
takatost
2024-07-16 16:37:37 +08:00
parent 00fb23d0c9
commit 00ec36d47c
17 changed files with 1122 additions and 904 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/workflow/nodes/llm/llm_node.py
View File

@ -170,7 +170,7 @@ class LLMNode(BaseNode):
model_instance: ModelInstance,
prompt_messages: list[PromptMessage],
stop: Optional[list[str]] = None) \
-> Generator[RunStreamChunkEvent | "ModelInvokeCompleted", None, None]:
-> Generator["RunStreamChunkEvent | ModelInvokeCompleted", None, None]:
"""
Invoke large language model
:param node_data_model: node data model
@ -204,7 +204,7 @@ class LLMNode(BaseNode):
self.deduct_llm_quota(tenant_id=self.tenant_id, model_instance=model_instance, usage=usage)
def _handle_invoke_result(self, invoke_result: LLMResult | Generator) \
-> Generator[RunStreamChunkEvent | "ModelInvokeCompleted", None, None]:
-> Generator["RunStreamChunkEvent | ModelInvokeCompleted", None, None]:
"""
Handle invoke result
:param invoke_result: invoke result