content: /*
   *  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;
      }
  }
  