Set content type for tool responses.

yishuangP · copybara-github · commit a77be48a9f33 · 2026-03-06T21:38:14.000-08:00
LiteRT-LM-PiperOrigin-RevId: 879964350
diff --git a/kotlin/java/com/google/ai/edge/litertlm/Conversation.kt b/kotlin/java/com/google/ai/edge/litertlm/Conversation.kt
@@ -320,7 +320,12 @@ class Conversation(
           addProperty("name", functionName)
           add("response", result)
         }
-      toolResponsesJSONArray.add(toolResponseJSONObject)
+      val toolResponseWithTypeJSONObject =
+        JsonObject().apply {
+          addProperty("type", "tool_response")
+          add("tool_response", toolResponseJSONObject)
+        }
+      toolResponsesJSONArray.add(toolResponseWithTypeJSONObject)
     }
     return JsonObject().apply {
       addProperty("role", "tool")

Original file line number	Diff line number	Diff line change
`@@ -320,7 +320,12 @@ class Conversation(`
`320`	`320`	`addProperty("name", functionName)`
`321`	`321`	`add("response", result)`
`322`	`322`	`}`
`323`		`- toolResponsesJSONArray.add(toolResponseJSONObject)`
	`323`	`+ val toolResponseWithTypeJSONObject =`
	`324`	`+ JsonObject().apply {`
	`325`	`+ addProperty("type", "tool_response")`
	`326`	`+ add("tool_response", toolResponseJSONObject)`
	`327`	`+ }`
	`328`	`+ toolResponsesJSONArray.add(toolResponseWithTypeJSONObject)`
`324`	`329`	`}`
`325`	`330`	`return JsonObject().apply {`
`326`	`331`	`addProperty("role", "tool")`