@cogitator-ai/core 0.7.0 → 0.8.0

package/dist/tools/web-search.js

@@ -0,0 +1,176 @@
+import { z } from 'zod';
+import { tool } from '../tool';
+const webSearchParams = z.object({
+    query: z.string().min(1).describe('Search query'),
+    provider: z
+        .enum(['tavily', 'brave', 'serper'])
+        .optional()
+        .describe('Search provider (default: auto-detect from available API keys)'),
+    maxResults: z
+        .number()
+        .int()
+        .min(1)
+        .max(20)
+        .optional()
+        .describe('Maximum number of results (default: 5, max: 20)'),
+    searchDepth: z
+        .enum(['basic', 'advanced'])
+        .optional()
+        .describe('Search depth for Tavily (default: basic)'),
+    includeAnswer: z
+        .boolean()
+        .optional()
+        .describe('Include AI-generated answer summary (Tavily only, default: false)'),
+});
+async function searchTavily(query, maxResults, searchDepth, includeAnswer, apiKey) {
+    const response = await fetch('https://api.tavily.com/search', {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+        },
+        body: JSON.stringify({
+            api_key: apiKey,
+            query,
+            max_results: maxResults,
+            search_depth: searchDepth,
+            include_answer: includeAnswer,
+        }),
+    });
+    if (!response.ok) {
+        const text = await response.text();
+        throw new Error(`Tavily API error: ${response.status} ${text}`);
+    }
+    const data = (await response.json());
+    return {
+        query,
+        provider: 'tavily',
+        results: data.results.map((r) => ({
+            title: r.title,
+            url: r.url,
+            snippet: r.content,
+            score: r.score,
+        })),
+        answer: data.answer,
+    };
+}
+async function searchBrave(query, maxResults, apiKey) {
+    const params = new URLSearchParams({
+        q: query,
+        count: maxResults.toString(),
+    });
+    const response = await fetch(`https://api.search.brave.com/res/v1/web/search?${params}`, {
+        headers: {
+            Accept: 'application/json',
+            'X-Subscription-Token': apiKey,
+        },
+    });
+    if (!response.ok) {
+        const text = await response.text();
+        throw new Error(`Brave Search API error: ${response.status} ${text}`);
+    }
+    const data = (await response.json());
+    return {
+        query,
+        provider: 'brave',
+        results: (data.web?.results ?? []).map((r) => ({
+            title: r.title,
+            url: r.url,
+            snippet: r.description,
+        })),
+    };
+}
+async function searchSerper(query, maxResults, apiKey) {
+    const response = await fetch('https://google.serper.dev/search', {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            'X-API-KEY': apiKey,
+        },
+        body: JSON.stringify({
+            q: query,
+            num: maxResults,
+        }),
+    });
+    if (!response.ok) {
+        const text = await response.text();
+        throw new Error(`Serper API error: ${response.status} ${text}`);
+    }
+    const data = (await response.json());
+    return {
+        query,
+        provider: 'serper',
+        results: data.organic.map((r) => ({
+            title: r.title,
+            url: r.link,
+            snippet: r.snippet,
+        })),
+        answer: data.answerBox?.answer ?? data.answerBox?.snippet,
+    };
+}
+function detectProvider() {
+    const tavily = process.env.TAVILY_API_KEY;
+    if (tavily)
+        return { provider: 'tavily', apiKey: tavily };
+    const brave = process.env.BRAVE_API_KEY;
+    if (brave)
+        return { provider: 'brave', apiKey: brave };
+    const serper = process.env.SERPER_API_KEY;
+    if (serper)
+        return { provider: 'serper', apiKey: serper };
+    return null;
+}
+function getApiKey(provider) {
+    switch (provider) {
+        case 'tavily':
+            return process.env.TAVILY_API_KEY ?? null;
+        case 'brave':
+            return process.env.BRAVE_API_KEY ?? null;
+        case 'serper':
+            return process.env.SERPER_API_KEY ?? null;
+    }
+}
+export const webSearch = tool({
+    name: 'web_search',
+    description: 'Search the web using Tavily, Brave, or Serper APIs. Returns relevant results with titles, URLs, and snippets. Requires API key in environment (TAVILY_API_KEY, BRAVE_API_KEY, or SERPER_API_KEY).',
+    parameters: webSearchParams,
+    category: 'web',
+    tags: ['search', 'web', 'internet'],
+    execute: async ({ query, provider: requestedProvider, maxResults = 5, searchDepth = 'basic', includeAnswer = false, }) => {
+        let provider;
+        let apiKey;
+        if (requestedProvider) {
+            const key = getApiKey(requestedProvider);
+            if (!key) {
+                return {
+                    error: `API key not found for ${requestedProvider}. Set ${requestedProvider.toUpperCase()}_API_KEY environment variable.`,
+                };
+            }
+            provider = requestedProvider;
+            apiKey = key;
+        }
+        else {
+            const detected = detectProvider();
+            if (!detected) {
+                return {
+                    error: 'No search API key found. Set one of: TAVILY_API_KEY, BRAVE_API_KEY, or SERPER_API_KEY',
+                };
+            }
+            provider = detected.provider;
+            apiKey = detected.apiKey;
+        }
+        try {
+            switch (provider) {
+                case 'tavily':
+                    return await searchTavily(query, maxResults, searchDepth, includeAnswer, apiKey);
+                case 'brave':
+                    return await searchBrave(query, maxResults, apiKey);
+                case 'serper':
+                    return await searchSerper(query, maxResults, apiKey);
+            }
+        }
+        catch (err) {
+            return { error: err.message, query, provider };
+        }
+    },
+});
+//# sourceMappingURL=web-search.js.map

package/dist/tools/web-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"web-search.js","sourceRoot":"","sources":["../../src/tools/web-search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAE/B,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC;IACjD,QAAQ,EAAE,CAAC;SACR,IAAI,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;SACnC,QAAQ,EAAE;SACV,QAAQ,CAAC,gEAAgE,CAAC;IAC7E,UAAU,EAAE,CAAC;SACV,MAAM,EAAE;SACR,GAAG,EAAE;SACL,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,EAAE,CAAC;SACP,QAAQ,EAAE;SACV,QAAQ,CAAC,iDAAiD,CAAC;IAC9D,WAAW,EAAE,CAAC;SACX,IAAI,CAAC,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;SAC3B,QAAQ,EAAE;SACV,QAAQ,CAAC,0CAA0C,CAAC;IACvD,aAAa,EAAE,CAAC;SACb,OAAO,EAAE;SACT,QAAQ,EAAE;SACV,QAAQ,CAAC,mEAAmE,CAAC;CACjF,CAAC,CAAC;AAgBH,KAAK,UAAU,YAAY,CACzB,KAAa,EACb,UAAkB,EAClB,WAAiC,EACjC,aAAsB,EACtB,MAAc;IAEd,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,+BAA+B,EAAE;QAC5D,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;SACnC;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;YACnB,OAAO,EAAE,MAAM;YACf,KAAK;YACL,WAAW,EAAE,UAAU;YACvB,YAAY,EAAE,WAAW;YACzB,cAAc,EAAE,aAAa;SAC9B,CAAC;KACH,CAAC,CAAC;IAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC,CAAC;IAClE,CAAC;IAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAGlC,CAAC;IAEF,OAAO;QACL,KAAK;QACL,QAAQ,EAAE,QAAQ;QAClB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAChC,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,GAAG,EAAE,CAAC,CAAC,GAAG;YACV,OAAO,EAAE,CAAC,CAAC,OAAO;YAClB,KAAK,EAAE,CAAC,CAAC,KAAK;SACf,CAAC,CAAC;QACH,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,WAAW,CACxB,KAAa,EACb,UAAkB,EAClB,MAAc;IAEd,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC;QACjC,CAAC,EAAE,KAAK;QACR,KAAK,EAAE,UAAU,CAAC,QAAQ,EAAE;KAC7B,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,kDAAkD,MAAM,EAAE,EAAE;QACvF,OAAO,EAAE;YACP,MAAM,EAAE,kBAAkB;YAC1B,sBAAsB,EAAE,MAAM;SAC/B;KACF,CAAC,CAAC;IAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,2BAA2B,QAAQ,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC,CAAC;IACxE,CAAC;IAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAElC,CAAC;IAEF,OAAO;QACL,KAAK;QACL,QAAQ,EAAE,OAAO;QACjB,OAAO,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC7C,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,GAAG,EAAE,CAAC,CAAC,GAAG;YACV,OAAO,EAAE,CAAC,CAAC,WAAW;SACvB,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,YAAY,CACzB,KAAa,EACb,UAAkB,EAClB,MAAc;IAEd,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,kCAAkC,EAAE;QAC/D,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,WAAW,EAAE,MAAM;SACpB;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;YACnB,CAAC,EAAE,KAAK;YACR,GAAG,EAAE,UAAU;SAChB,CAAC;KACH,CAAC,CAAC;IAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC,CAAC;IAClE,CAAC;IAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAGlC,CAAC;IAEF,OAAO;QACL,KAAK;QACL,QAAQ,EAAE,QAAQ;QAClB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAChC,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,GAAG,EAAE,CAAC,CAAC,IAAI;YACX,OAAO,EAAE,CAAC,CAAC,OAAO;SACnB,CAAC,CAAC;QACH,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,MAAM,IAAI,IAAI,CAAC,SAAS,EAAE,OAAO;KAC1D,CAAC;AACJ,CAAC;AAED,SAAS,cAAc;IACrB,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IAC1C,IAAI,MAAM;QAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;IAE1D,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;IACxC,IAAI,KAAK;QAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;IAEvD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IAC1C,IAAI,MAAM;QAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;IAE1D,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,SAAS,CAAC,QAAuC;IACxD,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,QAAQ;YACX,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,IAAI,CAAC;QAC5C,KAAK,OAAO;YACV,OAAO,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,IAAI,CAAC;QAC3C,KAAK,QAAQ;YACX,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,IAAI,CAAC;IAC9C,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,SAAS,GAAG,IAAI,CAAC;IAC5B,IAAI,EAAE,YAAY;IAClB,WAAW,EACT,mMAAmM;IACrM,UAAU,EAAE,eAAe;IAC3B,QAAQ,EAAE,KAAK;IACf,IAAI,EAAE,CAAC,QAAQ,EAAE,KAAK,EAAE,UAAU,CAAC;IACnC,OAAO,EAAE,KAAK,EAAE,EACd,KAAK,EACL,QAAQ,EAAE,iBAAiB,EAC3B,UAAU,GAAG,CAAC,EACd,WAAW,GAAG,OAAO,EACrB,aAAa,GAAG,KAAK,GACtB,EAAE,EAAE;QACH,IAAI,QAAuC,CAAC;QAC5C,IAAI,MAAc,CAAC;QAEnB,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,GAAG,GAAG,SAAS,CAAC,iBAAiB,CAAC,CAAC;YACzC,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,OAAO;oBACL,KAAK,EAAE,yBAAyB,iBAAiB,SAAS,iBAAiB,CAAC,WAAW,EAAE,gCAAgC;iBAC1H,CAAC;YACJ,CAAC;YACD,QAAQ,GAAG,iBAAiB,CAAC;YAC7B,MAAM,GAAG,GAAG,CAAC;QACf,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,cAAc,EAAE,CAAC;YAClC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO;oBACL,KAAK,EACH,uFAAuF;iBAC1F,CAAC;YACJ,CAAC;YACD,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC;YAC7B,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;QAC3B,CAAC;QAED,IAAI,CAAC;YACH,QAAQ,QAAQ,EAAE,CAAC;gBACjB,KAAK,QAAQ;oBACX,OAAO,MAAM,YAAY,CAAC,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,CAAC,CAAC;gBACnF,KAAK,OAAO;oBACV,OAAO,MAAM,WAAW,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;gBACtD,KAAK,QAAQ;oBACX,OAAO,MAAM,YAAY,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;YACzD,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;QAC5D,CAAC;IACH,CAAC;CACF,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cogitator-ai/core",
-  "version": "0.7.0",
+  "version": "0.8.0",
   "description": "Core runtime for Cogitator AI agents",
   "type": "module",
   "main": "./dist/index.js",
@@ -22,9 +22,9 @@
     "openai": "^4.24.0",
     "zod": "^3.22.4",
     "zod-to-json-schema": "^3.22.4",
-    "@cogitator-ai/memory": "0.4.2",
-    "@cogitator-ai/models": "6.0.0",
-    "@cogitator-ai/types": "0.8.0"
+    "@cogitator-ai/memory": "0.4.3",
+    "@cogitator-ai/types": "0.8.1",
+    "@cogitator-ai/models": "6.0.1"
   },
   "optionalDependencies": {
     "pg": "^8.11.3"
@@ -37,7 +37,7 @@
   "peerDependencies": {
     "@aws-sdk/client-bedrock-runtime": "^3.0.0",
     "pg": "^8.0.0",
-    "@cogitator-ai/sandbox": "0.2.5"
+    "@cogitator-ai/sandbox": "0.2.6"
   },
   "peerDependenciesMeta": {
     "@aws-sdk/client-bedrock-runtime": {