diff --git a/easy-agents-embedding/easy-agents-embedding-openai/pom.xml b/easy-agents-embedding/easy-agents-embedding-openai/pom.xml index 78d5db0..9f79b2d 100644 --- a/easy-agents-embedding/easy-agents-embedding-openai/pom.xml +++ b/easy-agents-embedding/easy-agents-embedding-openai/pom.xml @@ -21,6 +21,11 @@ com.easyagents easy-agents-core + + junit + junit + test + diff --git a/easy-agents-embedding/easy-agents-embedding-openai/src/main/java/com/easyagents/embedding/openai/OpenAIEmbeddingModel.java b/easy-agents-embedding/easy-agents-embedding-openai/src/main/java/com/easyagents/embedding/openai/OpenAIEmbeddingModel.java index 8e0f044..afbc8c2 100644 --- a/easy-agents-embedding/easy-agents-embedding-openai/src/main/java/com/easyagents/embedding/openai/OpenAIEmbeddingModel.java +++ b/easy-agents-embedding/easy-agents-embedding-openai/src/main/java/com/easyagents/embedding/openai/OpenAIEmbeddingModel.java @@ -1,20 +1,7 @@ -/* - * Copyright (c) 2023-2026, Easy-Agents (fuhai999@gmail.com). - *

- * 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 - *

- * http://www.apache.org/licenses/LICENSE-2.0 - *

- * 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.easyagents.embedding.openai; +import com.alibaba.fastjson2.JSON; +import com.alibaba.fastjson2.JSONObject; import com.easyagents.core.document.Document; import com.easyagents.core.model.client.HttpClient; import com.easyagents.core.model.embedding.BaseEmbeddingModel; @@ -24,10 +11,9 @@ import com.easyagents.core.store.VectorData; import com.easyagents.core.util.JSONUtil; import com.easyagents.core.util.Maps; import com.easyagents.core.util.StringUtil; -import com.alibaba.fastjson2.JSON; -import com.alibaba.fastjson2.JSONObject; import java.util.HashMap; +import java.util.Locale; import java.util.Map; public class OpenAIEmbeddingModel extends BaseEmbeddingModel { @@ -69,19 +55,63 @@ public class OpenAIEmbeddingModel extends BaseEmbeddingModel headers, String payload) { + return "{\"data\":[{}]}"; + } + }); + + ModelException exception = Assert.assertThrows( + ModelException.class, + () -> model.embed(Document.of("hello")) + ); + + Assert.assertTrue(exception.getMessage().contains("data[0].embedding")); + } +}