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()