feat(core): add context provider plugin system (AIPexStudio#54)

Author: 0x5457

packages/browser-ext/src/components/chatbot/__tests__/use-chat.test.ts

@@ -179,7 +179,10 @@ describe("useChat", () => {
       await result.current.sendMessage("Hi there");
     });
 
-    expect(agent.chat).toHaveBeenCalledWith("Hi there", undefined);
+    expect(agent.chat).toHaveBeenCalledWith("Hi there", {
+      sessionId: undefined,
+      contexts: undefined,
+    });
     expect(result.current.sessionId).toBe("session-1");
     expect(result.current.messages[0].role).toBe("user");
   });

packages/browser-ext/src/hooks/use-agent.ts

@@ -1,12 +1,14 @@
 import {
   AIPex,
   aisdk,
+  ContextManager,
   IndexedDBStorage,
   SessionStorage,
 } from "@aipexstudio/aipex-core";
 import { useEffect, useMemo, useState } from "react";
 import { SYSTEM_PROMPT } from "~/components/chatbot/constants";
 import { createAIProvider } from "~/lib/ai-provider";
+import { allBrowserProviders } from "~/lib/context/providers";
 import { allBrowserTools } from "~/tools";
 import type { ChatSettings } from "~/types";
 
@@ -64,6 +66,12 @@ export function useAgent({
         }),
       );
 
+      // Create context manager with browser providers
+      const contextManager = new ContextManager({
+        providers: allBrowserProviders,
+        autoInitialize: true,
+      });
+
       // Get all available tools
       const tools = [...allBrowserTools];
 
@@ -74,6 +82,7 @@ export function useAgent({
         model,
         tools,
         storage,
+        contextManager,
         maxTurns: 10,
       });
 

packages/browser-ext/src/hooks/use-chat.ts

@@ -1,4 +1,4 @@
-import type { AgentEvent, AIPex } from "@aipexstudio/aipex-core";
+import type { AgentEvent, AIPex, Context } from "@aipexstudio/aipex-core";
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
 import { ChatAdapter } from "../adapters/chat-adapter";
 import type { ChatConfig, ChatStatus, ContextItem, UIMessage } from "../types";
@@ -190,19 +190,21 @@ export function useChat(
       handlersRef.current?.onMessageSent?.(userMessage);
       adapter.setStatus("submitted");
 
-      // Build the input text with context
-      let inputText = text;
-      if (contexts && contexts.length > 0) {
-        const contextText = contexts
-          .map((ctx) => `[${ctx.type}: ${ctx.label}]\n${ctx.value}`)
-          .join("\n\n");
-        inputText = `${contextText}\n\n${text}`;
-      }
-
-      const events = agent.chat(
-        inputText,
-        sessionId ? { sessionId } : undefined,
-      );
+      // Convert ContextItem to core Context type
+      const coreContexts: Context[] | undefined = contexts?.map((ctx) => ({
+        id: ctx.id,
+        type: ctx.type as Context["type"],
+        providerId: "ui-selected",
+        label: ctx.label,
+        value: ctx.value,
+        metadata: ctx.metadata,
+        timestamp: Date.now(),
+      }));
+
+      const events = agent.chat(text, {
+        sessionId: sessionId ?? undefined,
+        contexts: coreContexts,
+      });
       await processAgentEvents(events);
     },
     [adapter, agent, sessionId, processAgentEvents],

packages/browser-ext/src/hooks/use-tabs-sync.ts

@@ -3,9 +3,78 @@
  * Monitors tab changes and updates available contexts accordingly
  */
 
+import type { Context } from "@aipexstudio/aipex-core";
 import { useCallback, useEffect, useRef } from "react";
 import type { ContextItem } from "@/components/ai-elements/prompt-input";
-import { getAllAvailableContexts } from "~/lib/context-providers";
+import {
+  BookmarksProvider,
+  CurrentPageProvider,
+  TabsProvider,
+} from "~/lib/context/providers";
+
+const currentPageProvider = new CurrentPageProvider();
+const tabsProvider = new TabsProvider();
+const bookmarksProvider = new BookmarksProvider();
+
+/**
+ * Get all available contexts from providers
+ * Converts core Context to UI ContextItem
+ */
+async function getAllAvailableContexts(): Promise<ContextItem[]> {
+  const results = await Promise.allSettled([
+    currentPageProvider.getContexts(),
+    tabsProvider.getContexts(),
+    bookmarksProvider.getContexts(),
+  ]);
+
+  const contexts: Context[] = [];
+  let currentPageTabId: number | null = null;
+  let currentPageUrl: string | null = null;
+
+  // Current page - add first and record its tab ID and URL
+  if (results[0].status === "fulfilled" && results[0].value.length > 0) {
+    const currentPage = results[0].value[0];
+    contexts.push(currentPage);
+
+    // Extract tab ID from metadata
+    currentPageTabId = currentPage.metadata?.tabId as number | null;
+    currentPageUrl = currentPage.metadata?.url as string | null;
+  }
+
+  // Tabs - exclude the current page tab
+  if (results[1].status === "fulfilled") {
+    const allTabs = results[1].value;
+    const filteredTabs = allTabs.filter((tab) => {
+      const tabId = tab.metadata?.tabId as number | undefined;
+      if (currentPageTabId !== null && tabId === currentPageTabId) {
+        return false;
+      }
+      return true;
+    });
+    contexts.push(...filteredTabs);
+  }
+
+  // Bookmarks - exclude if URL matches current page
+  if (results[2].status === "fulfilled") {
+    const allBookmarks = results[2].value;
+    const filteredBookmarks = allBookmarks.filter((bookmark) => {
+      if (currentPageUrl && bookmark.metadata?.url === currentPageUrl) {
+        return false;
+      }
+      return true;
+    });
+    contexts.push(...filteredBookmarks);
+  }
+
+  // Convert core Context to UI ContextItem
+  return contexts.map((ctx) => ({
+    id: ctx.id,
+    type: ctx.type as ContextItem["type"],
+    label: ctx.label,
+    value: typeof ctx.value === "string" ? ctx.value : "",
+    metadata: ctx.metadata,
+  }));
+}
 
 interface UseTabsSyncOptions {
   /**