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.llm.embedding;
  
  import com.agentsflex.core.document.Document;
  import com.agentsflex.core.llm.exception.LlmException;
  import com.agentsflex.core.store.VectorData;
  
  public interface EmbeddingModel {
  
      default VectorData embed(String text) {
          return embed(Document.of(text), EmbeddingOptions.DEFAULT);
      }
  
      default VectorData embed(Document document) {
          return embed(document, EmbeddingOptions.DEFAULT);
      }
  
      VectorData embed(Document document, EmbeddingOptions options);
  
      default int dimensions() {
          VectorData vectorData = embed(Document.of("agents-flex"));
          if (vectorData == null) {
              throw new LlmException("Embedding model does not contain vector data, maybe config is not correct.");
          }
          double[] vector = vectorData.getVector();
          if (vector == null || vector.length == 0) {
              throw new LlmException("Embedding model does not contain vector data, maybe config is not correct.");
          }
          return vector.length;
      }
  }
  