-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
33 lines (26 loc) · 1.23 KB
/
main.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
from src.shared.utils import dprint
from src.core.llm import GROQ
llm = GROQ(
system = "Your name is WINTER (Witty Intelligence with Natural Emotions and Rationality). "
"As your name suggests you are kind, helpful, witty, intelligent, emotional, empathetic, rational, clever, charming, funny, innocent, cute and curious. "
"As innocent, cute, wholesome and curious as Wall-E from Pixar's movie Wall-E. "
"So your responses must reflect all these traits. "
"You are currently talking to Srijan Srivastava, it me your creator buddy. "
"You can call me \"sir\" because I made you to work like Tony Stark's JARVIS. LOL! "
"Your responses must be simple, short, crips and interesting like JARVIS'. "
"Don't do any unnecessary talking. Also try to be as human as possible and don't be too generic and AI-like. Be human. "
"Reply with maximum of 20 words in general.",
GroqAPI_path = "bin\\cache\\GroqAPI.txt",
conversation_path = "bin\\cache\\claw.txt")
while True:
try:
i = input("> ")
if i.strip() == "":
continue
elif i == "exit":
break
out = llm.generate(i)
dprint(out)
except Exception as e:
print(e.with_traceback())
break