Add support for logging data or custom responses from LLMs (#1191)

firebase · Nov 6, 2024 · 0fcdb38 · 0fcdb38
1 parent c487e47
commit 0fcdb38
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/js/plugins/google-cloud/src/telemetry/generate.ts b/js/plugins/google-cloud/src/telemetry/generate.ts
@@ -349,6 +349,9 @@ class GenerateTelemetry implements Telemetry {
     if (part.text) {
       return this.toPartLogText(part.text);
     }
+    if (part.data) {
+      return this.toPartLogText(JSON.stringify(part.data));
+    }
     if (part.media) {
       return this.toPartLogMedia(part);
     }
@@ -358,6 +361,9 @@ class GenerateTelemetry implements Telemetry {
     if (part.toolResponse) {
       return this.toPartLogToolResponse(part);
     }
+    if (part.custom) {
+      return this.toPartLogText(JSON.stringify(part.custom));
+    }
     return '<unknown format>';
   }