Add agentops integration

python/composio/tools/toolset.py

@@ -4,12 +4,15 @@
 
 import base64
 import binascii
+from functools import wraps
 import hashlib
+import importlib
 import itertools
 import json
 import os
 import time
 import typing as t
+from importlib.util import find_spec
 
 import typing_extensions as te
 from pydantic import BaseModel
@@ -46,7 +49,7 @@
 from composio.tools.local import load_local_tools
 from composio.tools.local.handler import LocalClient
 from composio.utils.enums import get_enum_key
-from composio.utils.logging import LogLevel, WithLogger
+from composio.utils.logging import LogLevel, WithLogger, get_logger
 from composio.utils.url import get_api_url_base
 
 
@@ -69,6 +72,30 @@ class ProcessorsType(te.TypedDict):
     """Response processors."""
 
 
+def _check_agentops() -> bool:
+    """Check if AgentOps is installed and initialized."""
+    logger = get_logger()
+    logger.info("Checking if AgentOps is installed and initialized")
+    if find_spec("agentops") is None:
+        logger.info("AgentOps is not installed")
+        return False
+    logger.info("AgentOps is installed")
+    import agentops
+    logger.info("AgentOps is initialized")
+    return agentops.get_api_key() is not None
+
+def _record_action_if_available(func: t.Callable) -> t.Callable:
+    @wraps(func)
+    def wrapper(self, *args, **kwargs):
+        if _check_agentops():
+            import agentops
+            action_name = str(kwargs.get('action', 'unknown_action'))
+            return agentops.record_action(action_name)(func)(self, *args, **kwargs)
+        else:
+            return func(self, *args, **kwargs)
+    return wrapper
+
+
 class ComposioToolSet(WithLogger):
     """Composio toolset."""
 
@@ -486,6 +513,7 @@ def _process_respone(self, action: Action, response: t.Dict) -> t.Dict:
             type_="post",
         )
 
+    @_record_action_if_available
     def execute_action(
         self,
         action: ActionType,

python/plugins/crew_ai/crewai_demo.py

@@ -9,6 +9,9 @@
 from langchain_openai import ChatOpenAI
 
 from composio_crewai import App, ComposioToolSet
+import agentops
+
+agentops.init()
 
 
 # Load environment variables from .env

python/plugins/langchain/langchain_demo.py

@@ -9,11 +9,14 @@
 from langchain import hub  # type: ignore
 from langchain.agents import AgentExecutor, create_openai_functions_agent
 from langchain_openai import ChatOpenAI
+import agentops
 
 
 # Load environment variables from .env
 dotenv.load_dotenv()
 
+agentops.init()
+
 # Pull relevant agent model.
 prompt = hub.pull("hwchase17/openai-functions-agent")