From af9d06b4ba153523fed366d3f8b1b19d1fc5f5e8 Mon Sep 17 00:00:00 2001 From: Yateng Hong Date: Wed, 17 Jan 2024 03:27:24 +0800 Subject: [PATCH] [FormRecognizer] Add telemetry header to track usage (#488) --- scripts/data_utils.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/scripts/data_utils.py b/scripts/data_utils.py index 97ac3d5481..697e7b53c7 100644 --- a/scripts/data_utils.py +++ b/scripts/data_utils.py @@ -1029,7 +1029,8 @@ def __new__(cls, *args, **kwargs): url = os.getenv("FORM_RECOGNIZER_ENDPOINT") key = os.getenv("FORM_RECOGNIZER_KEY") if url and key: - cls.instance = DocumentAnalysisClient(endpoint=url, credential=AzureKeyCredential(key)) + cls.instance = DocumentAnalysisClient( + endpoint=url, credential=AzureKeyCredential(key), headers={"x-ms-useragent": "sample-app-aoai-chatgpt/1.0.0"}) else: print("SingletonFormRecognizerClient: Skipping since credentials not provided. Assuming NO form recognizer extensions(like .pdf) in directory") cls.instance = object() # dummy object @@ -1040,4 +1041,4 @@ def __getstate__(self): def __setstate__(self, state): url, key = state - self.instance = DocumentAnalysisClient(endpoint=url, credential=AzureKeyCredential(key)) \ No newline at end of file + self.instance = DocumentAnalysisClient(endpoint=url, credential=AzureKeyCredential(key), headers={"x-ms-useragent": "sample-app-aoai-chatgpt/1.0.0"})