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 dd2ba3ffa..06df969cd 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 8a3fbc670..55861dd71 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 0d75d233e..bf5ba92dd 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 2609db2ab..2f6d3e267 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 1827e42ba..31586b069 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 56d3a1ff0..68b41c670 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 bd46b1c1c..3051c3aaa 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 021675e85..bc12b858f 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 eea551918..430c3189e 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 f8cd79556..4fd5e080e 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 137143afd..0548e4fd9 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 828c11438..7882b5381 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 b4a6268b0..4af5e2d8a 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 65e3c1907..2ffb14d4f 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 43ca48401..4b6d8d813 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"