Skip to content

Commit 36a3123

Browse files
committed
code refactoring
1 parent 7074d00 commit 36a3123

File tree

3 files changed

+2
-11
lines changed

3 files changed

+2
-11
lines changed
Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,9 @@
11
package com.javaaidev.agent;
22

3-
import com.javaaidev.chatagent.model.ChatAgentRequest;
4-
import com.javaaidev.chatagent.springai.ModelAdapter;
5-
import java.util.List;
6-
import org.springframework.ai.chat.messages.Message;
7-
83
public abstract class AbstractChatAgentController {
94

105
public static final String SYSTEM_TEXT = """
116
You are a chef who is proficient in various cuisines. Please answer users' questions about cooking.
127
For other unrelated inputs, simply tell the user that you don't know.
138
""";
14-
15-
protected List<Message> chatRequestToMessages(ChatAgentRequest request) {
16-
return ModelAdapter.fromRequest(request);
17-
}
189
}

src/main/java/com/javaaidev/agent/ChatAgentController.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ public Flux<ServerSentEvent<ChatAgentResponse>> chat(@RequestBody ChatAgentReque
2727
if (request == null) {
2828
return Flux.empty();
2929
}
30-
var messages = chatRequestToMessages(request);
30+
var messages = ModelAdapter.fromRequest(request);
3131
var chatResponse = chatClient.prompt().system(SYSTEM_TEXT)
3232
.messages(messages.toArray(new Message[0]))
3333
.call()

src/main/java/com/javaaidev/agent/ChatAgentStreamingController.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ public ChatAgentStreamingController(ChatClient.Builder builder) {
2626
@PostMapping(produces = MediaType.TEXT_EVENT_STREAM_VALUE)
2727
public Flux<ServerSentEvent<ChatAgentResponse>> chatStreaming(
2828
@RequestBody ChatAgentRequest request) {
29-
var messages = chatRequestToMessages(request);
29+
var messages = ModelAdapter.fromRequest(request);
3030
return ModelAdapter.toStreamingResponse(
3131
chatClient.prompt()
3232
.system(SYSTEM_TEXT)

0 commit comments

Comments
 (0)