Skip to content

Commit 5136548

Browse files
dstibbekpavlov
authored andcommitted
Updated tests and updated api (#158)
1 parent 7a2ae41 commit 5136548

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

kotlin-sdk-core/src/commonTest/kotlin/io/modelcontextprotocol/kotlin/sdk/TypesTest.kt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ class TypesTest {
194194
val decoded = McpJson.decodeFromString<ContentBlock>(json)
195195

196196
assertIs<ResourceLink>(decoded)
197-
with(resourceLink) {
197+
with(decoded) {
198198
assertEquals("application/pdf", mimeType)
199199
assertEquals("This pdf is meant to be a resource link test", description)
200200
assertEquals("file01", name)

kotlin-sdk-test/src/jvmTest/kotlin/io/modelcontextprotocol/kotlin/sdk/integration/kotlin/PromptIntegrationTest.kt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
11
package io.modelcontextprotocol.kotlin.sdk.integration.kotlin
22

3+
import io.modelcontextprotocol.kotlin.sdk.ContentBlock
34
import io.modelcontextprotocol.kotlin.sdk.GetPromptRequest
45
import io.modelcontextprotocol.kotlin.sdk.GetPromptResult
56
import io.modelcontextprotocol.kotlin.sdk.ImageContent
67
import io.modelcontextprotocol.kotlin.sdk.PromptArgument
78
import io.modelcontextprotocol.kotlin.sdk.PromptMessage
8-
import io.modelcontextprotocol.kotlin.sdk.PromptMessageContent
99
import io.modelcontextprotocol.kotlin.sdk.Role
1010
import io.modelcontextprotocol.kotlin.sdk.ServerCapabilities
1111
import io.modelcontextprotocol.kotlin.sdk.TextContent
@@ -93,7 +93,7 @@ class PromptIntegrationTest : KotlinTestBase() {
9393
),
9494
)
9595

96-
val assistantContents = mutableListOf<PromptMessageContent>()
96+
val assistantContents = mutableListOf<ContentBlock>()
9797
assistantContents.add(TextContent(text = "I'd be happy to discuss $topic with you."))
9898

9999
if (includeImage) {

kotlin-sdk-test/src/jvmTest/kotlin/io/modelcontextprotocol/kotlin/sdk/integration/kotlin/ToolIntegrationTest.kt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@ package io.modelcontextprotocol.kotlin.sdk.integration.kotlin
33
import io.kotest.assertions.json.shouldEqualJson
44
import io.modelcontextprotocol.kotlin.sdk.CallToolResult
55
import io.modelcontextprotocol.kotlin.sdk.CallToolResultBase
6+
import io.modelcontextprotocol.kotlin.sdk.ContentBlock
67
import io.modelcontextprotocol.kotlin.sdk.ImageContent
7-
import io.modelcontextprotocol.kotlin.sdk.PromptMessageContent
88
import io.modelcontextprotocol.kotlin.sdk.ServerCapabilities
99
import io.modelcontextprotocol.kotlin.sdk.TextContent
1010
import io.modelcontextprotocol.kotlin.sdk.Tool
@@ -276,7 +276,7 @@ class ToolIntegrationTest : KotlinTestBase() {
276276
val text = (request.arguments["text"] as? JsonPrimitive)?.content ?: "Default text"
277277
val includeImage = (request.arguments["includeImage"] as? JsonPrimitive)?.content?.toBoolean() ?: true
278278

279-
val content = mutableListOf<PromptMessageContent>(
279+
val content = mutableListOf<ContentBlock>(
280280
TextContent(text = "Text content: $text"),
281281
)
282282

0 commit comments

Comments
 (0)