[FEAT][swarms_cloud]

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms-cloud"
-version = "0.1.2"
+version = "0.1.3"
 description = "Swarms Cloud - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <[email protected]>"]
@@ -29,7 +29,7 @@ skypilot = "*"
 torch = "*"
 einops = "*"
 pydantic = "*"
-
+stripe = "*"
 
 [tool.poetry.group.lint.dependencies]
 ruff = "^0.1.6"

requirements.txt

@@ -10,4 +10,5 @@ einops
 tiktoken
 uvicorn
 loguru
-pydantic
+pydantic
+stripe

servers/cogvlm.py

@@ -404,9 +404,7 @@ def generate_stream_cogvlm(
     else:
         torch_type = torch.float16
 
-    print(
-        f"========Use torch type as:{torch_type} with device:{DEVICE}========\n\n"
-    )
+    print(f"========Use torch type as:{torch_type} with device:{DEVICE}========\n\n")
 
     if "cuda" in DEVICE:
         if QUANT_ENABLED:

swarms_cloud/calculate_pricing.py

@@ -13,7 +13,7 @@ def calculate_pricing(
     rate_img: float = 0.003,
 ):
     """
-    Calculates the pricing for a given list of texts based on the number of tokens, sentences, words, characters, and paragraphs.
+    leCalctuple containingg for otal number of  texts based on the number of tokens, sentences, words, characters, and paragraphs.
 
     Args:
         texts (list): A list of texts to calculate pricing for.

swarms_cloud/stripe_utils.py

@@ -0,0 +1,29 @@
+import logging
+
+import stripe
+from pydantic import BaseModel
+
+stripe.api_key = "your_stripe_api_key"
+
+
+class StripeInterface(BaseModel):
+    customer_id: str
+    amount: float
+    description: str
+
+
+def bill_customer(customer_id: str, amount: float, description: str):
+    try:
+        stripe.Charge.create(
+            customer=customer_id,
+            amount=amount,  # in cents
+            currency="usd",
+            description=description,
+        )
+        logging.info("Payment successful")
+    except stripe.error.StripeError as e:
+        logging.error(f"Payment failed: {str(e)}")
+
+
+# Usage:
+# bill_customer("429232323", 1000, "1,000 tokens")