Skip to content

Commit bb23a48

Browse files
committed
fix: address PR #4 review feedback
1 parent 6fcb5a3 commit bb23a48

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

src/main/java/com/steve/ai/llm/GroqClient.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,8 @@ public String sendRequest(String systemPrompt, String userPrompt) {
5353
messages.add(userMessage);
5454

5555
requestBody.add("messages", messages);
56-
requestBody.addProperty("max_tokens", 500); // Keep it short for speed
57-
requestBody.addProperty("temperature", 0.7);
56+
requestBody.addProperty("max_tokens", SteveConfig.MAX_TOKENS.get());
57+
requestBody.addProperty("temperature", SteveConfig.TEMPERATURE.get());
5858

5959
HttpRequest request = HttpRequest.newBuilder()
6060
.uri(URI.create(GROQ_API_URL))

src/main/java/com/steve/ai/llm/TaskPlanner.java

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,12 @@ public TaskPlanner() {
7070
maxTokens,
7171
temperature
7272
);
73-
AsyncLLMClient baseDeepSeek = new AsyncDeepSeekClient();
73+
AsyncLLMClient baseDeepSeek = new AsyncDeepSeekClient(
74+
SteveConfig.DEEPSEEK_API_KEY.get(),
75+
SteveConfig.DEEPSEEK_MODEL.get(),
76+
maxTokens,
77+
temperature
78+
);
7479

7580
// Wrap with resilience patterns (caching, retries, circuit breaker)
7681
tempAsyncOpenAI = new ResilientLLMClient(baseOpenAI, tempCache, tempFallback);
@@ -171,6 +176,8 @@ public CompletableFuture<ResponseParser.ParsedResponse> planTasksAsync(SteveEnti
171176
// Build params map with provider-specific model
172177
String modelForProvider = switch (provider) {
173178
case "deepseek" -> SteveConfig.DEEPSEEK_MODEL.get();
179+
case "groq" -> SteveConfig.GROQ_MODEL.get();
180+
case "gemini" -> SteveConfig.GEMINI_MODEL.get();
174181
default -> SteveConfig.OPENAI_MODEL.get();
175182
};
176183

0 commit comments

Comments
 (0)