Skip to content

Commit a47a535

Browse files
committed
fix: user_id_extractor for credit
1 parent 4bd80a4 commit a47a535

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

nilai-api/src/nilai_api/credit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ class LLMResponse(BaseModel):
9494

9595
def user_id_extractor() -> Callable[[Request], Awaitable[str]]:
9696
if CONFIG.auth.auth_strategy == "nuc":
97-
return UserIdExtractors.from_nuc_bearer_token()
97+
return from_nuc_bearer_root_token()
9898
else:
9999
extractor = UserIdExtractors.from_header("Authorization")
100100

@@ -145,7 +145,7 @@ async def calculator(request: Request, response_data: dict) -> float:
145145

146146

147147
LLMMeter = create_metering_dependency(
148-
user_id_extractor=from_nuc_bearer_root_token(),
148+
user_id_extractor=user_id_extractor(),
149149
estimated_cost=2.0,
150150
cost_calculator=llm_cost_calculator(MyCostDictionary),
151151
)

0 commit comments

Comments
 (0)