Fix NPE

wso2 · Sep 11, 2024 · f86956a · f86956a
1 parent 9869b3b
commit f86956a
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/...forcer/org.wso2.apk.enforcer/src/main/java/org/wso2/apk/enforcer/security/AuthFilter.java b/...forcer/org.wso2.apk.enforcer/src/main/java/org/wso2/apk/enforcer/security/AuthFilter.java
@@ -128,10 +128,16 @@ public boolean handleRequest(RequestContext requestContext) {
  // Any auth token has been provided for application-level security or not
  boolean canAuthenticated = false;
  if (requestContext.getMatchedAPI() != null && requestContext.getMatchedAPI().getAiProvider() != null) {
- requestContext.addMetadataToMap("aitoken:prompttokenid", requestContext.getMatchedAPI().getAiProvider().getPromptTokens().getValue());
- requestContext.addMetadataToMap("aitoken:completiontokenid", requestContext.getMatchedAPI().getAiProvider().getCompletionToken().getValue());
- requestContext.addMetadataToMap("aitoken:totaltokenid", requestContext.getMatchedAPI().getAiProvider().getTotalToken().getValue());
- requestContext.addMetadataToMap("aitoken:extracttokenfrom", requestContext.getMatchedAPI().getAiProvider().getCompletionToken().getIn());
+ if (requestContext.getMatchedAPI().getAiProvider().getPromptTokens() != null) {
+ requestContext.addMetadataToMap("aitoken:prompttokenid", requestContext.getMatchedAPI().getAiProvider().getPromptTokens().getValue());
+ }
+ if (requestContext.getMatchedAPI().getAiProvider().getCompletionToken() != null) {
+ requestContext.addMetadataToMap("aitoken:extracttokenfrom", requestContext.getMatchedAPI().getAiProvider().getCompletionToken().getIn());
+ requestContext.addMetadataToMap("aitoken:completiontokenid", requestContext.getMatchedAPI().getAiProvider().getCompletionToken().getValue());
+ }
+ if (requestContext.getMatchedAPI().getAiProvider().getTotalToken() != null) {
+ requestContext.addMetadataToMap("aitoken:totaltokenid", requestContext.getMatchedAPI().getAiProvider().getTotalToken().getValue());
+ }
  }
  for (Authenticator authenticator : authenticators) {
  if (authenticator.canAuthenticate(requestContext)) {