langchain/templates/elastic-query-generator/ingest.py
2023-10-26 19:44:30 -07:00

23 lines
649 B
Python

import os
from elasticsearch import Elasticsearch
es_host = os.environ["ELASTIC_SEARCH_SERVER"]
es_password = os.environ["ELASTIC_PASSWORD"]
db = Elasticsearch(
es_host,
http_auth=('elastic', es_password),
ca_certs='http_ca.crt' # Replace with your actual path
)
customers = [
{"firstname": "Jennifer", "lastname": "Walters"},
{"firstname": "Monica","lastname":"Rambeau"},
{"firstname": "Carol","lastname":"Danvers"},
{"firstname": "Wanda","lastname":"Maximoff"},
{"firstname": "Jennifer","lastname":"Takeda"},
]
for i, customer in enumerate(customers):
db.create(index="customers", document=customer, id=i)