ok/ChatMastermind
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ai_factory: fixed argument parsing bug

Browse Source
This commit is contained in:
juk0de 2023-09-13 07:52:05 +02:00
parent 310cb9421e
commit a7345cbc41
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
chatmastermind/ai_factory.py
View File

@ -17,7 +17,7 @@ def create_ai(args: argparse.Namespace, config: Config) -> AI: # noqa: 11
is not found, it uses the first AI in the list.
"""
ai_conf: AIConfig
if 'AI' in args and args.AI:
if hasattr(args, 'AI') and args.AI:
try:
ai_conf = config.ais[args.AI]
except KeyError:
@ -32,11 +32,11 @@ def create_ai(args: argparse.Namespace, config: Config) -> AI: # noqa: 11
if ai_conf.name == 'openai':
ai = OpenAI(cast(OpenAIConfig, ai_conf))
if 'model' in args and args.model:
if hasattr(args, 'model') and args.model:
ai.config.model = args.model
if 'max_tokens' in args and args.max_tokens:
if hasattr(args, 'max_tokens') and args.max_tokens:
ai.config.max_tokens = args.max_tokens
if 'temperature' in args and args.temperature:
if hasattr(args, 'temperature') and args.temperature:
ai.config.temperature = args.temperature
return ai
else: