langchain handler fixes

AgentOps-AI · May 23, 2024 · 827e152 · 827e152
1 parent c628ee0
commit 827e152
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/agentops/__init__.py b/agentops/__init__.py
@@ -10,7 +10,7 @@
 from .agent import track_agent
 from .log_config import logger
 try:
-    from .langchain_callback_handler import LangchainCallbackHandler, AsyncLangchainCallbackHandler
+    from .partners.langchain_callback_handler import LangchainCallbackHandler, AsyncLangchainCallbackHandler
 except ModuleNotFoundError:
     pass
 

diff --git a/agentops/partners/langchain_callback_handler.py b/agentops/partners/langchain_callback_handler.py
@@ -14,7 +14,7 @@
 from agentops import ActionEvent, LLMEvent, ToolEvent, ErrorEvent
 from agentops.helpers import get_ISO_time
 
-from .helpers import debug_print_function_params
+from ..helpers import debug_print_function_params
 
 
 def get_model_from_kwargs(kwargs: any) -> str: