From 5372e2f57e6a0e30788af4659cf664153d8d8a98 Mon Sep 17 00:00:00 2001 From: Gal Kleinman Date: Sat, 9 Mar 2024 18:14:04 +0200 Subject: [PATCH] fix: add entry points for auto-instrumentation --- .../opentelemetry-instrumentation-anthropic/pyproject.toml | 4 +++- packages/opentelemetry-instrumentation-bedrock/pyproject.toml | 3 +++ .../opentelemetry-instrumentation-chromadb/pyproject.toml | 3 +++ packages/opentelemetry-instrumentation-cohere/pyproject.toml | 3 +++ .../opentelemetry-instrumentation-haystack/pyproject.toml | 3 +++ .../opentelemetry-instrumentation-langchain/pyproject.toml | 3 +++ .../opentelemetry-instrumentation-llamaindex/pyproject.toml | 4 +++- packages/opentelemetry-instrumentation-openai/pyproject.toml | 3 +++ .../opentelemetry-instrumentation-pinecone/pyproject.toml | 3 +++ packages/opentelemetry-instrumentation-qdrant/pyproject.toml | 3 +++ .../opentelemetry-instrumentation-replicate/pyproject.toml | 3 +++ .../opentelemetry-instrumentation-transformers/pyproject.toml | 3 +++ .../opentelemetry-instrumentation-vertexai/pyproject.toml | 3 +++ packages/opentelemetry-instrumentation-watsonx/pyproject.toml | 3 +++ .../opentelemetry-instrumentation-weaviate/pyproject.toml | 4 +++- 15 files changed, 45 insertions(+), 3 deletions(-) diff --git a/packages/opentelemetry-instrumentation-anthropic/pyproject.toml b/packages/opentelemetry-instrumentation-anthropic/pyproject.toml index dd2ba3ffa4..06df969cd1 100644 --- a/packages/opentelemetry-instrumentation-anthropic/pyproject.toml +++ b/packages/opentelemetry-instrumentation-anthropic/pyproject.toml @@ -42,7 +42,9 @@ vcrpy = "^6.0.1" pytest-recording = "^0.13.1" opentelemetry-sdk = "^1.23.0" - [build-system] requires = ["poetry-core"] build-backend = "poetry.core.masonry.api" + +[project.entry-points.opentelemetry_instrumentor] +anthropic = "opentelemetry.instrumentation.anthropic:AnthropicInstrumentor" diff --git a/packages/opentelemetry-instrumentation-bedrock/pyproject.toml b/packages/opentelemetry-instrumentation-bedrock/pyproject.toml index 8a3fbc6709..55861dd713 100644 --- a/packages/opentelemetry-instrumentation-bedrock/pyproject.toml +++ b/packages/opentelemetry-instrumentation-bedrock/pyproject.toml @@ -37,3 +37,6 @@ 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/pyproject.toml b/packages/opentelemetry-instrumentation-chromadb/pyproject.toml index 0d75d233e6..bf5ba92ddc 100644 --- a/packages/opentelemetry-instrumentation-chromadb/pyproject.toml +++ b/packages/opentelemetry-instrumentation-chromadb/pyproject.toml @@ -44,3 +44,6 @@ opentelemetry-sdk = "^1.23.0" [build-system] requires = ["poetry-core"] build-backend = "poetry.core.masonry.api" + +[project.entry-points.opentelemetry_instrumentor] +chromadb = "opentelemetry.instrumentation.chromadb:ChromaInstrumentor" diff --git a/packages/opentelemetry-instrumentation-cohere/pyproject.toml b/packages/opentelemetry-instrumentation-cohere/pyproject.toml index 2609db2abd..2f6d3e2671 100644 --- a/packages/opentelemetry-instrumentation-cohere/pyproject.toml +++ b/packages/opentelemetry-instrumentation-cohere/pyproject.toml @@ -45,3 +45,6 @@ opentelemetry-sdk = "^1.23.0" [build-system] requires = ["poetry-core"] build-backend = "poetry.core.masonry.api" + +[project.entry-points.opentelemetry_instrumentor] +cohere = "opentelemetry.instrumentation.cohere:CohereInstrumentor" diff --git a/packages/opentelemetry-instrumentation-haystack/pyproject.toml b/packages/opentelemetry-instrumentation-haystack/pyproject.toml index 1827e42ba6..31586b0695 100644 --- a/packages/opentelemetry-instrumentation-haystack/pyproject.toml +++ b/packages/opentelemetry-instrumentation-haystack/pyproject.toml @@ -45,3 +45,6 @@ opentelemetry-sdk = "^1.23.0" [build-system] requires = ["poetry-core"] build-backend = "poetry.core.masonry.api" + +[project.entry-points.opentelemetry_instrumentor] +haystack = "opentelemetry.instrumentation.haystack:HaystackInstrumentor" diff --git a/packages/opentelemetry-instrumentation-langchain/pyproject.toml b/packages/opentelemetry-instrumentation-langchain/pyproject.toml index 56d3a1ff04..68b41c670e 100644 --- a/packages/opentelemetry-instrumentation-langchain/pyproject.toml +++ b/packages/opentelemetry-instrumentation-langchain/pyproject.toml @@ -50,3 +50,6 @@ opentelemetry-instrumentation-openai = {path="../opentelemetry-instrumentation-o [build-system] requires = ["poetry-core"] build-backend = "poetry.core.masonry.api" + +[project.entry-points.opentelemetry_instrumentor] +langchain = "opentelemetry.instrumentation.langchain:LangchainInstrumentor" diff --git a/packages/opentelemetry-instrumentation-llamaindex/pyproject.toml b/packages/opentelemetry-instrumentation-llamaindex/pyproject.toml index bd46b1c1ca..3051c3aaa2 100644 --- a/packages/opentelemetry-instrumentation-llamaindex/pyproject.toml +++ b/packages/opentelemetry-instrumentation-llamaindex/pyproject.toml @@ -48,7 +48,9 @@ opentelemetry-instrumentation-chromadb = {path="../opentelemetry-instrumentation sqlalchemy = "^2.0.27" llama-index-agent-openai = "^0.1.5" - [build-system] requires = ["poetry-core"] build-backend = "poetry.core.masonry.api" + +[project.entry-points.opentelemetry_instrumentor] +llama_index = "opentelemetry.instrumentation.llamaindex:LlamaIndexInstrumentor" diff --git a/packages/opentelemetry-instrumentation-openai/pyproject.toml b/packages/opentelemetry-instrumentation-openai/pyproject.toml index 021675e85b..bc12b858f6 100644 --- a/packages/opentelemetry-instrumentation-openai/pyproject.toml +++ b/packages/opentelemetry-instrumentation-openai/pyproject.toml @@ -44,3 +44,6 @@ pytest-asyncio = "^0.23.5" [build-system] requires = [ "poetry-core" ] build-backend = "poetry.core.masonry.api" + +[project.entry-points.opentelemetry_instrumentor] +openai = "opentelemetry.instrumentation.openai:OpenAIInstrumentor" diff --git a/packages/opentelemetry-instrumentation-pinecone/pyproject.toml b/packages/opentelemetry-instrumentation-pinecone/pyproject.toml index eea551918b..430c3189ec 100644 --- a/packages/opentelemetry-instrumentation-pinecone/pyproject.toml +++ b/packages/opentelemetry-instrumentation-pinecone/pyproject.toml @@ -47,3 +47,6 @@ pinecone-client = "^2.2.4" [build-system] requires = [ "poetry-core" ] build-backend = "poetry.core.masonry.api" + +[project.entry-points.opentelemetry_instrumentor] +pinecone = "opentelemetry.instrumentation.pinecone:PineconeInstrumentor" diff --git a/packages/opentelemetry-instrumentation-qdrant/pyproject.toml b/packages/opentelemetry-instrumentation-qdrant/pyproject.toml index f8cd795561..4fd5e080e0 100644 --- a/packages/opentelemetry-instrumentation-qdrant/pyproject.toml +++ b/packages/opentelemetry-instrumentation-qdrant/pyproject.toml @@ -41,3 +41,6 @@ opentelemetry-sdk = "^1.23.0" [build-system] requires = [ "poetry-core" ] build-backend = "poetry.core.masonry.api" + +[project.entry-points.opentelemetry_instrumentor] +qdrant_client = "opentelemetry.instrumentation.qdrant:QdrantInstrumentor" diff --git a/packages/opentelemetry-instrumentation-replicate/pyproject.toml b/packages/opentelemetry-instrumentation-replicate/pyproject.toml index 137143afdd..0548e4fd96 100644 --- a/packages/opentelemetry-instrumentation-replicate/pyproject.toml +++ b/packages/opentelemetry-instrumentation-replicate/pyproject.toml @@ -41,3 +41,6 @@ replicate = ">=0.23.1,<0.25.0" [build-system] requires = ["poetry-core"] build-backend = "poetry.core.masonry.api" + +[project.entry-points.opentelemetry_instrumentor] +replicate = "opentelemetry.instrumentation.replicate:ReplicateInstrumentor" diff --git a/packages/opentelemetry-instrumentation-transformers/pyproject.toml b/packages/opentelemetry-instrumentation-transformers/pyproject.toml index 828c114384..7882b53816 100644 --- a/packages/opentelemetry-instrumentation-transformers/pyproject.toml +++ b/packages/opentelemetry-instrumentation-transformers/pyproject.toml @@ -37,3 +37,6 @@ 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" diff --git a/packages/opentelemetry-instrumentation-vertexai/pyproject.toml b/packages/opentelemetry-instrumentation-vertexai/pyproject.toml index b4a6268b0c..4af5e2d8ae 100644 --- a/packages/opentelemetry-instrumentation-vertexai/pyproject.toml +++ b/packages/opentelemetry-instrumentation-vertexai/pyproject.toml @@ -46,3 +46,6 @@ google-cloud-aiplatform = "^1.39.0" [build-system] requires = ["poetry-core"] build-backend = "poetry.core.masonry.api" + +[project.entry-points.opentelemetry_instrumentor] +vertexai = "opentelemetry.instrumentation.vertexai:VertexAIInstrumentor" diff --git a/packages/opentelemetry-instrumentation-watsonx/pyproject.toml b/packages/opentelemetry-instrumentation-watsonx/pyproject.toml index 65e3c19076..2ffb14d4fe 100644 --- a/packages/opentelemetry-instrumentation-watsonx/pyproject.toml +++ b/packages/opentelemetry-instrumentation-watsonx/pyproject.toml @@ -33,3 +33,6 @@ 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" diff --git a/packages/opentelemetry-instrumentation-weaviate/pyproject.toml b/packages/opentelemetry-instrumentation-weaviate/pyproject.toml index 43ca48401f..4b6d8d8132 100644 --- a/packages/opentelemetry-instrumentation-weaviate/pyproject.toml +++ b/packages/opentelemetry-instrumentation-weaviate/pyproject.toml @@ -45,7 +45,9 @@ opentelemetry-sdk = "^1.22.0" opentelemetry-instrumentation-openai = {path="../opentelemetry-instrumentation-openai", develop=true} weaviate-client = "^3.26.0" - [build-system] requires = ["poetry-core"] build-backend = "poetry.core.masonry.api" + +[project.entry-points.opentelemetry_instrumentor] +weaviate = "opentelemetry.instrumentation.weaviate:WeaviateInstrumentor"