Update api_usage.py

api_usage.py

@@ -1,6 +1,7 @@
 import requests
 import os
 import openai
+import anthropic
 
 BASE_URL = 'https://api.openai.com/v1'
 GPT_TYPES = ["gpt-3.5-turbo", "gpt-4", "gpt-4-32k"]
@@ -87,6 +88,24 @@ def check_key_availability():
     except:
         return False
 
+def check_key_ant_availability(ant):
+    try:
+        ant.completions.create(
+            prompt=f"{anthropic.HUMAN_PROMPT}Hi{anthropic.AI_PROMPT}",
+            max_tokens_to_sample=1,
+            model="claude-instant-v1",
+        )
+        return True, "Working"
+    except anthropic.APIConnectionError as e:
+        print(e.__cause__)  # an underlying Exception, likely raised within httpx.
+        return False, "Error: The server could not be reached"
+    except anthropic.RateLimitError as e:
+        return True, "Error: 429, rate limited; we should back off a bit."
+    except anthropic.APIStatusError as e:
+        err_msg = e.body.get('error', {}).get('message', '')
+        return False, f"Error: {e.status_code}, {err_msg}"
+
 if __name__ == "__main__":
     key = os.getenv("OPENAI_API_KEY")
+    key_ant = os.getenv("ANTHROPIC_API_KEY")
     results = get_subscription(key)