diff --git a/haystack/preview/components/rankers/similarity.py b/haystack/preview/components/rankers/similarity.py index 66f620294..5d5125006 100644 --- a/haystack/preview/components/rankers/similarity.py +++ b/haystack/preview/components/rankers/similarity.py @@ -56,6 +56,12 @@ class SimilarityRanker: self.model = None self.tokenizer = None + def _get_telemetry_data(self) -> Dict[str, Any]: + """ + Data that is sent to Posthog for usage analytics. + """ + return {"model": str(self.model_name_or_path)} + def warm_up(self): """ Warm up the model and tokenizer used in scoring the documents. diff --git a/haystack/preview/components/readers/extractive.py b/haystack/preview/components/readers/extractive.py index 081646dd2..be18b5698 100644 --- a/haystack/preview/components/readers/extractive.py +++ b/haystack/preview/components/readers/extractive.py @@ -67,6 +67,12 @@ class ExtractiveReader: self.no_answer = no_answer self.calibration_factor = calibration_factor + def _get_telemetry_data(self) -> Dict[str, Any]: + """ + Data that is sent to Posthog for usage analytics. + """ + return {"model": self.model_name_or_path} + def to_dict(self) -> Dict[str, Any]: """ Serialize this component to a dictionary. diff --git a/haystack/preview/components/retrievers/memory_bm25_retriever.py b/haystack/preview/components/retrievers/memory_bm25_retriever.py index 5ae1224e6..564f2fa08 100644 --- a/haystack/preview/components/retrievers/memory_bm25_retriever.py +++ b/haystack/preview/components/retrievers/memory_bm25_retriever.py @@ -41,6 +41,12 @@ class MemoryBM25Retriever: self.top_k = top_k self.scale_score = scale_score + def _get_telemetry_data(self) -> Dict[str, Any]: + """ + Data that is sent to Posthog for usage analytics. + """ + return {"document_store": type(self.document_store).__name__} + def to_dict(self) -> Dict[str, Any]: """ Serialize this component to a dictionary. diff --git a/haystack/preview/components/retrievers/memory_embedding_retriever.py b/haystack/preview/components/retrievers/memory_embedding_retriever.py index 1d5f1cea2..17a709fdf 100644 --- a/haystack/preview/components/retrievers/memory_embedding_retriever.py +++ b/haystack/preview/components/retrievers/memory_embedding_retriever.py @@ -44,6 +44,12 @@ class MemoryEmbeddingRetriever: self.scale_score = scale_score self.return_embedding = return_embedding + def _get_telemetry_data(self) -> Dict[str, Any]: + """ + Data that is sent to Posthog for usage analytics. + """ + return {"document_store": type(self.document_store).__name__} + def to_dict(self) -> Dict[str, Any]: """ Serialize this component to a dictionary. diff --git a/haystack/preview/components/writers/document_writer.py b/haystack/preview/components/writers/document_writer.py index 5ce8c9d4c..5b3d2f4ac 100644 --- a/haystack/preview/components/writers/document_writer.py +++ b/haystack/preview/components/writers/document_writer.py @@ -19,6 +19,12 @@ class DocumentWriter: self.document_store = document_store self.policy = policy + def _get_telemetry_data(self) -> Dict[str, Any]: + """ + Data that is sent to Posthog for usage analytics. + """ + return {"document_store": type(self.document_store).__name__} + def to_dict(self) -> Dict[str, Any]: """ Serialize this component to a dictionary.