@firebase/ai 2.8.0 → 2.9.0-canary.78384d32c

package/dist/src/methods/chat-session.d.ts

@@ -14,7 +14,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { Content, GenerateContentResult, GenerateContentStreamResult, Part, RequestOptions, SingleRequestOptions, StartChatParams } from '../types';
+import { Content, FunctionCall, FunctionResponsePart, GenerateContentRequest, GenerateContentResponse, GenerateContentResult, GenerateContentStreamResult, Part, RequestOptions, SingleRequestOptions, StartChatParams } from '../types';
 import { ApiSettings } from '../types/internal';
 import { ChromeAdapter } from '../types/chrome-adapter';
 /**
@@ -42,6 +42,12 @@ export declare class ChatSession {
      * to history.
      */
     getHistory(): Promise<Content[]>;
+    /**
+     * Format Content into a request for generateContent or
+     * generateContentStream.
+     * @internal
+     */
+    _formatRequest(incomingContent: Content, tempHistory: Content[]): GenerateContentRequest;
     /**
      * Sends a chat message and receives a non-streaming
      * {@link GenerateContentResult}
@@ -53,4 +59,19 @@ export declare class ChatSession {
      * and a response promise.
      */
     sendMessageStream(request: string | Array<string | Part>, singleRequestOptions?: SingleRequestOptions): Promise<GenerateContentStreamResult>;
+    /**
+     * Get function calls that the SDK has references to actually call.
+     * This is all-or-nothing. If the model is requesting multiple
+     * function calls, all of them must have references in order for
+     * automatic function calling to work.
+     *
+     * @internal
+     */
+    _getCallableFunctionCalls(response?: GenerateContentResponse): FunctionCall[] | undefined;
+    /**
+     * Call user-defined functions if requested by the model, and return
+     * the response that should be sent to the model.
+     * @internal
+     */
+    _callFunctionsAsNeeded(functionCalls: FunctionCall[]): Promise<FunctionResponsePart[]>;
 }

package/dist/src/methods/generate-content.d.ts

@@ -14,10 +14,12 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { GenerateContentRequest, GenerateContentResult, GenerateContentStreamResult, SingleRequestOptions } from '../types';
+import { GenerateContentRequest, GenerateContentResponse, GenerateContentResult, GenerateContentStreamResult, SingleRequestOptions } from '../types';
 import { ApiSettings } from '../types/internal';
 import { ChromeAdapter } from '../types/chrome-adapter';
-export declare function generateContentStream(apiSettings: ApiSettings, model: string, params: GenerateContentRequest, chromeAdapter?: ChromeAdapter, singleRequestOptions?: SingleRequestOptions): Promise<GenerateContentStreamResult>;
+export declare function generateContentStream(apiSettings: ApiSettings, model: string, params: GenerateContentRequest, chromeAdapter?: ChromeAdapter, singleRequestOptions?: SingleRequestOptions): Promise<GenerateContentStreamResult & {
+    firstValue?: GenerateContentResponse;
+}>;
 export declare function templateGenerateContent(apiSettings: ApiSettings, templateId: string, templateParams: object, singleRequestOptions?: SingleRequestOptions): Promise<GenerateContentResult>;
 export declare function templateGenerateContentStream(apiSettings: ApiSettings, templateId: string, templateParams: object, singleRequestOptions?: SingleRequestOptions): Promise<GenerateContentStreamResult>;
 export declare function generateContent(apiSettings: ApiSettings, model: string, params: GenerateContentRequest, chromeAdapter?: ChromeAdapter, singleRequestOptions?: SingleRequestOptions): Promise<GenerateContentResult>;

package/dist/src/requests/response-helpers.d.ts

@@ -36,7 +36,7 @@ export declare function getText(response: GenerateContentResponse, partFilter: (
 /**
  * Returns every {@link FunctionCall} associated with first candidate.
  */
-export declare function getFunctionCalls(response: GenerateContentResponse): FunctionCall[] | undefined;
+export declare function getFunctionCalls(response?: GenerateContentResponse): FunctionCall[] | undefined;
 /**
  * Returns every {@link InlineDataPart} in the first candidate if present.
  *

package/dist/src/requests/stream-reader.d.ts

@@ -25,7 +25,9 @@ import { InferenceSource } from '../public-types';
  *
  * @param response - Response from a fetch call
  */
-export declare function processStream(response: Response, apiSettings: ApiSettings, inferenceSource?: InferenceSource): GenerateContentStreamResult;
+export declare function processStream(response: Response, apiSettings: ApiSettings, inferenceSource?: InferenceSource): Promise<GenerateContentStreamResult & {
+    firstValue?: GenerateContentResponse;
+}>;
 /**
  * Reads a raw string stream, buffers incomplete chunks, and yields parsed JSON objects.
  */

package/dist/src/types/content.d.ts

@@ -243,6 +243,7 @@ export interface FunctionResponse {
     id?: string;
     name: string;
     response: object;
+    parts?: Part[];
 }
 /**
  * Interface for sending an image.

package/dist/src/types/requests.d.ts

@@ -231,6 +231,15 @@ export interface RequestOptions {
      * (used regardless of your chosen Gemini API provider).
      */
     baseUrl?: string;
+    /**
+     * Limits amount of sequential function calls the SDK can make during automatic
+     * function calling, in order to prevent infinite loops. If not specified,
+     * this value defaults to 10.
+     *
+     * When it reaches this limit, it will return the last response received
+     * from the model, whether it is a text response or further function calls.
+     */
+    maxSequentalFunctionCalls?: number;
 }
 /**
  * Options that can be provided per-request.
@@ -304,6 +313,11 @@ export interface FunctionDeclaration {
      * case-sensitive. For a function with no parameters, this can be left unset.
      */
     parameters?: ObjectSchema | ObjectSchemaRequest;
+    /**
+     * Reference to an actual function to call. Specifying this will cause the
+     * function to be called automatically when requested by the model.
+     */
+    functionReference?: Function;
 }
 /**
  * A tool that allows a Gemini model to connect to Google Search to access and incorporate

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@firebase/ai",
-  "version": "2.8.0",
+  "version": "2.9.0-canary.78384d32c",
   "description": "The Firebase AI SDK",
   "author": "Firebase <firebase-support@google.com> (https://firebase.google.com/)",
   "engines": {
@@ -48,19 +48,19 @@
     "trusted-type-check": "tsec -p tsconfig.json --noEmit"
   },
   "peerDependencies": {
-    "@firebase/app": "0.x",
-    "@firebase/app-types": "0.x"
+    "@firebase/app": "0.14.9-canary.78384d32c",
+    "@firebase/app-types": "0.9.3-canary.78384d32c"
   },
   "dependencies": {
-    "@firebase/app-check-interop-types": "0.3.3",
-    "@firebase/component": "0.7.0",
-    "@firebase/logger": "0.5.0",
-    "@firebase/util": "1.13.0",
+    "@firebase/app-check-interop-types": "0.3.3-canary.78384d32c",
+    "@firebase/component": "0.7.1-canary.78384d32c",
+    "@firebase/logger": "0.5.0-canary.78384d32c",
+    "@firebase/util": "1.14.0-canary.78384d32c",
     "tslib": "^2.1.0"
   },
   "license": "Apache-2.0",
   "devDependencies": {
-    "@firebase/app": "0.14.8",
+    "@firebase/app": "0.14.9-canary.78384d32c",
     "@rollup/plugin-json": "6.1.0",
     "rollup": "2.79.2",
     "rollup-plugin-replace": "2.2.0",