Skip to content

Commit

Permalink
feat: (gpt) add model arg
Browse files Browse the repository at this point in the history
>sdb
  • Loading branch information
yufeikang committed Feb 22, 2024
1 parent bf848f6 commit 73ab615
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/ai_cli/bot/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,7 @@ class GPTBot(Bot):
def __init__(self, setting: Setting, args=None, **kwargs):
super().__init__(setting, **kwargs)
self.init_env(setting, args=args)
self.model = setting.model.get_value()
self.model = args.model or setting.model.get_value()
self.max_tokens = setting.max_tokens.get_value()
init_kwargs = {
"api_key": self.api_key,
Expand Down

0 comments on commit 73ab615

Please sign in to comment.