Bläddra i källkod

[autofix.ci] apply automated fixes

tags/2.0.0-beta.1
autofix-ci[bot] 2 månader sedan
förälder
incheckning
ff76adc88a
Inget konto är kopplat till bidragsgivarens mejladress

+ 1
- 1
api/controllers/console/datasets/rag_pipeline/rag_pipeline.py Visa fil

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

+ 1
- 0
api/controllers/console/wraps.py Visa fil

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

return decorated

+ 0
- 1
api/core/app/apps/pipeline/pipeline_runner.py Visa fil

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

+ 1
- 1
api/core/workflow/entities/variable_pool.py Visa fil

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

+ 2
- 2
api/core/workflow/nodes/datasource/datasource_node.py Visa fil

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

+ 2
- 0
api/services/feature_service.py Visa fil

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

Laddar…
Avbryt
Spara