lda.chat/lda-wf
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ruff check fix

Browse Source
This commit is contained in:
lda
2026-07-30 01:27:46 +07:00 Verified
parent a3a4fb9250
commit 981edd3cf5
42 changed files with 155 additions and 153 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/thesis/generate_agent_challenge_evaluation.py
+1 -1
View File
@@ -9,7 +9,7 @@ ROOT = THESIS_DIR.parents[1]
if str(ROOT) not in sys.path:
sys.path.insert(0, str(ROOT))
from examples.agent_challenges.evaluation import ( # noqa: E402
from examples.agent_challenges.evaluation import (
load_evaluation_cohort,
render_evaluation_figures,
render_evaluation_markdown,
examples/agent_challenges/runner.py
+6 -6
View File
@@ -30,29 +30,29 @@ def _utf8_subprocess_env() -> dict[str, str]:
return env
from examples.agent_challenges.names import ( # noqa: E402
from examples.agent_challenges.names import (
short_challenge_name,
short_model_name,
)
from examples.agent_challenges.opencode_io import ( # noqa: E402
from examples.agent_challenges.opencode_io import (
build_opencode_command,
opencode_text_results,
parse_opencode_output,
result_text,
)
from examples.agent_challenges.opencode_resume import ( # noqa: E402
from examples.agent_challenges.opencode_resume import (
build_resume_command,
extract_session_id,
resume_prompt_for_result,
)
from examples.agent_challenges.report_models import ( # noqa: E402
from examples.agent_challenges.report_models import (
build_trial_report,
)
from examples.agent_challenges.reports import ( # noqa: E402
from examples.agent_challenges.reports import (
save_report_from_result_payload,
write_trial_report_projections,
)
from examples.agent_challenges.workspace import ( # noqa: E402
from examples.agent_challenges.workspace import (
ChallengeDef,
TrialConfig,
_display_path,
src/wf_api/__init__.py
+16 -16
View File
@@ -71,17 +71,13 @@ from .wrapper_hints import (
__all__ = [
"AUTH_ID_PATTERN",
"DEFAULT_CALL_STEP_ID",
"AuthRecord",
"AuthStore",
"CapabilityStepUpdate",
"builtin_sources",
"get_qualified_spec",
"matches_query",
"paged_list_payload",
"qualify_spec",
"DEFAULT_ERROR_OUTCOME",
"DEFAULT_ERROR_STEP_ID",
"DEFAULT_OK_OUTCOME",
"RUNTIME_ERROR_CAPABILITY",
"AuthRecord",
"AuthStore",
"CapabilityStepUpdate",
"MissingDecision",
"MissingDecisionKind",
"NextActionPatchExample",
@@ -89,10 +85,10 @@ __all__ = [
"NextActions",
"OutcomeCandidate",
"OutcomeCandidateKind",
"RUNTIME_ERROR_CAPABILITY",
"RawWorkflowPlan",
"RouteSource",
"RuntimeDependencies",
"SourceResourceRef",
"TraceRange",
"WorkflowAdminApi",
"WorkflowAdminAuthProvider",
@@ -112,10 +108,9 @@ __all__ = [
"WorkflowEventRecorder",
"WorkflowLiveSourceChecker",
"WorkflowOperationContext",
"WorkflowRuntimeRunner",
"WorkflowRunApi",
"WorkflowRunSurface",
"SourceResourceRef",
"WorkflowRuntimeRunner",
"WorkflowSourceAdminApi",
"WorkflowSourceAdminSurface",
"WorkflowSourceRegistryApi",
@@ -124,17 +119,22 @@ __all__ = [
"WorkflowSourceRegistryProvider",
"WorkflowSourceRegistrySurface",
"WorkflowSpecProvider",
"WorkflowStores",
"WorkflowSurfaceCapabilityId",
"WrapperAuthoringHints",
"WrapperHintConfidence",
"WrapperOutcomePolicy",
"builtin_sources",
"durable_workflow_api",
"file_workflow_stores",
"get_qualified_spec",
"matches_query",
"paged_list_payload",
"parse_workflow_surface_capability_id",
"qualify_spec",
"require_workflow_stores",
"resolve_runtime_dependencies",
"validate_auth_id",
"workflow_output_schema_for_authoring",
"wrapper_hints_for_capability",
"resolve_runtime_dependencies",
"WorkflowStores",
"file_workflow_stores",
"durable_workflow_api",
"require_workflow_stores",
]
src/wf_api/schema_projection.py
+1 -3
View File
@@ -291,9 +291,7 @@ def _resolve_local_reference(
raise ValueError(f"cyclic reference {reference!r} at schema path {label!r}")
seen.add(reference)
if reference.startswith("#/$defs/"):
pointer = reference.removeprefix("#/")
elif reference.startswith("#/definitions/"):
if reference.startswith("#/$defs/") or reference.startswith("#/definitions/"):
pointer = reference.removeprefix("#/")
else:
raise ValueError(
src/wf_api/source_registry.py
+1 -1
View File
@@ -102,10 +102,10 @@ class AtomicJsonRegistryStore(Generic[RegistryT]):
__all__ = [
"SOURCE_REGISTRY_ID_PATTERN",
"AtomicJsonRegistryStore",
"SourceRegistryBaseModel",
"SourceRegistryStore",
"SOURCE_REGISTRY_ID_PATTERN",
"validate_source_registry_id",
"validate_unique_source_ids",
]
src/wf_artifacts/__init__.py
+15 -15
View File
@@ -56,46 +56,46 @@ from .store import FileWorkflowArtifactStore, WorkflowArtifactStore
from .validation import validate_deployment_dependencies
__all__ = [
"AvailableCapability",
"ArtifactKind",
"AvailableCapability",
"AvailableSource",
"CheckpointReason",
"DependencyDiagnostic",
"DiagnosticSeverity",
"DriftPolicy",
"DraftWorkspaceConflictError",
"DraftWorkspaceStore",
"DriftPolicy",
"FileDraftWorkspaceStore",
"FileWorkflowArtifactStore",
"FileRunStore",
"FileWorkflowArtifactStore",
"PinnedRunEnvironment",
"RequiredCapability",
"ResumeReadiness",
"RunCheckpoint",
"RunStore",
"SourceBinding",
"StoredRunStatus",
"WorkflowArtifact",
"WorkflowArtifactCatalogEntry",
"WorkflowCapabilityRef",
"WorkflowDraftWorkspace",
"WorkflowArtifactStore",
"WorkflowRunRecord",
"WorkflowCapabilityRef",
"WorkflowDeployment",
"RunStore",
"RunCheckpoint",
"CheckpointReason",
"PinnedRunEnvironment",
"ResumeReadiness",
"StoredRunStatus",
"ensure_run_id",
"WorkflowDraftWorkspace",
"WorkflowRunRecord",
"artifact_catalog_entry",
"artifact_node_name",
"compile_workflow_draft",
"create_draft_workspace",
"create_workflow_artifact_from_plan",
"compile_workflow_draft",
"ensure_run_id",
"ensure_workspace_id",
"get_draft_workspace",
"logical_ref_for_concrete_ref",
"normalize_plan_node_refs",
"patch_draft_workspace",
"patch_workflow_draft",
"replace_draft_workspace_document",
"replace_validated_draft_document",
"patch_workflow_draft",
"summarize_draft_workspace",
"validate_deployment_dependencies",
"validate_workflow_draft",
src/wf_artifacts/drafts/__init__.py
+2 -2
View File
@@ -21,9 +21,9 @@ from .models import (
)
__all__ = [
"DraftDiagnostic",
"DraftChooseClause",
"DraftChooseStep",
"DraftDiagnostic",
"DraftEndStep",
"DraftForeachStep",
"DraftInterruptStep",
@@ -31,8 +31,8 @@ __all__ = [
"DraftMatchCase",
"DraftMatchStep",
"DraftSubgraphStep",
"DraftWhenStep",
"DraftUseStep",
"DraftWhenStep",
"WorkflowDraft",
"build_workflow_from_draft",
"compile_workflow_draft",
src/wf_authoring/__init__.py
+22 -22
View File
@@ -88,10 +88,8 @@ from .schemas import StateFieldMetadata, state_field
from .subgraph import async_subgraph_node, subgraph_node, subgraph_ref
__all__ = [
"NodeCatalog",
"NodeCatalogEntry",
"AsyncRegistryHandler",
"AuthoredReducer",
"async_subgraph_node",
"BoolOutput",
"CoalesceInput",
"ConcatInput",
@@ -105,47 +103,48 @@ __all__ = [
"FilterItemsPresentInput",
"GraphPath",
"ItemOutput",
"MappingOutput",
"MappingItemsOutput",
"MappingOutput",
"MaybeItemOutput",
"NoOutput",
"NodeCatalog",
"NodeCatalogEntry",
"NodeReturn",
"NodeSpec",
"Nothing",
"PickKeyInput",
"PickPathInput",
"ProjectFieldsInput",
"ReducerCatalog",
"RenameFieldsInput",
"RuntimeErrorInput",
"NoOutput",
"NodeReturn",
"NodeSpec",
"Nothing",
"AsyncRegistryHandler",
"SyncRegistryHandler",
"SequenceInput",
"StateFieldMetadata",
"SyncRegistryHandler",
"TextOutput",
"TruthyInput",
"ValueOutput",
"ValuesOutput",
"WorkflowBuilder",
"async_subgraph_node",
"bind_fields",
"bind_state",
"build_async_registry",
"build_registry",
"build_extract_text_content_spec",
"build_extract_text_content_workflow",
"bind_state",
"build_registry",
"coalesce",
"concat",
"constant",
"context",
"context_path",
"default_if_none",
"exists",
"expr",
"extract_field",
"extract_text_content",
"filter_items",
"filter_items_present",
"merge_maps",
"context",
"context_path",
"expr",
"exists",
"first_item",
"first_item_maybe",
"first_item_or_none",
@@ -158,16 +157,17 @@ __all__ = [
"last_item",
"last_item_or_none",
"length",
"pick_key",
"pick_path",
"project_fields",
"rename_fields",
"runtime_error",
"merge_maps",
"node",
"not_",
"outcome",
"output_to",
"pick_key",
"pick_path",
"project_fields",
"reducer",
"rename_fields",
"runtime_error",
"state",
"state_field",
"state_path",
src/wf_authoring/catalog.py
+2 -2
View File
@@ -16,7 +16,7 @@ class NodeCatalogEntry:
output_schema: dict[str, Any]
@classmethod
def from_spec(cls, spec: NodeSpec[Any, Any]) -> "NodeCatalogEntry":
def from_spec(cls, spec: NodeSpec[Any, Any]) -> NodeCatalogEntry:
input_schema = (
spec.input_schema_contract or spec.input_model.model_json_schema()
)
@@ -38,7 +38,7 @@ class NodeCatalog:
specs: dict[str, NodeSpec[Any, Any]]
@classmethod
def from_specs(cls, *specs: NodeSpec[Any, Any]) -> "NodeCatalog":
def from_specs(cls, *specs: NodeSpec[Any, Any]) -> NodeCatalog:
return cls(specs={spec.name: spec for spec in specs})
def entries(self) -> list[NodeCatalogEntry]:
src/wf_authoring/nodes/__init__.py
+1 -1
View File
@@ -24,8 +24,8 @@ __all__ = [
"AsyncRegistryHandler",
"ContextNodeCallable",
"InputT",
"NodeCallable",
"NoOutput",
"NodeCallable",
"NodeReturn",
"NodeSpec",
"Nothing",
src/wf_authoring/nodes/spec.py
+1 -1
View File
@@ -22,7 +22,7 @@ from .result import NodeReturn, Nothing
from .schema import schema_ref_for
def _default_outcome(spec: "NodeSpec[Any, Any]") -> str:
def _default_outcome(spec: NodeSpec[Any, Any]) -> str:
return spec.outcomes[0]
src/wf_authoring/ops/__init__.py
+4 -4
View File
@@ -55,17 +55,17 @@ __all__ = [
"FilterItemsInput",
"FilterItemsPresentInput",
"ItemOutput",
"MappingOutput",
"MappingItemsOutput",
"MappingOutput",
"MaybeItemOutput",
"PickPathInput",
"PickKeyInput",
"PickPathInput",
"ProjectFieldsInput",
"RenameFieldsInput",
"RuntimeErrorInput",
"SequenceInput",
"TruthyInput",
"TextOutput",
"TruthyInput",
"ValueOutput",
"ValuesOutput",
"coalesce",
@@ -82,8 +82,8 @@ __all__ = [
"last_item",
"last_item_or_none",
"length",
"pick_path",
"pick_key",
"pick_path",
"project_fields",
"rename_fields",
"runtime_error",
src/wf_authoring/reducers/catalog.py
+1 -1
View File
@@ -15,7 +15,7 @@ class ReducerCatalog:
definitions: dict[str, ReducerDefinition]
@classmethod
def from_reducers(cls, *reducers: AuthoredReducer) -> "ReducerCatalog":
def from_reducers(cls, *reducers: AuthoredReducer) -> ReducerCatalog:
return cls(
definitions={
reducer.definition.spec.name: reducer.definition for reducer in reducers
src/wf_authoring/schemas.py
+2 -2
View File
@@ -1,8 +1,8 @@
from __future__ import annotations
from collections.abc import Mapping
from collections.abc import Iterator, Mapping
from dataclasses import dataclass, field
from typing import Any, Iterator
from typing import Any
from pydantic import BaseModel, TypeAdapter
src/wf_config/__init__.py
+1 -1
View File
@@ -22,7 +22,6 @@ from .models import (
)
__all__ = [
"load_workflow_config",
"AuthConfig",
"ClientConfig",
"FilesystemStoreConfig",
@@ -40,4 +39,5 @@ __all__ = [
"StdioSourceTransportConfig",
"StdlibSourceConfig",
"WorkflowConfigFile",
"load_workflow_config",
]
src/wf_core/__init__.py
+20 -20
View File
@@ -57,56 +57,56 @@ from .validation import (
)
__all__ = [
"END",
"START",
"AsyncNodeHandler",
"ConditionNode",
"Edge",
"EndNode",
"ExecutionFrame",
"ForeachConcurrentPolicy",
"ForeachItemErrorPolicy",
"ForeachNode",
"FrameStatus",
"InterruptNode",
"InterruptRequest",
"InterruptRoute",
"JoinNode",
"NodeDef",
"NodeHandler",
"NodeResult",
"NodeUse",
"PersistedRunState",
"PreparedSubgraph",
"ReducerRef",
"ReducerSpec",
"RunState",
"RunStatus",
"RuntimeContext",
"SchemaRef",
"SiblingWritePolicy",
"StateField",
"StateSchema",
"SubgraphNode",
"AsyncNodeHandler",
"NodeHandler",
"PreparedSubgraph",
"ExecutionFrame",
"FrameStatus",
"RunState",
"RunStatus",
"RuntimeContext",
"StepExecutionResult",
"SubgraphNode",
"TraceEntry",
"InterruptRoute",
"InterruptRequest",
"PersistedRunState",
"dump_run_state",
"load_run_state",
"START",
"END",
"ValidationIssue",
"ValidationIssueCode",
"ValidationReport",
"Workflow",
"WorkflowRef",
"WorkflowExecutionError",
"WorkflowRef",
"coerce_node_result",
"dump_run_state",
"execute_workflow",
"execute_workflow_async",
"execute_workflow_result_async",
"execute_workflow",
"load_run_state",
"resume_workflow",
"resume_workflow_async",
"resume_workflow_result_async",
"resume_workflow",
"step_workflow_async",
"step_workflow",
"step_workflow_async",
"validate_workflow",
"workflow_ref_from",
]
src/wf_core/models/__init__.py
+2 -2
View File
@@ -41,12 +41,12 @@ __all__ = [
"LiteralOperand",
"NodeDef",
"NodeResult",
"ReducerRef",
"ReducerSpec",
"NodeUse",
"NotCondition",
"Operand",
"PathOperand",
"ReducerRef",
"ReducerSpec",
"SchemaRef",
"SiblingWritePolicy",
"StateField",
src/wf_core/models/conditions.py
+2 -2
View File
@@ -34,14 +34,14 @@ class NotCondition(BaseModel):
"""Condition that negates another condition expression."""
op: Literal["not"]
arg: "Condition"
arg: Condition
class VariadicCondition(BaseModel):
"""Condition that combines one or more child conditions."""
op: Literal["and", "or"]
args: list["Condition"] = Field(min_length=1)
args: list[Condition] = Field(min_length=1)
class BinaryCondition(BaseModel):
src/wf_core/models/workflow.py
+1 -1
View File
@@ -43,7 +43,7 @@ class Workflow(BaseModel):
nodes: list[Step]
edges: list[Edge]
def validate_structure(self) -> "ValidationReport":
def validate_structure(self) -> ValidationReport:
"""Return all structural validation issues for this workflow."""
validation = import_module("wf_core.validation.core")
return cast("ValidationReport", validation.validate_workflow(self))
src/wf_core/runtime/__init__.py
+1 -1
View File
@@ -21,6 +21,7 @@ from .subgraphs import PreparedSubgraph
__all__ = [
"AsyncNodeHandler",
"NodeHandler",
"PreparedSubgraph",
"WorkflowExecutionError",
"coerce_node_result",
"complete_step",
@@ -34,5 +35,4 @@ __all__ = [
"resume_workflow_result_async",
"step_workflow",
"step_workflow_async",
"PreparedSubgraph",
]
src/wf_core/runtime/scheduler.py
+2 -2
View File
@@ -15,7 +15,7 @@ class BlockedOnChildren:
child_frame_ids: tuple[str, ...]
@classmethod
def from_frame(cls, frame: ExecutionFrame) -> "BlockedOnChildren | None":
def from_frame(cls, frame: ExecutionFrame) -> BlockedOnChildren | None:
raw = frame.metadata.get("blocked_on")
if raw is None:
return None
@@ -46,7 +46,7 @@ class ForeachIterationMetadata:
loop_alias: str
@classmethod
def from_frame(cls, frame: ExecutionFrame) -> "ForeachIterationMetadata | None":
def from_frame(cls, frame: ExecutionFrame) -> ForeachIterationMetadata | None:
if frame.kind != "foreach_iteration":
return None
metadata = frame.metadata
src/wf_core/tokens.py
+1 -1
View File
@@ -1,4 +1,4 @@
START = "__start__"
END = "__end__"
__all__ = ["START", "END"]
__all__ = ["END", "START"]
src/wf_mcp/broker/__init__.py
+1 -1
View File
@@ -31,8 +31,8 @@ __all__ = [
"discover_connection_capabilities",
"load_broker_config",
"make_event",
"normalize_transport",
"snapshot_from_specs",
"specs_from_discovered_tools",
"normalize_transport",
"workflow_server_from_service",
]
src/wf_mcp/proxy/__init__.py
+1 -1
View File
@@ -7,7 +7,7 @@ from .runtime import (
__all__ = [
"ProxyRuntime",
"register_proxy_admin_tools",
"create_proxy_client",
"create_proxy_server",
"register_proxy_admin_tools",
]
src/wf_mcp/shared/__init__.py
+1 -1
View File
@@ -17,8 +17,8 @@ __all__ = [
"LdaNamespace",
"ProxyNamespace",
"ProxyToolName",
"connection_id_to_resource_path",
"clamp_limit",
"connection_id_to_resource_path",
"error_payload",
"is_admin_tool_name",
"make_cursor",
src/wf_mcp/source_registry.py
+6 -6
View File
@@ -53,7 +53,7 @@ def workflow_mcp_source_to_connection_config(source: object) -> ConnectionConfig
for field in ("id", "provider", "account", "enabled", "ownership", "transport"):
if getattr(source, field, None) is None:
raise ValueError(f"wf_config MCP source missing required field: {field}")
transport = getattr(source, "transport")
transport = source.transport
metadata = dict(getattr(source, "metadata", {}))
if transport.kind == "stdio":
metadata.update(
@@ -83,12 +83,12 @@ def workflow_mcp_source_to_connection_config(source: object) -> ConnectionConfig
if auth_ref is not None:
metadata["auth_ref"] = auth_ref
return ConnectionConfig(
id=getattr(source, "id"),
server=getattr(source, "provider"),
account=getattr(source, "account"),
enabled=getattr(source, "enabled"),
id=source.id,
server=source.provider,
account=source.account,
enabled=source.enabled,
metadata=metadata,
source_config_ownership=getattr(source, "ownership"),
source_config_ownership=source.ownership,
)
src/wf_mcp/workflow_surface/runtime_dependencies.py
+1 -1
View File
@@ -6,7 +6,7 @@ older MCP workflow-surface imports keep working until callers migrate.
from __future__ import annotations
from wf_api.runtime_dependencies import ( # noqa: F401
from wf_api.runtime_dependencies import (
RuntimeDependencies,
resolve_runtime_dependencies,
)
src/wf_mcp/workflow_surface/tools.py
+2 -1
View File
@@ -1,6 +1,7 @@
from __future__ import annotations
from typing import Annotated, Any, Mapping
from collections.abc import Mapping
from typing import Annotated, Any
from fastmcp import FastMCP
from pydantic import Field
src/wf_openapi/__init__.py
+1 -1
View File
@@ -10,8 +10,8 @@ from .spec import load_openapi_document, load_openapi_operations
__all__ = [
"OPENAPI_OUTCOMES",
"OpenApiExecutionConfig",
"OpenApiOperationOutput",
"OpenApiOperation",
"OpenApiOperationOutput",
"build_openapi_capability_source",
"call_openapi_operation",
"load_openapi_document",
src/wf_openapi/request.py
+2 -1
View File
@@ -1,7 +1,8 @@
from __future__ import annotations
from collections.abc import Mapping
from dataclasses import dataclass, field
from typing import Any, Mapping
from typing import Any
from urllib.parse import quote
from .models import OpenApiOperation
src/wf_platform/__init__.py
+2 -2
View File
@@ -49,8 +49,8 @@ _LAZY_EXPORTS = {
__all__ = [
"CapabilityBuckets",
"CapabilitySource",
"CapabilityRef",
"CapabilitySource",
"DocumentationPrompt",
"DocumentationResource",
"NodeSpecInventory",
@@ -63,10 +63,10 @@ __all__ = [
"SourcePermissionsSnapshot",
"SourcePolicy",
"SourcePolicySnapshot",
"SourceRef",
"SourceStatus",
"SourceVisibility",
"SourceVisibilitySnapshot",
"SourceRef",
"build_documentation_source",
"hash_json_schema",
"page_items",
src/wf_sources_mcp/__init__.py
+5 -5
View File
@@ -65,17 +65,19 @@ __all__ = [
"AdapterLookupRef",
"AuthRecord",
"DiscoveredConnectionCapabilities",
"LegacyAdapterRef",
"LegacyConnectionConfigLike",
"SourceAdapterRef",
"FileSourceRegistryStore",
"HttpSourceTransport",
"LegacyAdapterRef",
"LegacyConnectionConfigLike",
"McpSourceConnection",
"McpSourceRegistryEntry",
"SourceAdapterRef",
"SourceRegistryFile",
"SourceRegistryStore",
"SourceTransport",
"StdioSourceTransport",
"ToolWrapperEvent",
"ToolWrapperEventSink",
"auth_missing_diagnostic",
"auth_ref_for_connection",
"connection_auth_diagnostic",
@@ -92,8 +94,6 @@ __all__ = [
"specs_from_discovered_tools",
"tool_call_completed_event",
"tool_call_started_event",
"ToolWrapperEvent",
"ToolWrapperEventSink",
"wrap_discovered_tool",
]
src/wf_sources_mcp/client/transport.py
+8 -6
View File
@@ -77,14 +77,16 @@ async def open_mcp_session(
headers=bound.headers or None,
auth=bound.auth,
)
async with http_client:
async with streamable_http_client(
async with (
http_client,
streamable_http_client(
str(transport.url),
http_client=http_client,
) as (read_stream, write_stream, _get_session_id):
async with ClientSession(read_stream, write_stream) as session:
await session.initialize()
yield session
) as (read_stream, write_stream, _get_session_id),
):
async with ClientSession(read_stream, write_stream) as session:
await session.initialize()
yield session
return
raise ValueError(f"unsupported MCP transport {transport.kind!r}")
src/wf_sources_python/loader.py
+1 -1
View File
@@ -180,7 +180,7 @@ def _module_file_under_root(root: Path, module: str) -> Path | None:
def _synthetic_module_root(*, source_id: str, root: Path) -> str:
digest = sha256(f"{source_id}\0{root}".encode("utf-8")).hexdigest()[:16]
digest = sha256(f"{source_id}\0{root}".encode()).hexdigest()[:16]
safe_source = source_id.replace(".", "_").replace("-", "_")
return f"_wf_source_{safe_source}_{digest}"
src/wf_transport_rpc_http/__init__.py
+3 -3
View File
@@ -91,12 +91,13 @@ __all__ = [
"PatchDraftParams",
"PatchDraftWorkspaceParams",
"ReadRunTraceParams",
"ReplaceDraftWorkspaceDocumentParams",
"RemoveDraftBindingParams",
"RemoveDraftRouteParams",
"RemoveDraftStepParams",
"RouteSourceParams",
"ReplaceDraftWorkspaceDocumentParams",
"ResumeRunParams",
"RouteSourceParams",
"RpcWorkflowApiClient",
"SaveArtifactParams",
"SaveDeploymentParams",
"SetDraftContractParams",
@@ -117,5 +118,4 @@ __all__ = [
"ValidateDraftWorkspaceParams",
"WorkflowRpcError",
"create_rpc_app",
"RpcWorkflowApiClient",
]
tests/wf_api/test_operation_context.py
+1 -1
View File
@@ -112,4 +112,4 @@ def test_context_runtime_runner_uses_workflow_runtime_service(tmp_path: Path) ->
service = WfMcpService(store=FileStore(tmp_path / "context_runtime"))
context = context_from_service(service)
assert getattr(context.runtime, "runtime") is service.workflow_runtime
assert context.runtime.runtime is service.workflow_runtime
tests/wf_cli/test_caps.py
+1 -1
View File
@@ -6,8 +6,8 @@ from typing import Any
import pytest
from typer.testing import CliRunner
import wf_cli.commands.caps as caps
from wf_cli.app import app
from wf_cli.commands import caps
from wf_cli.commands.caps import (
CapCallOutputFormat,
render_cap_call_output,
tests/wf_mcp/service/test_catalog.py
+6 -6
View File
@@ -320,8 +320,8 @@ def test_source_catalog_service_registers_and_lists_sources_directly(
server="demo",
account="personal",
),
connection_list_enabled=lambda: [],
connection_list_all=lambda: [],
connection_list_enabled=list,
connection_list_all=list,
tool_executor_for=unused_tool_executor,
load_auth=lambda connection_id: None,
emit_event=lambda event: None,
@@ -362,8 +362,8 @@ def test_source_catalog_service_excludes_hidden_sources_from_planner_catalog(
server="demo",
account="personal",
),
connection_list_enabled=lambda: [],
connection_list_all=lambda: [],
connection_list_enabled=list,
connection_list_all=list,
tool_executor_for=unused_tool_executor,
load_auth=lambda connection_id: None,
emit_event=lambda event: None,
@@ -538,8 +538,8 @@ def test_source_catalog_uses_catalog_store_only(tmp_path: Path) -> None:
server="demo",
account="personal",
),
connection_list_enabled=lambda: [],
connection_list_all=lambda: [],
connection_list_enabled=list,
connection_list_all=list,
tool_executor_for=lambda connection: (_ for _ in ()).throw(
AssertionError("unexpected executor")
),
tests/wf_mcp/service/test_event_recorder.py
+1 -1
View File
@@ -106,7 +106,7 @@ def test_wfmcpservice_uses_broker_event_recorder() -> None:
event_bus=bus,
)
service._record_event( # noqa: SLF001
service._record_event(
make_event("connection_registered", connection_id="demo.personal")
)
tests/wf_mcp/service/test_upstream_transport.py
+2 -2
View File
@@ -240,8 +240,8 @@ async def test_upstream_transport_live_diagnostics_report_missing_connection() -
source_catalog = SourceCatalogService(
store=transport.catalog_store,
connection_lookup=_raise_missing_connection,
connection_list_enabled=lambda: [],
connection_list_all=lambda: [],
connection_list_enabled=list,
connection_list_all=list,
tool_executor_for=transport.tool_executor_for,
load_auth=transport.load_connection_auth,
emit_event=lambda event: None,
tests/wf_sources_mcp/test_auth.py
+1 -1
View File
@@ -100,7 +100,7 @@ async def test_httpx_oauth_refresher_posts_refresh_token_grant(
def __init__(self, **kwargs: object) -> None:
assert kwargs["timeout"] == 10.0
async def __aenter__(self) -> "_Client":
async def __aenter__(self) -> _Client:
return self
async def __aexit__(self, *args: object) -> None:
tests/wf_sources_mcp/test_client_transport.py
+4 -4
View File
@@ -274,10 +274,10 @@ async def test_open_mcp_session_uses_binder_for_http_headers(
def __init__(self, **kwargs: Any) -> None:
captured_clients.append(kwargs)
async def __aenter__(self) -> "_CapturingClient":
async def __aenter__(self) -> _CapturingClient:
return self
async def __aexit__(self, *args: Any) -> None:
async def __aexit__(self, *args: object) -> None:
return None
import wf_sources_mcp.client.transport as mod
@@ -319,7 +319,7 @@ async def test_open_mcp_session_refreshes_oauth_record_for_http(
def __init__(self, **kwargs: Any) -> None:
captured_clients.append(kwargs)
async def __aenter__(self) -> "_CapturingClient":
async def __aenter__(self) -> _CapturingClient:
return self
async def __aexit__(self, *args: object) -> None:
@@ -365,7 +365,7 @@ async def test_open_mcp_session_refreshes_oauth_record_for_http(
@pytest.mark.asyncio
async def test_open_mcp_session_uses_binder_for_stdio_env() -> None:
import wf_sources_mcp.client.transport as mod # noqa: I001
import wf_sources_mcp.client.transport as mod
from wf_api.auth import EnvAuth, StoredAuthRecord
captured_params: list[Any] = []