mirror of https://github.com/dapr/java-sdk.git
fxi things
Signed-off-by: siri-varma <siri.varma@outlook.com>
This commit is contained in:
siri-varma
parent
3c79ce8baf
commit
57fad34c1a
|
|
@ -0,0 +1,386 @@
|
||||||
|
/*
|
||||||
|
* Copyright 2025 The Dapr Authors
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package io.dapr.it.testcontainers.conversations;
|
||||||
|
|
||||||
|
import io.dapr.client.DaprPreviewClient;
|
||||||
|
import io.dapr.client.domain.AssistantMessage;
|
||||||
|
import io.dapr.client.domain.ConversationInputAlpha2;
|
||||||
|
import io.dapr.client.domain.ConversationMessage;
|
||||||
|
import io.dapr.client.domain.ConversationMessageContent;
|
||||||
|
import io.dapr.client.domain.ConversationRequestAlpha2;
|
||||||
|
import io.dapr.client.domain.ConversationResponseAlpha2;
|
||||||
|
import io.dapr.client.domain.ConversationResultAlpha2;
|
||||||
|
import io.dapr.client.domain.ConversationResultChoices;
|
||||||
|
import io.dapr.client.domain.ConversationToolCalls;
|
||||||
|
import io.dapr.client.domain.ConversationToolCallsOfFunction;
|
||||||
|
import io.dapr.client.domain.ConversationTools;
|
||||||
|
import io.dapr.client.domain.ConversationToolsFunction;
|
||||||
|
import io.dapr.client.domain.DeveloperMessage;
|
||||||
|
import io.dapr.client.domain.SystemMessage;
|
||||||
|
import io.dapr.client.domain.ToolMessage;
|
||||||
|
import io.dapr.client.domain.UserMessage;
|
||||||
|
import io.dapr.it.testcontainers.DaprPreviewClientConfiguration;
|
||||||
|
import io.dapr.testcontainers.Component;
|
||||||
|
import io.dapr.testcontainers.DaprContainer;
|
||||||
|
import io.dapr.testcontainers.DaprLogLevel;
|
||||||
|
import org.junit.jupiter.api.BeforeEach;
|
||||||
|
import org.junit.jupiter.api.Tag;
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.boot.test.context.SpringBootTest;
|
||||||
|
import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
|
||||||
|
import org.springframework.test.context.DynamicPropertyRegistry;
|
||||||
|
import org.springframework.test.context.DynamicPropertySource;
|
||||||
|
import org.testcontainers.containers.Network;
|
||||||
|
import org.testcontainers.junit.jupiter.Container;
|
||||||
|
import org.testcontainers.junit.jupiter.Testcontainers;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Random;
|
||||||
|
|
||||||
|
import static io.dapr.it.testcontainers.ContainerConstants.DAPR_RUNTIME_IMAGE_TAG;
|
||||||
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
||||||
|
import static org.junit.jupiter.api.Assertions.assertNotNull;
|
||||||
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
||||||
|
|
||||||
|
@SpringBootTest(
|
||||||
|
webEnvironment = WebEnvironment.RANDOM_PORT,
|
||||||
|
classes = {
|
||||||
|
DaprPreviewClientConfiguration.class,
|
||||||
|
TestConversationApplication.class
|
||||||
|
}
|
||||||
|
)
|
||||||
|
@Testcontainers
|
||||||
|
@Tag("testcontainers")
|
||||||
|
public class DaprConversationAlpha2IT {
|
||||||
|
|
||||||
|
private static final Network DAPR_NETWORK = Network.newNetwork();
|
||||||
|
private static final Random RANDOM = new Random();
|
||||||
|
private static final int PORT = RANDOM.nextInt(1000) + 8000;
|
||||||
|
|
||||||
|
@Container
|
||||||
|
private static final DaprContainer DAPR_CONTAINER = new DaprContainer(DAPR_RUNTIME_IMAGE_TAG)
|
||||||
|
.withAppName("conversation-alpha2-dapr-app")
|
||||||
|
.withComponent(new Component("echo", "conversation.echo", "v1", new HashMap<>()))
|
||||||
|
.withNetwork(DAPR_NETWORK)
|
||||||
|
.withDaprLogLevel(DaprLogLevel.DEBUG)
|
||||||
|
.withLogConsumer(outputFrame -> System.out.println(outputFrame.getUtf8String()))
|
||||||
|
.withAppChannelAddress("host.testcontainers.internal")
|
||||||
|
.withAppPort(PORT);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Expose the Dapr port to the host.
|
||||||
|
*
|
||||||
|
* @param registry the dynamic property registry
|
||||||
|
*/
|
||||||
|
@DynamicPropertySource
|
||||||
|
static void daprProperties(DynamicPropertyRegistry registry) {
|
||||||
|
registry.add("dapr.http.endpoint", DAPR_CONTAINER::getHttpEndpoint);
|
||||||
|
registry.add("dapr.grpc.endpoint", DAPR_CONTAINER::getGrpcEndpoint);
|
||||||
|
registry.add("server.port", () -> PORT);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private DaprPreviewClient daprPreviewClient;
|
||||||
|
|
||||||
|
@BeforeEach
|
||||||
|
public void setUp() {
|
||||||
|
org.testcontainers.Testcontainers.exposeHostPorts(PORT);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testConverseAlpha2WithUserMessage() {
|
||||||
|
// Create a user message
|
||||||
|
UserMessage userMessage = new UserMessage(List.of(new ConversationMessageContent("Hello, how are you?")));
|
||||||
|
userMessage.setName("TestUser");
|
||||||
|
|
||||||
|
// Create input with the message
|
||||||
|
ConversationInputAlpha2 input = new ConversationInputAlpha2(List.of(userMessage));
|
||||||
|
|
||||||
|
ConversationRequestAlpha2 request = new ConversationRequestAlpha2("echo", List.of(input));
|
||||||
|
|
||||||
|
ConversationResponseAlpha2 response = daprPreviewClient.converseAlpha2(request).block();
|
||||||
|
|
||||||
|
assertNotNull(response);
|
||||||
|
assertNotNull(response.getOutputs());
|
||||||
|
assertEquals(1, response.getOutputs().size());
|
||||||
|
|
||||||
|
ConversationResultAlpha2 result = response.getOutputs().get(0);
|
||||||
|
assertNotNull(result.getChoices());
|
||||||
|
assertTrue(result.getChoices().size() > 0);
|
||||||
|
|
||||||
|
ConversationResultChoices choice = result.getChoices().get(0);
|
||||||
|
assertNotNull(choice.getMessage());
|
||||||
|
assertNotNull(choice.getMessage().getContent());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testConverseAlpha2WithAllMessageTypes() {
|
||||||
|
List<ConversationMessage> messages = new ArrayList<>();
|
||||||
|
|
||||||
|
// System message
|
||||||
|
SystemMessage systemMsg = new SystemMessage(List.of(new ConversationMessageContent("You are a helpful assistant.")));
|
||||||
|
systemMsg.setName("system");
|
||||||
|
messages.add(systemMsg);
|
||||||
|
|
||||||
|
// User message
|
||||||
|
UserMessage userMsg = new UserMessage(List.of(new ConversationMessageContent("Hello!")));
|
||||||
|
userMsg.setName("user");
|
||||||
|
messages.add(userMsg);
|
||||||
|
|
||||||
|
// Assistant message
|
||||||
|
AssistantMessage assistantMsg = new AssistantMessage(List.of(new ConversationMessageContent("Hi there!")),
|
||||||
|
List.of(new ConversationToolCalls(
|
||||||
|
new ConversationToolCallsOfFunction("get_weather", "{\"location\": \"New York\"}"))));
|
||||||
|
assistantMsg.setName("assistant");
|
||||||
|
messages.add(assistantMsg);
|
||||||
|
|
||||||
|
// Tool message
|
||||||
|
ToolMessage toolMsg = new ToolMessage(List.of(new ConversationMessageContent("Weather data: 72F")));
|
||||||
|
toolMsg.setName("tool");
|
||||||
|
messages.add(toolMsg);
|
||||||
|
|
||||||
|
// Developer message
|
||||||
|
DeveloperMessage devMsg = new DeveloperMessage(List.of(new ConversationMessageContent("Debug info")));
|
||||||
|
devMsg.setName("developer");
|
||||||
|
messages.add(devMsg);
|
||||||
|
|
||||||
|
ConversationInputAlpha2 input = new ConversationInputAlpha2(messages);
|
||||||
|
ConversationRequestAlpha2 request = new ConversationRequestAlpha2("echo", List.of(input));
|
||||||
|
|
||||||
|
ConversationResponseAlpha2 response = daprPreviewClient.converseAlpha2(request).block();
|
||||||
|
|
||||||
|
assertNotNull(response);
|
||||||
|
assertNotNull(response.getOutputs());
|
||||||
|
assertTrue(response.getOutputs().size() > 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testConverseAlpha2WithScrubPII() {
|
||||||
|
// Create a user message with PII
|
||||||
|
UserMessage userMessage = new UserMessage(List.of(new ConversationMessageContent("My email is test@example.com and phone is +1234567890")));
|
||||||
|
|
||||||
|
ConversationInputAlpha2 input = new ConversationInputAlpha2(List.of(userMessage));
|
||||||
|
input.setScrubPii(true);
|
||||||
|
|
||||||
|
ConversationRequestAlpha2 request = new ConversationRequestAlpha2("echo", List.of(input));
|
||||||
|
request.setScrubPii(true);
|
||||||
|
|
||||||
|
ConversationResponseAlpha2 response = daprPreviewClient.converseAlpha2(request).block();
|
||||||
|
|
||||||
|
assertNotNull(response);
|
||||||
|
assertNotNull(response.getOutputs());
|
||||||
|
assertTrue(response.getOutputs().size() > 0);
|
||||||
|
|
||||||
|
// Verify response structure (actual PII scrubbing depends on echo component implementation)
|
||||||
|
ConversationResultChoices choice = response.getOutputs().get(0).getChoices().get(0);
|
||||||
|
assertNotNull(choice.getMessage());
|
||||||
|
assertNotNull(choice.getMessage().getContent());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testConverseAlpha2WithTools() {
|
||||||
|
// Create a tool function
|
||||||
|
Map<String, Object> parameters = new HashMap<>();
|
||||||
|
parameters.put("location", "string");
|
||||||
|
parameters.put("unit", "celsius");
|
||||||
|
ConversationToolsFunction function = new ConversationToolsFunction("get_weather", parameters);
|
||||||
|
function.setDescription("Get current weather information");
|
||||||
|
|
||||||
|
ConversationTools tool = new ConversationTools(function);
|
||||||
|
|
||||||
|
// Create user message
|
||||||
|
UserMessage userMessage = new UserMessage(List.of(new ConversationMessageContent("What's the weather like?")));
|
||||||
|
|
||||||
|
ConversationInputAlpha2 input = new ConversationInputAlpha2(List.of(userMessage));
|
||||||
|
|
||||||
|
ConversationRequestAlpha2 request = new ConversationRequestAlpha2("echo", List.of(input));
|
||||||
|
request.setTools(List.of(tool));
|
||||||
|
request.setToolChoice("auto");
|
||||||
|
|
||||||
|
ConversationResponseAlpha2 response = daprPreviewClient.converseAlpha2(request).block();
|
||||||
|
|
||||||
|
assertNotNull(response);
|
||||||
|
assertNotNull(response.getOutputs());
|
||||||
|
assertTrue(response.getOutputs().size() > 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testConverseAlpha2WithMetadataAndParameters() {
|
||||||
|
UserMessage userMessage = new UserMessage(List.of(new ConversationMessageContent("Hello world")));
|
||||||
|
|
||||||
|
ConversationInputAlpha2 input = new ConversationInputAlpha2(List.of(userMessage));
|
||||||
|
|
||||||
|
// Set metadata and parameters
|
||||||
|
Map<String, String> metadata = new HashMap<>();
|
||||||
|
metadata.put("request-id", "test-123");
|
||||||
|
metadata.put("source", "integration-test");
|
||||||
|
|
||||||
|
Map<String, Object> parameters = new HashMap<>();
|
||||||
|
parameters.put("max_tokens", "1000");
|
||||||
|
parameters.put("temperature", "0.7");
|
||||||
|
|
||||||
|
ConversationRequestAlpha2 request = new ConversationRequestAlpha2("echo", List.of(input));
|
||||||
|
request.setContextId("test-context-123");
|
||||||
|
request.setTemperature(0.8);
|
||||||
|
request.setMetadata(metadata);
|
||||||
|
request.setParameters(parameters);
|
||||||
|
|
||||||
|
ConversationResponseAlpha2 response = daprPreviewClient.converseAlpha2(request).block();
|
||||||
|
|
||||||
|
assertNotNull(response);
|
||||||
|
assertNotNull(response.getOutputs());
|
||||||
|
assertTrue(response.getOutputs().size() > 0);
|
||||||
|
|
||||||
|
// Verify context ID is handled properly
|
||||||
|
// Note: actual context ID behavior depends on echo component implementation
|
||||||
|
assertNotNull(response.getContextId());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testConverseAlpha2WithAssistantToolCalls() {
|
||||||
|
// Create a tool call
|
||||||
|
ConversationToolCallsOfFunction toolFunction =
|
||||||
|
new ConversationToolCallsOfFunction("get_weather", "{\"location\": \"New York\"}");
|
||||||
|
ConversationToolCalls toolCall = new ConversationToolCalls(toolFunction);
|
||||||
|
toolCall.setId("call_123");
|
||||||
|
|
||||||
|
// Create assistant message with tool calls
|
||||||
|
AssistantMessage assistantMsg = new AssistantMessage(List.of(new ConversationMessageContent("Hi there!")),
|
||||||
|
List.of(new ConversationToolCalls(
|
||||||
|
new ConversationToolCallsOfFunction("get_weather", "{\"location\": \"New York\"}")))); // Note: Current implementation doesn't support setting tool calls in constructor
|
||||||
|
// This tests the structure and ensures no errors occur
|
||||||
|
|
||||||
|
ConversationInputAlpha2 input = new ConversationInputAlpha2(List.of(assistantMsg));
|
||||||
|
|
||||||
|
ConversationRequestAlpha2 request = new ConversationRequestAlpha2("echo", List.of(input));
|
||||||
|
|
||||||
|
ConversationResponseAlpha2 response = daprPreviewClient.converseAlpha2(request).block();
|
||||||
|
|
||||||
|
assertNotNull(response);
|
||||||
|
assertNotNull(response.getOutputs());
|
||||||
|
assertTrue(response.getOutputs().size() > 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testConverseAlpha2WithComplexScenario() {
|
||||||
|
List<ConversationMessage> messages = new ArrayList<>();
|
||||||
|
|
||||||
|
// System message setting context
|
||||||
|
SystemMessage systemMsg = new SystemMessage(List.of(new ConversationMessageContent("You are a helpful weather assistant.")));
|
||||||
|
systemMsg.setName("WeatherBot");
|
||||||
|
messages.add(systemMsg);
|
||||||
|
|
||||||
|
// User asking for weather
|
||||||
|
UserMessage userMsg = new UserMessage(List.of(new ConversationMessageContent("What's the weather in San Francisco?")));
|
||||||
|
userMsg.setName("User123");
|
||||||
|
messages.add(userMsg);
|
||||||
|
|
||||||
|
// Assistant response
|
||||||
|
AssistantMessage assistantMsg = new AssistantMessage(List.of(new ConversationMessageContent("Hi there!")),
|
||||||
|
List.of(new ConversationToolCalls(
|
||||||
|
new ConversationToolCallsOfFunction("get_weather", "{\"location\": \"New York\"}"))));
|
||||||
|
assistantMsg.setName("WeatherBot");
|
||||||
|
messages.add(assistantMsg);
|
||||||
|
|
||||||
|
// Tool response
|
||||||
|
ToolMessage toolMsg = new ToolMessage(List.of(new ConversationMessageContent("{\"temperature\": \"68F\", \"condition\": \"sunny\"}")));
|
||||||
|
toolMsg.setName("weather_api");
|
||||||
|
messages.add(toolMsg);
|
||||||
|
|
||||||
|
ConversationInputAlpha2 input = new ConversationInputAlpha2(messages);
|
||||||
|
input.setScrubPii(false);
|
||||||
|
|
||||||
|
// Create tools
|
||||||
|
Map<String, Object> functionParams = new HashMap<>();
|
||||||
|
functionParams.put("location", "string");
|
||||||
|
functionParams.put("unit", "fahrenheit");
|
||||||
|
ConversationToolsFunction weatherFunction = new ConversationToolsFunction("get_current_weather",
|
||||||
|
functionParams);
|
||||||
|
weatherFunction.setDescription("Get current weather for a location");
|
||||||
|
|
||||||
|
|
||||||
|
ConversationTools weatherTool = new ConversationTools(weatherFunction);
|
||||||
|
|
||||||
|
// Set up complete request
|
||||||
|
Map<String, String> metadata = new HashMap<>();
|
||||||
|
metadata.put("conversation-type", "weather-query");
|
||||||
|
metadata.put("user-session", "session-456");
|
||||||
|
|
||||||
|
Map<String, Object> parameters = new HashMap<>();
|
||||||
|
parameters.put("max_tokens", "2000");
|
||||||
|
parameters.put("response_format", "json");
|
||||||
|
|
||||||
|
ConversationRequestAlpha2 request = new ConversationRequestAlpha2("echo", List.of(input));
|
||||||
|
request.setContextId("weather-conversation-789");
|
||||||
|
request.setTemperature(0.7);
|
||||||
|
request.setScrubPii(false);
|
||||||
|
request.setTools(List.of(weatherTool));
|
||||||
|
request.setToolChoice("auto");
|
||||||
|
request.setMetadata(metadata);
|
||||||
|
request.setParameters(parameters);
|
||||||
|
|
||||||
|
ConversationResponseAlpha2 response = daprPreviewClient.converseAlpha2(request).block();
|
||||||
|
|
||||||
|
assertNotNull(response);
|
||||||
|
assertNotNull(response.getOutputs());
|
||||||
|
assertTrue(response.getOutputs().size() > 0);
|
||||||
|
|
||||||
|
ConversationResultAlpha2 result = response.getOutputs().get(0);
|
||||||
|
assertNotNull(result.getChoices());
|
||||||
|
assertTrue(result.getChoices().size() > 0);
|
||||||
|
|
||||||
|
ConversationResultChoices choice = result.getChoices().get(0);
|
||||||
|
assertNotNull(choice.getFinishReason());
|
||||||
|
assertTrue(choice.getIndex() >= 0);
|
||||||
|
|
||||||
|
if (choice.getMessage() != null) {
|
||||||
|
assertNotNull(choice.getMessage().getContent());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testConverseAlpha2MultipleInputs() {
|
||||||
|
// Create multiple conversation inputs
|
||||||
|
List<ConversationInputAlpha2> inputs = new ArrayList<>();
|
||||||
|
|
||||||
|
// First input - greeting
|
||||||
|
UserMessage greeting = new UserMessage(List.of(new ConversationMessageContent("Hello!")));
|
||||||
|
ConversationInputAlpha2 input1 = new ConversationInputAlpha2(List.of(greeting));
|
||||||
|
inputs.add(input1);
|
||||||
|
|
||||||
|
// Second input - question
|
||||||
|
UserMessage question = new UserMessage(List.of(new ConversationMessageContent("How are you?")));
|
||||||
|
ConversationInputAlpha2 input2 = new ConversationInputAlpha2(List.of(question));
|
||||||
|
input2.setScrubPii(true);
|
||||||
|
inputs.add(input2);
|
||||||
|
|
||||||
|
ConversationRequestAlpha2 request = new ConversationRequestAlpha2("echo", inputs);
|
||||||
|
|
||||||
|
ConversationResponseAlpha2 response = daprPreviewClient.converseAlpha2(request).block();
|
||||||
|
|
||||||
|
assertNotNull(response);
|
||||||
|
assertNotNull(response.getOutputs());
|
||||||
|
assertTrue(response.getOutputs().size() > 0);
|
||||||
|
|
||||||
|
// Should handle multiple inputs appropriately
|
||||||
|
for (ConversationResultAlpha2 result : response.getOutputs()) {
|
||||||
|
assertNotNull(result.getChoices());
|
||||||
|
assertTrue(result.getChoices().size() > 0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -132,3 +132,4 @@ public class DaprConversationIT {
|
||||||
response.getConversationOutputs().get(1).getResult());
|
response.getConversationOutputs().get(1).getResult());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue