format and docs

Author: dlarocque

docs-devsite/ai.templategenerativemodel.md

@@ -39,8 +39,8 @@ export declare class TemplateGenerativeModel
 
 |  Method | Modifiers | Description |
 |  --- | --- | --- |
-|  [generateContent(templateId, templateVariables)](./ai.templategenerativemodel.md#templategenerativemodelgeneratecontent) |  | <b><i>(Public Preview)</i></b> Makes a single non-streaming call to the model and returns an object containing a single [GenerateContentResponse](./ai.generatecontentresponse.md#generatecontentresponse_interface)<!-- -->. |
-|  [generateContentStream(templateId, templateVariables)](./ai.templategenerativemodel.md#templategenerativemodelgeneratecontentstream) |  | <b><i>(Public Preview)</i></b> Makes a single streaming call to the model and returns an object containing an iterable stream that iterates over all chunks in the streaming response as well as a promise that returns the final aggregated response. |
+|  [generateContent(templateId, templateVariables, singleRequestOptions)](./ai.templategenerativemodel.md#templategenerativemodelgeneratecontent) |  | <b><i>(Public Preview)</i></b> Makes a single non-streaming call to the model and returns an object containing a single [GenerateContentResponse](./ai.generatecontentresponse.md#generatecontentresponse_interface)<!-- -->. |
+|  [generateContentStream(templateId, templateVariables, singleRequestOptions)](./ai.templategenerativemodel.md#templategenerativemodelgeneratecontentstream) |  | <b><i>(Public Preview)</i></b> Makes a single streaming call to the model and returns an object containing an iterable stream that iterates over all chunks in the streaming response as well as a promise that returns the final aggregated response. |
 
 ## TemplateGenerativeModel.(constructor)
 
@@ -85,7 +85,8 @@ Makes a single non-streaming call to the model and returns an object containing
 <b>Signature:</b>
 
 ```typescript
-generateContent(templateId: string, templateVariables: object): Promise<GenerateContentResult>;
+generateContent(templateId: string, templateVariables: object, // anything!
+    singleRequestOptions?: SingleRequestOptions): Promise<GenerateContentResult>;
 ```
 
 #### Parameters
@@ -94,6 +95,7 @@ generateContent(templateId: string, templateVariables: object): Promise<Generate
 |  --- | --- | --- |
 |  templateId | string | The ID of the server-side template to execute. |
 |  templateVariables | object | A key-value map of variables to populate the template with. |
+|  singleRequestOptions | [SingleRequestOptions](./ai.singlerequestoptions.md#singlerequestoptions_interface) |  |
 
 <b>Returns:</b>
 
@@ -109,7 +111,7 @@ Makes a single streaming call to the model and returns an object containing an i
 <b>Signature:</b>
 
 ```typescript
-generateContentStream(templateId: string, templateVariables: object): Promise<GenerateContentStreamResult>;
+generateContentStream(templateId: string, templateVariables: object, singleRequestOptions?: SingleRequestOptions): Promise<GenerateContentStreamResult>;
 ```
 
 #### Parameters
@@ -118,6 +120,7 @@ generateContentStream(templateId: string, templateVariables: object): Promise<Ge
 |  --- | --- | --- |
 |  templateId | string | The ID of the server-side template to execute. |
 |  templateVariables | object | A key-value map of variables to populate the template with. |
+|  singleRequestOptions | [SingleRequestOptions](./ai.singlerequestoptions.md#singlerequestoptions_interface) |  |
 
 <b>Returns:</b>
 

docs-devsite/ai.templateimagenmodel.md

@@ -39,7 +39,7 @@ export declare class TemplateImagenModel
 
 |  Method | Modifiers | Description |
 |  --- | --- | --- |
-|  [generateImages(templateId, templateVariables)](./ai.templateimagenmodel.md#templateimagenmodelgenerateimages) |  | <b><i>(Public Preview)</i></b> Makes a single call to the model and returns an object containing a single [ImagenGenerationResponse](./ai.imagengenerationresponse.md#imagengenerationresponse_interface)<!-- -->. |
+|  [generateImages(templateId, templateVariables, singleRequestOptions)](./ai.templateimagenmodel.md#templateimagenmodelgenerateimages) |  | <b><i>(Public Preview)</i></b> Makes a single call to the model and returns an object containing a single [ImagenGenerationResponse](./ai.imagengenerationresponse.md#imagengenerationresponse_interface)<!-- -->. |
 
 ## TemplateImagenModel.(constructor)
 
@@ -84,7 +84,7 @@ Makes a single call to the model and returns an object containing a single [Imag
 <b>Signature:</b>
 
 ```typescript
-generateImages(templateId: string, templateVariables: object): Promise<ImagenGenerationResponse<ImagenInlineImage>>;
+generateImages(templateId: string, templateVariables: object, singleRequestOptions?: SingleRequestOptions): Promise<ImagenGenerationResponse<ImagenInlineImage>>;
 ```
 
 #### Parameters
@@ -93,6 +93,7 @@ generateImages(templateId: string, templateVariables: object): Promise<ImagenGen
 |  --- | --- | --- |
 |  templateId | string | The ID of the server-side template to execute. |
 |  templateVariables | object | A key-value map of variables to populate the template with. |
+|  singleRequestOptions | [SingleRequestOptions](./ai.singlerequestoptions.md#singlerequestoptions_interface) |  |
 
 <b>Returns:</b>
 

packages/ai/integration/prompt-templates.test.ts

@@ -66,7 +66,7 @@ describe('Prompt templates', function () {
               testConfig.ai.backend.backendType
             )}`,
             { animal: 'Rhino' },
-            { timeout: 100}
+            { timeout: 100 }
           );
           expect(images.length).to.equal(2); // We ask for two images in the prompt template
         });

packages/ai/src/requests/request.test.ts

@@ -531,9 +531,7 @@ describe('request methods', () => {
       await clock.tickAsync(0);
       controller.abort(abortReason);
 
-      await expect(requestPromise).to.be.rejectedWith(
-        'AbortError',
-      );
+      await expect(requestPromise).to.be.rejectedWith('AbortError');
     });
 
     it('should abort fetch if timeout expires during request', async () => {
@@ -564,7 +562,9 @@ describe('request methods', () => {
 
       expect(internalSignal?.aborted).to.be.true;
       expect((internalSignal?.reason as Error).name).to.equal('AbortError');
-      expect((internalSignal?.reason as Error).message).to.equal('Timeout has expired.');
+      expect((internalSignal?.reason as Error).message).to.equal(
+        'Timeout has expired.'
+      );
     });
 
     it('should succeed and clear timeout if fetch completes before timeout', async () => {
@@ -648,7 +648,10 @@ describe('request methods', () => {
       await clock.tickAsync(timeoutDuration / 2);
       controller.abort(abortReason);
 
-      await expect(requestPromise).to.be.rejectedWith('AbortError', abortReason);
+      await expect(requestPromise).to.be.rejectedWith(
+        'AbortError',
+        abortReason
+      );
     });
 
     it('should use timeout reason if it occurs before external signal abort', async () => {

packages/ai/src/requests/request.ts

@@ -189,7 +189,9 @@ export async function makeRequest(
       : DEFAULT_FETCH_TIMEOUT_MS;
   const internalAbortController = new AbortController();
   const fetchTimeoutId = setTimeout(() => {
-    internalAbortController.abort(new DOMException(TIMEOUT_EXPIRED_MESSAGE, ABORT_ERROR_NAME));
+    internalAbortController.abort(
+      new DOMException(TIMEOUT_EXPIRED_MESSAGE, ABORT_ERROR_NAME)
+    );
     logger.debug(
       `Aborting request to ${url} due to timeout (${timeoutMillis}ms)`
     );
@@ -198,7 +200,9 @@ export async function makeRequest(
   const externalAbortListener = (): void => {
     logger.debug(`Aborting request to ${url} due to external abort signal.`);
     // If this listener was invoked, an external signal was aborted, so externalSignal must be defined.
-    internalAbortController.abort(new DOMException(externalSignal!.reason, ABORT_ERROR_NAME));
+    internalAbortController.abort(
+      new DOMException(externalSignal!.reason, ABORT_ERROR_NAME)
+    );
   };
 
   if (externalSignal) {