andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-11 00:58:30 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

use langchain callback handler to support all LLMs

Browse Source
This commit is contained in:
Braelyn Boynton
2024-05-03 15:07:17 -07:00
parent 498bf77f08
commit b9d6ec5721
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/crewai/agent.py
View File

@@ -25,7 +25,9 @@ from crewai.memory.contextual.contextual_memory import ContextualMemory
from crewai.utilities import I18N, Logger, Prompts, RPMController from crewai.utilities import I18N, Logger, Prompts, RPMController
from crewai.utilities.token_counter_callback import TokenCalcHandler, TokenProcess from crewai.utilities.token_counter_callback import TokenCalcHandler, TokenProcess
agentops = None
try: try:
import agentops
from agentops import track_agent from agentops import track_agent
except ImportError: except ImportError:
@@ -195,6 +197,12 @@ class Agent(BaseModel):
): ):
self.llm.callbacks.append(token_handler) self.llm.callbacks.append(token_handler)
if agentops and not any(
isinstance(handler, agentops.LangchainCallbackHandler) for handler in self.llm.callbacks
):
agentops.stop_instrumenting()
self.llm.callbacks.append(agentops.LangchainCallbackHandler())
if not self.agent_executor: if not self.agent_executor:
if not self.cache_handler: if not self.cache_handler:
self.cache_handler = CacheHandler() self.cache_handler = CacheHandler()