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

Refactor ModerationResult builder methods

Browse Source 
- Deprecate the existing builder methods to remove the prefix `with`
 - Update references
This commit is contained in:
Ilayaperumal Gopinathan
2024-12-18 13:06:27 +00:00
committed by Mark Pollack
parent 5f300c237e
commit 1c72f14a69
2 changed files with 30 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
models/spring-ai-openai/src/main/java/org/springframework/ai/openai/OpenAiModerationModel.java
View File

@@ -147,9 +147,9 @@ public class OpenAiModerationModel implements ModerationModel {
.build();
}
ModerationResult moderationResult = ModerationResult.builder()
.withCategories(categories)
.withCategoryScores(categoryScores)
.withFlagged(result.flagged())
.categories(categories)
.categoryScores(categoryScores)
.flagged(result.flagged())
.build();
moderationResults.add(moderationResult);
}

27
spring-ai-core/src/main/java/org/springframework/ai/moderation/ModerationResult.java
View File

@@ -100,16 +100,43 @@ public final class ModerationResult {
private CategoryScores categoryScores;
public Builder flagged(boolean flagged) {
this.flagged = flagged;
return this;
}
public Builder categories(Categories categories) {
this.categories = categories;
return this;
}
public Builder categoryScores(CategoryScores categoryScores) {
this.categoryScores = categoryScores;
return this;
}
/**
* @deprecated use {@link #flagged(boolean)} instead.
*/
@Deprecated(forRemoval = true, since = "1.0.0-M5")
public Builder withFlagged(boolean flagged) {
this.flagged = flagged;
return this;
}
/**
* @deprecated use {@link #categories(Categories)} instead.
*/
@Deprecated(forRemoval = true, since = "1.0.0-M5")
public Builder withCategories(Categories categories) {
this.categories = categories;
return this;
}
/**
* @deprecated use {@link #categoryScores(CategoryScores)} instead.
*/
@Deprecated(forRemoval = true, since = "1.0.0-M5")
public Builder withCategoryScores(CategoryScores categoryScores) {
this.categoryScores = categoryScores;
return this;