pjandro/OmniRoute
1
0
Fork 0
mirror of https://github.com/diegosouzapw/OmniRoute.git synced 2026-08-03 13:52:09 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix(glm): add dedicated coding transport (#2087)

Browse Source 
Integrated into release/v3.8.0
This commit is contained in:
Raxxoor
2026-05-10 04:52:00 +01:00
committed by GitHub
parent 73fc6e3ca6
commit a8106bbadd
51 changed files with 2326 additions and 398 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
src/app/api/providers/[id]/models/route.ts
View File

@@ -22,7 +22,10 @@ import { getProviderOutboundGuard } from "@/shared/network/outboundUrlGuard";
import { getStaticQoderModels } from "@omniroute/open-sse/services/qoderCli.ts";
import { getAntigravityHeaders } from "@omniroute/open-sse/services/antigravityHeaders.ts";
import { getAntigravityModelsDiscoveryUrls } from "@omniroute/open-sse/config/antigravityUpstream.ts";
import { getGlmModelsUrl } from "@omniroute/open-sse/config/glmProvider.ts";
import {
buildGlmCodingHeaders,
buildGlmModelsUrl,
} from "@omniroute/open-sse/config/glmProvider.ts";
import { getImageProvider } from "@omniroute/open-sse/config/imageRegistry.ts";
import { getVideoProvider } from "@omniroute/open-sse/config/videoRegistry.ts";
import { resolveAntigravityVersion } from "@omniroute/open-sse/services/antigravityVersion.ts";
@@ -1551,40 +1554,73 @@ export async function GET(
}
}
if (provider === "glm" || provider === "glmt") {
if (provider === "glm" || provider === "glm-cn" || provider === "glmt") {
const cachedResponse = maybeReturnCachedDiscovery();
if (cachedResponse) return cachedResponse;
const autoFetchDisabledResponse = maybeReturnAutoFetchDisabled();
if (autoFetchDisabledResponse) return autoFetchDisabledResponse;
const url = getGlmModelsUrl(connection.providerSpecificData);
const token = apiKey || accessToken;
const glmProviderSpecificData = {
...asRecord(connection.providerSpecificData),
...(provider === "glm-cn" ? { apiRegion: "china" } : {}),
};
const discoveredTargets = [
{
transport: "openai" as const,
url: buildGlmModelsUrl(glmProviderSpecificData, "openai"),
},
{
transport: "anthropic" as const,
url: buildGlmModelsUrl(glmProviderSpecificData, "anthropic"),
},
];
const discoveryTargets = discoveredTargets.filter(
(target, index, all) => all.findIndex((other) => other.url === target.url) === index
);
let response: Response;
let response: Response | null = null;
try {
response = await safeOutboundFetch(url, {
...SAFE_OUTBOUND_FETCH_PRESETS.modelsDiscovery,
guard: getProviderOutboundGuard(),
proxyConfig: proxy,
method: "GET",
headers: {
"Content-Type": "application/json",
...(token ? { Authorization: `Bearer ${token}` } : {}),
},
});
for (const target of discoveryTargets) {
response = await safeOutboundFetch(target.url, {
...SAFE_OUTBOUND_FETCH_PRESETS.modelsDiscovery,
guard: getProviderOutboundGuard(),
proxyConfig: proxy,
method: "GET",
headers:
target.transport === "openai"
? token
? buildGlmCodingHeaders(token, false)
: { "Content-Type": "application/json", Accept: "application/json" }
: {
"Content-Type": "application/json",
Accept: "application/json",
...(token ? { "x-api-key": token } : {}),
"anthropic-version": "2023-06-01",
},
});
if (response.ok) break;
if (response.status === 401 || response.status === 403) break;
}
} catch (error) {
const fallback = buildDiscoveryErrorFallbackResponse(error);
if (fallback) return fallback;
throw error;
}
if (!response.ok) {
if (!response?.ok) {
if (response?.status === 401 || response?.status === 403) {
return NextResponse.json(
{ error: `Failed to fetch models: ${response.status}` },
{ status: response.status }
);
}
const fallback = buildDiscoveryFallbackResponse();
if (fallback) return fallback;
return NextResponse.json(
{ error: `Failed to fetch models: ${response.status}` },
{ status: response.status }
{ error: `Failed to fetch models: ${response?.status || 502}` },
{ status: response?.status || 502 }
);
}