Small fixes.
This commit is contained in:
@@ -14,11 +14,11 @@ def create_ai(args: argparse.Namespace, config: Config) -> AI:
|
||||
Creates an AI subclass instance from the given arguments
|
||||
and configuration file.
|
||||
"""
|
||||
if args.ai:
|
||||
if args.AI:
|
||||
try:
|
||||
ai_conf = config.ais[args.ai]
|
||||
ai_conf = config.ais[args.AI]
|
||||
except KeyError:
|
||||
raise AIError(f"AI ID '{args.ai}' does not exist in this configuration")
|
||||
raise AIError(f"AI ID '{args.AI}' does not exist in this configuration")
|
||||
elif default_ai_ID in config.ais:
|
||||
ai_conf = config.ais[default_ai_ID]
|
||||
else:
|
||||
@@ -34,4 +34,4 @@ def create_ai(args: argparse.Namespace, config: Config) -> AI:
|
||||
ai.config.temperature = args.temperature
|
||||
return ai
|
||||
else:
|
||||
raise AIError(f"AI '{args.ai}' is not supported")
|
||||
raise AIError(f"AI '{args.AI}' is not supported")
|
||||
|
||||
@@ -13,7 +13,7 @@ def create_message(chat: ChatDB, args: argparse.Namespace) -> Message:
|
||||
Creates (and writes) a new message from the given arguments.
|
||||
"""
|
||||
question_parts = []
|
||||
question_list = args.question if args.question is not None else []
|
||||
question_list = args.ask if args.ask is not None else []
|
||||
source_list = args.source if args.source is not None else []
|
||||
code_list = args.source_code if args.source_code is not None else []
|
||||
|
||||
@@ -35,7 +35,7 @@ def create_message(chat: ChatDB, args: argparse.Namespace) -> Message:
|
||||
|
||||
message = Message(question=Question(full_question),
|
||||
tags=args.output_tags, # FIXME
|
||||
ai=args.ai,
|
||||
ai=args.AI,
|
||||
model=args.model)
|
||||
chat.add_to_cache([message])
|
||||
return message
|
||||
@@ -59,7 +59,7 @@ def question_cmd(args: argparse.Namespace, config: Config) -> None:
|
||||
response: AIResponse = ai.request(message,
|
||||
chat,
|
||||
args.num_answers, # FIXME
|
||||
args.otags) # FIXME
|
||||
args.output_tags) # FIXME
|
||||
assert response
|
||||
# TODO:
|
||||
# * add answer to the message above (and create
|
||||
|
||||
Reference in New Issue
Block a user