mirror of
https://github.com/run-llama/mongodb-demo.git
synced 2026-07-01 20:44:05 -04:00
Laurie Voss
28 lines
890 B
Python
28 lines
890 B
Python
## This script imports the tinytweets.json file into your mongo database
|
|
## It will work for any json file containing a single array of objects
|
|
## There's nothing specific to llamaindex going on here
|
|
## You can get your data into mongo any way you like.
|
|
|
|
json_file = 'tinytweets.json'
|
|
|
|
# Load environment variables from local .env file
|
|
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
|
|
import os
|
|
import json
|
|
from pymongo.mongo_client import MongoClient
|
|
from pymongo.server_api import ServerApi
|
|
|
|
# Load the tweets from a local file
|
|
with open(json_file, 'r') as f:
|
|
tweets = json.load(f)
|
|
|
|
# Create a new client and connect to the server
|
|
client = MongoClient(os.getenv('MONGODB_URI'), server_api=ServerApi('1'))
|
|
db = client[os.getenv("MONGODB_DATABASE")]
|
|
collection = db[os.getenv("MONGODB_COLLECTION")]
|
|
|
|
# Insert the tweets into mongo
|
|
collection.insert_many(tweets)
|