Skip to content

Commit d1c62f4

Browse files
committed
Remove redundant dependencies and JSON schema generator registrations in MistralAI client
1 parent b083fc0 commit d1c62f4

File tree

2 files changed

+2
-10
lines changed

2 files changed

+2
-10
lines changed

prompt/prompt-executor/prompt-executor-clients/prompt-executor-mistralai-client/build.gradle.kts

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,6 @@ kotlin {
1313
commonMain {
1414
dependencies {
1515
api(project(":prompt:prompt-executor:prompt-executor-clients:prompt-executor-openai-client-base"))
16-
api(project(":prompt:prompt-structure"))
17-
api(project(":prompt:prompt-executor:prompt-executor-clients:prompt-executor-openai-client"))
1816
implementation(libs.oshai.kotlin.logging)
1917
}
2018
}

prompt/prompt-executor/prompt-executor-clients/prompt-executor-mistralai-client/src/commonMain/kotlin/ai/koog/prompt/executor/clients/mistralai/MistralAILLMClient.kt

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -25,18 +25,13 @@ import ai.koog.prompt.executor.clients.openai.base.models.OpenAIStaticContent
2525
import ai.koog.prompt.executor.clients.openai.base.models.OpenAITool
2626
import ai.koog.prompt.executor.clients.openai.base.models.OpenAIToolChoice
2727
import ai.koog.prompt.executor.clients.openai.base.models.OpenAIUsage
28-
import ai.koog.prompt.executor.clients.openai.structure.OpenAIBasicJsonSchemaGenerator
29-
import ai.koog.prompt.executor.clients.openai.structure.OpenAIStandardJsonSchemaGenerator
3028
import ai.koog.prompt.executor.model.LLMChoice
3129
import ai.koog.prompt.llm.LLMCapability
3230
import ai.koog.prompt.llm.LLMProvider
3331
import ai.koog.prompt.llm.LLModel
3432
import ai.koog.prompt.message.Message
3533
import ai.koog.prompt.params.LLMParams
3634
import ai.koog.prompt.streaming.StreamFrameFlowBuilder
37-
import ai.koog.prompt.structure.RegisteredBasicJsonSchemaGenerators
38-
import ai.koog.prompt.structure.RegisteredStandardJsonSchemaGenerators
39-
import ai.koog.prompt.structure.annotations.InternalStructuredOutputApi
4035
import io.github.oshai.kotlinlogging.KotlinLogging
4136
import io.ktor.client.HttpClient
4237
import kotlinx.datetime.Clock
@@ -77,13 +72,12 @@ public open class MistralAILLMClient(
7772
),
7873
LLMEmbeddingProvider {
7974

80-
@OptIn(InternalStructuredOutputApi::class)
8175
private companion object {
8276
private val staticLogger = KotlinLogging.logger { }
8377

8478
init {
85-
RegisteredBasicJsonSchemaGenerators[LLMProvider.MistralAI] = OpenAIBasicJsonSchemaGenerator
86-
RegisteredStandardJsonSchemaGenerators[LLMProvider.MistralAI] = OpenAIStandardJsonSchemaGenerator
79+
// On class load register custom OpenAI JSON schema generators for structured output.
80+
registerOpenAIJsonSchemaGenerators(LLMProvider.DeepSeek)
8781
}
8882
}
8983

0 commit comments

Comments
 (0)