@f5-sales-demo/pi-ai 19.51.2

package/package.json

@@ -0,0 +1,135 @@
+{
+	"type": "module",
+	"name": "@f5-sales-demo/pi-ai",
+	"version": "19.51.2",
+	"description": "Unified LLM API with automatic model discovery and provider configuration",
+	"homepage": "https://github.com/f5-sales-demo/xcsh",
+	"author": "Can Boluk",
+	"contributors": [
+		"Mario Zechner"
+	],
+	"license": "MIT",
+	"repository": {
+		"type": "git",
+		"url": "git+https://github.com/f5-sales-demo/xcsh.git",
+		"directory": "packages/ai"
+	},
+	"bugs": {
+		"url": "https://github.com/f5-sales-demo/xcsh/issues"
+	},
+	"keywords": [
+		"ai",
+		"llm",
+		"openai",
+		"anthropic",
+		"gemini",
+		"unified",
+		"api"
+	],
+	"main": "./src/index.ts",
+	"types": "./src/index.ts",
+	"bin": {
+		"pi-ai": "./src/cli.ts"
+	},
+	"scripts": {
+		"check": "biome check . && bun run check:types",
+		"check:types": "tsgo -p tsconfig.json --noEmit",
+		"lint": "biome lint .",
+		"test": "bun test",
+		"fix": "biome check --write --unsafe .",
+		"fmt": "biome format --write .",
+		"generate-models": "bun scripts/generate-models.ts"
+	},
+	"dependencies": {
+		"@anthropic-ai/sdk": "^0.78",
+		"@aws-sdk/client-bedrock-runtime": "^3",
+		"@bufbuild/protobuf": "^2.11",
+		"@google/genai": "^1.43",
+		"@f5-sales-demo/pi-utils": "workspace:*",
+		"@sinclair/typebox": "^0.34",
+		"@smithy/node-http-handler": "^4.4",
+		"ajv": "^8.20",
+		"ajv-formats": "^3.0",
+		"openai": "^6.25",
+		"partial-json": "^0.1",
+		"zod": "^4.3"
+	},
+	"devDependencies": {
+		"@types/bun": "^1.3"
+	},
+	"engines": {
+		"bun": ">=1.3.7"
+	},
+	"files": [
+		"src",
+		"README.md",
+		"CHANGELOG.md"
+	],
+	"exports": {
+		".": {
+			"types": "./src/index.ts",
+			"import": "./src/index.ts"
+		},
+		"./*": {
+			"types": "./src/*.ts",
+			"import": "./src/*.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,95 @@
+/**
+ * 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",
+	"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();
+}