[breadboard-web/gemini-kit] Fix JSONata calls to account for LLM Cont…

…ent (breadboard-ai#1579)
eltociear · Apr 30, 2024 · 5369037 · 5369037
1 parent e736f37
commit 5369037
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 3 deletions.
diff --git a/.changeset/twelve-lamps-tie.md b/.changeset/twelve-lamps-tie.md
@@ -0,0 +1,6 @@
+---
+"@google-labs/breadboard-web": patch
+"@google-labs/gemini-kit": patch
+---
+
+Fix JSONata calls to account for LLM Content
diff --git a/packages/breadboard-web/public/graphs/gemini-pro-vision.json b/packages/breadboard-web/public/graphs/gemini-pro-vision.json
@@ -262,7 +262,7 @@
       "id": "makeBody",
       "type": "jsonata",
       "configuration": {
-        "expression": "{ \"contents\": { \"parts\": $.parts }}"
+        "expression": "{ \"contents\": $.parts[0] }"
       }
     },
     {

diff --git a/packages/breadboard-web/src/boards/gemini-pro-vision.ts b/packages/breadboard-web/src/boards/gemini-pro-vision.ts
@@ -81,7 +81,7 @@ export default await board(() => {
 
   const makeBody = json.jsonata({
     $id: "makeBody",
-    expression: `{ "contents": { "parts": $.parts }}`,
+    expression: `{ "contents": $.parts[0] }`,
     parts: parameters,
   });
 

diff --git a/packages/gemini-kit/src/boards/gemini-pro-vision.ts b/packages/gemini-kit/src/boards/gemini-pro-vision.ts
@@ -81,7 +81,7 @@ export default await board(() => {
 
   const makeBody = json.jsonata({
     $id: "makeBody",
-    expression: `{ "contents": { "parts": $.parts }}`,
+    expression: `{ "contents": $.parts[0] }`,
     parts: parameters,
   });