(self, user_id: int, key: str, value: Any)
| 90 | return user_dict[key] |
| 91 | |
| 92 | def set_user_attribute(self, user_id: int, key: str, value: Any): |
| 93 | self.check_if_user_exists(user_id, raise_exception=True) |
| 94 | self.user_collection.update_one({"_id": user_id}, {"$set": {key: value}}) |
| 95 | |
| 96 | def update_n_used_tokens(self, user_id: int, model: str, n_input_tokens: int, n_output_tokens: int): |
| 97 | n_used_tokens_dict = self.get_user_attribute(user_id, "n_used_tokens") |
no test coverage detected