[CLEANUP]

The-Swarm-Corporation · Mar 18, 2024 · cec7e82 · cec7e82
1 parent 66c65b0
commit cec7e82
Show file tree

Hide file tree

Showing 10 changed files with 5 additions and 5 deletions.
diff --git a/swarms_cloud/__init__.py b/swarms_cloud/__init__.py
@@ -1,6 +1,6 @@
-from swarms_cloud.api_key_generator import generate_api_key
+from swarms_cloud.utils.api_key_generator import generate_api_key
 from swarms_cloud.func_api_wrapper import SwarmCloud
-from swarms_cloud.rate_limiter import rate_limiter
+from swarms_cloud.utils.rate_limiter import rate_limiter
 from swarms_cloud.sky_api import SkyInterface
 from swarms_cloud.schema.openai_protocol import (  # noqa: E501
     ChatCompletionRequest,

diff --git a/swarms_cloud/loggers/__init__.py b/swarms_cloud/loggers/__init__.py
diff --git a/swarms_cloud/supabase_handler.py → swarms_cloud/loggers/supabase_handler.py b/swarms_cloud/supabase_handler.py → swarms_cloud/loggers/supabase_handler.py
diff --git a/swarms_cloud/supabase_logger.py → swarms_cloud/loggers/supabase_logger.py b/swarms_cloud/supabase_logger.py → swarms_cloud/loggers/supabase_logger.py
diff --git a/swarms_cloud/utils/__init__.py b/swarms_cloud/utils/__init__.py
diff --git a/swarms_cloud/api_key_generator.py → swarms_cloud/utils/api_key_generator.py b/swarms_cloud/api_key_generator.py → swarms_cloud/utils/api_key_generator.py
diff --git a/swarms_cloud/rate_limiter.py → swarms_cloud/utils/rate_limiter.py b/swarms_cloud/rate_limiter.py → swarms_cloud/utils/rate_limiter.py
diff --git a/tests/test_api_generator.py b/tests/test_api_generator.py
@@ -1,6 +1,6 @@
 import pytest
 from swarms.structs.agent import Agent
-from swarms_cloud.api_key_generator import generate_api_key
+from swarms_cloud.utils.api_key_generator import generate_api_key
 
 
 # Basic tests

diff --git a/tests/test_rate_limiter.py b/tests/test_rate_limiter.py
@@ -3,7 +3,7 @@
 from fastapi import FastAPI, HTTPException
 from time import time
 import logging
-from swarms_cloud.rate_limiter import rate_limiter
+from swarms_cloud.utils.rate_limiter import rate_limiter
 
 
 class MockClient:

diff --git a/tests/test_supabase_pro.py b/tests/test_supabase_pro.py
@@ -2,7 +2,7 @@
 
 import pytest
 from supabase import Client
-from swarms_cloud.supabase_handler import SwarmCloudUsageLogger
+from swarms_cloud.loggers.supabase_handler import SwarmCloudUsageLogger
 
 
 # Create a mock Supabase client for testing