Update app.py
Browse files
app.py
CHANGED
@@ -12,6 +12,9 @@ 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")
|
17 |
openai_api_key = os.getenv("OPENAI_API_KEY")
|
@@ -32,6 +35,7 @@ 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)
|
|
|
12 |
|
13 |
from llama_index.vector_stores import PineconeVectorStore
|
14 |
|
15 |
+
from llama_index import VectorStoreIndex
|
16 |
+
from llama_index.retrievers import VectorIndexRetriever
|
17 |
+
|
18 |
# Set OpenAI API key from Streamlit secrets
|
19 |
pinecone_api_key = os.getenv("PINECONE_API_KEY")
|
20 |
openai_api_key = os.getenv("OPENAI_API_KEY")
|
|
|
35 |
|
36 |
pinecone_index.describe_index_stats()
|
37 |
|
38 |
+
|
39 |
# Initialize vector index and retriever
|
40 |
vector_index = VectorStoreIndex.from_vector_store(vector_store=vector_store)
|
41 |
retriever = VectorIndexRetriever(index=vector_index, similarity_top_k=5)
|