Fix ResourceNotFound in chat model type for AzureOpenAI (#337)

* Updated endpoint in model.py * Update spacy_llm/models/rest/azure/model.py * Update spacy_llm/models/rest/azure/model.py --------- Co-authored-by: Raphael Mitsch <[email protected]>
explosion · Oct 20, 2023 · 7a0460c · 7a0460c
1 parent 5c2b565
commit 7a0460c
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}/{self._model_type.value}"
+            + f"openai/deployments/{self._name}/{'' if self._model_type == ModelType.COMPLETION else 'chat/'}"
+            f"completions"
         )
 
     @property