pratikshahp commited on
Commit
18f84bc
·
verified ·
1 Parent(s): dd80727

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -7
app.py CHANGED
@@ -6,7 +6,7 @@ from langchain.embeddings import HuggingFaceEmbeddings
6
  from langchain.document_loaders import WhatsAppChatLoader
7
  from langchain.text_splitter import RecursiveCharacterTextSplitter
8
  from dotenv import load_dotenv
9
- import pinecone
10
 
11
  load_dotenv()
12
 
@@ -15,12 +15,16 @@ pinecone_key = os.getenv("PINECONE_API_KEY")
15
  pinecone.init(api_key=pinecone_key, environment='us-west1-gcp')
16
  index_name = "whatsapp-chat-index"
17
 
18
- if index_name not in pinecone.list_indexes():
19
- pinecone.create_index(
20
- name=index_name,
21
- dimension=384, # change as per embedding model
22
- metric="cosine"
23
- )
 
 
 
 
24
 
25
  index = pinecone.Index(index_name)
26
 
 
6
  from langchain.document_loaders import WhatsAppChatLoader
7
  from langchain.text_splitter import RecursiveCharacterTextSplitter
8
  from dotenv import load_dotenv
9
+ from pinecone import Pinecone, ServerlessSpec
10
 
11
  load_dotenv()
12
 
 
15
  pinecone.init(api_key=pinecone_key, environment='us-west1-gcp')
16
  index_name = "whatsapp-chat-index"
17
 
18
+ if 'index_name' not in pc.list_indexes().names():
19
+ pc.create_index(
20
+ name=index_name,
21
+ dimension=1536,
22
+ metric='euclidean',
23
+ spec=ServerlessSpec(
24
+ cloud='aws',
25
+ region='us-west-2'
26
+ )
27
+ )
28
 
29
  index = pinecone.Index(index_name)
30