import lancedb
import config, embedding
db = lancedb.connect(config.memories_db_path)
table = db.open_table(config.memories_table)
docs = [
    {
        "id": "wifi-001",
        "title": "Perubahan password WiFi menjadi login portal",
        "story": (
            "Password WiFi kantor telah berubah. "
            "Sekarang akses WiFi menggunakan login portal. "
            "Akun login akan diberikan melalui chat pribadi masing-masing user."
        ),
    },
    {
        "id": "m365-001",
        "title": "Cara cek spam pada Outlook Group",
        "story": (
            "Untuk mengecek spam pada Outlook Group, buka Outlook Web. "
            "Masuk ke menu Groups, pilih group terkait, lalu cek folder Junk Email atau Spam."
        ),
    },
    {
        "id": "printer-001",
        "title": "Printer tidak terdeteksi di komputer",
        "story": (
            "Jika printer tidak terdeteksi, cek koneksi kabel atau jaringan. "
            "Pastikan driver printer sudah terinstall, lalu coba restart service Print Spooler."
        ),
    },
]
for doc in docs:
    row = {
        "id"            : doc["id"     ],
        "title"         : doc["title"  ],
        "story"         : doc["story"  ],
        "vector_title"  : embedding.embed_text(config.model_url, config.model_name, doc["title"]),
    }
    table.add([row])
    print(f"Menambahkan record: {doc['title']}")
print("\nTable berhasil diproses.")
print("Nama table:", config.memories_table)
print("Jumlah row:", table.count_rows())