(iotdb) branch master updated: finish

[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 9bde5e80d31 finish
9bde5e80d31 is described below

commit 9bde5e80d31a1321c0824686e35c0362ebd0dad7
Author: Yongzao <532741...@qq.com>
AuthorDate: Tue Jan 27 13:31:18 2026 +0800

    finish
---
 .../org/apache/iotdb/ainode/it/AINodeCallInferenceIT.java  | 14 ++++++++++++++
 .../execution/operator/process/ai/InferenceOperator.java   |  7 +++++++
 2 files changed, 21 insertions(+)

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 9bf0927bb62..852827aa8ab 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
@@ -40,6 +40,7 @@ import java.sql.Statement;
 
 import static org.apache.iotdb.ainode.utils.AINodeTestUtils.BUILTIN_MODEL_MAP;
 import static org.apache.iotdb.ainode.utils.AINodeTestUtils.checkHeader;
+import static org.apache.iotdb.ainode.utils.AINodeTestUtils.errorTest;
 import static org.apache.iotdb.ainode.utils.AINodeTestUtils.prepareDataInTree;
 
 @RunWith(IoTDBTestRunner.class)
@@ -72,6 +73,7 @@ public class AINodeCallInferenceIT {
           Statement statement = connection.createStatement()) {
         callInferenceTest(statement, modelInfo);
         callInferenceByDefaultTest(statement, modelInfo);
+        callInferenceErrorTest(statement, modelInfo);
       }
     }
   }
@@ -118,4 +120,16 @@ public class AINodeCallInferenceIT {
       }
     }
   }
+
+  public static void callInferenceErrorTest(
+      Statement statement, AINodeTestUtils.FakeModelInfo modelInfo) {
+    String multiVariateSQL =
+        String.format(
+            "CALL INFERENCE(%s, \"SELECT s0,s1 FROM root.AI LIMIT 128\", 
generateTime=true, outputLength=10)",
+            modelInfo.getModelId());
+    errorTest(
+        statement,
+        multiVariateSQL,
+        "701: Call inference function should not contain more than one input 
column, found [2] input columns.");
+  }
 }
diff --git 
a/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/operator/process/ai/InferenceOperator.java
 
b/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/operator/process/ai/InferenceOperator.java
index 8bf9f5ead14..d39d81aff5b 100644
--- 
a/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/operator/process/ai/InferenceOperator.java
+++ 
b/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/operator/process/ai/InferenceOperator.java
@@ -24,6 +24,7 @@ import org.apache.iotdb.ainode.rpc.thrift.TInferenceResp;
 import org.apache.iotdb.commons.client.exception.ClientManagerException;
 import org.apache.iotdb.db.exception.ainode.AINodeConnectionException;
 import org.apache.iotdb.db.exception.runtime.ModelInferenceProcessException;
+import org.apache.iotdb.db.exception.sql.SemanticException;
 import org.apache.iotdb.db.protocol.client.an.AINodeClient;
 import org.apache.iotdb.db.protocol.client.an.AINodeClientManager;
 import org.apache.iotdb.db.queryengine.execution.MemoryEstimationHelper;
@@ -225,6 +226,12 @@ public class InferenceOperator implements ProcessOperator {
         maxTimestamp = Math.max(maxTimestamp, timestamp);
       }
       timeColumnBuilder.writeLong(timestamp);
+      if (inputTsBlock.getValueColumnCount() > 1) {
+        throw new SemanticException(
+            String.format(
+                "Call inference function should not contain more than one 
input column, found [%d] input columns.",
+                inputTsBlock.getValueColumnCount()));
+      }
       for (int columnIndex = 0; columnIndex < 
inputTsBlock.getValueColumnCount(); columnIndex++) {
         
columnBuilders[columnIndexes[columnIndex]].write(inputTsBlock.getColumn(columnIndex),
 i);
       }