Removed debug prints, and other cleanup.

andrewyng · Jan 22, 2025 · 1a7be38 · 1a7be38
1 parent 70ae662
commit 1a7be38
Show file tree

Hide file tree

Showing 4 changed files with 191 additions and 86 deletions.
diff --git a/aisuite/providers/aws_provider.py b/aisuite/providers/aws_provider.py
@@ -36,10 +36,6 @@ def convert_request(
             for message in messages
         ]
 
-        import pprint
-
-        pprint.pprint(messages)
-
         # Handle system message
         system_message = []
         if messages and messages[0]["role"] == "system":
@@ -68,8 +64,6 @@ def convert_request(
                     }
                 )
 
-        pprint.pprint(formatted_messages)
-
         return system_message, formatted_messages
 
     @staticmethod

diff --git a/aisuite/providers/fireworks_provider.py b/aisuite/providers/fireworks_provider.py
@@ -113,8 +113,6 @@ def chat_completions_create(self, model, messages, **kwargs):
             "Content-Type": "application/json",
         }
 
-        print(data)
-
         try:
             # Make the request to Fireworks AI endpoint.
             response = httpx.post(

diff --git a/aisuite/providers/huggingface_provider.py b/aisuite/providers/huggingface_provider.py
@@ -64,12 +64,9 @@ def chat_completions_create(self, model, messages, **kwargs):
                 **kwargs,  # Include other parameters like temperature, max_tokens, etc.
             }
 
-            print(transformed_messages)
-
             # Make the API call using the client
             response = self.client.chat_completion(model=model, **payload)
 
-            print(response)
             return self._normalize_response(response)
 
         except Exception as e: