added maas headers

microsoft · Aug 14, 2024 · 21cbedd · 21cbedd
1 parent 63e383d
commit 21cbedd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/runtime/prompty/prompty/serverless/executor.py b/runtime/prompty/prompty/serverless/executor.py
@@ -46,6 +46,7 @@ def invoke(self, data: any) -> any:
             response = ChatCompletionsClient(
                 endpoint=self.endpoint,
                 credential=AzureKeyCredential(self.key),
+                user_agent=f"prompty/{VERSION}"
             ).complete(
                 model=self.model,
                 messages=data if isinstance(data, list) else [data],
@@ -61,6 +62,7 @@ def invoke(self, data: any) -> any:
             response = EmbeddingsClient(
                 endpoint=self.endpoint,
                 credential=AzureKeyCredential(self.key),
+                user_agent=f"prompty/{VERSION}",
             ).complete(
                 model=self.model,
                 input=data if isinstance(data, list) else [data],