@jeffusion/bungee-core 6.3.0 → 6.4.0

package/dist/plugins/token-metering/index.js

@@ -2675,7 +2675,7 @@ class ResponsesWebSocketMetering {
     this.responseCount++;
     this.callbacks.prepareRequest(response.requestId);
     try {
-      if (await this.emit(response, event, { phase: "selected" }) && await this.emit(response, event, { phase: "request", url: event.upstreamUrl, body: { model, stream: true } }))
+      if (await this.emit(response, event, { phase: "selected" }) && await this.emit(response, event, { phase: "request", url: event.upstreamUrl, upstreamHost: new URL(event.upstreamUrl).host, body: { model, stream: true } }))
         return response;
     } catch (error) {
       this.responseCount--;
@@ -2695,7 +2695,7 @@ class ResponsesWebSocketMetering {
     response.attemptId = uuid("attempt");
     response.bound = true;
     this.callbacks.prepareRequest(response.requestId);
-    return await this.emit(response, event, { phase: "selected" }) && await this.emit(response, event, { phase: "request", url: event.upstreamUrl, body: { model: response.model, stream: true } });
+    return await this.emit(response, event, { phase: "selected" }) && await this.emit(response, event, { phase: "request", url: event.upstreamUrl, upstreamHost: new URL(event.upstreamUrl).host, body: { model: response.model, stream: true } });
   }
   async emit(response, event, phase, settlement) {
     if (this.closed || !event.isActive()) {
@@ -2895,6 +2895,8 @@ function getAttempt(attempts, event, logger, onDrop) {
     attempt.keyId = event.keyId;
   attempt.routeId = event.routeId || attempt.routeId;
   attempt.upstreamId = event.upstreamId || attempt.upstreamId;
+  if (event.phase === "request")
+    attempt.upstreamHost = event.upstreamHost;
   return attempt;
 }
 function ensureSession(attempt, provider) {
@@ -3016,6 +3018,7 @@ function finalizeAttempt(input) {
     attemptId: input.attemptId,
     routeId: input.routeId,
     upstreamId: input.upstreamId,
+    upstreamHost: input.upstreamHost,
     provider: input.provider,
     model: input.model,
     pricingProvider: input.pricingProvider,
@@ -3308,6 +3311,7 @@ var TokenMeteringPlugin = definePlugin(class {
         keyId: attempt.keyId ?? null,
         routeId: attempt.routeId,
         upstreamId: attempt.upstreamId,
+        upstreamHost: attempt.upstreamHost,
         provider: attempt.provider,
         model: attempt.model,
         pricingProvider: attempt.pricingProvider,
@@ -3343,4 +3347,4 @@ export {
   server_default as default
 };
 
-//# debugId=DB115B0B5828A5AC64756E2164756E21
+//# debugId=C8C9AA1DD73B7BC264756E2164756E21