Merge pull request #40 from jazkarta/task/add-metadata-when-guardrail-triggers

web search will be now managed by admin from dashboard api

Author: sudipto68

api/server/controllers/agents/request.js

@@ -13,7 +13,7 @@ const { disposeClient, clientRegistry, requestDataMap } = require('~/server/clea
 const { handleAbortError } = require('~/server/middleware');
 const { logViolation } = require('~/cache');
 const { saveMessage } = require('~/models');
-const { syncChatToAyo, extractResponseText, loadAyoGuardrails } = require('~/server/services/ayoDashboard');
+const { syncChatToAyo, extractResponseText, loadAyoGuardrails, loadAyoGlobalConfig } = require('~/server/services/ayoDashboard');
 
 function createCloseHandler(abortController) {
   return function (manual) {
@@ -158,6 +158,12 @@ const ResumableAgentController = async (req, res, next, initializeClient, addTit
       logger.warn('[ayoDashboard] loadAyoGuardrails failed', { status: err.status, message: err.message });
     }
 
+    try {
+      req.ayoWebSearch = (await loadAyoGlobalConfig(req))?.webSearch ?? false;
+    } catch (err) {
+      logger.warn('[ayoDashboard] loadAyoGlobalConfig failed', { status: err.status, message: err.message });
+    }
+
     /** @type {{ client: TAgentClient; userMCPAuthMap?: Record<string, Record<string, string>> }} */
     const result = await initializeClient({
       req,
@@ -585,6 +591,12 @@ const _LegacyAgentController = async (req, res, next, initializeClient, addTitle
       logger.warn('[ayoDashboard] loadAyoGuardrails failed', { status: err.status, message: err.message });
     }
 
+    try {
+      req.ayoWebSearch = (await loadAyoGlobalConfig(req))?.webSearch ?? false;
+    } catch (err) {
+      logger.warn('[ayoDashboard] loadAyoGlobalConfig failed', { status: err.status, message: err.message });
+    }
+
     /** @type {{ client: TAgentClient; userMCPAuthMap?: Record<string, Record<string, string>> }} */
     const result = await initializeClient({
       req,

api/server/services/ayoDashboard.js

@@ -231,6 +231,33 @@ const loadAyoGuardrails = async (req, conversationId) => {
   return value;
 };
 
+const getAyoGlobalConfig = async (token) => {
+  const url = `${getBaseUrl()}/api/configurations/global-configurations/`;
+  const res = await fetch(url, {
+    method: 'GET',
+    headers: { 'Content-Type': 'application/json', Authorization: `Bearer ${token}` },
+  });
+  if (!res.ok) {
+    const body = await res.text().catch(() => '');
+    const err = new Error(`getAyoGlobalConfig failed: ${res.status} ${body}`);
+    err.status = res.status;
+    throw err;
+  }
+  const data = await res.json();
+  return { webSearch: data?.web_search === true };
+};
+
+const loadAyoGlobalConfig = async (req) => {
+  if (!getBaseUrl()) {
+    return { webSearch: false };
+  }
+  const { accessToken, refreshToken } = getTokensFromReq(req);
+  if (!accessToken) {
+    return { webSearch: false };
+  }
+  return callWithRefresh(req, accessToken, refreshToken, (t) => getAyoGlobalConfig(t));
+};
+
 const markConversationDeleted = async (token, conversationId) => {
   const url = `${getBaseUrl()}/api/chats/conversations/mark-deleted/`;
   const res = await fetch(url, {
@@ -439,4 +466,4 @@ const syncChatToAyo = async ({
   }
 };
 
-module.exports = { syncChatToAyo, syncConversationDeleteToAyo, syncChatMetadataToAyo, updateConversationTitle, refreshAccessToken, isTokenExpired, getCurrentUserInfo, extractResponseText, loadAyoGuardrails };
+module.exports = { syncChatToAyo, syncConversationDeleteToAyo, syncChatMetadataToAyo, updateConversationTitle, refreshAccessToken, isTokenExpired, getCurrentUserInfo, extractResponseText, loadAyoGuardrails, loadAyoGlobalConfig };

packages/api/src/endpoints/custom/initialize.ts

@@ -23,14 +23,15 @@ const { PROXY } = process.env;
 function buildCustomOptions(
   endpointConfig: Partial<TEndpoint>,
   ayoGuardrails: string[],
+  ayoWebSearch: boolean,
   appConfig?: AppConfig,
   endpointTokenConfig?: Record<string, unknown>,
 ) {
   const customOptions: Record<string, unknown> = {
     headers: endpointConfig.headers,
     addParams:
       endpointConfig.name === 'LiteLLM'
-        ? { ...(endpointConfig.addParams ?? {}), guardrails: ayoGuardrails }
+        ? { ...(endpointConfig.addParams ?? {}), guardrails: ayoGuardrails, web_search: ayoWebSearch }
         : endpointConfig.addParams,
     dropParams: endpointConfig.dropParams,
     customParams: endpointConfig.customParams,
@@ -164,9 +165,11 @@ export async function initializeCustom({
   }
 
   const AYO_GUARDRAILS = req.ayoGuardrails ?? [];
+  const AYO_WEB_SEARCH = req.ayoWebSearch ?? false;
   const customOptions = buildCustomOptions(
     endpointConfig,
     AYO_GUARDRAILS,
+    AYO_WEB_SEARCH,
     appConfig,
     endpointTokenConfig,
   );

packages/api/src/types/http.ts

@@ -22,4 +22,5 @@ export type ServerRequest = Request<unknown, unknown, RequestBody> & {
   user?: IUser;
   config?: AppConfig;
   ayoGuardrails?: string[];
+  ayoWebSearch?: boolean;
 };