[AI Assistance] Support multiple function chunks

Bug: 360751542
Change-Id: I0ab88b303c2f56c007a635e610bd0a375d94b64a
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/6074691
Commit-Queue: Alex Rudenko <[email protected]>
Auto-Submit: Nikolay Vitkov <[email protected]>
Reviewed-by: Alex Rudenko <[email protected]>

Author: Lightning00Blade

front_end/core/host/AidaClient.test.ts

@@ -172,7 +172,7 @@ describeWithEnvironment('AidaClient', () => {
       {
         explanation: 'hello brave new world!',
         metadata: {rpcGlobalId: 123},
-        functionCall: undefined,
+        functionCalls: undefined,
         completed: true,
       },
     ]);
@@ -200,7 +200,7 @@ describeWithEnvironment('AidaClient', () => {
       {
         explanation: 'hello world',
         metadata: {rpcGlobalId: 123},
-        functionCall: undefined,
+        functionCalls: undefined,
         completed: true,
       },
     ]);
@@ -281,7 +281,7 @@ describeWithEnvironment('AidaClient', () => {
             'If it were so, it was a grievous fault,\n' +
             'And grievously hath Caesar answer’d it.\n',
         metadata: {rpcGlobalId: 123},
-        functionCall: undefined,
+        functionCalls: undefined,
         completed: true,
       },
     ]);
@@ -342,7 +342,7 @@ describeWithEnvironment('AidaClient', () => {
             },
           ],
         },
-        functionCall: undefined,
+        functionCalls: undefined,
         completed: true,
       },
     ]);

front_end/core/host/AidaClient.ts

@@ -213,7 +213,7 @@ export interface AidaResponseMetadata {
 export interface AidaResponse {
   explanation: string;
   metadata: AidaResponseMetadata;
-  functionCall?: AidaFunctionCallResponse;
+  functionCalls?: [AidaFunctionCallResponse, ...AidaFunctionCallResponse[]];
   completed: boolean;
 }
 
@@ -322,7 +322,7 @@ export class AidaClient {
     let chunk;
     const text = [];
     let inCodeChunk = false;
-    let functionCall: AidaFunctionCallResponse|undefined = undefined;
+    const functionCalls: AidaFunctionCallResponse[] = [];
     const metadata: AidaResponseMetadata = {rpcGlobalId: 0};
     while ((chunk = await stream.read())) {
       let textUpdated = false;
@@ -378,10 +378,10 @@ export class AidaClient {
           text.push(result.codeChunk.code);
           textUpdated = true;
         } else if ('functionCallChunk' in result) {
-          functionCall = {
+          functionCalls.push({
             name: result.functionCallChunk.functionCall.name,
             args: result.functionCallChunk.functionCall.args,
-          };
+          });
         } else if ('error' in result) {
           throw new Error(`Server responded: ${JSON.stringify(result)}`);
         } else {
@@ -399,7 +399,8 @@ export class AidaClient {
     yield {
       explanation: text.join('') + (inCodeChunk ? CODE_CHUNK_SEPARATOR : ''),
       metadata,
-      functionCall,
+      functionCalls: functionCalls.length ? functionCalls as [AidaFunctionCallResponse, ...AidaFunctionCallResponse[]] :
+                                            undefined,
       completed: true,
     };
   }

front_end/panels/freestyler/AiAgent.ts

@@ -272,7 +272,7 @@ export abstract class AiAgent<T> {
       response = aidaResponse.explanation;
       rpcId = aidaResponse.metadata.rpcGlobalId ?? rpcId;
 
-      if (aidaResponse.functionCall) {
+      if (aidaResponse.functionCalls) {
         throw new Error('Function calling not supported yet');
       }
 
@@ -342,7 +342,7 @@ export abstract class AiAgent<T> {
   }
 
   parseResponse(response: Host.AidaClient.AidaResponse): ParsedResponse {
-    if (response.functionCall && response.completed) {
+    if (response.functionCalls && response.completed) {
       throw new Error('Function calling not supported yet');
     }
     return {

front_end/panels/freestyler/FreestylerAgent.ts

@@ -265,7 +265,7 @@ export class FreestylerAgent extends AiAgent<SDK.DOMModel.DOMNode> {
   }
 
   override parseResponse(response: Host.AidaClient.AidaResponse): ParsedResponse {
-    if (response.functionCall) {
+    if (response.functionCalls) {
       throw new Error('Function calling not supported yet');
     }