Rename rate_limiter.py to api_call_counter.py

api_call_counter.py

@@ -0,0 +1,13 @@
+import time 
+
+class APICallCounter:
+    """
+    litellmのAPI呼び出し回数をカウントする
+    """
+    def __init__(self):
+        """呼び出し回数を初期化する"""
+        self._call_count = 0 # 呼び出し回数
+
+    def reset(self):
+        """呼び出し回数をリセットする"""
+        self._call_count = 0

rate_limiter.py

@@ -1,32 +0,0 @@
-import time 
-
-class RateLimiter:
-    """
-    litellmのAPI呼び出し回数をカウントして、
-    一定回数のAPI呼び出しが行われるたびに一定秒間待機する
-    """
-    def __init__(self, calls_per_wait: int = 10, seconds_to_wait: int = 60):
-        """
-        RateLimiterを初期化する
-        
-        Args:
-            calls_per_wait (int): 待機するまでの呼び出し回数
-            seconds_to_wait (int): 待機する秒数
-        """
-        self.calls_per_wait = calls_per_wait
-        self.seconds_to_wait = seconds_to_wait
-
-        self._call_count = 0 # 呼び出し回数
-
-    def sleep_if_needed(self):
-        """
-        呼び出し回数が`calls_per_wait`を超えていたら、
-        APIの呼び出し回数をリセットして`seconds_to_wait`秒待機する
-        その後、呼び出し回数を1増やす
-        """
-        if self._call_count >= self.calls_per_wait:    
-            print(f"呼び出し回数が{self.calls_per_wait}回を超えたので、{self.seconds_to_wait}秒待機します")
-            time.sleep(self.seconds_to_wait)
-            self._call_count = 0
-            
-        self._call_count += 1