fix(api): fix variable truncation related field not returned.

api/controllers/console/datasets/rag_pipeline/rag_pipeline_draft_variable.py

@@ -10,6 +10,10 @@ from controllers.console import api
 from controllers.console.app.error import (
     DraftWorkflowNotExist,
 )
+from controllers.console.app.workflow_draft_variable import (
+    _WORKFLOW_DRAFT_VARIABLE_FIELDS,
+    _WORKFLOW_DRAFT_VARIABLE_WITHOUT_VALUE_FIELDS,
+)
 from controllers.console.datasets.wraps import get_rag_pipeline
 from controllers.console.wraps import account_initialization_required, setup_required
 from controllers.web.error import InvalidArgumentError, NotFoundError
@@ -70,38 +74,6 @@ def _create_pagination_parser():
     return parser
 
 
-_WORKFLOW_DRAFT_VARIABLE_WITHOUT_VALUE_FIELDS = {
-    "id": fields.String,
-    "type": fields.String(attribute=lambda model: model.get_variable_type()),
-    "name": fields.String,
-    "description": fields.String,
-    "selector": fields.List(fields.String, attribute=lambda model: model.get_selector()),
-    "value_type": fields.String,
-    "edited": fields.Boolean(attribute=lambda model: model.edited),
-    "visible": fields.Boolean,
-}
-
-_WORKFLOW_DRAFT_VARIABLE_FIELDS = dict(
-    _WORKFLOW_DRAFT_VARIABLE_WITHOUT_VALUE_FIELDS,
-    value=fields.Raw(attribute=_serialize_var_value),
-)
-
-_WORKFLOW_DRAFT_ENV_VARIABLE_FIELDS = {
-    "id": fields.String,
-    "type": fields.String(attribute=lambda _: "env"),
-    "name": fields.String,
-    "description": fields.String,
-    "selector": fields.List(fields.String, attribute=lambda model: model.get_selector()),
-    "value_type": fields.String,
-    "edited": fields.Boolean(attribute=lambda model: model.edited),
-    "visible": fields.Boolean,
-}
-
-_WORKFLOW_DRAFT_ENV_VARIABLE_LIST_FIELDS = {
-    "items": fields.List(fields.Nested(_WORKFLOW_DRAFT_ENV_VARIABLE_FIELDS)),
-}
-
-
 def _get_items(var_list: WorkflowDraftVariableList) -> list[WorkflowDraftVariable]:
     return var_list.variables
 

api/controllers/console/datasets/rag_pipeline/rag_pipeline_workflow.py

@@ -984,6 +984,7 @@ class RagPipelineDatasourceVariableApi(Resource):
         )
         return workflow_node_execution
 
+
 class RagPipelineRecommendedPluginApi(Resource):
     @setup_required
     @login_required
@@ -993,6 +994,7 @@ class RagPipelineRecommendedPluginApi(Resource):
         recommended_plugins = rag_pipeline_service.get_recommended_plugins()
         return recommended_plugins
 
+
 api.add_resource(
     DraftRagPipelineApi,
     "/rag/pipelines/<uuid:pipeline_id>/workflows/draft",
@@ -1105,4 +1107,4 @@ api.add_resource(
 api.add_resource(
     RagPipelineRecommendedPluginApi,
     "/rag/pipelines/recommended-plugins",
-)
+)

api/services/rag_pipeline/rag_pipeline.py

@@ -1307,4 +1307,3 @@ class RagPipelineService:
             "installed_recommended_plugins": installed_plugin_list,
             "uninstalled_recommended_plugins": uninstalled_plugin_list,
         }
-