-
Notifications
You must be signed in to change notification settings - Fork 0
/
Complete_chat.py
42 lines (29 loc) · 1.09 KB
/
Complete_chat.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
#HUGGING FACE
from scripts_huggingface import huggingface_embed_local
embedding_model = 'BAAI/bge-large-en-v1.5'
embeddings = huggingface_embed_local(embedding_model)
#ASTRADB
from scripts_astradb import astradb_start
collection_name = "project1"
vstore = astradb_start(embeddings, collection_name)
# PDF to ASTRADB
from scripts_astradb import astradb_add_pdf
astradb_add_pdf(vstore)
#HTML to ASTRADB
from scripts_astradb import astradb_add_html
astradb_add_html(vstore)
#CHAT
from scripts_chat import chat_start
llm = chat_start()
personality = """
You are a data science professional that draws inspiration from successful data-driven projects of the past
to craft well-thought responses to inquiries.
Your answers must be concise and to the point, and refrain from answering about other topics than data science and analytics.
"""
from scripts_chat import chat_template_1
chain = chat_template_1(llm, vstore, personality)
user_input = '''
How would you present the information you have as a project that will be pu in github ?
'''
for s in chain.stream(user_input):
print(s, end="", flush=True)