@kevisual/query 0.0.7 → 0.0.9-alpha.1

package/dist/query-adapter.d.ts

@@ -0,0 +1,19 @@
+type AdapterOpts = {
+    url: string;
+    headers?: Record<string, string>;
+    body?: Record<string, any>;
+    timeout?: number;
+};
+/**
+ *
+ * @param opts
+ * @param overloadOpts 覆盖fetch的默认配置
+ * @returns
+ */
+declare const adapter: (opts: AdapterOpts, overloadOpts?: RequestInit) => Promise<any>;
+/**
+ * adapter
+ */
+declare const queryFetch: (opts: AdapterOpts, overloadOpts?: RequestInit) => Promise<any>;
+
+export { adapter, queryFetch };

package/dist/query-adapter.js

@@ -0,0 +1,53 @@
+/**
+ *
+ * @param opts
+ * @param overloadOpts 覆盖fetch的默认配置
+ * @returns
+ */
+const adapter = async (opts, overloadOpts) => {
+    const controller = new AbortController();
+    const signal = controller.signal;
+    const timeout = opts.timeout || 60000 * 3; // 默认超时时间为 60s * 3
+    const timer = setTimeout(() => {
+        controller.abort();
+    }, timeout);
+    return fetch(opts.url, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            ...opts.headers,
+        },
+        body: JSON.stringify(opts.body),
+        signal,
+        ...overloadOpts,
+    })
+        .then((response) => {
+        // 获取 Content-Type 头部信息
+        const contentType = response.headers.get('Content-Type');
+        // 判断返回的数据类型
+        if (contentType && contentType.includes('application/json')) {
+            return response.json(); // 解析为 JSON
+        }
+        else {
+            return response.text(); // 解析为文本
+        }
+    })
+        .catch((err) => {
+        if (err.name === 'AbortError') {
+            console.log('Request timed out and was aborted');
+        }
+        console.error(err);
+        return {
+            code: 500,
+        };
+    })
+        .finally(() => {
+        clearTimeout(timer);
+    });
+};
+/**
+ * adapter
+ */
+const queryFetch = adapter;
+
+export { adapter, queryFetch };

package/dist/query-ai.d.ts

@@ -0,0 +1,26 @@
+import OpenAI, { ClientOptions } from 'openai';
+export { default as OpenAI } from 'openai';
+import { RequestOptions } from 'openai/core.mjs';
+
+type QueryOpts = {
+    /**
+     * OpenAI model name, example: deepseek-chat
+     */
+    model: string;
+    /**
+     * OpenAI client options
+     * QueryAi.init() will be called with these options
+     */
+    openAiOpts?: ClientOptions;
+    openai?: OpenAI;
+};
+declare class QueryAI {
+    private openai;
+    model?: string;
+    constructor(opts?: QueryOpts);
+    init(opts: ClientOptions): void;
+    query(prompt: string, opts?: RequestOptions): Promise<any>;
+    queryAsync(prompt: string, opts?: RequestOptions): Promise<any>;
+}
+
+export { QueryAI };