[MCP] Consult Firebase Tool (#8494)

Author: nohe427

src/gif/fdcExperience.ts

@@ -3,13 +3,16 @@ import { cloudCompanionOrigin } from "../api";
 
 const apiClient = new Client({ urlPrefix: cloudCompanionOrigin(), auth: true });
 const schemaGeneratorExperience = "/appeco/firebase/fdc-schema-generator";
+const geminiInFirebaseChatExperience = "/appeco/firebase/firebase-chat/free";
 const operationGeneratorExperience = "/appeco/firebase/fdc-query-generator";
 
 export interface GenerateSchemaRequest {
   input: { messages: { content: string; author: string }[] };
   experienceContext: { experience: string };
 }
 
+export type ChatExperienceRequest = GenerateSchemaRequest;
+
 export interface GenerateSchemaResponse {
   output: { messages: { content: string }[] };
   displayContext: {
@@ -20,6 +23,24 @@ export interface GenerateSchemaResponse {
   };
 }
 
+export interface ChatExperienceResponse {
+  output: { messages: { content: string; author: string }[] };
+  outputDataContext: {
+    additionalContext: { "@type": string };
+    attributionContext: {
+      citationMetadata: {
+        citations: {
+          startIndex: number;
+          endIndex: number;
+          url: string;
+          title: string;
+          license: string;
+        }[];
+      };
+    };
+  };
+}
+
 export interface GenerateOperationRequest {
   input: { messages: { content: string; author: string }[] };
   experienceContext: { experience: string };
@@ -55,6 +76,28 @@ export async function generateSchema(prompt: string, project: string): Promise<s
   return res.body.output.messages[0].content;
 }
 
+/**
+ * chatWithFirebase interacts with the Gemini in Firebase integration providing deeper knowledge on Firebase.
+ * @param prompt the interaction that the user would like to have with the service.
+ * @param project project identifier.
+ * @return ChatExperienceResponse includes not only the message from the service but also links to the resources used by the service.
+ */
+export async function chatWithFirebase(
+  prompt: string,
+  project: string,
+): Promise<ChatExperienceResponse> {
+  const res = await apiClient.post<ChatExperienceRequest, ChatExperienceResponse>(
+    `/v1beta/projects/${project}/locations/global/instances/default:completeTask`,
+    {
+      input: { messages: [{ content: prompt, author: "USER" }] },
+      experienceContext: {
+        experience: geminiInFirebaseChatExperience,
+      },
+    },
+  );
+  return res.body;
+}
+
 /**
  * generateOperation generates an operation based on the users prompt and deployed Firebase Data Connect Service.
  * @param prompt description of the operation the user would like to generate.

src/mcp/tools/core/consult_firebase_assistant.ts

@@ -0,0 +1,30 @@
+import { z } from "zod";
+import { tool } from "../../tool.js";
+import { toContent } from "../../util.js";
+import { chatWithFirebase } from "../../../gif/fdcExperience.js";
+
+export const consult_firebase_assistant = tool(
+  {
+    name: "consult_firebase_assistant",
+    description:
+      "Send a question to an AI assistant specifically enhanced to answer Firebase questions.",
+    inputSchema: z.object({
+      prompt: z
+        .string()
+        .describe("A description of what the user is trying to do or learn with Firebase."),
+    }),
+    annotations: {
+      title: "Helps answer queries and provide information related to Firebase.",
+      readOnlyHint: true,
+    },
+    _meta: {
+      requiresProject: true,
+      requiresAuth: true,
+      // TODO: Create an endpoint to check for GiF activiation.
+    },
+  },
+  async ({ prompt }, { projectId }) => {
+    const schema = await chatWithFirebase(prompt, projectId!);
+    return toContent(schema);
+  },
+);

src/mcp/tools/core/index.ts

@@ -1,4 +1,9 @@
+import { consult_firebase_assistant } from "./consult_firebase_assistant";
 import { get_firebase_directory } from "./get_firebase_directory";
 import { set_firebase_directory } from "./set_firebase_directory";
 
-export const coreTools = [get_firebase_directory, set_firebase_directory];
+export const coreTools = [
+  get_firebase_directory,
+  set_firebase_directory,
+  consult_firebase_assistant,
+];