diff --git a/trustgraph-flow/trustgraph/model/text_completion/azure_openai/llm.py b/trustgraph-flow/trustgraph/model/text_completion/azure_openai/llm.py index 9d803c90..def44bd4 100755 --- a/trustgraph-flow/trustgraph/model/text_completion/azure_openai/llm.py +++ b/trustgraph-flow/trustgraph/model/text_completion/azure_openai/llm.py @@ -20,7 +20,7 @@ default_ident = "text-completion" default_temperature = 0.0 default_max_output = 4192 -default_api = "2024-12-01-preview" +default_api = os.getenv("AZURE_API_VERSION", "2024-12-01-preview") default_endpoint = os.getenv("AZURE_ENDPOINT", None) default_token = os.getenv("AZURE_TOKEN", None) default_model = os.getenv("AZURE_MODEL", None)