refactor: compartmentalize mcp models

This commit is contained in:
lda
2026-06-06 13:56:50 +07:00 Verified
parent c25c4cfee5
commit 2c31fa678a
9 changed files with 105 additions and 59 deletions
+1 -3
View File
@@ -16,6 +16,7 @@ from wf_core import (
Workflow,
)
from wf_api.models import RawWorkflowPlan
from wf_mcp.capabilities import CatalogNodeEntry, CatalogPromptEntry, CatalogResourceEntry
from wf_platform import (
CapabilitySource,
)
@@ -23,9 +24,6 @@ from ...connections import ConnectionRegistry
from ...events import EventBus, McpEvent
from ...models import (
AuthRecord,
CatalogNodeEntry,
CatalogPromptEntry,
CatalogResourceEntry,
CatalogSnapshot,
BrokerConfig,
ConnectionConfig,
+1 -3
View File
@@ -7,6 +7,7 @@ from typing import Any
from pydantic import BaseModel
from wf_authoring import NodeReturn, NodeSpec
from wf_mcp.capabilities import CatalogNodeEntry, CatalogPromptEntry, CatalogResourceEntry
from wf_platform import (
CapabilityBuckets,
CapabilitySource,
@@ -21,9 +22,6 @@ from ...connections import ConnectionConfig, qualify_node_name
from ...events import McpEvent, make_event
from ...models import (
AuthRecord,
CatalogNodeEntry,
CatalogPromptEntry,
CatalogResourceEntry,
CatalogSnapshot,
)
from ...runtime import ToolExecutor