ababio commited on
Commit
95d85ed
·
verified ·
1 Parent(s): 7052798

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import os
2
  import streamlit as st
3
  from dotenv import load_dotenv
4
- from pinecone import PineconeClient
 
5
  from llama_index.embeddings import OpenAIEmbedding
6
  from llama_index.ingestion import IngestionPipeline
7
  from llama_index.query_engine import RetrieverQueryEngine
@@ -20,7 +21,7 @@ index_name = os.getenv("INDEX_NAME")
20
  embed_model = OpenAIEmbedding(api_key=openai_api_key)
21
 
22
  # Initialize connection to Pinecone
23
- pinecone_client = PineconeClient(api_key=pinecone_api_key)
24
  pinecone_index = pinecone_client.Index(index_name)
25
  vector_store = PineconeVectorStore(pinecone_index=pinecone_index)
26
 
 
1
  import os
2
  import streamlit as st
3
  from dotenv import load_dotenv
4
+ from pinecone.grpc import PineconeGRPC
5
+ from pinecone import ServerlessSpec
6
  from llama_index.embeddings import OpenAIEmbedding
7
  from llama_index.ingestion import IngestionPipeline
8
  from llama_index.query_engine import RetrieverQueryEngine
 
21
  embed_model = OpenAIEmbedding(api_key=openai_api_key)
22
 
23
  # Initialize connection to Pinecone
24
+ pinecone_client = PineconeGRPC(api_key=pinecone_api_key)
25
  pinecone_index = pinecone_client.Index(index_name)
26
  vector_store = PineconeVectorStore(pinecone_index=pinecone_index)
27