Update api_call.py

api_call.py

@@ -12,7 +12,6 @@ class Chain:
         # Load environment variables
         load_dotenv()
         self.mistral_api_key = os.getenv("MISTRAL_API_KEY")
-        print(self.mistral_api_key, os.getenv("MISTRAL_API_KEY"))
         self.model = model
         # Initialize model and parser
         self.model = ChatMistralAI(model=self.model,  api_key=self.mistral_api_key)
@@ -51,8 +50,7 @@ class Chain:
         Comparison of two Tiletels with retry mechanism for rate limiting
         """
         retries = 0
-        print('titles', end='\n ------- \n\n')
-        print(self.mistral_api_key, end='\n ------- \n\n')
+
         while retries < max_retries:
             try:
                 response = self.chain_single.invoke({"titles": titles})
@@ -69,7 +67,6 @@ class Chain:
 
 
     def classify(self, title_0, title_1, max_retries=10):
-        print('mistral: ', title_0, title_1)
         return self._classify_with_backoff(f"{title_0} and {title_1}", max_retries)