[autofix.ci] apply automated fixes

This commit is contained in:
autofix-ci[bot]
2025-08-15 03:12:08 +00:00
committed by GitHub
parent c685663242
commit 71e511f396
6 changed files with 7 additions and 5 deletions

View File

@@ -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

View File

@@ -270,4 +270,5 @@ def knowledge_pipeline_publish_enabled(view):
if features.knowledge_pipeline.publish_enabled:
return view(*args, **kwargs)
abort(403)
return decorated

View File

@@ -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

View File

@@ -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.

View File

@@ -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,

View File

@@ -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"