omni_bot / docs /examples /load-and-query-pinecone.md
Zack Zitting Bradshaw
Upload folder using huggingface_hub
4962437
```python
import hashlib
import json
from urllib.request import urlopen
from decouple import config
from swarms.drivers import PineconeVectorStoreDriver
def load_data(driver: PineconeVectorStoreDriver) -> None:
response = urlopen(
"https://raw.githubusercontent.com/wedeploy-examples/"
"supermarket-web-example/master/products.json"
)
for product in json.loads(response.read()):
driver.upsert_text(
product["description"],
vector_id=hashlib.md5(product["title"].encode()).hexdigest(),
meta={
"title": product["title"],
"description": product["description"],
"type": product["type"],
"price": product["price"],
"rating": product["rating"]
},
namespace="supermarket-products"
)
vector_driver = PineconeVectorStoreDriver(
api_key=config("PINECONE_API_KEY"),
environment=config("PINECONE_ENVIRONMENT"),
index_name=config("PINECONE_INDEX_NAME")
)
load_data(vector_driver)
result = vector_driver.query(
"fruit",
count=3,
filter={
"price": {"$lte": 15},
"rating": {"$gte": 4}
},
namespace="supermarket-products"
)
print(result)
```