fix: prompt 값 수정

9oormthon-univ · Dec 7, 2024 · 97ed5ea · 97ed5ea
1 parent ad207cd
commit 97ed5ea
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/main/java/com/groom/orbit/ai/app/openai/OpenAiService.java b/src/main/java/com/groom/orbit/ai/app/openai/OpenAiService.java
@@ -2,6 +2,8 @@
 
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
+import java.util.stream.Stream;
 
 import org.springframework.ai.chat.model.ChatModel;
 import org.springframework.ai.chat.model.ChatResponse;
@@ -88,7 +90,13 @@ public RecommendGoalListResponseDto recommendGoal(Long memberId) {
                 "goalList",
                 String.join(
                     PARAMETER_LIST_DELIMITER,
-                    othersVector.stream().flatMap(vector -> vector.goals().stream()).toList()),
+                    othersVector.stream()
+                        .filter(Objects::nonNull)
+                        .flatMap(
+                            vector ->
+                                vector.goals() != null ? vector.goals().stream() : Stream.empty())
+                        .filter(Objects::nonNull)
+                        .toList()),
                 "format",
                 format));
     return converter.convert(response);