AgentOps ENG-525: Decouple CrewAI and AgentOps (#1033)

* Make AgentOps import optional upon AGENTOPS_API_KEY
    being set

Co-authored-by: João Moura <joaomdmoura@gmail.com>
This commit is contained in:
Vikram Guhan Subbiah
2024-08-10 05:47:13 -07:00
committed by GitHub
parent d1770b3749
commit c0005e112e
4 changed files with 42 additions and 20 deletions

View File

@@ -1,3 +1,4 @@
import os
from typing import List
from langchain_openai import ChatOpenAI
@@ -6,16 +7,23 @@ from pydantic import BaseModel, Field
from crewai.utilities import Converter
from crewai.utilities.pydantic_schema_parser import PydanticSchemaParser
agentops = None
try:
from agentops import track_agent
except ImportError:
def track_agent(name):
def mock_agent_ops_provider():
def track_agent():
def noop(f):
return f
return noop
return track_agent
agentops = None
if os.environ.get("AGENTOPS_API_KEY"):
try:
from agentops import track_agent
except ImportError:
track_agent = mock_agent_ops_provider()
else:
track_agent = mock_agent_ops_provider()
class Entity(BaseModel):