fix function name

AgentOps-AI · Aug 17, 2024 · 6bc844f · 6bc844f
1 parent 7a17a4c
commit 6bc844f
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/agentops/llms/anthropic.py b/agentops/llms/anthropic.py
@@ -21,7 +21,7 @@ def __init__(self, client):
         self.tool_event = {}
         self.tool_id = ""
 
-    def _handle_response(
+    def handle_response(
         self, response, kwargs, init_timestamp, session: Optional[Session] = None
     ):
         """Handle responses for Anthropic"""
@@ -206,13 +206,13 @@ def patched_function(*args, **kwargs):
                         f"{pprint.pformat(completion_override)}"
                     )
                     return None
-                return self._handle_response(
+                return self.handle_response(
                     result_model, kwargs, init_timestamp, session=session
                 )
 
             # Call the original function with its original arguments
             result = original_create(*args, **kwargs)
-            return self._handle_response(
+            return self.handle_response(
                 result, kwargs, init_timestamp, session=session
             )
 
@@ -274,12 +274,12 @@ async def patched_function(*args, **kwargs):
                     )
                     return None
 
-                return self._handle_response(
+                return self.handle_response(
                     result_model, kwargs, init_timestamp, session=session
                 )
 
             result = await original_create_async(*args, **kwargs)
-            return self._handle_response(
+            return self.handle_response(
                 result, kwargs, init_timestamp, session=session
             )