fix: Pass deployment name as param for azure api (#406)

This commit is contained in:
Taranjeet Singh
2023-08-08 23:55:26 -07:00
committed by GitHub
parent 030e3521a9
commit 1f0f0c93b7
4 changed files with 20 additions and 5 deletions

View File

@@ -62,6 +62,7 @@ class QueryConfig(BaseConfig):
top_p=None,
history=None,
stream: bool = False,
deployment_name=None,
):
"""
Initializes the QueryConfig instance.
@@ -106,6 +107,7 @@ class QueryConfig(BaseConfig):
self.max_tokens = max_tokens if max_tokens else 1000
self.model = model
self.top_p = top_p if top_p else 1
self.deployment_name = deployment_name
if self.validate_template(template):
self.template = template