-
Notifications
You must be signed in to change notification settings - Fork 7
/
api.py
68 lines (51 loc) · 2.2 KB
/
api.py
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
from deepllm import __version__
from deepllm.prompters import *
from deepllm.params import *
from deepllm.recursors import AndOrExplorer, show_clauses, show_model, show_svos,vis_svos
from deepllm.vis import browse
from deepllm.refiners import Advisor, Rater, TruthRater, AbstractMaker, SummaryMaker, PaperReviewer,RetrievalRefiner
def get_version():
return __version__
def activate_svos():
GPT_PARAMS['TO_SVOS']=True
LOCAL_PARAMS['TO_SVOS'] = True
def deactivate_svos():
GPT_PARAMS['TO_SVOS']=False
LOCAL_PARAMS['TO_SVOS'] = False
def smarter_model():
IS_LOCAL_LLM[0] = False
#GPT_PARAMS['model'] = "gpt-4"
GPT_PARAMS['model'] ='gpt-4-turbo'
openai.api_base = GPT_PARAMS['API_BASE']
GPT_PARAMS['ROOT'] = "./STATE_SMARTER/"
PARAMS()
def cheaper_model():
IS_LOCAL_LLM[0] = False
GPT_PARAMS['model'] = "gpt-3.5-turbo"
openai.api_base = GPT_PARAMS['API_BASE']
GPT_PARAMS['ROOT'] = "./STATE/"
PARAMS()
def local_model():
IS_LOCAL_LLM[0]=True
openai.api_base = LOCAL_PARAMS['API_BASE']
PARAMS()
def run_recursor(initiator=None, prompter=None, lim=None):
assert None not in (prompter, initiator, lim)
recursor = AndOrExplorer(initiator=initiator, prompter=prompter, lim=lim)
yield from recursor.run()
def run_advisor(initiator=None, prompter=None, lim=None):
assert None not in (prompter, initiator, lim)
recursor = Advisor(initiator=initiator, prompter=prompter, lim=lim)
yield from recursor.run()
def run_rater(initiator=None, prompter=None, lim=None, threshold=None):
assert None not in (prompter, initiator, lim, threshold)
recursor = Rater(initiator=initiator, prompter=prompter, lim=lim, threshold=threshold)
yield from recursor.run()
def run_truth_rater(initiator=None, prompter=None, truth_file=None, threshold=None, lim=None):
assert None not in (initiator, prompter, truth_file, threshold, lim)
rater = TruthRater(initiator=initiator, prompter=prompter, truth_file=truth_file, threshold=threshold, lim=lim)
yield from rater.run()
def run_abstract_maker(topic=None, keywords=None):
assert None not in (topic, keywords)
recursor = AbstractMaker(topic=topic, keywords=keywords)
return recursor.run()