@aigne/aigne-hub 0.10.17-beta.1 → 1.74.0-beta

package/lib/esm/utils/model.js

@@ -1,197 +0,0 @@
-import { AnthropicChatModel } from "@aigne/anthropic";
-import { BedrockChatModel } from "@aigne/bedrock";
-import { DeepSeekChatModel } from "@aigne/deepseek";
-import { DoubaoChatModel, DoubaoImageModel } from "@aigne/doubao";
-import { GeminiChatModel, GeminiImageModel, GeminiVideoModel } from "@aigne/gemini";
-import { IdeogramImageModel } from "@aigne/ideogram";
-import { OllamaChatModel } from "@aigne/ollama";
-import { OpenRouterChatModel } from "@aigne/open-router";
-import { OpenAIChatModel, OpenAIImageModel, OpenAIVideoModel, } from "@aigne/openai";
-import { PoeChatModel } from "@aigne/poe";
-import { XAIChatModel } from "@aigne/xai";
-import { NodeHttpHandler, streamCollector } from "@smithy/node-http-handler";
-import { HttpsProxyAgent } from "https-proxy-agent";
-import { AIGNEHubImageModel } from "../aigne-hub-image-model.js";
-import { AIGNEHubChatModel } from "../aigne-hub-model.js";
-import { AIGNEHubVideoModel } from "../aigne-hub-video-model.js";
-import { AIGNE_HUB_PROVIDER } from "./constants.js";
-const getClientOptions = () => {
-    const proxy = ["HTTPS_PROXY", "https_proxy", "HTTP_PROXY", "http_proxy", "ALL_PROXY", "all_proxy"]
-        .map((i) => process.env[i])
-        .filter(Boolean)[0];
-    const httpAgent = proxy ? new HttpsProxyAgent(proxy) : undefined;
-    const clientOptions = {
-        fetchOptions: {
-            // @ts-ignore
-            agent: httpAgent,
-        },
-    };
-    return {
-        clientOptions,
-        httpAgent,
-    };
-};
-const GOOGLE = "google";
-export function availableModels() {
-    const { clientOptions, httpAgent } = getClientOptions();
-    return [
-        {
-            name: OpenAIChatModel.name,
-            apiKeyEnvName: "OPENAI_API_KEY",
-            create: (params) => new OpenAIChatModel({ ...params, clientOptions }),
-        },
-        {
-            name: AnthropicChatModel.name,
-            apiKeyEnvName: "ANTHROPIC_API_KEY",
-            create: (params) => new AnthropicChatModel({ ...params, clientOptions }),
-        },
-        {
-            name: BedrockChatModel.name,
-            apiKeyEnvName: "AWS_ACCESS_KEY_ID",
-            create: (params) => new BedrockChatModel({
-                ...params,
-                clientOptions: {
-                    requestHandler: NodeHttpHandler.create({ httpAgent, httpsAgent: httpAgent }),
-                    streamCollector,
-                },
-            }),
-        },
-        {
-            name: DeepSeekChatModel.name,
-            apiKeyEnvName: "DEEPSEEK_API_KEY",
-            create: (params) => new DeepSeekChatModel({ ...params, clientOptions }),
-        },
-        {
-            name: [GeminiChatModel.name, GOOGLE],
-            apiKeyEnvName: ["GEMINI_API_KEY", "GOOGLE_API_KEY"],
-            create: (params) => new GeminiChatModel({
-                ...params,
-                clientOptions: { httpOptions: clientOptions.fetchOptions },
-            }),
-        },
-        {
-            name: OllamaChatModel.name,
-            apiKeyEnvName: "OLLAMA_API_KEY",
-            create: (params) => new OllamaChatModel({ ...params, clientOptions }),
-        },
-        {
-            name: OpenRouterChatModel.name,
-            apiKeyEnvName: "OPEN_ROUTER_API_KEY",
-            create: (params) => new OpenRouterChatModel({ ...params, clientOptions }),
-        },
-        {
-            name: XAIChatModel.name,
-            apiKeyEnvName: "XAI_API_KEY",
-            create: (params) => new XAIChatModel({ ...params, clientOptions }),
-        },
-        {
-            name: DoubaoChatModel.name,
-            apiKeyEnvName: "DOUBAO_API_KEY",
-            create: (params) => new DoubaoChatModel({ ...params, clientOptions }),
-        },
-        {
-            name: PoeChatModel.name,
-            apiKeyEnvName: "POE_API_KEY",
-            create: (params) => new PoeChatModel({ ...params, clientOptions }),
-        },
-        {
-            name: AIGNEHubChatModel.name,
-            apiKeyEnvName: "AIGNE_HUB_API_KEY",
-            create: (params) => new AIGNEHubChatModel({ ...params, clientOptions }),
-        },
-    ];
-}
-export function availableImageModels() {
-    const { clientOptions } = getClientOptions();
-    return [
-        {
-            name: OpenAIImageModel.name,
-            apiKeyEnvName: "OPENAI_API_KEY",
-            create: (params) => new OpenAIImageModel({ ...params, clientOptions }),
-        },
-        {
-            name: [GeminiImageModel.name, GOOGLE],
-            apiKeyEnvName: "GEMINI_API_KEY",
-            create: (params) => new GeminiImageModel({ ...params, clientOptions }),
-        },
-        {
-            name: IdeogramImageModel.name,
-            apiKeyEnvName: "IDEOGRAM_API_KEY",
-            create: (params) => new IdeogramImageModel({ ...params }),
-        },
-        {
-            name: DoubaoImageModel.name,
-            apiKeyEnvName: "DOUBAO_API_KEY",
-            create: (params) => new DoubaoImageModel({ ...params, clientOptions }),
-        },
-        {
-            name: AIGNEHubImageModel.name,
-            apiKeyEnvName: "AIGNE_HUB_API_KEY",
-            create: (params) => new AIGNEHubImageModel({ ...params, clientOptions }),
-        },
-    ];
-}
-export function availableVideoModels() {
-    const { clientOptions } = getClientOptions();
-    return [
-        {
-            name: OpenAIVideoModel.name,
-            apiKeyEnvName: "OPENAI_API_KEY",
-            create: (params) => new OpenAIVideoModel({ ...params, clientOptions }),
-        },
-        {
-            name: [GeminiVideoModel.name, GOOGLE],
-            apiKeyEnvName: "GEMINI_API_KEY",
-            create: (params) => new GeminiVideoModel({ ...params, clientOptions }),
-        },
-        {
-            name: AIGNEHubVideoModel.name,
-            apiKeyEnvName: "AIGNE_HUB_API_KEY",
-            create: (params) => new AIGNEHubVideoModel({ ...params, clientOptions }),
-        },
-    ];
-}
-export function findModel(provider) {
-    provider = provider.toLowerCase().replace(/-/g, "");
-    const all = availableModels();
-    const match = all.find((m) => {
-        if (typeof m.name === "string") {
-            return m.name.toLowerCase().includes(provider);
-        }
-        return m.name.some((n) => n.toLowerCase().includes(provider));
-    });
-    return { all, match };
-}
-export function findImageModel(provider) {
-    provider = provider.toLowerCase().replace(/-/g, "");
-    const all = availableImageModels();
-    const match = all.find((m) => {
-        if (typeof m.name === "string") {
-            return m.name.toLowerCase().includes(provider);
-        }
-        return m.name.some((n) => n.toLowerCase().includes(provider));
-    });
-    return { all, match };
-}
-export function findVideoModel(provider) {
-    provider = provider.toLowerCase().replace(/-/g, "");
-    const all = availableVideoModels();
-    const match = all.find((m) => {
-        if (typeof m.name === "string") {
-            return m.name.toLowerCase().includes(provider);
-        }
-        return m.name.some((n) => n.toLowerCase().includes(provider));
-    });
-    return { all, match };
-}
-export const parseModel = (model) => {
-    // replace first ':' with '/' to compatible with `provider:model-name` format
-    model = model.replace(/^([\w-]+):/, "$1/");
-    let { provider, name } = model.match(/(?<provider>[^/]*)(\/(?<name>.*))?/)?.groups ?? {};
-    provider = provider?.replace(/-/g, "");
-    const match = provider ? findModel(provider)?.match : undefined;
-    if (match) {
-        return { provider, model: name };
-    }
-    return { provider: AIGNE_HUB_PROVIDER, model };
-};

package/lib/esm/utils/type.d.ts

@@ -1,78 +0,0 @@
-import type { ChatModelOptions, ImageModelOptions, VideoModelOptions } from "@aigne/core";
-import type { OpenAIChatModelOptions } from "@aigne/openai";
-import { z } from "zod";
-export declare const aigneHubModelOptionsSchema: z.ZodObject<{
-    baseURL: z.ZodOptional<z.ZodString>;
-    apiKey: z.ZodOptional<z.ZodString>;
-    model: z.ZodOptional<z.ZodString>;
-    modelOptions: z.ZodOptional<z.ZodObject<{
-        model: z.ZodOptional<z.ZodString>;
-        temperature: z.ZodOptional<z.ZodNumber>;
-        topP: z.ZodOptional<z.ZodNumber>;
-        frequencyPenalty: z.ZodOptional<z.ZodNumber>;
-        presencePenalty: z.ZodOptional<z.ZodNumber>;
-        parallelToolCalls: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
-    }, "strip", z.ZodTypeAny, {
-        parallelToolCalls: boolean;
-        model?: string | undefined;
-        temperature?: number | undefined;
-        topP?: number | undefined;
-        frequencyPenalty?: number | undefined;
-        presencePenalty?: number | undefined;
-    }, {
-        model?: string | undefined;
-        temperature?: number | undefined;
-        topP?: number | undefined;
-        frequencyPenalty?: number | undefined;
-        presencePenalty?: number | undefined;
-        parallelToolCalls?: boolean | undefined;
-    }>>;
-    clientOptions: z.ZodOptional<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>>;
-}, "strip", z.ZodTypeAny, {
-    model?: string | undefined;
-    baseURL?: string | undefined;
-    apiKey?: string | undefined;
-    modelOptions?: {
-        parallelToolCalls: boolean;
-        model?: string | undefined;
-        temperature?: number | undefined;
-        topP?: number | undefined;
-        frequencyPenalty?: number | undefined;
-        presencePenalty?: number | undefined;
-    } | undefined;
-    clientOptions?: {} | undefined;
-}, {
-    model?: string | undefined;
-    baseURL?: string | undefined;
-    apiKey?: string | undefined;
-    modelOptions?: {
-        model?: string | undefined;
-        temperature?: number | undefined;
-        topP?: number | undefined;
-        frequencyPenalty?: number | undefined;
-        presencePenalty?: number | undefined;
-        parallelToolCalls?: boolean | undefined;
-    } | undefined;
-    clientOptions?: {} | undefined;
-}>;
-export interface AIGNEHubChatModelOptions extends ChatModelOptions {
-    baseURL?: string;
-    apiKey?: string;
-    clientOptions?: OpenAIChatModelOptions["clientOptions"] & {
-        clientId?: string;
-    };
-}
-export interface AIGNEHubImageModelOptions extends ImageModelOptions {
-    baseURL?: string;
-    apiKey?: string;
-    clientOptions?: OpenAIChatModelOptions["clientOptions"] & {
-        clientId?: string;
-    };
-}
-export interface AIGNEHubVideoModelOptions extends VideoModelOptions {
-    baseURL?: string;
-    apiKey?: string;
-    clientOptions?: OpenAIChatModelOptions["clientOptions"] & {
-        clientId?: string;
-    };
-}

package/lib/esm/utils/type.js

@@ -1,17 +0,0 @@
-import { z } from "zod";
-export const aigneHubModelOptionsSchema = z.object({
-    baseURL: z.string().optional(),
-    apiKey: z.string().optional(),
-    model: z.string().optional(),
-    modelOptions: z
-        .object({
-        model: z.string().optional(),
-        temperature: z.number().optional(),
-        topP: z.number().optional(),
-        frequencyPenalty: z.number().optional(),
-        presencePenalty: z.number().optional(),
-        parallelToolCalls: z.boolean().optional().default(true),
-    })
-        .optional(),
-    clientOptions: z.object({}).optional(),
-});