create api_call_counter.py

api_call_counter.py

@@ -0,0 +1,19 @@
+class APICallCounter:
+    """
+    litellmのAPI呼び出し回数をカウントするクラス。
+    """
+    def __init__(self):
+        """カウンターを0で初期化する"""
+        self.total_calls = 0
+
+    def count_successful_call(self, kwargs, completion_obj, start_time, end_time):
+        """API呼び出しが成功したときに呼び出され、カウンターを1増やす。"""
+        self.total_calls += 1
+
+    def count_failed_call(self, kwargs, completion_obj, start_time, end_time):
+        """API呼び出しが失敗したときに呼び出され、カウンターを1増やす。"""
+        self.total_calls += 1
+
+    def get_total_calls(self):
+        """現在のカウント状況を返す。"""
+        return self.total_calls