@qingflow-tech/qingflow-app-user-mcp 1.0.1 → 1.0.3

package/src/qingflow_mcp/tools/code_block_tools.py

@@ -91,7 +91,7 @@ class CodeBlockTools(RecordTools):
         def record_code_block_run(
             profile: str = DEFAULT_PROFILE,
             app_key: str = "",
-            record_id: int = 0,
+            record_id: str = "",
             code_block_field: str = "",
             role: int = 1,
             workflow_node_id: int | None = None,
@@ -197,7 +197,7 @@ class CodeBlockTools(RecordTools):
         *,
         profile: str,
         app_key: str,
-        record_id: int,
+        record_id: int | str,
         code_block_field: str,
         role: int = 1,
         workflow_node_id: int | None = None,

package/src/qingflow_mcp/tools/import_tools.py

@@ -783,6 +783,8 @@ class ImportTools(ToolBase):
                 error_code="CONFIG_ERROR",
                 message="record_import_status_get accepts import_id or process_id_str, but not both at the same time",
                 extra={
+                    "import_id": normalized_import_id,
+                    "process_id_str": normalized_process_id,
                     "details": {
                         "fix_hint": "Use only one of `import_id` or `process_id_str`. You may pass `app_key` as an optional routing hint for direct method compatibility.",
                     }
@@ -793,6 +795,8 @@ class ImportTools(ToolBase):
                 error_code="CONFIG_ERROR",
                 message="record_import_status_get requires at least one selector: process_id_str, import_id, or app_key",
                 extra={
+                    "import_id": normalized_import_id,
+                    "process_id_str": normalized_process_id,
                     "details": {
                         "fix_hint": "Use `process_id_str` or `import_id` for a known import, or use only `app_key` to inspect the latest import in that app.",
                     }
@@ -806,6 +810,9 @@ class ImportTools(ToolBase):
             if local_job is None and normalized_process_id:
                 matches = [item for item in self._job_store.list() if _normalize_optional_text(item.get("process_id_str")) == normalized_process_id]
                 local_job = matches[0] if len(matches) == 1 else None
+            effective_process_id = normalized_process_id
+            if effective_process_id is None and isinstance(local_job, dict):
+                effective_process_id = _normalize_optional_text(local_job.get("process_id_str"))
             resolved_app_key = normalized_app_key
             if not resolved_app_key and isinstance(local_job, dict):
                 resolved_app_key = str(local_job.get("app_key") or "").strip()
@@ -814,6 +821,8 @@ class ImportTools(ToolBase):
                     error_code="CONFIG_ERROR",
                     message="record_import_status_get could not determine app_key from the provided selector",
                     extra={
+                        "import_id": normalized_import_id,
+                        "process_id_str": effective_process_id,
                         "details": {
                             "fix_hint": "Use the original `app_key`, or call import status with the latest-import mode: only `app_key`.",
                         }
@@ -832,13 +841,18 @@ class ImportTools(ToolBase):
             matched_record, matched_by = _match_import_record(
                 records,
                 local_job=local_job,
-                process_id_str=normalized_process_id,
+                import_id=normalized_import_id,
+                process_id_str=effective_process_id,
             )
             if matched_record is None:
                 return self._failed_status_result(
                     error_code="IMPORT_STATUS_AMBIGUOUS",
                     message="could not uniquely resolve an import record from the provided identifiers",
-                    extra={"matched_by": matched_by},
+                    extra={
+                        "import_id": normalized_import_id,
+                        "process_id_str": effective_process_id,
+                        "matched_by": matched_by,
+                    },
                 )
             normalized_process = _normalize_optional_text(
                 matched_record.get("processIdStr") or matched_record.get("processId") or matched_record.get("process_id_str")
@@ -2118,6 +2132,7 @@ def _match_import_record(
     records: list[JSONObject],
     *,
     local_job: dict[str, Any] | None,
+    import_id: str | None,
     process_id_str: str | None,
 ) -> tuple[JSONObject | None, str | None]:
     if process_id_str:
@@ -2130,6 +2145,16 @@ def _match_import_record(
             return exact[0], "process_id_str"
         if len(exact) > 1:
             return None, "process_id_str"
+    if import_id:
+        exact = [
+            item
+            for item in records
+            if import_id in _extract_import_record_ids(item)
+        ]
+        if len(exact) == 1:
+            return exact[0], "import_id"
+        if len(exact) > 1:
+            return None, "import_id"
     if isinstance(local_job, dict):
         source_file_name = _normalize_optional_text(local_job.get("source_file_name"))
         started_at = _parse_utc(local_job.get("started_at"))
@@ -2160,6 +2185,15 @@ def _match_import_record(
     return None, None
 
 
+def _extract_import_record_ids(record: JSONObject) -> set[str]:
+    identifiers: set[str] = set()
+    for key in ("importId", "import_id", "dataImportId", "data_import_id"):
+        normalized = _normalize_optional_text(record.get(key))
+        if normalized:
+            identifiers.add(normalized)
+    return identifiers
+
+
 def _parse_utc(value: Any) -> datetime | None:
     text = _normalize_optional_text(value)
     if text is None: