mirror of
https://github.com/OMGeeky/gpt-pilot.git
synced 2026-01-04 18:30:20 +01:00
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
# user_tasks.py
|
|
from termcolor import colored
|
|
|
|
from utils.utils import execute_step, find_role_from_step, generate_app_data, step_already_finished
|
|
from database.database import save_progress, get_progress_steps
|
|
from logger.logger import logger
|
|
from prompts.prompts import get_additional_info_from_user
|
|
from const.function_calls import USER_TASKS
|
|
|
|
|
|
def get_user_tasks(convo, args):
|
|
current_step = 'user_tasks'
|
|
|
|
# If this app_id already did this step, just get all data from DB and don't ask user again
|
|
step = get_progress_steps(args['app_id'], current_step)
|
|
if step and not execute_step(args['step'], current_step):
|
|
step_already_finished(args, step)
|
|
return step['user_tasks']
|
|
|
|
# USER TASKS
|
|
print(colored(f"Generating user tasks...\n", "green"))
|
|
logger.info(f"Generating user tasks...")
|
|
|
|
user_tasks = convo.send_message('user_stories/user_tasks.prompt',
|
|
{}, USER_TASKS)
|
|
|
|
logger.info(user_tasks)
|
|
user_tasks = get_additional_info_from_user(user_tasks, 'product_owner')
|
|
|
|
logger.info(f"Final user tasks: {user_tasks}")
|
|
|
|
save_progress(args['app_id'], current_step, {
|
|
"messages": convo.get_messages(),
|
|
"user_tasks": user_tasks,
|
|
"app_data": generate_app_data(args)
|
|
})
|
|
|
|
return user_tasks
|
|
# USER TASKS END |