(iotdb) branch master updated: [AINode] Fix call inference bug (#17011)

[yongzao]

This is an automated email from the ASF dual-hosted git repository.

yongzao pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/iotdb.git


The following commit(s) were added to refs/heads/master by this push:
     new db973ea3217 [AINode] Fix call inference bug (#17011)
db973ea3217 is described below

commit db973ea32172c78d6cc4c0af7bbb55fd8ded6e32
Author: Yongzao <yong...@apache.org>
AuthorDate: Mon Jan 12 17:26:40 2026 +0800

    [AINode] Fix call inference bug (#17011)
---
 .../iotdb/ainode/it/AINodeCallInferenceIT.java     | 22 ++++++++++++++++++++++
 .../iotdb/ainode/core/manager/inference_manager.py |  6 +++++-
 2 files changed, 27 insertions(+), 1 deletion(-)

diff --git 
a/integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeCallInferenceIT.java
 
b/integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeCallInferenceIT.java
index 3131e398059..9bf0927bb62 100644
--- 
a/integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeCallInferenceIT.java
+++ 
b/integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeCallInferenceIT.java
@@ -48,6 +48,8 @@ public class AINodeCallInferenceIT {
 
   private static final String CALL_INFERENCE_SQL_TEMPLATE =
       "CALL INFERENCE(%s, \"SELECT s%d FROM root.AI LIMIT %d\", 
generateTime=true, outputLength=%d)";
+  private static final String CALL_INFERENCE_BY_DEFAULT_SQL_TEMPLATE =
+      "CALL INFERENCE(%s, \"SELECT s%d FROM root.AI LIMIT 256\")";
   private static final int DEFAULT_INPUT_LENGTH = 256;
   private static final int DEFAULT_OUTPUT_LENGTH = 48;
 
@@ -69,6 +71,7 @@ public class AINodeCallInferenceIT {
       try (Connection connection = 
EnvFactory.getEnv().getConnection(BaseEnv.TREE_SQL_DIALECT);
           Statement statement = connection.createStatement()) {
         callInferenceTest(statement, modelInfo);
+        callInferenceByDefaultTest(statement, modelInfo);
       }
     }
   }
@@ -96,4 +99,23 @@ public class AINodeCallInferenceIT {
       }
     }
   }
+
+  public static void callInferenceByDefaultTest(
+      Statement statement, AINodeTestUtils.FakeModelInfo modelInfo) throws 
SQLException {
+    // Invoke call inference for specified models, there should exist result.
+    for (int i = 0; i < 4; i++) {
+      String callInferenceSQL =
+          String.format(CALL_INFERENCE_BY_DEFAULT_SQL_TEMPLATE, 
modelInfo.getModelId(), i);
+      try (ResultSet resultSet = statement.executeQuery(callInferenceSQL)) {
+        ResultSetMetaData resultSetMetaData = resultSet.getMetaData();
+        checkHeader(resultSetMetaData, "output");
+        int count = 0;
+        while (resultSet.next()) {
+          count++;
+        }
+        // Ensure the call inference return results
+        Assert.assertTrue(count > 0);
+      }
+    }
+  }
 }
diff --git a/iotdb-core/ainode/iotdb/ainode/core/manager/inference_manager.py 
b/iotdb-core/ainode/iotdb/ainode/core/manager/inference_manager.py
index ada641dd54c..b758908c5e4 100644
--- a/iotdb-core/ainode/iotdb/ainode/core/manager/inference_manager.py
+++ b/iotdb-core/ainode/iotdb/ainode/core/manager/inference_manager.py
@@ -265,7 +265,11 @@ class InferenceManager:
             req,
             data_getter=lambda r: r.dataset,
             extract_attrs=lambda r: {
-                "output_length": int(r.inferenceAttributes.pop("outputLength", 
96)),
+                "output_length": (
+                    96
+                    if r.inferenceAttributes is None
+                    else int(r.inferenceAttributes.pop("outputLength", 96))
+                ),
                 **(r.inferenceAttributes or {}),
             },
             resp_cls=TInferenceResp,