Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Revert default OpenAI embedding to text-embedding-ada-002 as some vector stores have problems with 3-small

Browse Source
This commit is contained in:
Christian Tzolov
2024-02-16 11:19:34 +01:00
parent 5391505098
commit fdd3fbb2f9
4 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
vector-stores/spring-ai-milvus-store/src/test/java/org/springframework/ai/vectorstore/MilvusVectorStoreIT.java
View File

@@ -310,8 +310,10 @@ public class MilvusVectorStoreIT {
@Bean
public EmbeddingClient embeddingClient() {
return new OpenAiEmbeddingClient(new OpenAiApi(System.getenv("OPENAI_API_KEY")), MetadataMode.EMBED,
OpenAiEmbeddingOptions.builder().withModel("text-embedding-ada-002").build());
return new OpenAiEmbeddingClient(new OpenAiApi(System.getenv("OPENAI_API_KEY")));
// return new OpenAiEmbeddingClient(new
// OpenAiApi(System.getenv("OPENAI_API_KEY")), MetadataMode.EMBED,
// OpenAiEmbeddingOptions.builder().withModel("text-embedding-ada-002").build());
}
}