@truefoundry/trueforge-ui 0.3.1 → 0.4.0-rc.0

package/dist/plugins/trueforge-agent-server-adapter/index.js

@@ -1,6 +1,6 @@
 import {
-  createTrueFoundryServer
-} from "../../chunk-7AAO4UDQ.js";
+  createTrueForgeServer
+} from "../../chunk-HCGDDQ4G.js";
 import {
   readSessionIsCreateAgent
 } from "../../chunk-RDXBMTP5.js";
@@ -1112,6 +1112,99 @@ function createSkillCatalog(client) {
   };
 }
 
+// src/plugins/trueforge-agent-server-adapter/catalogs/webSearchProviderCatalog.ts
+import { TrueForgeApi as TrueForgeApi3 } from "@truefoundry/trueforge-sdk";
+var PARALLEL_TYPE = "parallel";
+var PARALLEL_DISPLAY_NAME = "Parallel";
+function displayNameForType2(type) {
+  if (type === PARALLEL_TYPE) {
+    return PARALLEL_DISPLAY_NAME;
+  }
+  return type;
+}
+function toUiCatalogEntry4(provider) {
+  return {
+    id: provider.type,
+    name: displayNameForType2(provider.type),
+    type: provider.type
+  };
+}
+function toUiWebSearchProvider(provider) {
+  return {
+    id: provider.type,
+    name: displayNameForType2(provider.type),
+    catalogId: provider.type,
+    isConnected: true
+  };
+}
+function filterUiWebSearchProviders({
+  providers,
+  query
+}) {
+  const normalizedQuery = query?.trim().toLowerCase();
+  if (normalizedQuery === void 0 || normalizedQuery === "") {
+    return providers;
+  }
+  return providers.filter(
+    (provider) => provider.name.toLowerCase().includes(normalizedQuery) || provider.id.toLowerCase().includes(normalizedQuery)
+  );
+}
+function toHarnessManifest4({
+  type,
+  apiKey
+}) {
+  if (type !== PARALLEL_TYPE) {
+    throw new Error(`Unsupported web-search provider type: ${type}`);
+  }
+  return {
+    type: PARALLEL_TYPE,
+    auth: { apiKey }
+  };
+}
+function createWebSearchProviderCatalog(client) {
+  async function resolveApiKey(apiKey) {
+    const trimmed = apiKey?.trim();
+    if (trimmed !== void 0 && trimmed !== "") {
+      return trimmed;
+    }
+    const existing = await client.settings.webSearchProviders.get();
+    return existing.data.manifest.auth.apiKey;
+  }
+  return {
+    getWebSearchProviderCatalog: async () => {
+      const body = await client.catalogs.webSearchProviders.list();
+      return body.data.map(toUiCatalogEntry4);
+    },
+    listWebSearchProviders: async (req) => {
+      let providers;
+      try {
+        const body = await client.settings.webSearchProviders.get();
+        providers = [toUiWebSearchProvider(body.data.manifest)];
+      } catch (err) {
+        if (err instanceof TrueForgeApi3.NotFoundError) {
+          providers = [];
+        } else {
+          throw err;
+        }
+      }
+      return filterUiWebSearchProviders({ providers, query: req?.query });
+    },
+    createWebSearchProvider: async (req) => {
+      const body = await client.settings.webSearchProviders.createOrUpdate({
+        manifest: toHarnessManifest4({ type: req.type, apiKey: req.apiKey })
+      });
+      return toUiWebSearchProvider(body.data.manifest);
+    },
+    updateWebSearchProvider: async (req) => {
+      const apiKey = await resolveApiKey(req.apiKey);
+      const body = await client.settings.webSearchProviders.createOrUpdate({
+        manifest: toHarnessManifest4({ type: PARALLEL_TYPE, apiKey })
+      });
+      return toUiWebSearchProvider(body.data.manifest);
+    }
+  };
+}
+
 // src/plugins/trueforge-agent-server-adapter/permissionsServer.ts
 function createHarnessPermissionsServer(options = {}) {
   const client = options.client ?? createTrueForgeClient(options);
@@ -1249,9 +1342,10 @@ function createTrueForgeAgentUIServer(options = {}) {
     modelCatalog: createModelProviderCatalog(client),
     connectorCatalog: createConnectorCatalog(client),
     skillCatalog: createSkillCatalog(client),
-    sandboxCatalog: createSandboxProviderCatalog(client)
+    sandboxCatalog: createSandboxProviderCatalog(client),
+    webSearchCatalog: createWebSearchProviderCatalog(client)
   };
-  return createTrueFoundryServer({
+  return createTrueForgeServer({
     chatServer: createHarnessChatServer({ client }),
     ...createHarnessBuilderServer({ client }),
     catalog,
@@ -1275,7 +1369,9 @@ export {
   createSkillCatalog,
   createTrueForgeAgentUIServer,
   createTrueForgeClient,
+  createWebSearchProviderCatalog,
   filterUiSandboxProviders,
+  filterUiWebSearchProviders,
   getCapabilities,
   listConfiguredMcpServers,
   listModels,
@@ -1287,6 +1383,7 @@ export {
   toHarnessModelProvider,
   toHarnessManifest2 as toHarnessSandboxManifest,
   toHarnessManifest3 as toHarnessSkillManifest,
+  toHarnessManifest4 as toHarnessWebSearchManifest,
   toModelSelection,
   toUiAgentSpec,
   toUiCatalogModelProviderEntry,
@@ -1295,6 +1392,8 @@ export {
   toUiModelProvider,
   toUiSandboxProvider,
   toUiSandboxProviderListEntry,
-  toUiSkill2 as toUiSkill
+  toUiSkill2 as toUiSkill,
+  toUiCatalogEntry4 as toUiWebSearchCatalogEntry,
+  toUiWebSearchProvider
 };
 //# sourceMappingURL=index.js.map