Add additionalInfo field to ResponseMetaInfo (#367)

Author: BLannoo

agents/agents-features/agents-features-event-handler/src/jvmTest/kotlin/ai/koog/agents/features/eventHandler/feature/EventHandlerFeatureTest.kt

@@ -84,9 +84,9 @@ class EventHandlerTest {
             "OnBeforeNode (node: __start__, input: ${agentInput})",
             "OnAfterNode (node: __start__, input: ${agentInput}, output: ${agentInput})",
             "OnBeforeNode (node: test LLM call, input: Test LLM call prompt)",
-            "OnBeforeLLMCall (prompt: [System(content=Test system message, metaInfo=RequestMetaInfo(timestamp=$ts)), User(content=Test user message, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[]), Assistant(content=Test assistant response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null), attachment=null, finishReason=null), User(content=Test LLM call prompt, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[])], tools: [])",
+            "OnBeforeLLMCall (prompt: [System(content=Test system message, metaInfo=RequestMetaInfo(timestamp=$ts)), User(content=Test user message, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[]), Assistant(content=Test assistant response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null, additionalInfo={}), attachment=null, finishReason=null), User(content=Test LLM call prompt, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[])], tools: [])",
             "OnAfterLLMCall (responses: [Assistant: Default test response])",
-            "OnAfterNode (node: test LLM call, input: Test LLM call prompt, output: Assistant(content=Default test response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null), attachment=null, finishReason=null))",
+            "OnAfterNode (node: test LLM call, input: Test LLM call prompt, output: Assistant(content=Default test response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null, additionalInfo={}), attachment=null, finishReason=null))",
             "OnStrategyFinished (strategy: $strategyName, result: $agentResult)",
             "OnAgentFinished (strategy: $strategyName, result: $agentResult)",
         )
@@ -128,9 +128,9 @@ class EventHandlerTest {
             "OnBeforeNode (node: __start__, input: ${agentInput})",
             "OnAfterNode (node: __start__, input: ${agentInput}, output: ${agentInput})",
             "OnBeforeNode (node: test LLM call, input: Test LLM call prompt)",
-            "OnBeforeLLMCall (prompt: [System(content=Test system message, metaInfo=RequestMetaInfo(timestamp=$ts)), User(content=Test user message, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[]), Assistant(content=Test assistant response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null), attachment=null, finishReason=null), User(content=Test LLM call prompt, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[])], tools: [dummy])",
+            "OnBeforeLLMCall (prompt: [System(content=Test system message, metaInfo=RequestMetaInfo(timestamp=$ts)), User(content=Test user message, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[]), Assistant(content=Test assistant response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null, additionalInfo={}), attachment=null, finishReason=null), User(content=Test LLM call prompt, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[])], tools: [dummy])",
             "OnAfterLLMCall (responses: [Assistant: Default test response])",
-            "OnAfterNode (node: test LLM call, input: Test LLM call prompt, output: Assistant(content=Default test response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null), attachment=null, finishReason=null))",
+            "OnAfterNode (node: test LLM call, input: Test LLM call prompt, output: Assistant(content=Default test response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null, additionalInfo={}), attachment=null, finishReason=null))",
             "OnStrategyFinished (strategy: $strategyName, result: $agentResult)",
             "OnAgentFinished (strategy: $strategyName, result: $agentResult)",
         )
@@ -174,13 +174,13 @@ class EventHandlerTest {
             "OnBeforeNode (node: __start__, input: ${agentInput})",
             "OnAfterNode (node: __start__, input: ${agentInput}, output: ${agentInput})",
             "OnBeforeNode (node: test LLM call, input: Test LLM call prompt)",
-            "OnBeforeLLMCall (prompt: [System(content=Test system message, metaInfo=RequestMetaInfo(timestamp=$ts)), User(content=Test user message, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[]), Assistant(content=Test assistant response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null), attachment=null, finishReason=null), User(content=Test LLM call prompt, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[])], tools: [dummy])",
+            "OnBeforeLLMCall (prompt: [System(content=Test system message, metaInfo=RequestMetaInfo(timestamp=$ts)), User(content=Test user message, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[]), Assistant(content=Test assistant response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null, additionalInfo={}), attachment=null, finishReason=null), User(content=Test LLM call prompt, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[])], tools: [dummy])",
             "OnAfterLLMCall (responses: [Assistant: Default test response])",
-            "OnAfterNode (node: test LLM call, input: Test LLM call prompt, output: Assistant(content=Default test response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null), attachment=null, finishReason=null))",
+            "OnAfterNode (node: test LLM call, input: Test LLM call prompt, output: Assistant(content=Default test response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null, additionalInfo={}), attachment=null, finishReason=null))",
             "OnBeforeNode (node: test LLM call with tools, input: Test LLM call with tools prompt)",
-            "OnBeforeLLMCall (prompt: [System(content=Test system message, metaInfo=RequestMetaInfo(timestamp=$ts)), User(content=Test user message, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[]), Assistant(content=Test assistant response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null), attachment=null, finishReason=null), User(content=Test LLM call prompt, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[]), Assistant(content=Default test response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null), attachment=null, finishReason=null), User(content=Test LLM call with tools prompt, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[])], tools: [dummy])",
+            "OnBeforeLLMCall (prompt: [System(content=Test system message, metaInfo=RequestMetaInfo(timestamp=$ts)), User(content=Test user message, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[]), Assistant(content=Test assistant response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null, additionalInfo={}), attachment=null, finishReason=null), User(content=Test LLM call prompt, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[]), Assistant(content=Default test response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null, additionalInfo={}), attachment=null, finishReason=null), User(content=Test LLM call with tools prompt, metaInfo=RequestMetaInfo(timestamp=$ts), attachments=[])], tools: [dummy])",
             "OnAfterLLMCall (responses: [Assistant: Default test response])",
-            "OnAfterNode (node: test LLM call with tools, input: Test LLM call with tools prompt, output: Assistant(content=Default test response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null), attachment=null, finishReason=null))",
+            "OnAfterNode (node: test LLM call with tools, input: Test LLM call with tools prompt, output: Assistant(content=Default test response, metaInfo=ResponseMetaInfo(timestamp=$ts, totalTokensCount=null, inputTokensCount=null, outputTokensCount=null, additionalInfo={}), attachment=null, finishReason=null))",
             "OnStrategyFinished (strategy: $strategyName, result: $agentResult)",
             "OnAgentFinished (strategy: $strategyName, result: $agentResult)",
         )

prompt/prompt-model/src/commonMain/kotlin/ai/koog/prompt/message/Message.kt

@@ -246,6 +246,8 @@ public data class RequestMetaInfo(
  * @property totalTokensCount The total number of tokens involved in the response, including both input and output tokens, or null if not available.
  * @property inputTokensCount The number of tokens used in the input, or null if not available.
  * @property outputTokensCount The number of tokens generated in the output, or null if not available.
+ * @property additionalInfo Additional metadata as a map of string keys to string values.
+ *                          This can be used to store custom metadata that doesn't fit into the standard fields.
  * @property timestamp The timestamp indicating when the response was created.
  * Defaults to the current system time if not explicitly set.
  */
@@ -255,6 +257,7 @@ public data class ResponseMetaInfo(
     public val totalTokensCount: Int? = null,
     public val inputTokensCount: Int? = null,
     public val outputTokensCount: Int? = null,
+    public val additionalInfo: Map<String, String> = emptyMap(),
 ) : MessageMetaInfo {
     /**
      * Companion object for the ResponseMetaInfo class.
@@ -272,8 +275,9 @@ public data class ResponseMetaInfo(
             clock: Clock,
             totalTokensCount: Int? = null,
             inputTokensCount: Int? = null,
-            outputTokensCount: Int? = null
+            outputTokensCount: Int? = null,
+            additionalInfo: Map<String, String> = emptyMap()
         ): ResponseMetaInfo =
-            ResponseMetaInfo(clock.now(), totalTokensCount, inputTokensCount, outputTokensCount)
+            ResponseMetaInfo(clock.now(), totalTokensCount, inputTokensCount, outputTokensCount, additionalInfo)
     }
 }