@aryee337/aery-ai 0.2.28 → 0.2.29

package/package.json

@@ -1,106 +1,141 @@
 {
-  "name": "@aryee337/aery-ai",
-  "version": "0.2.28",
-  "description": "Unified LLM API with automatic model discovery and provider configuration",
-  "type": "module",
-  "main": "./dist/index.js",
-  "types": "./dist/index.d.ts",
-  "exports": {
-    ".": {
-      "types": "./dist/index.d.ts",
-      "import": "./dist/index.js"
-    },
-    "./anthropic": {
-      "types": "./dist/providers/anthropic.d.ts",
-      "import": "./dist/providers/anthropic.js"
-    },
-    "./azure-openai-responses": {
-      "types": "./dist/providers/azure-openai-responses.d.ts",
-      "import": "./dist/providers/azure-openai-responses.js"
-    },
-    "./google": {
-      "types": "./dist/providers/google.d.ts",
-      "import": "./dist/providers/google.js"
-    },
-    "./google-vertex": {
-      "types": "./dist/providers/google-vertex.d.ts",
-      "import": "./dist/providers/google-vertex.js"
-    },
-    "./mistral": {
-      "types": "./dist/providers/mistral.d.ts",
-      "import": "./dist/providers/mistral.js"
-    },
-    "./openai-codex-responses": {
-      "types": "./dist/providers/openai-codex-responses.d.ts",
-      "import": "./dist/providers/openai-codex-responses.js"
-    },
-    "./openai-completions": {
-      "types": "./dist/providers/openai-completions.d.ts",
-      "import": "./dist/providers/openai-completions.js"
-    },
-    "./openai-responses": {
-      "types": "./dist/providers/openai-responses.d.ts",
-      "import": "./dist/providers/openai-responses.js"
-    },
-    "./oauth": {
-      "types": "./dist/oauth.d.ts",
-      "import": "./dist/oauth.js"
-    },
-    "./bedrock-provider": {
-      "types": "./dist/bedrock-provider.d.ts",
-      "import": "./dist/bedrock-provider.js"
-    }
-  },
-  "bin": {
-    "pi-ai": "./dist/cli.js"
-  },
-  "files": [
-    "dist",
-    "README.md"
-  ],
-  "scripts": {
-    "clean": "shx rm -rf dist",
-    "generate-models": "node scripts/generate-models.ts",
-    "generate-image-models": "node scripts/generate-image-models.ts",
-    "build": "npm run generate-models && npm run generate-image-models && tsgo -p tsconfig.build.json",
-    "test": "vitest --run",
-    "prepublishOnly": "npm run clean && npm run build"
-  },
-  "dependencies": {
-    "@anthropic-ai/sdk": "0.91.1",
-    "@aws-sdk/client-bedrock-runtime": "3.1048.0",
-    "@smithy/node-http-handler": "4.7.3",
-    "@google/genai": "1.52.0",
-    "@mistralai/mistralai": "2.2.1",
-    "http-proxy-agent": "7.0.2",
-    "https-proxy-agent": "7.0.6",
-    "openai": "6.26.0",
-    "partial-json": "0.1.7",
-    "typebox": "1.1.38"
-  },
-  "keywords": [
-    "ai",
-    "llm",
-    "openai",
-    "anthropic",
-    "gemini",
-    "bedrock",
-    "unified",
-    "api"
-  ],
-  "author": "Mario Zechner",
-  "license": "MIT",
-  "repository": {
-    "type": "git",
-    "url": "git+https://github.com/eminent337/aery.git",
-    "directory": "packages/ai"
-  },
-  "engines": {
-    "node": ">=22.19.0"
-  },
-  "devDependencies": {
-    "@types/node": "24.12.4",
-    "canvas": "3.2.3",
-    "vitest": "3.2.4"
-  }
-}
+	"type": "module",
+	"name": "@aryee337/aery-ai",
+	"version": "0.2.29",
+	"description": "Unified LLM API with automatic model discovery and provider configuration",
+	"homepage": "https://aery.dev",
+	"author": "aryee337",
+	"contributors": [
+		
+	],
+	"license": "MIT",
+	"repository": {
+		"type": "git",
+		"url": "git+https://github.com/eminent337/aery.git",
+		"directory": "packages/ai"
+	},
+	"bugs": {
+		"url": "https://github.com/eminent337/aery/issues"
+	},
+	"keywords": [
+		"ai",
+		"llm",
+		"openai",
+		"anthropic",
+		"gemini",
+		"unified",
+		"api"
+	],
+	"main": "./src/index.ts",
+	"types": "./src/index.ts",
+	"scripts": {
+		"check": "biome check . && bun run check:types",
+		"check:types": "tsgo -p tsconfig.json --noEmit",
+		"lint": "biome lint .",
+		"test": "bun test --parallel",
+		"fix": "biome check --write --unsafe .",
+		"fmt": "biome format --write .",
+		"generate-models": "bun scripts/generate-models.ts"
+	},
+	"dependencies": {
+		"@bufbuild/protobuf": "catalog:",
+		"@aryee337/aery-utils": "catalog:",
+		"openai": "catalog:",
+		"partial-json": "catalog:",
+		"zod": "catalog:"
+	},
+	"devDependencies": {
+		"@types/bun": "catalog:"
+	},
+	"engines": {
+		"bun": ">=1.3.14"
+	},
+	"files": [
+		"src",
+		"README.md",
+		"CHANGELOG.md"
+	],
+	"exports": {
+		".": {
+			"types": "./src/index.ts",
+			"import": "./src/index.ts"
+		},
+		"./*": {
+			"types": "./src/*.ts",
+			"import": "./src/*.ts"
+		},
+		"./auth-broker": {
+			"types": "./src/auth-broker/index.ts",
+			"import": "./src/auth-broker/index.ts"
+		},
+		"./auth-broker/*": {
+			"types": "./src/auth-broker/*.ts",
+			"import": "./src/auth-broker/*.ts"
+		},
+		"./auth-gateway": {
+			"types": "./src/auth-gateway/index.ts",
+			"import": "./src/auth-gateway/index.ts"
+		},
+		"./auth-gateway/*": {
+			"types": "./src/auth-gateway/*.ts",
+			"import": "./src/auth-gateway/*.ts"
+		},
+		"./models.json": {
+			"types": "./src/models.json.d.ts",
+			"import": "./src/models.json"
+		},
+		"./provider-models": {
+			"types": "./src/provider-models/index.ts",
+			"import": "./src/provider-models/index.ts"
+		},
+		"./provider-models/*": {
+			"types": "./src/provider-models/*.ts",
+			"import": "./src/provider-models/*.ts"
+		},
+		"./providers/*": {
+			"types": "./src/providers/*.ts",
+			"import": "./src/providers/*.ts"
+		},
+		"./providers/cursor/gen/*": {
+			"types": "./src/providers/cursor/gen/*.ts",
+			"import": "./src/providers/cursor/gen/*.ts"
+		},
+		"./providers/openai-codex/*": {
+			"types": "./src/providers/openai-codex/*.ts",
+			"import": "./src/providers/openai-codex/*.ts"
+		},
+		"./usage/*": {
+			"types": "./src/usage/*.ts",
+			"import": "./src/usage/*.ts"
+		},
+		"./utils/*": {
+			"types": "./src/utils/*.ts",
+			"import": "./src/utils/*.ts"
+		},
+		"./utils/discovery": {
+			"types": "./src/utils/discovery/index.ts",
+			"import": "./src/utils/discovery/index.ts"
+		},
+		"./utils/discovery/*": {
+			"types": "./src/utils/discovery/*.ts",
+			"import": "./src/utils/discovery/*.ts"
+		},
+		"./utils/oauth": {
+			"types": "./src/utils/oauth/index.ts",
+			"import": "./src/utils/oauth/index.ts"
+		},
+		"./utils/oauth/*": {
+			"types": "./src/utils/oauth/*.ts",
+			"import": "./src/utils/oauth/*.ts"
+		},
+		"./utils/schema": {
+			"types": "./src/utils/schema/index.ts",
+			"import": "./src/utils/schema/index.ts"
+		},
+		"./utils/schema/*": {
+			"types": "./src/utils/schema/*.ts",
+			"import": "./src/utils/schema/*.ts"
+		},
+		"./*.js": "./src/*.ts"
+	}
+}

package/src/api-registry.ts

@@ -0,0 +1,96 @@
+/**
+ * Custom API provider registry.
+ *
+ * Allows extensions to register streaming functions for custom API types
+ * (e.g., "vertex-claude-api") that are not built into stream.ts.
+ */
+import type {
+	Api,
+	AssistantMessageEventStream,
+	Context,
+	KnownApi,
+	Model,
+	SimpleStreamOptions,
+	StreamOptions,
+} from "./types";
+
+const BUILTIN_APIS = new Set<KnownApi>([
+	"openai-completions",
+	"openai-responses",
+	"openai-codex-responses",
+	"azure-openai-responses",
+	"anthropic-messages",
+	"bedrock-converse-stream",
+	"google-generative-ai",
+	"google-gemini-cli",
+	"google-vertex",
+	"ollama-chat",
+	"cursor-agent",
+]);
+
+export type CustomStreamFn = (
+	model: Model<Api>,
+	context: Context,
+	options?: StreamOptions,
+) => AssistantMessageEventStream;
+export type CustomStreamSimpleFn = (
+	model: Model<Api>,
+	context: Context,
+	options?: SimpleStreamOptions,
+) => AssistantMessageEventStream;
+
+export interface RegisteredCustomApi {
+	stream: CustomStreamFn;
+	streamSimple: CustomStreamSimpleFn;
+	sourceId?: string;
+}
+
+const customApiRegistry = new Map<string, RegisteredCustomApi>();
+
+function assertCustomApiName(api: string): void {
+	if (BUILTIN_APIS.has(api as KnownApi)) {
+		throw new Error(`Cannot register custom API "${api}": built-in API names are reserved.`);
+	}
+}
+
+/**
+ * Register a custom API streaming function.
+ */
+export function registerCustomApi(
+	api: string,
+	streamSimple: CustomStreamSimpleFn,
+	sourceId?: string,
+	stream?: CustomStreamFn,
+): void {
+	assertCustomApiName(api);
+	customApiRegistry.set(api, {
+		stream: stream ?? ((model, context, options) => streamSimple(model, context, options as SimpleStreamOptions)),
+		streamSimple,
+		sourceId,
+	});
+}
+
+/**
+ * Get a custom API provider by API identifier.
+ */
+export function getCustomApi(api: string): RegisteredCustomApi | undefined {
+	return customApiRegistry.get(api);
+}
+
+/**
+ * Remove all custom APIs registered by a specific source (e.g., extension path).
+ */
+export function unregisterCustomApis(sourceId: string): void {
+	for (const [api, entry] of customApiRegistry.entries()) {
+		if (entry.sourceId === sourceId) {
+			customApiRegistry.delete(api);
+		}
+	}
+}
+
+/**
+ * Clear all custom API registrations.
+ */
+export function clearCustomApis(): void {
+	customApiRegistry.clear();
+}