[CLEANUP]

send_local_request_to_cogvlm.py

@@ -49,7 +49,7 @@ def image_to_base64(image_path):
 }
 
 # Specify the URL of your FastAPI application
-url = f"http://localhost:8100/v1/chat/completions"
+url = "http://localhost:8100/v1/chat/completions"
 
 # Start the timer
 start_time = time.time()

servers/cogvlm/cogvlm.py

@@ -38,7 +38,6 @@
 from swarms_cloud.calculate_pricing import calculate_pricing, count_tokens
 from swarms_cloud.auth_with_swarms_cloud import fetch_api_key_info
 from swarms_cloud.log_api_request_to_supabase import log_to_supabase, ModelAPILogEntry
-from swarms_cloud import verify_token
 
 # from exa import calculate_workers
 # import torch.distributed as dist

swarms_cloud/auth_with_swarms_cloud.py

@@ -102,16 +102,12 @@ def authenticate_user(
     return token
 
 
-
 def verify_token(req: Request):
     token = req.headers["Authorization"]
     # Here your code for verifying the token or whatever you use
     token = token.split("Bearer ")[1]
     token = token.strip()
     valid = is_token_valid(token)
     if valid is False:
-        raise HTTPException(
-            status_code=401,
-            detail="Unauthorized"
-        )
-    return True
+        raise HTTPException(status_code=401, detail="Unauthorized")
+    return True

swarms_cloud/calculate_pricing.py

@@ -4,9 +4,7 @@
 from transformers import AutoTokenizer, PreTrainedTokenizer
 
 
-def count_tokens(
-    texts: List[str], tokenizer: PreTrainedTokenizer, model: str
-) -> int:
+def count_tokens(texts: List[str], tokenizer: PreTrainedTokenizer, model: str) -> int:
     """
     Counts the total number of tokens in a list of texts using a tokenizer.