summaryrefslogtreecommitdiff
path: root/config/pi/extensions/provider-discovery.ts.disabled
diff options
context:
space:
mode:
authorIgor Soarez <igor@soarez.org>2026-09-17 19:45:23 +0100
committerIgor Soarez <igor@soarez.org>2026-09-17 19:45:23 +0100
commit13cc065ec014b86369dfe4ad60d1a43fa06fd251 (patch)
tree684690c61b7b59758c2aac132c65b81c86de0608 /config/pi/extensions/provider-discovery.ts.disabled
dotfiles
Shell, editor and terminal configuration, the tools that go with them, and szecret, an age-based secret manager whose store lives elsewhere. History restarts here. What came before was not written to be read by anyone else, and carried machine names, addresses and an encrypted secret store; it is kept privately instead. What is left is meant to be safe on any machine, including one whose owner is not me: see CLAUDE.md for the rule it is written to.
Diffstat (limited to 'config/pi/extensions/provider-discovery.ts.disabled')
-rw-r--r--config/pi/extensions/provider-discovery.ts.disabled108
1 files changed, 108 insertions, 0 deletions
diff --git a/config/pi/extensions/provider-discovery.ts.disabled b/config/pi/extensions/provider-discovery.ts.disabled
new file mode 100644
index 0000000..759a2fb
--- /dev/null
+++ b/config/pi/extensions/provider-discovery.ts.disabled
@@ -0,0 +1,108 @@
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+
+interface ProviderConfig {
+ id: string;
+ baseUrl: string;
+ envKey: string;
+ suffix: string;
+}
+
+const PROVIDERS: ProviderConfig[] = [
+ {
+ id: "siliconflow",
+ baseUrl: "https://api.siliconflow.com/v1",
+ envKey: "SILICONFLOW_API_KEY",
+ suffix: "SF"
+ },
+ {
+ id: "deepinfra",
+ baseUrl: "https://api.deepinfra.com/v1/openai",
+ envKey: "DEEPINFRA_API_KEY",
+ suffix: "DI"
+ }
+];
+
+// Patterns matched in order; first match wins.
+const CONTEXT_OVERRIDES: Array<[RegExp, number]> = [
+ [/qwen3.*coder/i, 262144],
+ [/qwen3.*next/i, 262144],
+ [/qwen3-235b|qwen3-32b|qwen3-14b|qwen3-8b|qwen3-30b/i, 131072],
+ [/qwq-32b/i, 131072],
+ [/qwen2\.5-.*-128k/i, 131072],
+ [/qwen2\.5/i, 32768],
+ [/deepseek-v3|deepseek-r1/i, 131072],
+ [/kimi-k2/i, 131072],
+ [/glm-4\.[567]|glm-5/i, 131072],
+ [/gpt-oss/i, 131072],
+ [/minimax/i, 245760],
+ [/hunyuan/i, 32768],
+];
+
+function resolveContextWindow(modelId: string): number {
+ for (const [pattern, ctx] of CONTEXT_OVERRIDES) {
+ if (pattern.test(modelId)) return ctx;
+ }
+ return 32768; // conservative fallback
+}
+
+export default async function (pi: ExtensionAPI) {
+ console.log("Provider discovery: Starting...");
+
+ await Promise.all(PROVIDERS.map(async (config) => {
+ const apiKey = process.env[config.envKey];
+ if (!apiKey) {
+ console.warn(`Provider discovery: ${config.envKey} not found. Skipping ${config.id}.`);
+ return;
+ }
+
+ try {
+ const response = await fetch(`${config.baseUrl}/models`, {
+ headers: { "Authorization": `Bearer ${apiKey}` }
+ });
+
+ if (!response.ok) {
+ throw new Error(`API error ${response.status}`);
+ }
+
+ const payload = await response.json();
+ const models = payload.data
+ .filter((model: any) => {
+ // Heuristic to filter for chat/text models
+ const id = model.id.toLowerCase();
+ return id.includes("/") || model.type === "text-generation" || id.includes("gpt") || id.includes("llama") || id.includes("claude");
+ })
+ .map((model: any) => {
+ const id = model.id;
+ const isReasoning = id.toLowerCase().includes("r1") ||
+ id.toLowerCase().includes("thinking") ||
+ id.toLowerCase().includes("reasoning") ||
+ id.toLowerCase().includes("qwq");
+
+ return {
+ id: id,
+ name: `${id.split('/').pop()} (${config.suffix})`,
+ reasoning: isReasoning,
+ input: ["text"],
+ contextWindow: model.context_window ?? resolveContextWindow(id),
+ maxTokens: 4096
+ };
+ });
+
+ pi.registerProvider(config.id, {
+ baseUrl: config.baseUrl,
+ apiKey: config.envKey,
+ api: "openai-completions",
+ models: models,
+ compat: {
+ supportsDeveloperRole: false,
+ supportsStrictMode: false,
+ requiresToolResultName: true
+ },
+ });
+
+ console.log(`Provider discovery: Registered ${models.length} models for ${config.id}.`);
+ } catch (error) {
+ console.error(`Provider discovery: Failed to fetch models for ${config.id}:`, error);
+ }
+ }));
+}