chat-agent-toolkit 1.2.126 → 1.2.127

package/dist/tools/index.d.ts

@@ -9,4 +9,5 @@
  * @author ai-research-agent contributors
  */
 export { AGENT_TOOLS } from './qwksearch-api-tools';
+export { QwkSearchMCPSession, getQwkSearchTools } from './qwksearch-mcp-tools';
 export * from './search';

package/dist/tools/qwksearch-mcp-tools.d.ts

@@ -0,0 +1,26 @@
+import { ToolSet } from 'ai';
+export interface QwkSearchMCPConfig {
+    command?: string;
+    args?: string[];
+    env?: Record<string, string>;
+}
+/**
+ * Long-lived MCP session for QwkSearch tools.
+ * Spawns a local MCP server process and keeps it alive across tool calls.
+ */
+export declare class QwkSearchMCPSession {
+    private client;
+    private config;
+    constructor(config?: QwkSearchMCPConfig);
+    getTools(): Promise<ToolSet>;
+    close(): Promise<void>;
+}
+/**
+ * One-shot helper to get QwkSearch MCP tools.
+ * Creates a session, fetches tools, and returns them.
+ * Caller should close the returned session when done.
+ */
+export declare function getQwkSearchTools(config?: QwkSearchMCPConfig): Promise<{
+    tools: ToolSet;
+    session: QwkSearchMCPSession;
+}>;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chat-agent-toolkit",
-  "version": "1.2.126",
+  "version": "1.2.127",
   "description": "Multi-provider AI agent toolkit: generate language responses, search the web, extract content, and manage memory across 10+ LLM providers.",
   "author": "vtempest <grokthiscontact@gmail.com>",
   "license": "rights.institute/PROSPER",

package/src/tools/index.ts

@@ -10,4 +10,5 @@
  */
 
 export { AGENT_TOOLS } from "./qwksearch-api-tools";
+export { QwkSearchMCPSession, getQwkSearchTools } from "./qwksearch-mcp-tools";
 export * from "./search";

package/src/tools/qwksearch-mcp-tools.ts

@@ -0,0 +1,57 @@
+import { createMCPClient } from "@ai-sdk/mcp";
+import { Experimental_StdioMCPTransport } from "@ai-sdk/mcp/mcp-stdio";
+import type { MCPClient } from "@ai-sdk/mcp";
+import type { ToolSet } from "ai";
+
+export interface QwkSearchMCPConfig {
+  command?: string;
+  args?: string[];
+  env?: Record<string, string>;
+}
+
+/**
+ * Long-lived MCP session for QwkSearch tools.
+ * Spawns a local MCP server process and keeps it alive across tool calls.
+ */
+export class QwkSearchMCPSession {
+  private client: MCPClient | null = null;
+  private config: QwkSearchMCPConfig;
+
+  constructor(config: QwkSearchMCPConfig = {}) {
+    this.config = config;
+  }
+
+  async getTools(): Promise<ToolSet> {
+    if (!this.client) {
+      this.client = await createMCPClient({
+        transport: new Experimental_StdioMCPTransport({
+          command: this.config.command || "bun",
+          args: this.config.args || ["run", "qwksearch-mcp-server/bin/qwksearch-mcp.ts"],
+          env: {
+            ...process.env as Record<string, string>,
+            ...this.config.env,
+          },
+        }),
+      });
+    }
+    return (await this.client.tools()) as ToolSet;
+  }
+
+  async close() {
+    await this.client?.close();
+    this.client = null;
+  }
+}
+
+/**
+ * One-shot helper to get QwkSearch MCP tools.
+ * Creates a session, fetches tools, and returns them.
+ * Caller should close the returned session when done.
+ */
+export async function getQwkSearchTools(
+  config?: QwkSearchMCPConfig
+): Promise<{ tools: ToolSet; session: QwkSearchMCPSession }> {
+  const session = new QwkSearchMCPSession(config);
+  const tools = await session.getTools();
+  return { tools, session };
+}