Spaces:
Sleeping
Sleeping
danicafisher
commited on
Commit
•
0fdee25
1
Parent(s):
5c7cc99
Update app.py
Browse files
app.py
CHANGED
@@ -7,8 +7,11 @@ from aimakerspace.openai_utils.prompts import (
|
|
7 |
from aimakerspace.vectordatabase import VectorDatabase
|
8 |
from aimakerspace.openai_utils.chatmodel import ChatOpenAI
|
9 |
from langchain_community.embeddings import OpenAIEmbeddings
|
|
|
10 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
11 |
from langchain_community.document_loaders import PyMuPDFLoader
|
|
|
|
|
12 |
import chainlit as cl
|
13 |
import nest_asyncio
|
14 |
nest_asyncio.apply()
|
@@ -27,6 +30,7 @@ split_documents_Blueprint = text_splitter.split_documents(pdf_loader_Blueprint)
|
|
27 |
# split_documents_NIST = text_splitter.split_texts(documents_NIST)
|
28 |
# split_documents_Blueprint = text_splitter.split_texts(documents_Blueprint)
|
29 |
|
|
|
30 |
|
31 |
RAG_PROMPT_TEMPLATE = """ \
|
32 |
Use the provided context to answer the user's query.
|
|
|
7 |
from aimakerspace.vectordatabase import VectorDatabase
|
8 |
from aimakerspace.openai_utils.chatmodel import ChatOpenAI
|
9 |
from langchain_community.embeddings import OpenAIEmbeddings
|
10 |
+
from langchain_community.vectorstores import Chroma
|
11 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
12 |
from langchain_community.document_loaders import PyMuPDFLoader
|
13 |
+
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
14 |
+
from langchain.prompts import SystemMessagePromptTemplate, HumanMessagePromptTemplate, ChatPromptTemplate
|
15 |
import chainlit as cl
|
16 |
import nest_asyncio
|
17 |
nest_asyncio.apply()
|
|
|
30 |
# split_documents_NIST = text_splitter.split_texts(documents_NIST)
|
31 |
# split_documents_Blueprint = text_splitter.split_texts(documents_Blueprint)
|
32 |
|
33 |
+
rag_documents = split_documents_NIST + split_documents_Blueprint
|
34 |
|
35 |
RAG_PROMPT_TEMPLATE = """ \
|
36 |
Use the provided context to answer the user's query.
|