question_cmd: fixes

This commit is contained in:
juk0de 2023-09-09 08:51:17 +02:00 committed by Oleksandr Kozachuk
parent c52713c833
commit c4f7bcc94e

View File

@ -1,5 +1,6 @@
import argparse import argparse
from pathlib import Path from pathlib import Path
from itertools import zip_longest
from ..configuration import Config from ..configuration import Config
from ..chat import ChatDB from ..chat import ChatDB
from ..message import Message, Question from ..message import Message, Question
@ -11,8 +12,26 @@ def create_message(chat: ChatDB, args: argparse.Namespace) -> Message:
""" """
Creates (and writes) a new message from the given arguments. Creates (and writes) a new message from the given arguments.
""" """
# FIXME: add sources to the question question_parts = []
message = Message(question=Question(args.question), question_list = args.question if args.question is not None else []
source_list = args.source if args.source is not None else []
# FIXME: don't surround all sourced files with ```
# -> do it only if '--source-code-only' is True and no source code
# could be extracted from that file
for question, source in zip_longest(question_list, source_list, fillvalue=None):
if question is not None and source is not None:
with open(source) as r:
question_parts.append(f"{question}\n\n```\n{r.read().strip()}\n```")
elif question is not None:
question_parts.append(question)
elif source is not None:
with open(source) as r:
question_parts.append(f"```\n{r.read().strip()}\n```")
full_question = '\n\n'.join(question_parts)
message = Message(question=Question(full_question),
tags=args.output_tags, # FIXME tags=args.output_tags, # FIXME
ai=args.ai, ai=args.ai,
model=args.model) model=args.model)