[FEAT][CLEANUP]

The-Swarm-Corporation · Apr 16, 2024 · 79a969d · 79a969d
1 parent 25c19df
commit 79a969d
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 26 deletions.
diff --git a/servers/text_to_video/sample_request.py b/servers/text_to_video/sample_request.py
@@ -16,4 +16,4 @@
 
 response = requests.post(url, headers=headers, data=json.dumps(data))
 
-print(response.json())
+print(response.json())
diff --git a/swarms_cloud/__init__.py b/swarms_cloud/__init__.py
@@ -1,15 +1,17 @@
-from swarms_cloud.api_utils import (
-    VariableInterface,
-    check_api_key,
-    check_request,
-    create_error_response,
-    get_model_list,
-)
 from swarms_cloud.auth_with_swarms_cloud import (
     authenticate_user,
     fetch_api_key_info,
     is_token_valid,
 )
+
+
+from swarms_cloud.check_model_list import (
+    get_model_list,
+    create_error_response,
+    check_request,
+)
+
+
 from swarms_cloud.calculate_pricing import calculate_pricing, count_tokens
 from swarms_cloud.func_api_wrapper import SwarmCloud
 from swarms_cloud.loggers.log_api_request_to_supabase import (
@@ -54,11 +56,19 @@
 from swarms_cloud.utils.rate_limiter import rate_limiter
 from swarms_cloud.auth_with_swarms_cloud import verify_token
 
+
 __all__ = [
-    "generate_api_key",
+    "authenticate_user",
+    "fetch_api_key_info",
+    "is_token_valid",
+    "get_model_list",
+    "create_error_response",
+    "check_request",
+    "calculate_pricing",
+    "count_tokens",
     "SwarmCloud",
-    "rate_limiter",
-    "SkyInterface",
+    "ModelAPILogEntry",
+    "log_to_supabase",
     "ChatCompletionRequest",
     "ChatCompletionRequestQos",
     "ChatCompletionResponse",
@@ -80,25 +90,16 @@
     "GenerateRequest",
     "GenerateRequestQos",
     "GenerateResponse",
+    "GenerationConfig",
     "ModelCard",
     "ModelList",
     "ModelPermission",
     "UsageInfo",
-    "GenerationConfig",
-    "VariableInterface",
-    "check_api_key",
-    "get_model_list",
-    "create_error_response",
-    "check_request",
     "InputOpenAISpec",
-    "OutputOpenAISpec",
     "OpenAIAPIWrapper",
-    "calculate_pricing",
-    "authenticate_user",
-    "is_token_valid",
-    "count_tokens",
-    "ModelAPILogEntry",
-    "log_to_supabase",
-    "fetch_api_key_info",
+    "OutputOpenAISpec",
+    "SkyInterface",
+    "generate_api_key",
+    "rate_limiter",
     "verify_token",
 ]
diff --git a/text_to_video.py b/text_to_video.py
@@ -93,7 +93,6 @@ def text_to_video(
     return out
 
 
-
 @app.post("/v1/chat/completions", response_model=TextToVideoResponse)
 async def create_chat_completion(
     request: TextToVideoRequest,  # token: str = Depends(authenticate_user)