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)
|