Skip to content

Add capabilities integration tests #564

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 1 commit into
base: develop
Choose a base branch
from
Draft

Add capabilities integration tests #564

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions ...tegration/tests/AIAgentIntegrationTest.kt → ...ion/tests/agent/AIAgentIntegrationTest.kt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package ai.koog.integration.tests
package ai.koog.integration.tests.agent

import ai.koog.agents.core.agent.AIAgent
import ai.koog.agents.core.agent.ToolCalls
Expand Down Expand Up @@ -57,6 +57,7 @@ import org.junit.jupiter.params.provider.Arguments
import org.junit.jupiter.params.provider.MethodSource
import java.nio.file.Path
import java.nio.file.Paths
import java.util.Base64
import java.util.stream.Stream
import kotlin.io.path.readBytes
import kotlin.reflect.typeOf
Expand Down Expand Up @@ -423,7 +424,7 @@ class AIAgentIntegrationTest {
val imageFile = testResourcesDir.resolve("test.png")

val imageBytes = imageFile.readBytes()
val base64Image = java.util.Base64.getEncoder().encodeToString(imageBytes)
val base64Image = Base64.getEncoder().encodeToString(imageBytes)

val promptWithImage = """
I'm sending you an image encoded in base64 format.
Expand Down
7 changes: 4 additions & 3 deletions ...ests/AIAgentMultipleLLMIntegrationTest.kt → ...gent/AIAgentMultipleLLMIntegrationTest.kt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package ai.koog.integration.tests
package ai.koog.integration.tests.agent

import ai.koog.agents.core.agent.AIAgent
import ai.koog.agents.core.agent.AIAgentException
Expand All @@ -22,7 +22,7 @@ import ai.koog.agents.core.tools.ToolResult
import ai.koog.agents.features.eventHandler.feature.EventHandler
import ai.koog.agents.features.eventHandler.feature.EventHandlerConfig
import ai.koog.agents.features.tracing.feature.Tracing
import ai.koog.integration.tests.ReportingLLMLLMClient.Event
import ai.koog.integration.tests.agent.ReportingLLMLLMClient.Event
import ai.koog.integration.tests.utils.Models
import ai.koog.integration.tests.utils.RetryUtils.withRetry
import ai.koog.integration.tests.utils.TestLogPrinter
Expand Down Expand Up @@ -60,6 +60,7 @@ import org.junit.jupiter.params.ParameterizedTest
import org.junit.jupiter.params.provider.Arguments
import org.junit.jupiter.params.provider.MethodSource
import java.io.File
import java.util.Base64
import java.util.stream.Stream
import kotlin.test.Ignore
import kotlin.test.assertContains
Expand Down Expand Up @@ -841,7 +842,7 @@ class AIAgentMultipleLLMIntegrationTest {
assertTrue(imageFile.exists(), "Image test file should exist")

val imageBytes = imageFile.readBytes()
val base64Image = java.util.Base64.getEncoder().encodeToString(imageBytes)
val base64Image = Base64.getEncoder().encodeToString(imageBytes)

withRetry {
val agent = when (model.provider) {
Expand Down
5 changes: 4 additions & 1 deletion ...ation/tests/OllamaAgentIntegrationTest.kt → ...tests/agent/OllamaAgentIntegrationTest.kt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package ai.koog.integration.tests
package ai.koog.integration.tests.agent

import ai.koog.agents.core.agent.AIAgent
import ai.koog.agents.core.agent.config.AIAgentConfig
Expand All @@ -13,6 +13,9 @@ import ai.koog.agents.core.dsl.extension.onAssistantMessage
import ai.koog.agents.core.dsl.extension.onToolCall
import ai.koog.agents.core.tools.ToolRegistry
import ai.koog.agents.features.eventHandler.feature.EventHandler
import ai.koog.integration.tests.InjectOllamaTestFixture
import ai.koog.integration.tests.OllamaTestFixture
import ai.koog.integration.tests.OllamaTestFixtureExtension
import ai.koog.integration.tests.tools.AnswerVerificationTool
import ai.koog.integration.tests.tools.GenericParameterTool
import ai.koog.integration.tests.tools.GeographyQueryTool
Expand Down
5 changes: 4 additions & 1 deletion ...tests/OllamaSimpleAgentIntegrationTest.kt → ...agent/OllamaSimpleAgentIntegrationTest.kt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,10 +1,13 @@
package ai.koog.integration.tests
package ai.koog.integration.tests.agent

import ai.koog.agents.core.agent.AIAgent
import ai.koog.agents.core.tools.ToolRegistry
import ai.koog.agents.ext.tool.SayToUser
import ai.koog.agents.features.eventHandler.feature.EventHandler
import ai.koog.agents.features.eventHandler.feature.EventHandlerConfig
import ai.koog.integration.tests.InjectOllamaTestFixture
import ai.koog.integration.tests.OllamaTestFixture
import ai.koog.integration.tests.OllamaTestFixtureExtension
import ai.koog.integration.tests.utils.annotations.Retry
import ai.koog.integration.tests.utils.annotations.RetryExtension
import kotlinx.coroutines.test.runTest
Expand Down
Loading
Loading