Skip to content

Commit 34db84e

Browse files
committed
Refactor integration tests: rename JavaAIAgentBuilderIntegrationTest for clearer context, update utility code to simplify Function usage, and adjust file organization for consistency
Signed-off-by: Sergey Karpov <[email protected]>
1 parent 73d5b48 commit 34db84e

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

integration-tests/src/jvmTest/java/ai/koog/integration/tests/agent/JavaAIAgentBuilderIntegrationTest.java renamed to integration-tests/src/jvmTest/java/ai/koog/integration/tests/agent/JavaAIAgentBuilderInteropIntegrationTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
/**
2626
* Integration tests for AIAgent builder with basic configuration, tools, and event handlers.
2727
*/
28-
public class JavaAIAgentBuilderIntegrationTest extends KoogJavaTestBase {
28+
public class JavaAIAgentBuilderInteropIntegrationTest extends KoogJavaTestBase {
2929

3030
private final List<AutoCloseable> resourcesToClose = new ArrayList<>();
3131

integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/utils/JavaInteropUtils.kt renamed to integration-tests/src/jvmTest/java/ai/koog/integration/tests/utils/JavaInteropUtils.kt

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ import ai.koog.prompt.llm.LLModel
2323
import ai.koog.prompt.message.Message
2424
import kotlinx.coroutines.runBlocking
2525
import kotlinx.serialization.json.Json
26+
import java.util.function.Function
2627

2728
/**
2829
*
@@ -139,13 +140,13 @@ object JavaInteropUtils {
139140
@JvmStatic
140141
fun <T> llmWriteSession(
141142
context: AIAgentFunctionalContext,
142-
action: java.util.function.Function<AIAgentLLMWriteSession, T>
143+
action: Function<AIAgentLLMWriteSession, T>
143144
): T = context.llm.writeSession(action)
144145

145146
@JvmStatic
146147
fun <T> llmReadSession(
147148
context: AIAgentFunctionalContext,
148-
action: java.util.function.Function<AIAgentLLMReadSession, T>
149+
action: Function<AIAgentLLMReadSession, T>
149150
): T = context.llm.readSession(action)
150151

151152
@JvmStatic

integration-tests/src/jvmTest/java/ai/koog/integration/tests/agent/AIAgentBuilderIntegrationTest.kt renamed to integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/agent/AIAgentBuilderIntegrationTest.kt

File renamed without changes.

0 commit comments

Comments
 (0)