ababio commited on
Commit
8d877c4
1 Parent(s): cd986d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -9
app.py CHANGED
@@ -6,7 +6,11 @@ from openai import OpenAI
6
  from llama_index.node_parser import SemanticSplitterNodeParser
7
  from llama_index.embeddings import OpenAIEmbedding
8
  from llama_index.ingestion import IngestionPipeline
9
- from pinecone import PineconeClient, Index, create_index
 
 
 
 
10
 
11
  # Set OpenAI API key from Streamlit secrets
12
  pinecone_api_key = os.getenv("PINECONE_API_KEY")
@@ -15,18 +19,19 @@ openai_api_key = os.getenv("OPENAI_API_KEY")
15
  # Initialize OpenAI client
16
  client = OpenAI(api_key=openai_api_key)
17
 
18
- # Initialize Pinecone connection
19
- pinecone_client = PineconeClient(api_key=pinecone_api_key)
20
- index_name = "annualreport"
21
 
22
- # Check if the index exists, if not, create it
23
- if index_name not in pinecone_client.list_indexes():
24
- create_index(name=index_name, dimension=1536) # Dimension should match your embedding model
25
 
26
- # Initialize Pinecone index
27
- pinecone_index = Index(index_name)
 
 
28
  vector_store = PineconeVectorStore(pinecone_index=pinecone_index)
29
 
 
 
30
  # Initialize vector index and retriever
31
  vector_index = VectorStoreIndex.from_vector_store(vector_store=vector_store)
32
  retriever = VectorIndexRetriever(index=vector_index, similarity_top_k=5)
 
6
  from llama_index.node_parser import SemanticSplitterNodeParser
7
  from llama_index.embeddings import OpenAIEmbedding
8
  from llama_index.ingestion import IngestionPipeline
9
+
10
+ from pinecone.grpc import PineconeGRPC
11
+ from pinecone import ServerlessSpec
12
+
13
+ from llama_index.vector_stores import PineconeVectorStore
14
 
15
  # Set OpenAI API key from Streamlit secrets
16
  pinecone_api_key = os.getenv("PINECONE_API_KEY")
 
19
  # Initialize OpenAI client
20
  client = OpenAI(api_key=openai_api_key)
21
 
 
 
 
22
 
23
+ # Initialize connection to Pinecone
24
+ pc = PineconeGRPC(api_key=pinecone_api_key)
25
+ index_name = "anualreport"
26
 
27
+ # Initialize your index
28
+ pinecone_index = pc.Index(index_name)
29
+
30
+ # Initialize VectorStore
31
  vector_store = PineconeVectorStore(pinecone_index=pinecone_index)
32
 
33
+ pinecone_index.describe_index_stats()
34
+
35
  # Initialize vector index and retriever
36
  vector_index = VectorStoreIndex.from_vector_store(vector_store=vector_store)
37
  retriever = VectorIndexRetriever(index=vector_index, similarity_top_k=5)