Skip to content

Commit fb4b672

Browse files
authored
Add deployments cli cmd (openai#165)
1 parent 0abf641 commit fb4b672

File tree

2 files changed

+40
-0
lines changed

2 files changed

+40
-0
lines changed

openai/cli.py

Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -183,6 +183,28 @@ def create(cls, args):
183183
sys.stdout.flush()
184184

185185

186+
class Deployment:
187+
@classmethod
188+
def get(cls, args):
189+
resp = openai.Deployment.retrieve(id=args.id)
190+
print(resp)
191+
192+
@classmethod
193+
def delete(cls, args):
194+
model = openai.Deployment.delete(args.id)
195+
print(model)
196+
197+
@classmethod
198+
def list(cls, args):
199+
models = openai.Deployment.list()
200+
print(models)
201+
202+
@classmethod
203+
def create(cls, args):
204+
models = openai.Deployment.create(model=args.model, scale_settings={"scale_type": args.scale_type})
205+
print(models)
206+
207+
186208
class Model:
187209
@classmethod
188210
def get(cls, args):
@@ -824,6 +846,23 @@ def help(args):
824846
)
825847
sub.set_defaults(func=Completion.create)
826848

849+
# Deployments
850+
sub = subparsers.add_parser("deployments.list")
851+
sub.set_defaults(func=Deployment.list)
852+
853+
sub = subparsers.add_parser("deployments.get")
854+
sub.add_argument("-i", "--id", required=True, help="The deployment ID")
855+
sub.set_defaults(func=Deployment.get)
856+
857+
sub = subparsers.add_parser("deployments.delete")
858+
sub.add_argument("-i", "--id", required=True, help="The deployment ID")
859+
sub.set_defaults(func=Deployment.delete)
860+
861+
sub = subparsers.add_parser("deployments.create")
862+
sub.add_argument("-m", "--model", required=True, help="The model ID")
863+
sub.add_argument("-s", "--scale_type", required=True, help="The scale type. Either 'manual' or 'standard'")
864+
sub.set_defaults(func=Deployment.create)
865+
827866
# Models
828867
sub = subparsers.add_parser("models.list")
829868
sub.set_defaults(func=Model.list)

openai/object_classes.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,5 @@
77
"file": api_resources.File,
88
"fine-tune": api_resources.FineTune,
99
"model": api_resources.Model,
10+
"deployment": api_resources.Deployment,
1011
}

0 commit comments

Comments
 (0)