merge new graph engine

This commit is contained in:
jyong 2025-08-27 18:33:38 +08:00
parent 4e2c3bfb05
commit 88abaa840c
3 changed files with 5 additions and 3 deletions

View File

@ -256,7 +256,6 @@ class RagPipelineGenerateEntity(WorkflowAppGenerateEntity):
batch: str
document_id: Optional[str] = None
start_node_id: Optional[str] = None
class SingleIterationRunEntity(BaseModel):
"""
Single Iteration Run Entity.
@ -276,3 +275,4 @@ class RagPipelineGenerateEntity(WorkflowAppGenerateEntity):
inputs: dict
single_loop_run: Optional[SingleLoopRunEntity] = None

View File

@ -538,6 +538,8 @@ class RagPipelineDslService:
account: Account,
dependencies: Optional[list[PluginDependency]] = None,
) -> Pipeline:
if not account.current_tenant_id:
raise ValueError("Tenant id is required")
"""Create a new app or update an existing one."""
pipeline_data = data.get("rag_pipeline", {})
# Set icon type

View File

@ -239,9 +239,9 @@ class WebsiteService:
return {"status": "active", "job_id": response.json().get("data", {}).get("taskId")}
@classmethod
def get_crawl_status(cls, job_id: str, provider: str, credential_id: Optional[str] = None) -> dict[str, Any]:
def get_crawl_status(cls, job_id: str, provider: str) -> dict[str, Any]:
"""Get crawl status using string parameters."""
api_request = WebsiteCrawlStatusApiRequest(provider=provider, job_id=job_id, credential_id=credential_id)
api_request = WebsiteCrawlStatusApiRequest(provider=provider, job_id=job_id)
return cls.get_crawl_status_typed(api_request)
@classmethod