diff --git a/packages/proxy/src/proxy.ts b/packages/proxy/src/proxy.ts index a80645e3..b66ec98c 100644 --- a/packages/proxy/src/proxy.ts +++ b/packages/proxy/src/proxy.ts @@ -376,14 +376,26 @@ export async function proxyV1({ baseAttributes.org_name = orgName; } - // Record total calls with model attributes - logHistogram?.({ - name: "aiproxy.requests", - value: 1, - attributes: baseAttributes, - }); + let requestLogged = false; + const logRequest = () => { + if (requestLogged) { + return; + } + + logHistogram?.({ + name: "aiproxy.requests", + value: 1, + attributes: baseAttributes, + }); + requestLogged = true; + }; + + if (orgName) { + logRequest(); + } if (url === "/credentials") { + logRequest(); let readable: ReadableStream | null = null; try { const key = await makeTempCredentials({ @@ -635,6 +647,7 @@ export async function proxyV1({ if (secrets.length > 0 && !orgName && secrets[0].org_name) { baseAttributes.org_name = secrets[0].org_name; } + logRequest(); if (endpointName) { return secrets.filter((s) => s.name === endpointName); @@ -1122,6 +1135,7 @@ export async function proxyV1({ }); } + logRequest(); logHistogram?.({ name: "aiproxy.requests_completed", value: 1,