Add getContent() method to ChatResponse

spring-projects · May 19, 2024 · bae3e51 · bae3e51
1 parent 6f4ab64
commit bae3e51
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/spring-ai-core/src/main/java/org/springframework/ai/chat/ChatResponse.java b/spring-ai-core/src/main/java/org/springframework/ai/chat/ChatResponse.java
@@ -78,6 +78,15 @@ public Generation getResult() {
 		return this.generations.get(0);
 	}
 
+	/**
+	 * Convenience method equivalent to {@code getResult().getOutput().getContent()}.
+	 * @return The content from the output of the first {@link Generation} in the
+	 * generations list.
+	 */
+	public String getContent() {
+		return getResult().getOutput().getContent();
+	}
+
 	/**
 	 * @return Returns {@link ChatResponseMetadata} containing information about the use
 	 * of the AI provider's API.