Skip to content

Commit 9e98962

Browse files
committed
Add additional ctor for OpenAiEmbeddingClient
1 parent 4c617e1 commit 9e98962

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed

models/spring-ai-openai/src/main/java/org/springframework/ai/openai/OpenAiEmbeddingClient.java

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,16 +53,43 @@ public class OpenAiEmbeddingClient extends AbstractEmbeddingClient {
5353

5454
private final MetadataMode metadataMode;
5555

56+
/**
57+
* Constructor for the OpenAiEmbeddingClient class.
58+
* @param openAiApi The OpenAiApi instance to use for making API requests.
59+
*/
5660
public OpenAiEmbeddingClient(OpenAiApi openAiApi) {
5761
this(openAiApi, MetadataMode.EMBED);
5862
}
5963

64+
/**
65+
* Initializes a new instance of the OpenAiEmbeddingClient class.
66+
* @param openAiApi The OpenAiApi instance to use for making API requests.
67+
* @param metadataMode The mode for generating metadata.
68+
*/
6069
public OpenAiEmbeddingClient(OpenAiApi openAiApi, MetadataMode metadataMode) {
6170
this(openAiApi, metadataMode,
6271
OpenAiEmbeddingOptions.builder().withModel(OpenAiApi.DEFAULT_EMBEDDING_MODEL).build(),
6372
RetryUtils.DEFAULT_RETRY_TEMPLATE);
6473
}
6574

75+
/**
76+
* Initializes a new instance of the OpenAiEmbeddingClient class.
77+
* @param openAiApi The OpenAiApi instance to use for making API requests.
78+
* @param metadataMode The mode for generating metadata.
79+
* @param openAiEmbeddingOptions The options for OpenAi embedding.
80+
*/
81+
public OpenAiEmbeddingClient(OpenAiApi openAiApi, MetadataMode metadataMode,
82+
OpenAiEmbeddingOptions openAiEmbeddingOptions) {
83+
this(openAiApi, metadataMode, openAiEmbeddingOptions, RetryUtils.DEFAULT_RETRY_TEMPLATE);
84+
}
85+
86+
/**
87+
* Initializes a new instance of the OpenAiEmbeddingClient class.
88+
* @param openAiApi - The OpenAiApi instance to use for making API requests.
89+
* @param metadataMode - The mode for generating metadata.
90+
* @param options - The options for OpenAI embedding.
91+
* @param retryTemplate - The RetryTemplate for retrying failed API requests.
92+
*/
6693
public OpenAiEmbeddingClient(OpenAiApi openAiApi, MetadataMode metadataMode, OpenAiEmbeddingOptions options,
6794
RetryTemplate retryTemplate) {
6895
Assert.notNull(openAiApi, "OpenAiService must not be null");

0 commit comments

Comments
 (0)