From c92d7cc5a1b1660c8d9a727df16f7b67e2102562 Mon Sep 17 00:00:00 2001 From: JackColquitt Date: Sun, 29 Sep 2024 12:23:00 -0700 Subject: [PATCH] Slight tweak to AzureAI --- trustgraph/model/text_completion/azure/llm.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/trustgraph/model/text_completion/azure/llm.py b/trustgraph/model/text_completion/azure/llm.py index 25339924..72c83e6c 100755 --- a/trustgraph/model/text_completion/azure/llm.py +++ b/trustgraph/model/text_completion/azure/llm.py @@ -64,6 +64,7 @@ class Processor(ConsumerProducer): self.token = token self.temperature = temperature self.max_output = max_output + self.model = "AzureAI" def build_prompt(self, system, content): @@ -140,7 +141,7 @@ class Processor(ConsumerProducer): print("Send response...", flush=True) - r = TextCompletionResponse(response=resp, error=None, in_token=inputtokens, out_token=outputtokens, model="AzureAI") + r = TextCompletionResponse(response=resp, error=None, in_token=inputtokens, out_token=outputtokens, model=self.model) self.producer.send(r, properties={"id": id}) except TooManyRequests: