AIX: Gemini: replay code execution part IDs

Author: enricoros

src/modules/aix/server/dispatch/chatGenerate/adapters/gemini.generateContent.ts

@@ -467,7 +467,7 @@ function _toGeminiContents(chatSequence: AixMessages_ChatMessage[], apiRequiresS
             case 'code_execution':
               if (invocation.language?.toLowerCase() !== 'python')
                 console.warn('Gemini only supports Python code execution, but got:', invocation.language);
-              parts.push(GeminiWire_ContentParts.ExecutableCodePart('PYTHON', invocation.code));
+              parts.push(GeminiWire_ContentParts.ExecutableCodePart('PYTHON', invocation.code, part.id));
               break;
             default:
               const _exhaustiveCheck: never = invocation;
@@ -501,7 +501,7 @@ function _toGeminiContents(chatSequence: AixMessages_ChatMessage[], apiRequiresS
               parts.push(GeminiWire_ContentParts.FunctionResponsePart({ id: part.id, name: part.response.name, response: functionResponseResponse }));
               break;
             case 'code_execution':
-              parts.push(GeminiWire_ContentParts.CodeExecutionResultPart(!part.error ? 'OUTCOME_OK' : 'OUTCOME_FAILED', toolErrorPrefix + part.response.result));
+              parts.push(GeminiWire_ContentParts.CodeExecutionResultPart(!part.error ? 'OUTCOME_OK' : 'OUTCOME_FAILED', toolErrorPrefix + part.response.result, part.id));
               break;
             default:
               const _exhaustiveCheck: never = part.response;

src/modules/aix/server/dispatch/wiretypes/gemini.wiretypes.ts

@@ -300,12 +300,12 @@ export namespace GeminiWire_ContentParts {
     return { functionResponse: { ...(id !== undefined ? { id } : {}), name, ...(response !== undefined ? { response } : {}) } };
   }
 
-  export function ExecutableCodePart(language: 'PYTHON', code: string): z.infer<typeof ExecutableCodePart_schema> {
-    return { executableCode: { language, code } };
+  export function ExecutableCodePart(language: 'PYTHON', code: string, id?: string): z.infer<typeof ExecutableCodePart_schema> {
+    return { executableCode: { ...(id !== undefined ? { id } : {}), language, code } };
   }
 
-  export function CodeExecutionResultPart(outcome: 'OUTCOME_OK' | 'OUTCOME_FAILED' | 'OUTCOME_DEADLINE_EXCEEDED', output?: string): z.infer<typeof CodeExecutionResultPart_schema> {
-    return { codeExecutionResult: { outcome, output } };
+  export function CodeExecutionResultPart(outcome: 'OUTCOME_OK' | 'OUTCOME_FAILED' | 'OUTCOME_DEADLINE_EXCEEDED', output?: string, id?: string): z.infer<typeof CodeExecutionResultPart_schema> {
+    return { codeExecutionResult: { ...(id !== undefined ? { id } : {}), outcome, output } };
   }
 
 }