Merge pull request #18 from paul-gauthier/issue-17

Added:  --openai-api-base OPENAI_API_BASE
This commit is contained in:
paul-gauthier 2023-06-10 10:54:22 -07:00 committed by GitHub
commit 39d24d9e24
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 3 deletions

View file

@ -50,11 +50,12 @@ class Coder:
map_tokens=1024, map_tokens=1024,
verbose=False, verbose=False,
openai_api_key=None, openai_api_key=None,
openai_api_base=None,
): ):
if openai_api_key: if not openai_api_key:
openai.api_key = openai_api_key
else:
raise MissingAPIKeyError("No OpenAI API key provided.") raise MissingAPIKeyError("No OpenAI API key provided.")
openai.api_key = openai_api_key
openai.api_base = openai_api_base
self.verbose = verbose self.verbose = verbose
self.abs_fnames = set() self.abs_fnames = set()

View file

@ -151,6 +151,12 @@ def main(args=None, input=None, output=None):
help="Specify the OpenAI API key", help="Specify the OpenAI API key",
env_var="OPENAI_API_KEY", env_var="OPENAI_API_KEY",
) )
parser.add_argument(
"--openai-api-base",
metavar="OPENAI_API_BASE",
default="https://api.openai.com/v1",
help="Specify the OpenAI API base endpoint (default: https://api.openai.com/v1)",
)
parser.add_argument( parser.add_argument(
"--dry-run", "--dry-run",
action="store_true", action="store_true",
@ -221,6 +227,7 @@ def main(args=None, input=None, output=None):
map_tokens=args.map_tokens, map_tokens=args.map_tokens,
verbose=args.verbose, verbose=args.verbose,
openai_api_key=args.openai_api_key, openai_api_key=args.openai_api_key,
openai_api_base=args.openai_api_base,
) )
if args.dirty_commits: if args.dirty_commits: