fix: auto-instrumentation support (#662)

traceloop · Mar 21, 2024 · 4a5df69 · 4a5df69
1 parent 9a82aab
commit 4a5df69
Show file tree

Hide file tree

Showing 28 changed files with 178 additions and 246 deletions.
diff --git a/packages/opentelemetry-instrumentation-anthropic/poetry.lock b/packages/opentelemetry-instrumentation-anthropic/poetry.lock
diff --git a/packages/opentelemetry-instrumentation-anthropic/pyproject.toml b/packages/opentelemetry-instrumentation-anthropic/pyproject.toml
@@ -46,5 +46,8 @@ opentelemetry-sdk = "^1.23.0"
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
-[project.entry-points.opentelemetry_instrumentor]
+[tool.poetry.extras]
+instruments = ["anthropic"]
+
+[tool.poetry.plugins."opentelemetry_instrumentor"]
 anthropic = "opentelemetry.instrumentation.anthropic:AnthropicInstrumentor"
diff --git a/packages/opentelemetry-instrumentation-bedrock/pyproject.toml b/packages/opentelemetry-instrumentation-bedrock/pyproject.toml
@@ -37,6 +37,3 @@ pytest-sugar = "1.0.0"
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
-
-[project.entry-points.opentelemetry_instrumentor]
-boto3 = "opentelemetry.instrumentation.bedrock:BedrockInstrumentor"
diff --git a/packages/opentelemetry-instrumentation-chromadb/poetry.lock b/packages/opentelemetry-instrumentation-chromadb/poetry.lock
diff --git a/packages/opentelemetry-instrumentation-chromadb/pyproject.toml b/packages/opentelemetry-instrumentation-chromadb/pyproject.toml
@@ -45,5 +45,8 @@ opentelemetry-sdk = "^1.23.0"
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
-[project.entry-points.opentelemetry_instrumentor]
+[tool.poetry.extras]
+instruments = ["chromadb"]
+
+[tool.poetry.plugins."opentelemetry_instrumentor"]
 chromadb = "opentelemetry.instrumentation.chromadb:ChromaInstrumentor"
diff --git a/packages/opentelemetry-instrumentation-cohere/poetry.lock b/packages/opentelemetry-instrumentation-cohere/poetry.lock
diff --git a/packages/opentelemetry-instrumentation-cohere/pyproject.toml b/packages/opentelemetry-instrumentation-cohere/pyproject.toml
@@ -46,5 +46,8 @@ opentelemetry-sdk = "^1.23.0"
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
-[project.entry-points.opentelemetry_instrumentor]
+[tool.poetry.extras]
+instruments = ["cohere"]
+
+[tool.poetry.plugins."opentelemetry_instrumentor"]
 cohere = "opentelemetry.instrumentation.cohere:CohereInstrumentor"
diff --git a/packages/opentelemetry-instrumentation-haystack/poetry.lock b/packages/opentelemetry-instrumentation-haystack/poetry.lock
diff --git a/packages/opentelemetry-instrumentation-haystack/pyproject.toml b/packages/opentelemetry-instrumentation-haystack/pyproject.toml
@@ -46,5 +46,8 @@ opentelemetry-sdk = "^1.23.0"
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
-[project.entry-points.opentelemetry_instrumentor]
-haystack = "opentelemetry.instrumentation.haystack:HaystackInstrumentor"
+[tool.poetry.extras]
+instruments = ["farm-haystack"]
+
+[tool.poetry.plugins."opentelemetry_instrumentor"]
+farm_haystack = "opentelemetry.instrumentation.haystack:HaystackInstrumentor"
diff --git a/packages/opentelemetry-instrumentation-langchain/poetry.lock b/packages/opentelemetry-instrumentation-langchain/poetry.lock
diff --git a/packages/opentelemetry-instrumentation-langchain/pyproject.toml b/packages/opentelemetry-instrumentation-langchain/pyproject.toml
@@ -51,5 +51,8 @@ opentelemetry-instrumentation-openai = {path="../opentelemetry-instrumentation-o
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
-[project.entry-points.opentelemetry_instrumentor]
+[tool.poetry.extras]
+instruments = ["langchain"]
+
+[tool.poetry.plugins."opentelemetry_instrumentor"]
 langchain = "opentelemetry.instrumentation.langchain:LangchainInstrumentor"
diff --git a/packages/opentelemetry-instrumentation-llamaindex/poetry.lock b/packages/opentelemetry-instrumentation-llamaindex/poetry.lock
diff --git a/packages/opentelemetry-instrumentation-llamaindex/pyproject.toml b/packages/opentelemetry-instrumentation-llamaindex/pyproject.toml
@@ -53,5 +53,8 @@ llama-index-vector-stores-chroma = "^0.1.5"
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
-[project.entry-points.opentelemetry_instrumentor]
-llama_index = "opentelemetry.instrumentation.llamaindex:LlamaIndexInstrumentor"
+[tool.poetry.extras]
+instruments = ["llama-index"]
+
+[tool.poetry.plugins."opentelemetry_instrumentor"]
+llama-index = "opentelemetry.instrumentation.llamaindex:LlamaIndexInstrumentor"
diff --git a/packages/opentelemetry-instrumentation-openai/poetry.lock b/packages/opentelemetry-instrumentation-openai/poetry.lock
diff --git a/packages/opentelemetry-instrumentation-openai/pyproject.toml b/packages/opentelemetry-instrumentation-openai/pyproject.toml
@@ -45,5 +45,8 @@ pytest-asyncio = "^0.23.5"
 requires = [ "poetry-core" ]
 build-backend = "poetry.core.masonry.api"
 
-[project.entry-points.opentelemetry_instrumentor]
+[tool.poetry.extras]
+instruments = ["openai"]
+
+[tool.poetry.plugins."opentelemetry_instrumentor"]
 openai = "opentelemetry.instrumentation.openai:OpenAIInstrumentor"
diff --git a/packages/opentelemetry-instrumentation-pinecone/poetry.lock b/packages/opentelemetry-instrumentation-pinecone/poetry.lock
diff --git a/packages/opentelemetry-instrumentation-pinecone/pyproject.toml b/packages/opentelemetry-instrumentation-pinecone/pyproject.toml
@@ -48,5 +48,8 @@ pinecone-client = "^2.2.4"
 requires = [ "poetry-core" ]
 build-backend = "poetry.core.masonry.api"
 
-[project.entry-points.opentelemetry_instrumentor]
-pinecone = "opentelemetry.instrumentation.pinecone:PineconeInstrumentor"
+[tool.poetry.extras]
+instruments = ["pinecone-client"]
+
+[tool.poetry.plugins."opentelemetry_instrumentor"]
+pinecone_client = "opentelemetry.instrumentation.pinecone:PineconeInstrumentor"
diff --git a/packages/opentelemetry-instrumentation-qdrant/poetry.lock b/packages/opentelemetry-instrumentation-qdrant/poetry.lock
diff --git a/packages/opentelemetry-instrumentation-qdrant/pyproject.toml b/packages/opentelemetry-instrumentation-qdrant/pyproject.toml
@@ -42,5 +42,8 @@ opentelemetry-sdk = "^1.23.0"
 requires = [ "poetry-core" ]
 build-backend = "poetry.core.masonry.api"
 
-[project.entry-points.opentelemetry_instrumentor]
+[tool.poetry.extras]
+instruments = ["qdrant-client"]
+
+[tool.poetry.plugins."opentelemetry_instrumentor"]
 qdrant_client = "opentelemetry.instrumentation.qdrant:QdrantInstrumentor"
diff --git a/packages/opentelemetry-instrumentation-replicate/poetry.lock b/packages/opentelemetry-instrumentation-replicate/poetry.lock
diff --git a/packages/opentelemetry-instrumentation-replicate/pyproject.toml b/packages/opentelemetry-instrumentation-replicate/pyproject.toml
@@ -42,5 +42,8 @@ replicate = ">=0.23.1,<0.25.0"
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
-[project.entry-points.opentelemetry_instrumentor]
+[tool.poetry.extras]
+instruments = ["replicate"]
+
+[tool.poetry.plugins."opentelemetry_instrumentor"]
 replicate = "opentelemetry.instrumentation.replicate:ReplicateInstrumentor"
diff --git a/packages/opentelemetry-instrumentation-transformers/pyproject.toml b/packages/opentelemetry-instrumentation-transformers/pyproject.toml
@@ -36,7 +36,4 @@ pytest-sugar = "1.0.0"
 
 [build-system]
 requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"
-
-[project.entry-points.opentelemetry_instrumentor]
-transformers = "opentelemetry.instrumentation.transformers:TransformersInstrumentor"
+build-backend = "poetry.core.masonry.api"
diff --git a/packages/opentelemetry-instrumentation-vertexai/poetry.lock b/packages/opentelemetry-instrumentation-vertexai/poetry.lock
diff --git a/packages/opentelemetry-instrumentation-vertexai/pyproject.toml b/packages/opentelemetry-instrumentation-vertexai/pyproject.toml
@@ -47,5 +47,8 @@ google-cloud-aiplatform = "^1.39.0"
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
-[project.entry-points.opentelemetry_instrumentor]
-vertexai = "opentelemetry.instrumentation.vertexai:VertexAIInstrumentor"
+[tool.poetry.extras]
+instruments = ["google-cloud-aiplatform"]
+
+[tool.poetry.plugins."opentelemetry_instrumentor"]
+google_cloud_aiplatform = "opentelemetry.instrumentation.vertexai:VertexAIInstrumentor"
diff --git a/packages/opentelemetry-instrumentation-watsonx/pyproject.toml b/packages/opentelemetry-instrumentation-watsonx/pyproject.toml
@@ -32,7 +32,4 @@ pytest-asyncio = "^0.23.5"
 
 [build-system]
 requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"
-
-[project.entry-points.opentelemetry_instrumentor]
-ibm_watson_machine_learning = "opentelemetry.instrumentation.watsonx:WatsonxInstrumentor"
+build-backend = "poetry.core.masonry.api"
diff --git a/packages/opentelemetry-instrumentation-weaviate/poetry.lock b/packages/opentelemetry-instrumentation-weaviate/poetry.lock