File size: 1,544 Bytes
f415c95 |
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 |
import type { MaxTokensCache } from "./index.js";
const FIREWORKS_API_URL = "https://api.fireworks.ai/inference/v1/models"; // Assumed
export async function fetchFireworksData(apiKey: string | undefined): Promise<MaxTokensCache["fireworks-ai"]> {
if (!apiKey) {
console.warn("Fireworks AI API key not provided. Skipping Fireworks AI fetch.");
return {};
}
try {
const response = await fetch(FIREWORKS_API_URL, {
headers: {
Authorization: `Bearer ${apiKey}`,
},
});
if (!response.ok) {
throw new Error(`Fireworks AI API request failed: ${response.status} ${response.statusText}`);
}
// eslint-disable-next-line @typescript-eslint/no-explicit-any
const data: any = await response.json(); // Assuming OpenAI structure { data: [ { id: string, ... } ] }
const modelsData: MaxTokensCache["fireworks-ai"] = {};
// Check if data and data.data exist and are an array
if (data?.data && Array.isArray(data.data)) {
for (const model of data.data) {
// Check for common context length fields (OpenAI uses context_window)
const contextLength = model.context_length ?? model.context_window ?? model.config?.max_tokens ?? null;
// Fireworks uses model.id
if (model.id && typeof contextLength === "number") {
modelsData[model.id] = contextLength;
}
}
} else {
console.warn("Unexpected response structure from Fireworks AI API:", data);
}
return modelsData;
} catch (error) {
console.error("Error fetching Fireworks AI data:", error);
return {}; // Return empty on error
}
}
|