Update app.py
Browse files
app.py
CHANGED
@@ -254,12 +254,10 @@ def generate_code_snippet(llm_choice, embed_model_choice, node_parser_choice, re
|
|
254 |
code_snippet += f"response_mode = '{response_mode}'\n\n"
|
255 |
|
256 |
# Vector store initialization
|
257 |
-
if vector_store_choice == "
|
258 |
-
code_snippet += "
|
259 |
-
code_snippet += "
|
260 |
-
code_snippet += "vector_store =
|
261 |
-
elif vector_store_choice == "Milvus":
|
262 |
-
code_snippet += "vector_store = MilvusVectorStore(dim=1536, overwrite=True)\n"
|
263 |
elif vector_store_choice == "Qdrant":
|
264 |
code_snippet += "client = qdrant_client.QdrantClient(location=':memory:')\n"
|
265 |
code_snippet += "vector_store = QdrantVectorStore(client=client, collection_name='sampledata')\n"
|
|
|
254 |
code_snippet += f"response_mode = '{response_mode}'\n\n"
|
255 |
|
256 |
# Vector store initialization
|
257 |
+
if vector_store_choice == "Pinecone":
|
258 |
+
code_snippet += "pc = Pinecone(api_key=os.environ['PINECONE_API_KEY'])\n"
|
259 |
+
code_snippet += "index = pc.Index('test')\n"
|
260 |
+
code_snippet += "vector_store = PineconeVectorStore(pinecone_index=index)\n"
|
|
|
|
|
261 |
elif vector_store_choice == "Qdrant":
|
262 |
code_snippet += "client = qdrant_client.QdrantClient(location=':memory:')\n"
|
263 |
code_snippet += "vector_store = QdrantVectorStore(client=client, collection_name='sampledata')\n"
|