File size: 594 Bytes
29cdc9d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | import click
from vitalis_ide.brain.thinker import ThinkingProcess
from vitalis_ide.brain.ghost_explain import GhostExplain
from vitalis_ide.brain.rag import LocalRAG
@click.group()
def cli():
pass
@cli.command()
@click.argument('input_text')
def think(input_text):
thinker = ThinkingProcess()
thinker.run(input_text)
@cli.command()
def explain():
explainer = GhostExplain()
click.echo(explainer.get_narrative())
@cli.command()
@click.argument('query')
def query_rag(query):
rag = LocalRAG()
click.echo(rag.retrieve(query))
if __name__ == '__main__':
cli()
|