question_cmd: fixes
This commit is contained in:
parent
c52713c833
commit
c4f7bcc94e
@ -1,5 +1,6 @@
|
||||
import argparse
|
||||
from pathlib import Path
|
||||
from itertools import zip_longest
|
||||
from ..configuration import Config
|
||||
from ..chat import ChatDB
|
||||
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.
|
||||
"""
|
||||
# FIXME: add sources to the question
|
||||
message = Message(question=Question(args.question),
|
||||
question_parts = []
|
||||
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
|
||||
ai=args.ai,
|
||||
model=args.model)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user