Spaces:
Paused
Paused
File size: 3,302 Bytes
0c8b3c0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 | /**
* Codex model discovery — probes backend endpoints for available models.
*/
import { getConfig } from "../config.js";
import { getTransport } from "../tls/transport.js";
import type { BackendModelEntry } from "../models/model-store.js";
let _firstModelFetchLogged = false;
export async function fetchModels(
headers: Record<string, string>,
proxyUrl?: string | null,
): Promise<BackendModelEntry[] | null> {
const config = getConfig();
const transport = getTransport();
const baseUrl = config.api.base_url;
const clientVersion = config.client.app_version;
const endpoints = [
`${baseUrl}/codex/models?client_version=${clientVersion}`,
`${baseUrl}/models`,
`${baseUrl}/sentinel/chat-requirements`,
];
headers["Accept"] = "application/json";
if (!transport.isImpersonate()) {
headers["Accept-Encoding"] = "gzip, deflate";
}
for (const url of endpoints) {
try {
const result = await transport.get(url, headers, 15, proxyUrl);
const parsed = JSON.parse(result.body) as Record<string, unknown>;
const sentinel = parsed.chat_models as Record<string, unknown> | undefined;
const models = sentinel?.models ?? parsed.models ?? parsed.data ?? parsed.categories;
if (Array.isArray(models) && models.length > 0) {
console.log(`[CodexApi] getModels() found ${models.length} entries from ${url}`);
if (!_firstModelFetchLogged) {
console.log(`[CodexApi] Raw response keys: ${Object.keys(parsed).join(", ")}`);
console.log(`[CodexApi] Raw model sample: ${JSON.stringify(models[0]).slice(0, 500)}`);
if (models.length > 1) {
console.log(`[CodexApi] Raw model sample[1]: ${JSON.stringify(models[1]).slice(0, 500)}`);
}
_firstModelFetchLogged = true;
}
// Flatten nested categories into a single list
const flattened: BackendModelEntry[] = [];
for (const item of models) {
if (item && typeof item === "object") {
const entry = item as Record<string, unknown>;
if (Array.isArray(entry.models)) {
for (const sub of entry.models as BackendModelEntry[]) {
flattened.push(sub);
}
} else {
flattened.push(item as BackendModelEntry);
}
}
}
if (flattened.length > 0) {
console.log(`[CodexApi] getModels() total after flatten: ${flattened.length} models`);
return flattened;
}
}
} catch (err) {
const msg = err instanceof Error ? err.message : String(err);
console.log(`[CodexApi] Probe ${url} failed: ${msg}`);
continue;
}
}
return null;
}
export async function probeEndpoint(
path: string,
headers: Record<string, string>,
proxyUrl?: string | null,
): Promise<Record<string, unknown> | null> {
const config = getConfig();
const transport = getTransport();
const url = `${config.api.base_url}${path}`;
headers["Accept"] = "application/json";
if (!transport.isImpersonate()) {
headers["Accept-Encoding"] = "gzip, deflate";
}
try {
const result = await transport.get(url, headers, 15, proxyUrl);
return JSON.parse(result.body) as Record<string, unknown>;
} catch {
return null;
}
}
|