Update entrypoint and service to aimode in FxA authentication

Updated with OpenAI pipeline changes from D271303

Point to stage FxA server

Author: Punam Dahiya

browser/app/profile/firefox.js

@@ -2325,7 +2325,7 @@ pref("identity.sync.tokenserver.uri", "https://token.services.mozilla.com/1.0/sy
 // [identity.fxaccounts.autoconfig.uri]/.well-known/fxa-client-configuration
 // This is now the prefered way of pointing to a custom FxA server, instead
 // of making changes to "identity.fxaccounts.*.uri".
-pref("identity.fxaccounts.autoconfig.uri", "");
+pref("identity.fxaccounts.autoconfig.uri", "https://accounts.stage.mozaws.net");
 
 // URL for help link about Send Tab.
 pref("identity.sendtabpromo.url", "https://support.mozilla.org/1/firefox/%VERSION%/%OS%/%LOCALE%/send-tab");

browser/base/content/browser-smart-window.js

@@ -128,6 +128,9 @@ var SmartWindow = {
                       type: "FXA_SMART_WINDOW_SIGNIN_FLOW",
                       data: {
                         entrypoint: "aimode",
+                        extraParams: {
+                          service: "aimode",
+                        }
                       },
                     },
                     gBrowser.selectedBrowser

browser/components/smartwindow/content/utils.mjs

@@ -14,6 +14,11 @@ ChromeUtils.defineESModuleGetters(lazy, {
 
 import { createEngine } from "chrome://global/content/ml/EngineProcess.sys.mjs";
 import { SmartAssistEngine } from "moz-src:///browser/components/genai/SmartAssistEngine.sys.mjs";
+import { getFxAccountsSingleton } from "resource://gre/modules/FxAccounts.sys.mjs";
+import {
+  SCOPE_PROFILE,
+  OAUTH_CLIENT_ID,
+} from "resource://gre/modules/FxAccountsCommon.sys.mjs";
 
 const { ChatHistoryMessage } = ChromeUtils.importESModule(
   "resource:///modules/smartwindow/ChatHistory.sys.mjs"
@@ -382,6 +387,20 @@ const get_page_content = async ({ url, mode, page }, allowedUrls) => {
   }
 };
 
+const getFxAccountToken = async () => {
+  try {
+    const fxAccounts = getFxAccountsSingleton();
+    const token = await fxAccounts.getOAuthToken({
+      scope: SCOPE_PROFILE,
+      client_id: OAUTH_CLIENT_ID,
+    });
+    return token;
+  } catch (error) {
+    console.error("Error obtaining FxA token:", error);
+    throw error;
+  }
+};
+
 /**
  * @param {PageExtractor} pageExtractor
  * @param {string} mode
@@ -459,6 +478,7 @@ async function runExtraction(pageExtractor, mode, page, label) {
  */
 export async function* fetchWithHistory(messages, allowedUrls) {
   const engineInstance = await createOpenAIEngine();
+  const fxAccountToken = await getFxAccountToken();
 
   // Normalize roles to lowercase and handle system messages
   let convo = Array.isArray(messages)
@@ -512,6 +532,7 @@ export async function* fetchWithHistory(messages, allowedUrls) {
   const streamModelResponse = () =>
     engineInstance.runWithGenerator({
       streamOptions: { enabled: true },
+      fxAccountToken,
       tool_choice: "auto",
       tools: toolsConfig,
       args: convo,

browser/components/smartwindow/content/welcome.js

@@ -69,6 +69,9 @@ XPCOMUtils.defineLazyPreferenceGetter(
                   {
                     data: {
                       entrypoint: "aimode",
+                      extraParams: {
+                        service: "aimode",
+                      }
                     },
                     type: "FXA_SMART_WINDOW_SIGNIN_FLOW",
                     navigate: true,

toolkit/components/ml/content/backends/OpenAIPipeline.mjs

@@ -327,10 +327,20 @@ export class OpenAIPipeline {
     lazy.console.debug("Running OpenAI pipeline");
     try {
       const { baseURL, apiKey, modelId } = this.#options;
+      let fxAccountToken = request.fxAccountToken
+        ? request.fxAccountToken
+        : null;
+
+      fxAccountToken='eyJhbGciOiJSUzI1NiIsInR5cCI6ImF0K0pXVCIsImtpZCI6IjIwMTkwNzMwLTE1ZTQ3M2ZkIn0.eyJpc3MiOiJodHRwczovL2FjY291bnRzLnN0YWdlLm1vemF3cy5uZXQiLCJhdWQiOiI1ODgyMzg2YzZkODAxNzc2IiwiY2xpZW50X2lkIjoiNTg4MjM4NmM2ZDgwMTc3NiIsImV4cCI6MTc2MjYwNzE5MywiaWF0IjoxNzYyNTYzOTkzLCJqdGkiOiI5MTk0MGUzNGU1MWNiNjJmY2E2ZjdmMWYyNjI5Mzk1MzEwNWI5MmM5ZWQzOTk1NzlmYTdjY2M4OGJjZjMwYjFjIiwic2NvcGUiOiJwcm9maWxlIiwic3ViIjoiYTcwMDVmNTA4ODQzNGM1YmI5YTM1YzYzNzM1NWZhYjUiLCJmeGEtZ2VuZXJhdGlvbiI6MTc2MjM3NzY4MDA0NiwiZnhhLXByb2ZpbGVDaGFuZ2VkQXQiOjE3NjIzNzc3Mjg3Nzd9.Y7CRqnC9Rc-FuSb-726OOKqGjC4T3FVB1YobJwg_m1r8ughEWScz0vam02IpLEC5WjE_5HRQD6URTcQ0epTXU2T5c-jkAo5kn8GMvkI-YieMOPePwU0LljnU7vGNMSJ7trdAjXhb8aCLnojP5wSDni7BM_khfr66yHWBWaOFV3TPUor8N9JDMb_5bDrqzvbigKpoeeJ23ihw17Mis3EGC4AjmyIqL92gGWVTvz2g9u_H0MOc8YfxEoO6aagz-OwYJrXVpVxCeRQ8wYbV9ce191a9NrLaUmBFn0NC6JfsbD2bLU7HDl7haUtIaofgnpzru-zlGmk8VoCA1dGJ--lPfg';
+      const defaultHeaders = fxAccountToken
+        ? { "authorization": `Bearer ${fxAccountToken}` }
+        : undefined;
+
       const client = new OpenAIPipeline.OpenAILib.OpenAI({
         baseURL: baseURL ? baseURL : "http://localhost:11434/v1",
         apiKey: apiKey || "ollama",
         maxRetries: 10,
+        ...(defaultHeaders ? { defaultHeaders } : {}),
       });
       const stream = request.streamOptions?.enabled || false;
       const tools = request.tools || [];