Add File

agents-flex-core/src/main/java/com/agentsflex/core/store/StoreOptions.java

@@ -0,0 +1,132 @@
+/*
+ *  Copyright (c) 2023-2025, Agents-Flex (fuhai999@gmail.com).
+ *  <p>
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *  <p>
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *  <p>
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ */
+package com.agentsflex.core.store;
+
+import com.agentsflex.core.llm.embedding.EmbeddingOptions;
+import com.agentsflex.core.util.Metadata;
+import com.agentsflex.core.util.StringUtil;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+/**
+ * Store Options, Each store can have its own Options implementation.
+ */
+public class StoreOptions extends Metadata {
+
+    public static final StoreOptions DEFAULT = new StoreOptions() {
+        @Override
+        public void setCollectionName(String collectionName) {
+            throw new IllegalStateException("Can not set collectionName to the default instance.");
+        }
+
+        @Override
+        public void setPartitionNames(List<String> partitionNames) {
+            throw new IllegalStateException("Can not set partitionName to the default instance.");
+        }
+
+        @Override
+        public void setEmbeddingOptions(EmbeddingOptions embeddingOptions) {
+            throw new IllegalStateException("Can not set embeddingOptions to the default instance.");
+        }
+    };
+
+    /**
+     * store collection name
+     */
+    private String collectionName;
+
+    /**
+     * store index name
+     */
+    private String indexName;
+
+    /**
+     * store partition name
+     */
+    private List<String> partitionNames;
+
+    /**
+     * store embedding options
+     */
+    private EmbeddingOptions embeddingOptions = EmbeddingOptions.DEFAULT;
+
+
+    public String getCollectionName() {
+        return collectionName;
+    }
+
+    public String getCollectionNameOrDefault(String other) {
+        return StringUtil.hasText(collectionName) ? collectionName : other;
+    }
+
+    public void setCollectionName(String collectionName) {
+        this.collectionName = collectionName;
+    }
+
+    public List<String> getPartitionNames() {
+        return partitionNames;
+    }
+
+    public String getPartitionName() {
+        return partitionNames != null && !partitionNames.isEmpty() ? partitionNames.get(0) : null;
+    }
+
+    public List<String> getPartitionNamesOrEmpty() {
+        return partitionNames == null ? Collections.emptyList() : partitionNames;
+    }
+
+    public void setPartitionNames(List<String> partitionNames) {
+        this.partitionNames = partitionNames;
+    }
+
+    public StoreOptions partitionName(String partitionName) {
+        if (this.partitionNames == null) {
+            this.partitionNames = new ArrayList<>(1);
+        }
+        this.partitionNames.add(partitionName);
+        return this;
+    }
+
+
+    public EmbeddingOptions getEmbeddingOptions() {
+        return embeddingOptions;
+    }
+
+    public void setEmbeddingOptions(EmbeddingOptions embeddingOptions) {
+        this.embeddingOptions = embeddingOptions;
+    }
+
+
+    public static StoreOptions ofCollectionName(String collectionName) {
+        StoreOptions storeOptions = new StoreOptions();
+        storeOptions.setCollectionName(collectionName);
+        return storeOptions;
+    }
+
+    public String getIndexName() {
+        return indexName;
+    }
+
+    public void setIndexName(String indexName) {
+        this.indexName = indexName;
+    }
+
+    public String getIndexNameOrDefault(String other) {
+        return StringUtil.hasText(indexName) ? indexName : other;
+    }
+}