Stop hardcoding python

google-gemini · Aug 22, 2024 · e8bce3e · e8bce3e
1 parent 2dfa412
commit e8bce3e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/.changeset/gorgeous-squids-wonder.md b/.changeset/gorgeous-squids-wonder.md
@@ -0,0 +1,5 @@
+---
+"@google/generative-ai": patch
+---
+
+Fix language marker in text helper for executable code results.
diff --git a/src/requests/response-helpers.ts b/src/requests/response-helpers.ts
@@ -125,7 +125,11 @@ export function getText(response: GenerateContentResponse): string {
       }
       if (part.executableCode) {
         textStrings.push(
-          "\n```python\n" + part.executableCode.code + "\n```\n",
+          "\n```" +
+            part.executableCode.language +
+            "\n" +
+            part.executableCode.code +
+            "\n```\n",
         );
       }
       if (part.codeExecutionResult) {