Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
D
dify
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
ai-tech
dify
Commits
fd8fe15d
Commit
fd8fe15d
authored
Mar 13, 2024
by
takatost
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
use answer node instead of end in advanced chatbot
parent
e80315f5
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
39 additions
and
28 deletions
+39
-28
workflow_converter.py
api/services/workflow/workflow_converter.py
+39
-28
No files found.
api/services/workflow/workflow_converter.py
View file @
fd8fe15d
...
...
@@ -19,7 +19,6 @@ from core.model_runtime.entities.llm_entities import LLMMode
from
core.model_runtime.utils.encoders
import
jsonable_encoder
from
core.prompt.simple_prompt_transform
import
SimplePromptTransform
from
core.workflow.entities.node_entities
import
NodeType
from
core.workflow.nodes.end.entities
import
EndNodeOutputType
from
events.app_event
import
app_was_created
from
extensions.ext_database
import
db
from
models.account
import
Account
...
...
@@ -149,10 +148,13 @@ class WorkflowConverter:
graph
=
self
.
_append_node
(
graph
,
llm_node
)
if
new_app_mode
==
AppMode
.
WORKFLOW
:
# convert to end node by app mode
end_node
=
self
.
_convert_to_end_node
(
app_model
=
app_model
)
end_node
=
self
.
_convert_to_end_node
()
graph
=
self
.
_append_node
(
graph
,
end_node
)
else
:
answer_node
=
self
.
_convert_to_answer_node
()
graph
=
self
.
_append_node
(
graph
,
answer_node
)
app_model_config_dict
=
app_config
.
app_model_config_dict
...
...
@@ -517,33 +519,42 @@ class WorkflowConverter:
}
}
def
_convert_to_end_node
(
self
,
app_model
:
App
)
->
dict
:
def
_convert_to_end_node
(
self
)
->
dict
:
"""
Convert to End Node
:param app_model: App instance
:return:
"""
if
app_model
.
mode
==
AppMode
.
CHAT
.
value
:
# for original completion app
return
{
"id"
:
"end"
,
"position"
:
None
,
"data"
:
{
"title"
:
"END"
,
"type"
:
NodeType
.
END
.
value
,
"outputs"
:
{
"variable"
:
"result"
,
"value_selector"
:
[
"llm"
,
"text"
]
}
}
elif
app_model
.
mode
==
AppMode
.
COMPLETION
.
value
:
# for original completion app
}
def
_convert_to_answer_node
(
self
)
->
dict
:
"""
Convert to Answer Node
:return:
"""
# for original chat app
return
{
"id"
:
"end
"
,
"id"
:
"answer
"
,
"position"
:
None
,
"data"
:
{
"title"
:
"END"
,
"type"
:
NodeType
.
END
.
value
,
"outputs"
:
{
"type"
:
EndNodeOutputType
.
PLAIN_TEXT
.
value
,
"plain_text_selector"
:
[
"llm"
,
"text"
]
}
"title"
:
"ANSWER"
,
"type"
:
NodeType
.
ANSWER
.
value
,
"variables"
:
{
"variable"
:
"text"
,
"value_selector"
:
[
"llm"
,
"text"
]
},
"answer"
:
"{{text}}"
}
}
...
...
@@ -582,7 +593,7 @@ class WorkflowConverter:
if
app_model
.
mode
==
AppMode
.
COMPLETION
.
value
:
return
AppMode
.
WORKFLOW
else
:
return
AppMode
.
value_of
(
app_model
.
mode
)
return
AppMode
.
ADVANCED_CHAT
def
_get_api_based_extension
(
self
,
tenant_id
:
str
,
api_based_extension_id
:
str
)
->
APIBasedExtension
:
"""
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment