File size: 1,594 Bytes
215f97f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | """
Seed script — inserts a base initial model into Firebase Firestore and Storage.
Usage:
python seed_model.py
python -m app.scripts.seed_model
"""
import asyncio
import os
import sys
# Ensure the project root is on sys.path so app.* imports work
BASE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".."))
if BASE_DIR not in sys.path:
sys.path.insert(0, BASE_DIR)
from app.core.firebase import init_firebase, get_firestore_db
from app.ml.serving.registry import DummyModel, save_model_version
async def seed():
"""Insert an initial base model record into Firebase."""
init_firebase()
db = get_firestore_db()
if db is None:
print("Firebase Firestore not initialized. Ensure FIREBASE_CREDENTIALS_PATH or JSON is set.")
return
# Check if an active model already exists
active_docs = db.collection("models").where("status", "==", "active").limit(1).get()
if active_docs:
doc = active_docs[0].to_dict()
print(f"Active model already exists in Firebase: version={doc.get('version', active_docs[0].id)}")
return
model_obj = DummyModel()
metrics = {
"accuracy": 0.85,
"f1": 0.88,
"note": "Initial base model.",
}
await save_model_version(model_obj, metrics, version="v1.0.0")
# Set status to active directly
db.collection("models").document("v1.0.0").update({"status": "active"})
print("✓ Initial base model seeded as active in Firebase (version=v1.0.0)")
def main():
asyncio.run(seed())
if __name__ == "__main__":
main()
|