Update langchain handler for v1

agentops/__init__.py

@@ -1,7 +1,7 @@
 # agentops/__init__.py
 
 from .client import Client
-from .event import ActionEvent, LLMEvent, ToolEvent, ErrorEvent
+from .event import Event, ActionEvent, LLMEvent, ToolEvent, ErrorEvent
 from .logger import AgentOpsLogger
 from .enums import Models, LLMMessageFormat
 from .decorators import record_function

agentops/enums.py

@@ -8,11 +8,11 @@ class EventType(Enum):
     TOOL = "tools"
     ERROR = "errors"
 
-
-class Result(Enum):
-    SUCCESS = "Success"
-    FAIL = "Fail"
-    INDETERMINATE = "Indeterminate"
+# TODO: Not used
+# class Result(Enum):
+#     SUCCESS = "Success"
+#     FAIL = "Fail"
+#     INDETERMINATE = "Indeterminate"
 
 
 class Models(Enum):

agentops/event.py

@@ -2,7 +2,7 @@
 AgentOps events.
 
 Data Class:
-    Event: Represents discrete events to be recorded.
+    Event: Repents discrete events to be recorded.
 """
 
 from dataclasses import dataclass, field

agentops/helpers.py

@@ -1,9 +1,12 @@
+from pprint import pprint, pformat
+from functools import wraps
 import time
 from datetime import datetime
 import json
 import inspect
 import logging
 from uuid import UUID
+import os
 
 
 def singleton(class_):
@@ -84,3 +87,42 @@ def check_call_stack_for_agent_id() -> str | None:
                 logging.debug('LLM call from agent named: ' + getattr(var, '_agent_ops_agent_name'))
                 return getattr(var, '_agent_ops_agent_id')
     return None
+
+# Function decorator that prints function name and its arguments to the console for debug purposes
+# Example output:
+    # <AGENTOPS_DEBUG_OUTPUT>
+    # on_llm_start called with arguments:
+    # run_id: UUID('5fda42fe-809b-4179-bad2-321d1a6090c7')
+    # parent_run_id: UUID('63f1c4da-3e9f-4033-94d0-b3ebed06668f')
+    # tags: []
+    # metadata: {}
+    # invocation_params: {'_type': 'openai-chat',
+    # 'model': 'gpt-3.5-turbo',
+    # 'model_name': 'gpt-3.5-turbo',
+    # 'n': 1,
+    # 'stop': ['Observation:'],
+    # 'stream': False,
+    # 'temperature': 0.7}
+    # options: {'stop': ['Observation:']}
+    # name: None
+    # batch_size: 1
+    # </AGENTOPS_DEBUG_OUTPUT>
+
+# regex to filter for just this:
+# <AGENTOPS_DEBUG_OUTPUT>([\s\S]*?)<\/AGENTOPS_DEBUG_OUTPUT>\n
+
+
+def debug_print_function_params(func):
+    @wraps(func)
+    def wrapper(self, *args, **kwargs):
+        if os.getenv('DEBUG_MODE') == 'Y':
+            print("\n<AGENTOPS_DEBUG_OUTPUT>")
+            print(f"{func.__name__} called with arguments:")
+
+            for key, value in kwargs.items():
+                print(f"{key}: {pformat(value)}")
+
+            print("</AGENTOPS_DEBUG_OUTPUT>\n")
+
+        return func(self, *args, **kwargs)
+    return wrapper