Remove old code.
This commit is contained in:
parent
33023d29f9
commit
17de0b9967
@ -18,110 +18,7 @@ from .commands.print import print_cmd
|
|||||||
|
|
||||||
def tags_completer(prefix: str, parsed_args: Any, **kwargs: Any) -> list[str]:
|
def tags_completer(prefix: str, parsed_args: Any, **kwargs: Any) -> list[str]:
|
||||||
config = Config.from_file(parsed_args.config)
|
config = Config.from_file(parsed_args.config)
|
||||||
return get_tags_unique(config, prefix)
|
return list(Message.tags_from_dir(Path(config.db), prefix=prefix))
|
||||||
|
|
||||||
|
|
||||||
def tags_cmd(args: argparse.Namespace, config: Config) -> None:
|
|
||||||
"""
|
|
||||||
Handler for the 'tags' command.
|
|
||||||
"""
|
|
||||||
chat = ChatDB.from_dir(cache_path=Path('.'),
|
|
||||||
db_path=Path(config.db))
|
|
||||||
if args.list:
|
|
||||||
tags_freq = chat.tags_frequency(args.prefix, args.contain)
|
|
||||||
for tag, freq in tags_freq.items():
|
|
||||||
print(f"- {tag}: {freq}")
|
|
||||||
# TODO: add renaming
|
|
||||||
|
|
||||||
|
|
||||||
def config_cmd(args: argparse.Namespace) -> None:
|
|
||||||
"""
|
|
||||||
Handler for the 'config' command.
|
|
||||||
"""
|
|
||||||
if args.create:
|
|
||||||
Config.create_default(Path(args.create))
|
|
||||||
|
|
||||||
|
|
||||||
def question_cmd(args: argparse.Namespace, config: Config) -> None:
|
|
||||||
"""
|
|
||||||
Handler for the 'question' command.
|
|
||||||
"""
|
|
||||||
chat = ChatDB.from_dir(cache_path=Path('.'),
|
|
||||||
db_path=Path(config.db))
|
|
||||||
# if it's a new question, create and store it immediately
|
|
||||||
if args.ask or args.create:
|
|
||||||
# FIXME: add sources to the question
|
|
||||||
message = Message(question=Question(args.question),
|
|
||||||
tags=args.ouput_tags, # FIXME
|
|
||||||
ai=args.ai,
|
|
||||||
model=args.model)
|
|
||||||
chat.add_to_cache([message])
|
|
||||||
if args.create:
|
|
||||||
return
|
|
||||||
|
|
||||||
# create the correct AI instance
|
|
||||||
ai: AI = create_ai(args, config)
|
|
||||||
if args.ask:
|
|
||||||
response: AIResponse = ai.request(message,
|
|
||||||
chat,
|
|
||||||
args.num_answers, # FIXME
|
|
||||||
args.otags) # FIXME
|
|
||||||
assert response
|
|
||||||
# TODO:
|
|
||||||
# * add answer to the message above (and create
|
|
||||||
# more messages for any additional answers)
|
|
||||||
pass
|
|
||||||
elif args.repeat:
|
|
||||||
lmessage = chat.latest_message()
|
|
||||||
assert lmessage
|
|
||||||
# TODO: repeat either the last question or the
|
|
||||||
# one(s) given in 'args.repeat' (overwrite
|
|
||||||
# existing ones if 'args.overwrite' is True)
|
|
||||||
pass
|
|
||||||
elif args.process:
|
|
||||||
# TODO: process either all questions without an
|
|
||||||
# answer or the one(s) given in 'args.process'
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
def hist_cmd(args: argparse.Namespace, config: Config) -> None:
|
|
||||||
"""
|
|
||||||
Handler for the 'hist' command.
|
|
||||||
"""
|
|
||||||
|
|
||||||
mfilter = MessageFilter(tags_or=args.or_tags,
|
|
||||||
tags_and=args.and_tags,
|
|
||||||
tags_not=args.exclude_tags,
|
|
||||||
question_contains=args.question,
|
|
||||||
answer_contains=args.answer)
|
|
||||||
chat = ChatDB.from_dir(Path('.'),
|
|
||||||
Path(config.db),
|
|
||||||
mfilter=mfilter)
|
|
||||||
chat.print(args.source_code_only,
|
|
||||||
args.with_tags,
|
|
||||||
args.with_files)
|
|
||||||
|
|
||||||
|
|
||||||
def print_cmd(args: argparse.Namespace, config: Config) -> None:
|
|
||||||
"""
|
|
||||||
Handler for the 'print' command.
|
|
||||||
"""
|
|
||||||
fname = Path(args.file)
|
|
||||||
try:
|
|
||||||
message = Message.from_file(fname)
|
|
||||||
if message:
|
|
||||||
print(message.to_str(source_code_only=args.source_code_only))
|
|
||||||
except MessageError:
|
|
||||||
print(f"File is not a valid message: {args.file}")
|
|
||||||
sys.exit(1)
|
|
||||||
if args.source_code_only:
|
|
||||||
display_source_code(data['answer'])
|
|
||||||
elif args.answer:
|
|
||||||
print(data['answer'].strip())
|
|
||||||
elif args.question:
|
|
||||||
print(data['question'].strip())
|
|
||||||
else:
|
|
||||||
print(dump_data(data).strip())
|
|
||||||
|
|
||||||
|
|
||||||
def create_parser() -> argparse.ArgumentParser:
|
def create_parser() -> argparse.ArgumentParser:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user