minor fix

LLMAgentBuilder · Oct 5, 2024 · 99085ce · 99085ce
1 parent 7f238ce
commit 99085ce
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/bootstrap/src/main/kotlin/io/github/llmagentbuilder/bootstrap/AgentBootstrap.kt b/bootstrap/src/main/kotlin/io/github/llmagentbuilder/bootstrap/AgentBootstrap.kt
@@ -50,7 +50,9 @@ object AgentBootstrap {
                 agentToolsProvider,
             )
         val observationRegistry = ObservationRegistry.create()
-        OpenTelemetryPlugin().install(agentConfig, observationRegistry)
+        if (agentConfig.observation?.enabled == true) {
+            OpenTelemetryPlugin().install(agentConfig, observationRegistry)
+        }
         val llmConfigs = agentConfig.llm
         val chatModel = ServiceLoader.load(ChatModelProvider::class.java)
             .stream()

diff --git a/core/src/main/kotlin/io/github/llmagentbuilder/core/tool/AgentToolsProvider.kt b/core/src/main/kotlin/io/github/llmagentbuilder/core/tool/AgentToolsProvider.kt
@@ -10,6 +10,7 @@ import io.github.llmagentbuilder.core.observation.DefaultAgentToolExecutionObser
 import io.micrometer.observation.ObservationRegistry
 import org.slf4j.LoggerFactory
 import org.springframework.ai.model.function.FunctionCallback
+import org.springframework.ai.model.function.FunctionCallbackContext
 import org.springframework.ai.model.function.FunctionCallbackWrapper
 import org.springframework.core.GenericTypeResolver
 import java.util.*
@@ -35,7 +36,7 @@ class AgentToolWrappersProvider(
             InstrumentedFunctionCallbackWrapper(
                 FunctionCallbackWrapper.builder(tool)
                     .withName(tool.name())
-                    .withSchemaType(FunctionCallbackWrapper.Builder.SchemaType.JSON_SCHEMA)
+                    .withSchemaType(FunctionCallbackContext.SchemaType.JSON_SCHEMA)
                     .withDescription(tool.description())
                     .withInputType(
                         types?.get(0)