mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-04-01 16:38:11 +02:00
a7acc07e79
* early work in progress * rename utility script * move actual data seeding to a shareable function * add test * make the test pass with the fix * fix comment --------- Co-authored-by: Richard Kuo (Danswer) <rkuo@onyx.app>
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
import argparse
|
|
import logging
|
|
from logging import getLogger
|
|
|
|
from onyx.db.seeding.chat_history_seeding import seed_chat_history
|
|
|
|
# Configure the logger
|
|
logging.basicConfig(
|
|
level=logging.INFO, # Set the log level (DEBUG, INFO, WARNING, ERROR, CRITICAL)
|
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", # Log format
|
|
handlers=[logging.StreamHandler()], # Output logs to console
|
|
)
|
|
|
|
logger = getLogger(__name__)
|
|
|
|
|
|
def go_main(num_sessions: int, num_messages: int, num_days: int) -> None:
|
|
seed_chat_history(num_sessions, num_messages, num_days)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Seed chat history")
|
|
parser.add_argument(
|
|
"--sessions",
|
|
type=int,
|
|
default=2048,
|
|
help="Number of chat sessions to seed",
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--messages",
|
|
type=int,
|
|
default=4,
|
|
help="Number of chat messages to seed per session",
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--days",
|
|
type=int,
|
|
default=90,
|
|
help="Number of days looking backwards over which to seed the timestamps with",
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
go_main(args.sessions, args.messages, args.days)
|