Update spacy_llm/models/rest/azure/model.py

explosion · Oct 20, 2023 · 97e13ff · 97e13ff
1 parent 5b04379
commit 97e13ff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/spacy_llm/models/rest/azure/model.py b/spacy_llm/models/rest/azure/model.py
@@ -48,7 +48,8 @@ def endpoint(self) -> str:
         return (
             self._endpoint
             + ("" if self._endpoint.endswith("/") else "/")
-            + f"openai/deployments/{self._name}/"+("" if self._model_type.value == "completions" else "chat/")+"completions"
+            + f"openai/deployments/{self._name}/{'' if self._model_type == ModelType.COMPLETION else 'chat/'}"
+              f"completions"
         )
 
     @property