Merge pull request #60 from pattern-tech/feat/wallet-accounting

fix: staked wallet bug fixed

Author: yasinfakhar

src/query_usage/routers/query_usage_router.py

@@ -124,7 +124,7 @@ def get_query_usage(
         }
     },
 )
-def get_user_query_usages(
+def get_user_daily_query_usages(
     provider: Optional[str] = None,
     user_id: UUID = Depends(authenticate_user),
     db: Session = Depends(get_db),

src/query_usage/services/query_usage_service.py

@@ -99,7 +99,7 @@ def get_user_max_query_allowance(self, db_session: Session, user_id: UUID) -> in
                 return max_allowed_query
 
         staked_morpheus = get_user_staked_tokens(
-            wallet_address=user.wallet_address, provider="morpheus")
+            wallet_address=user.wallet_address.lower(), provider="morpheus")
 
         if staked_morpheus == 0:
             raise RateLimitError(
@@ -112,7 +112,7 @@ def get_user_max_query_allowance(self, db_session: Session, user_id: UUID) -> in
         for setting in usage_setting:
             if setting.provider == "morpheus":
                 max_allowed_query = setting.max_query * \
-                    (int(staked_morpheus) / 1e18)
+                    int(int(staked_morpheus) / 1e18)
                 break
 
         return max_allowed_query