From fd5432cf1abf9c63a66ad17e71a7573d6bfc84ce Mon Sep 17 00:00:00 2001 From: Maciej Majek Date: Wed, 4 Sep 2024 15:09:39 +0200 Subject: [PATCH] chore: remove cost calculation, enahnce logging MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Bartłomiej Boczek --- src/rai/rai/cli/rai_cli.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/src/rai/rai/cli/rai_cli.py b/src/rai/rai/cli/rai_cli.py index 8f77a8235..e3b6ca796 100644 --- a/src/rai/rai/cli/rai_cli.py +++ b/src/rai/rai/cli/rai_cli.py @@ -92,24 +92,21 @@ def build_robot_identity(): documentation = str([doc.page_content for doc in docs]) n_tokens = len(documentation) // 4.0 logger.info( + "Building the robot docs vector store... " f"The documentation's length is {len(documentation)} chars, " f"approximately {n_tokens} tokens" ) - logger.warn( - f"Building the robot docs vector store will cost " - f"approximately {n_tokens / 1_000_000 * 0.1:.4f}$. " - "Do you want to continue? (y/n)" - ) + logger.warn("Do you want to continue? (y/n)") if input() == "y": build_docs_vector_store() else: logger.info("Skipping the robot docs vector store creation.") - logger.warn( - f"Building the robot identity will cost " - f"approximately {n_tokens / 1_000_000 * 0.15:.4f}$. " - "Do you want to continue? (y/n)" + logger.info( + "Building the robot identity... " + "You can do it manually by creating {save_dir}/robot_identity.txt " ) + logger.warn("Do you want to continue? (y/n)") if input() == "y": build_robot_identity() else: