Skip to content

Commit 63de8e5

Browse files
committed
fix(SAAI): update export() methods to return OptionMap
API now requires export() methods to return OptionMap. Updating the return type fixes compilation and runtime errors. Behavior remains unchanged.
1 parent ee47835 commit 63de8e5

File tree

8 files changed

+24
-24
lines changed

8 files changed

+24
-24
lines changed

anthropic/src/main/java/dev/bluetree242/saaiaddons/anthropic/AnthropicChatModelProvider.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -58,8 +58,8 @@ public Map<String, Object> getDefaultOptions() {
5858

5959
@NotNull
6060
@Override
61-
public Map<String, Object> export(@NotNull ChatModelContext context) {
62-
Map<String, Object> result = ChatModelProvider.super.export(context);
61+
public OptionMap export(@NotNull ChatModelContext context) {
62+
OptionMap result = ChatModelProvider.super.export(context);
6363
// Makes sure the "model" is always in the config even if it is not configured.
6464
result.putIfAbsent("model", "");
6565
return result;
@@ -88,4 +88,4 @@ public ChatResponse chat(ChatRequest request) {
8888
.build());
8989
}
9090
}
91-
}
91+
}

azure-openai/src/main/java/dev/bluetree242/saaiaddons/azure/AzureOpenAiChatModelProvider.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,8 @@ public Map<String, Object> getDefaultOptions() {
5353

5454
@NotNull
5555
@Override
56-
public Map<String, Object> export(@NotNull ChatModelContext context) {
57-
Map<String, Object> result = ChatModelProvider.super.export(context);
56+
public OptionMap export(@NotNull ChatModelContext context) {
57+
OptionMap result = ChatModelProvider.super.export(context);
5858
// Makes sure the "deployment_name" is always in the config even if it is not configured.
5959
result.putIfAbsent("deployment_name", "");
6060
return result;
@@ -65,4 +65,4 @@ public Map<String, Object> export(@NotNull ChatModelContext context) {
6565
public String getDisplayName(@Nullable ChatModelContext context) {
6666
return "Azure OpenAI";
6767
}
68-
}
68+
}

azure-openai/src/main/java/dev/bluetree242/saaiaddons/azure/AzureOpenAiEmbeddingProvider.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ public Map<String, Object> getDefaultOptions() {
4848

4949
@NotNull
5050
@Override
51-
public Map<String, Object> export(@NotNull EmbeddingContext context) {
52-
Map<String, Object> result = EmbeddingModelProvider.super.export(context);
51+
public OptionMap export(@NotNull EmbeddingContext context) {
52+
OptionMap result = EmbeddingModelProvider.super.export(context);
5353
// Makes sure the "deployment_name" is always in the config even if it is not configured.
5454
result.putIfAbsent("deployment_name", "");
5555
result.putIfAbsent("service_version", "");
@@ -62,4 +62,4 @@ public Map<String, Object> export(@NotNull EmbeddingContext context) {
6262
public String getDisplayName(@Nullable EmbeddingContext context) {
6363
return "Azure OpenAI";
6464
}
65-
}
65+
}

google-aistudio/src/main/java/dev/bluetree242/saaiaddons/aistudio/GoogleAiStudioChatModelProvider.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -70,8 +70,8 @@ public Map<String, Object> getDefaultOptions() {
7070

7171
@NotNull
7272
@Override
73-
public Map<String, Object> export(@NotNull ChatModelContext context) {
74-
Map<String, Object> result = ChatModelProvider.super.export(context);
73+
public OptionMap export(@NotNull ChatModelContext context) {
74+
OptionMap result = ChatModelProvider.super.export(context);
7575
result.putIfAbsent("model", "");
7676
return result;
7777
}
@@ -99,4 +99,4 @@ public ChatResponse chat(ChatRequest request) {
9999
.build());
100100
}
101101
}
102-
}
102+
}

hugging-face/src/main/java/dev/bluetree242/saaiaddons/huggingface/HuggingFaceChatModelProvider.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ public Map<String, Object> getDefaultOptions() {
4646

4747
@NotNull
4848
@Override
49-
public Map<String, Object> export(@NotNull ChatModelContext context) {
50-
Map<String, Object> result = ChatModelProvider.super.export(context);
49+
public OptionMap export(@NotNull ChatModelContext context) {
50+
OptionMap result = ChatModelProvider.super.export(context);
5151
result.putIfAbsent("model", "");
5252
return result;
5353
}
@@ -57,4 +57,4 @@ public Map<String, Object> export(@NotNull ChatModelContext context) {
5757
public String getDisplayName(@Nullable ChatModelContext context) {
5858
return "HuggingFace";
5959
}
60-
}
60+
}

hugging-face/src/main/java/dev/bluetree242/saaiaddons/huggingface/HuggingFaceEmbeddingProvider.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,8 @@ public Map<String, Object> getDefaultOptions() {
4141

4242
@NotNull
4343
@Override
44-
public Map<String, Object> export(@NotNull EmbeddingContext context) {
45-
Map<String, Object> result = EmbeddingModelProvider.super.export(context);
44+
public OptionMap export(@NotNull EmbeddingContext context) {
45+
OptionMap result = EmbeddingModelProvider.super.export(context);
4646
result.putIfAbsent("model", "");
4747
return result;
4848
}
@@ -52,4 +52,4 @@ public Map<String, Object> export(@NotNull EmbeddingContext context) {
5252
public String getDisplayName(@Nullable EmbeddingContext context) {
5353
return "HuggingFace";
5454
}
55-
}
55+
}

mistral-ai/src/main/java/dev/bluetree242/saaiaddons/mistralai/MistralAiChatModelProvider.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ public Map<String, Object> getDefaultOptions() {
4646

4747
@NotNull
4848
@Override
49-
public Map<String, Object> export(@NotNull ChatModelContext context) {
50-
Map<String, Object> result = ChatModelProvider.super.export(context);
49+
public OptionMap export(@NotNull ChatModelContext context) {
50+
OptionMap result = ChatModelProvider.super.export(context);
5151
// Makes sure the "model" is always in the config even if it is not configured.
5252
result.putIfAbsent("model", "");
5353
return result;
@@ -58,4 +58,4 @@ public Map<String, Object> export(@NotNull ChatModelContext context) {
5858
public String getDisplayName(@Nullable ChatModelContext context) {
5959
return "MistralAI";
6060
}
61-
}
61+
}

mistral-ai/src/main/java/dev/bluetree242/saaiaddons/mistralai/MistralAiEmbeddingProvider.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,8 @@ public Map<String, Object> getDefaultOptions() {
4444

4545
@NotNull
4646
@Override
47-
public Map<String, Object> export(@NotNull EmbeddingContext context) {
48-
Map<String, Object> result = EmbeddingModelProvider.super.export(context);
47+
public OptionMap export(@NotNull EmbeddingContext context) {
48+
OptionMap result = EmbeddingModelProvider.super.export(context);
4949
// Makes sure the "model" is always in the config even if it is not configured.
5050
result.putIfAbsent("model", "");
5151
return result;
@@ -56,4 +56,4 @@ public Map<String, Object> export(@NotNull EmbeddingContext context) {
5656
public String getDisplayName(@Nullable EmbeddingContext context) {
5757
return "MistralAI";
5858
}
59-
}
59+
}

0 commit comments

Comments
 (0)