fix compile

Author: linrrzqqq

be/src/core/column/predicate_column.h

@@ -325,7 +325,7 @@ class PredicateColumnType final : public COWHelper<IColumn, PredicateColumnType<
             for (size_t i = 0; i < n; i++) {
                 memcpy(dst, str.data(), str.size());
                 insert_string_value(dst, str.size());
-                dst += i * str.size();
+                dst += str.size();
             }
         } else if constexpr (Type == TYPE_LARGEINT) {
             const auto& v = x.get<TYPE_LARGEINT>();

be/src/exprs/aggregate/aggregate_function_ai_agg.h

@@ -43,11 +43,7 @@ class AggregateFunctionAIAggData {
 
     void add(StringRef ref) {
         auto delta_size = ref.size + (inited ? SEPARATOR_SIZE : 0);
-        if (handle_overflow(delta_size)) {
-            throw Exception(ErrorCode::OUT_OF_BOUND,
-                            "Failed to add data: combined context size exceeded "
-                            "maximum limit even after processing");
-        }
+        handle_overflow(delta_size);
         append_data(ref.data, ref.size);
     }
 
@@ -60,11 +56,7 @@ class AggregateFunctionAIAggData {
         _task = rhs._task;
 
         size_t delta_size = (inited ? SEPARATOR_SIZE : 0) + rhs.data.size();
-        if (handle_overflow(delta_size)) {
-            throw Exception(ErrorCode::OUT_OF_BOUND,
-                            "Failed to merge data: combined context size exceeded "
-                            "maximum limit even after processing");
-        }
+        handle_overflow(delta_size);
 
         if (!inited) {
             inited = true;
@@ -159,7 +151,7 @@ class AggregateFunctionAIAggData {
     const std::string& get_task() const { return _task; }
 
 #ifdef BE_TEST
-    static void normalize_endpoint_for_test(TAIResource& config) { normalize_endpoint(config); }
+    static void normalize_endpoint_for_test(AIResource& config) { normalize_endpoint(config); }
 #endif
 
 private:
@@ -195,17 +187,14 @@ class AggregateFunctionAIAggData {
         return client->execute_post_request(request_body, &response);
     }
 
-    // handle overflow situations when adding content.
-    bool handle_overflow(size_t additional_size) {
+    // Treat the context window as a soft batching trigger instead of a hard reject.
+    void handle_overflow(size_t additional_size) {
         const size_t max_context_size = get_ai_context_window_size();
-        if (additional_size + data.size() <= max_context_size) {
-            return false;
+        if (additional_size + data.size() <= max_context_size || !inited) {
+            return;
         }
 
         process_current_context();
-
-        // check if there is still an overflow after replacement.
-        return (additional_size + data.size() > max_context_size);
     }
 
     static size_t get_ai_context_window_size() {
@@ -214,7 +203,7 @@ class AggregateFunctionAIAggData {
         return static_cast<size_t>(_ctx->query_options().ai_context_window_size);
     }
 
-    static void normalize_endpoint(TAIResource& config) {
+    static void normalize_endpoint(AIResource& config) {
         if (iequal(config.provider_type, "GEMINI")) {
             if (!config.endpoint.ends_with("v1") && !config.endpoint.ends_with("v1beta")) {
                 return;

be/src/exprs/function/ai/ai_adapter.h

@@ -273,7 +273,6 @@ class VoyageAIAdapter : public AIAdapter {
     }
 
     Status build_embedding_request(const std::vector<std::string>& inputs,
-                                   const std::vector<std::string>& /*media_content_types*/,
                                    std::string& request_body) const override {
         rapidjson::Document doc;
         doc.SetObject();
@@ -503,7 +502,6 @@ class LocalAdapter : public AIAdapter {
     }
 
     Status build_embedding_request(const std::vector<std::string>& inputs,
-                                   const std::vector<std::string>& /*media_content_types*/,
                                    std::string& request_body) const override {
         rapidjson::Document doc;
         doc.SetObject();
@@ -909,7 +907,6 @@ class OpenAIAdapter : public VoyageAIAdapter {
 class DeepSeekAdapter : public OpenAIAdapter {
 public:
     Status build_embedding_request(const std::vector<std::string>& inputs,
-                                   const std::vector<std::string>& /*media_content_types*/,
                                    std::string& request_body) const override {
         return embed_not_supported_status();
     }
@@ -923,7 +920,6 @@ class DeepSeekAdapter : public OpenAIAdapter {
 class MoonShotAdapter : public OpenAIAdapter {
 public:
     Status build_embedding_request(const std::vector<std::string>& inputs,
-                                   const std::vector<std::string>& /*media_content_types*/,
                                    std::string& request_body) const override {
         return embed_not_supported_status();
     }
@@ -937,7 +933,6 @@ class MoonShotAdapter : public OpenAIAdapter {
 class MinimaxAdapter : public OpenAIAdapter {
 public:
     Status build_embedding_request(const std::vector<std::string>& inputs,
-                                   const std::vector<std::string>& /*media_content_types*/,
                                    std::string& request_body) const override {
         rapidjson::Document doc;
         doc.SetObject();
@@ -1269,7 +1264,6 @@ class GeminiAdapter : public AIAdapter {
     }
 
     Status build_embedding_request(const std::vector<std::string>& inputs,
-                                   const std::vector<std::string>& /*media_content_types*/,
                                    std::string& request_body) const override {
         rapidjson::Document doc;
         doc.SetObject();

be/test/ai/aggregate_function_ai_agg_test.cpp

@@ -425,7 +425,7 @@ TEST_F(AggregateFunctionAIAggTest, ai_context_window_size_session_variable_test)
 }
 
 TEST_F(AggregateFunctionAIAggTest, gemini_endpoint_normalize_to_generate_content_test) {
-    TAIResource resource;
+    AIResource resource;
     resource.provider_type = "GEMINI";
     resource.model_name = "gemini-pro";
     resource.endpoint = "https://generativelanguage.googleapis.com/v1beta";
@@ -435,7 +435,7 @@ TEST_F(AggregateFunctionAIAggTest, gemini_endpoint_normalize_to_generate_content
 }
 
 TEST_F(AggregateFunctionAIAggTest, openai_completions_endpoint_normalize_test) {
-    TAIResource resource;
+    AIResource resource;
     resource.provider_type = "OPENAI";
     resource.endpoint = "https://api.openai.com/v1/completions";
     AggregateFunctionAIAggData::normalize_endpoint_for_test(resource);

fe/fe-core/src/test/java/org/apache/doris/qe/SessionVariablesTest.java

@@ -20,8 +20,8 @@
 import org.apache.doris.analysis.IntLiteral;
 import org.apache.doris.analysis.SetType;
 import org.apache.doris.analysis.SetVar;
-import org.apache.doris.common.DdlException;
 import org.apache.doris.common.Config;
+import org.apache.doris.common.DdlException;
 import org.apache.doris.common.FeConstants;
 import org.apache.doris.nereids.parser.NereidsParser;
 import org.apache.doris.utframe.TestWithFeService;