2dd531114f
* Move kotaemon and ktem into same folder * Update docs * Update CI * Resolve mypy, isorts * Re-allow test pdf files
30 lines
666 B
Python
30 lines
666 B
Python
import time
|
|
|
|
from ktem.db.models import Conversation, Source, engine
|
|
from sqlmodel import Session
|
|
|
|
|
|
def add_conversation():
|
|
"""Add conversation to the manager."""
|
|
with Session(engine) as session:
|
|
c1 = Conversation(name="Conversation 1")
|
|
c2 = Conversation()
|
|
session.add(c1)
|
|
time.sleep(1)
|
|
session.add(c2)
|
|
time.sleep(1)
|
|
session.commit()
|
|
|
|
|
|
def add_files():
|
|
with Session(engine) as session:
|
|
s1 = Source(name="Source 1", path="Path 1")
|
|
s2 = Source(name="Source 2", path="Path 2")
|
|
session.add(s1)
|
|
session.add(s2)
|
|
session.commit()
|
|
|
|
|
|
# add_conversation()
|
|
add_files()
|