implement deployment id

This commit is contained in:
Paul Gauthier 2023-12-05 11:31:17 -08:00
parent bf03f43b44
commit b107db98fa
8 changed files with 21 additions and 13 deletions

View file

@ -189,7 +189,6 @@ def main(argv=None, input=None, output=None, force_git_root=None):
metavar="OPENAI_API_VERSION",
help="Specify the api_version",
)
# TODO: use deployment_id
model_group.add_argument(
"--openai-api-deployment-id",
metavar="OPENAI_API_DEPLOYMENT_ID",
@ -507,7 +506,9 @@ def main(argv=None, input=None, output=None, force_git_root=None):
client = openai.OpenAI(api_key=args.openai_api_key, **kwargs)
main_model = models.Model.create(args.model, client)
main_model = models.Model.create(
args.model, client, deployment_id=args.openai_api_deployment_id
)
try:
coder = Coder.create(