File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed
packages/traceloop-sdk/traceloop/sdk Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -15,6 +15,7 @@ class Instruments(Enum):
1515 LANGCHAIN = "langchain"
1616 LLAMA_INDEX = "llama_index"
1717 MARQO = "marqo"
18+ MCP = "mcp"
1819 MILVUS = "milvus"
1920 MISTRAL = "mistral"
2021 OLLAMA = "ollama"
@@ -32,4 +33,3 @@ class Instruments(Enum):
3233 VERTEXAI = "vertexai"
3334 WATSONX = "watsonx"
3435 WEAVIATE = "weaviate"
35- MCP = "mcp"
Original file line number Diff line number Diff line change @@ -403,6 +403,9 @@ def init_instrumentations(
403403 elif instrument == Instruments .MARQO :
404404 if init_marqo_instrumentor ():
405405 instrument_set = True
406+ elif instrument == Instruments .MCP :
407+ if init_mcp_instrumentor ():
408+ instrument_set = True
406409 elif instrument == Instruments .MILVUS :
407410 if init_milvus_instrumentor ():
408411 instrument_set = True
@@ -454,9 +457,6 @@ def init_instrumentations(
454457 elif instrument == Instruments .WEAVIATE :
455458 if init_weaviate_instrumentor ():
456459 instrument_set = True
457- elif instrument == Instruments .MCP :
458- if init_mcp_instrumentor ():
459- instrument_set = True
460460 else :
461461 print (Fore .RED + f"Warning: { instrument } instrumentation does not exist." )
462462 print (
You can’t perform that action at this time.
0 commit comments