Skip to content

Commit

Permalink
Merge pull request #236 from YiVal/auto3
Browse files Browse the repository at this point in the history
add name
  • Loading branch information
yje-arch committed Nov 19, 2023
2 parents 4ca8e95 + d0886de commit 6cbf2b2
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/yival/auto_prompt/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,7 @@ def main():
)
parser.parse_args()

name = input("Please enter the name of the experiment: ")
# Directly read the prompt and evaluation aspects using input()
prompt_input = input(
"Please enter the prompt (use {{}} to wrap around input variables): "
Expand All @@ -60,18 +61,17 @@ def main():
"Please provide evaluation aspects (optional): "
)

formatted_prompt = format_input(prompt_input)
aspects = evaluation_aspects_input.split(
","
) if evaluation_aspects_input else []

asyncio.run(auto_generate_config(formatted_prompt, aspects))
asyncio.run(auto_generate_config(prompt_input, aspects))

print(colored("\nGenerating configuration...", "yellow"))

subprocess.run([
"yival", "run", "auto_generated_config.yaml",
"--output_path=auto_generated.pkl"
f"--output_path={name}.pkl"
])

print(colored("\nProcess completed!", "green"))
Expand Down

0 comments on commit 6cbf2b2

Please sign in to comment.