perf: import tool

Author: c121914yu

packages/tool/packages/duckduckgo/search/config.ts

@@ -0,0 +1,42 @@
+import { defineTool } from '@tool/type';
+
+export default defineTool({
+  toolId: 'search',
+  versionList: [
+    {
+      version: '0.1.0',
+      description: 'Default version'
+    }
+  ],
+  type: 'search',
+  name: {
+    'zh-CN': 'DuckDuckGo 网络搜索',
+    en: 'DuckDuckGo Network Search'
+  },
+  description: {
+    'zh-CN': '使用 DuckDuckGo 进行网络搜索',
+    en: 'Use DuckDuckGo to search the web'
+  },
+  icon: 'core/workflow/template/duckduckgo',
+  inputs: [
+    {
+      renderTypeList: ['input', 'reference'],
+      selectedTypeIndex: 0,
+      valueType: 'string',
+      key: 'query',
+      label: 'query',
+      description: '检索词',
+      required: true,
+      toolDescription: '检索词'
+    }
+  ],
+  outputs: [
+    {
+      id: 'result',
+      valueType: 'string',
+      key: 'result',
+      label: '检索结果',
+      type: 'static'
+    }
+  ]
+});

packages/tool/packages/duckduckgo/search/index.ts

@@ -0,0 +1,9 @@
+import config from './config';
+import { InputType, tool as toolCb } from './src';
+import { exportTool } from '@tool/utils/tool';
+
+export default exportTool({
+  toolCb,
+  InputType,
+  config
+});

packages/tool/packages/duckduckgo/search/src/index.ts

@@ -0,0 +1,49 @@
+import { search, SafeSearchType } from 'duck-duck-scrape';
+import { z } from 'zod';
+import { getErrText } from '@tool/utils/err';
+import { delay } from '@tool/utils/delay';
+
+export const InputType = z.object({
+  query: z.string()
+});
+
+export const OutputType = z.object({
+  result: z.string()
+});
+
+const func = async (query: string, retry = 3) => {
+  try {
+    const searchResults = await search(query, {
+      safeSearch: SafeSearchType.STRICT,
+      time: 'y'
+    });
+
+    const result = searchResults.results
+      .map((item) => ({
+        title: item.title,
+        link: item.url,
+        snippet: item.description
+      }))
+      .slice(0, 10);
+
+    return {
+      result: JSON.stringify(result)
+    };
+  } catch (error) {
+    console.log(error);
+    if (retry <= 0) {
+      console.log('DuckDuckGo error', { error });
+      return {
+        result: getErrText(error, 'Failed to fetch data from DuckDuckGo')
+      };
+    }
+
+    await delay(Math.random() * 5000);
+    return func(query, retry - 1);
+  }
+};
+
+export async function tool(props: z.infer<typeof InputType>): Promise<z.infer<typeof OutputType>> {
+  const { result } = await func(props.query);
+  return { result };
+}

…go/children/searchImg/test/index.test.ts …ges/duckduckgo/search/test/index.test.tspackages/tool/packages/duckduckgo/children/searchImg/test/index.test.ts renamed to packages/tool/packages/duckduckgo/search/test/index.test.ts packages/tool/packages/duckduckgo/children/searchImg/test/index.test.ts renamed to packages/tool/packages/duckduckgo/search/test/index.test.ts

@@ -0,0 +1,43 @@
+import { defineTool } from '@tool/type';
+
+export default defineTool({
+  toolId: 'searchImg',
+  versionList: [
+    {
+      version: '0.1.0',
+      description: 'Default version'
+    }
+  ],
+  type: 'search',
+  name: {
+    'zh-CN': 'DuckDuckGo 图片搜索',
+    en: 'DockDuckGo Image Search'
+  },
+  description: {
+    'zh-CN': '使用 DuckDuckGo 进行图片搜索',
+    en: 'Use DuckDuckGo to search images'
+  },
+  icon: 'core/workflow/template/duckduckgo',
+  inputs: [
+    {
+      renderTypeList: ['input', 'reference'],
+      selectedTypeIndex: 0,
+      valueType: 'string',
+      key: 'query',
+      label: 'query',
+      description: '检索词',
+      required: true,
+      toolDescription: '检索词'
+    }
+  ],
+  outputs: [
+    {
+      id: 'result',
+      type: 'static',
+      valueType: 'string',
+      key: 'result',
+      label: 'result',
+      description: ' 检索结果'
+    }
+  ]
+});

packages/tool/packages/duckduckgo/searchImg/config.ts

@@ -0,0 +1,9 @@
+import config from './config';
+import { InputType, tool as toolCb } from './src';
+import { exportTool } from '@tool/utils/tool';
+
+export default exportTool({
+  toolCb,
+  InputType,
+  config
+});

packages/tool/packages/duckduckgo/searchImg/index.ts

@@ -0,0 +1,45 @@
+import { delay } from '@tool/utils/delay';
+import { getErrText } from '@tool/utils/err';
+import { SafeSearchType, searchImages } from 'duck-duck-scrape';
+import { z } from 'zod';
+
+export const InputType = z.object({
+  query: z.string()
+});
+
+export const OutputType = z.object({
+  result: z.string()
+});
+
+const func = async (query: string, retry = 3) => {
+  try {
+    const searchResults = await searchImages(query, {
+      safeSearch: SafeSearchType.STRICT
+    });
+
+    const result = searchResults.results
+      .map((item) => ({
+        title: item.title,
+        image: item.image
+      }))
+      .slice(0, 10);
+
+    return {
+      result: JSON.stringify(result)
+    };
+  } catch (error) {
+    if (retry <= 0) {
+      return {
+        result: getErrText(error, 'Failed to fetch data from DuckDuckGo')
+      };
+    }
+
+    await delay(Math.random() * 5000);
+    return func(query, retry - 1);
+  }
+};
+
+export async function tool(props: z.infer<typeof InputType>): Promise<z.infer<typeof OutputType>> {
+  const { result } = await func(props.query);
+  return { result };
+}

packages/tool/packages/duckduckgo/searchImg/src/index.ts

@@ -0,0 +1,28 @@
+import { expect, test, vi } from 'vitest';
+import { main } from '..';
+
+test('get current time', async () => {
+  // Mock Date constructor to return a fixed date
+  const date = new Date('2023-01-01T12:00:00.000');
+  vi.useFakeTimers();
+  vi.setSystemTime(date);
+  const res = await main({
+    formatStr: 'yyyy-MM-dd HH:mm:ss'
+  });
+  expect(res.output?.time).toBeDefined();
+  expect(res.output?.time).toEqual('2023-01-01 12:00:00');
+  vi.useRealTimers();
+});
+
+test('get current time with a invalid format', async () => {
+  // Mock Date constructor to return a fixed date
+  const date = new Date('2023-01-01T12:00:00.000');
+  vi.useFakeTimers();
+  vi.setSystemTime(date);
+  const res = await main({
+    formatStr: 'someFormatStrisInvalid'
+  });
+  expect(res.error).toBeDefined();
+  expect(res.output?.time).toBeUndefined();
+  vi.useRealTimers();
+});

packages/tool/packages/duckduckgo/searchImg/test/index.test.ts

@@ -0,0 +1,43 @@
+import { defineTool } from '@tool/type';
+
+export default defineTool({
+  toolId: 'searchNews',
+  versionList: [
+    {
+      version: '0.1.0',
+      description: 'Default version'
+    }
+  ],
+  type: 'search',
+  name: {
+    'zh-CN': 'DuckDuckGo 新闻检索',
+    en: 'DockDuckGo News Search'
+  },
+  description: {
+    'zh-CN': '使用 DuckDuckGo 进行新闻检索',
+    en: 'Use DuckDuckGo to search news'
+  },
+  icon: 'core/workflow/template/duckduckgo',
+  inputs: [
+    {
+      renderTypeList: ['input', 'reference'],
+      selectedTypeIndex: 0,
+      valueType: 'string',
+      key: 'query',
+      label: 'query',
+      description: '检索词',
+      required: true,
+      toolDescription: '检索词'
+    }
+  ],
+  outputs: [
+    {
+      id: 'result',
+      type: 'static',
+      valueType: 'string',
+      key: 'result',
+      label: 'result',
+      description: ' 检索结果'
+    }
+  ]
+});

packages/tool/packages/duckduckgo/searchNews/config.ts

@@ -0,0 +1,9 @@
+import config from './config';
+import { InputType, tool as toolCb } from './src';
+import { exportTool } from '@tool/utils/tool';
+
+export default exportTool({
+  toolCb,
+  InputType,
+  config
+});