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

feat: add backwards invoke node api

Browse Source
This commit is contained in:
Yeuoly
2024-09-24 18:03:48 +08:00
parent 592f85f7a9
commit 68c10a1672
5 changed files with 335 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
api/core/plugin/entities/request.py
View File

@ -14,6 +14,16 @@ from core.model_runtime.entities.message_entities import (
UserPromptMessage,
)
from core.model_runtime.entities.model_entities import ModelType
from core.workflow.nodes.question_classifier.entities import (
ClassConfig,
ModelConfig as QuestionClassifierModelConfig,
)
from core.workflow.nodes.parameter_extractor.entities import (
ModelConfig as ParameterExtractorModelConfig,
)
from core.workflow.nodes.parameter_extractor.entities import (
ParameterConfig,
)
class RequestInvokeTool(BaseModel):
@ -92,11 +102,27 @@ class RequestInvokeModeration(BaseModel):
"""
class RequestInvokeNode(BaseModel):
class RequestInvokeParameterExtractorNode(BaseModel):
"""
Request to invoke node
Request to invoke parameter extractor node
"""
parameters: list[ParameterConfig]
model: ParameterExtractorModelConfig
instruction: str
query: str
class RequestInvokeQuestionClassifierNode(BaseModel):
"""
Request to invoke question classifier node
"""
query: str
model: QuestionClassifierModelConfig
classes: list[ClassConfig]
instruction: str
class RequestInvokeApp(BaseModel):
"""