Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
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 |
-
#
|
23 |
-
|
24 |
-
|
25 |
|
26 |
-
# Initialize
|
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)
|