From 1b61337b7576fe650fe231e2814508852a0a743b Mon Sep 17 00:00:00 2001
From: Siddhant Rai <siddhant.rai.5686@gmail.com>
Date: Tue, 16 Apr 2024 01:08:39 +0530
Subject: [PATCH] fix: skip logging to db during tests

---
 application/usage.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/application/usage.py b/application/usage.py
index 95cd02f..131de66 100644
--- a/application/usage.py
+++ b/application/usage.py
@@ -1,3 +1,4 @@
+import sys
 from pymongo import MongoClient
 from datetime import datetime
 from application.core.settings import settings
@@ -9,6 +10,8 @@ usage_collection = db["token_usage"]
 
 
 def update_token_usage(api_key, token_usage):
+    if "pytest" in sys.modules:
+        return
     usage_data = {
         "api_key": api_key,
         "prompt_tokens": token_usage["prompt_tokens"],