danicafisher commited on
Commit
0fdee25
1 Parent(s): 5c7cc99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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.