| @@ -8,8 +8,8 @@ from controllers.console import api | |||
| from controllers.console.wraps import ( | |||
| account_initialization_required, | |||
| enterprise_license_required, | |||
| setup_required, | |||
| knowledge_pipeline_publish_enabled, | |||
| setup_required, | |||
| ) | |||
| from extensions.ext_database import db | |||
| from libs.login import login_required | |||
| @@ -270,4 +270,5 @@ def knowledge_pipeline_publish_enabled(view): | |||
| if features.knowledge_pipeline.publish_enabled: | |||
| return view(*args, **kwargs) | |||
| abort(403) | |||
| return decorated | |||
| @@ -13,7 +13,6 @@ from core.app.entities.app_invoke_entities import ( | |||
| from core.variables.variables import RAGPipelineVariable, RAGPipelineVariableInput | |||
| from core.workflow.callbacks import WorkflowCallback, WorkflowLoggingCallback | |||
| from core.workflow.entities.variable_pool import VariablePool | |||
| from core.workflow.enums import SystemVariableKey | |||
| from core.workflow.graph_engine.entities.event import GraphEngineEvent, GraphRunFailedEvent | |||
| from core.workflow.graph_engine.entities.graph import Graph | |||
| from core.workflow.system_variable import SystemVariable | |||
| @@ -18,6 +18,7 @@ from core.workflow.constants import ( | |||
| ) | |||
| from core.workflow.system_variable import SystemVariable | |||
| from factories import variable_factory | |||
| VariableValue = Union[str, int, float, dict, list, File] | |||
| VARIABLE_PATTERN = re.compile(r"\{\{#([a-zA-Z0-9_]{1,50}(?:\.[a-zA-Z_][a-zA-Z0-9_]{0,29}){1,10})#\}\}") | |||
| @@ -75,7 +76,6 @@ class VariablePool(BaseModel): | |||
| for key, value in rag_pipeline_variables_map.items(): | |||
| self.add((RAG_PIPELINE_VARIABLE_NODE_ID, key), value) | |||
| def add(self, selector: Sequence[str], value: Any, /) -> None: | |||
| """ | |||
| Add a variable to the variable pool. | |||
| @@ -138,7 +138,7 @@ class DatasourceNode(BaseNode): | |||
| datasource_runtime.get_online_document_page_content( | |||
| user_id=self.user_id, | |||
| datasource_parameters=GetOnlineDocumentPageContentRequest( | |||
| workspace_id=datasource_info.get("workspace_id"), | |||
| workspace_id=datasource_info.get("workspace_id"), | |||
| page_id=datasource_info.get("page").get("page_id"), | |||
| type=datasource_info.get("page").get("type"), | |||
| ), | |||
| @@ -205,7 +205,7 @@ class DatasourceNode(BaseNode): | |||
| storage_key=upload_file.key, | |||
| ) | |||
| variable_pool.add([self.node_id, "file"], file_info) | |||
| #variable_pool.add([self.node_id, "file"], file_info.to_dict()) | |||
| # variable_pool.add([self.node_id, "file"], file_info.to_dict()) | |||
| yield RunCompletedEvent( | |||
| run_result=NodeRunResult( | |||
| status=WorkflowNodeExecutionStatus.SUCCEEDED, | |||
| @@ -87,9 +87,11 @@ class WebAppAuthModel(BaseModel): | |||
| allow_email_code_login: bool = False | |||
| allow_email_password_login: bool = False | |||
| class KnowledgePipeline(BaseModel): | |||
| publish_enabled: bool = False | |||
| class PluginInstallationScope(StrEnum): | |||
| NONE = "none" | |||
| OFFICIAL_ONLY = "official_only" | |||