From b2d99d4de208b5cf8920e4d3f4ce5b54c362fc7d Mon Sep 17 00:00:00 2001 From: Darius Rapalis Date: Mon, 8 Sep 2025 14:07:14 +0300 Subject: [PATCH] giving user an option to pass in the prompt --- main.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index bbf3dd8..45cfb3e 100644 --- a/main.py +++ b/main.py @@ -1,4 +1,5 @@ import os +import sys from dotenv import load_dotenv from google import genai @@ -8,11 +9,15 @@ api_key = os.environ.get("GEMINI_API_KEY") client = genai.Client(api_key=api_key) def main(): - print("Hello from boot-dev-ai-agent!") + if len(sys.argv) < 2: + print("Error: No prompt provided.\nUsage: uv run main.py \"\"") + sys.exit(1) + + user_prompt = sys.argv[1] response = client.models.generate_content( model="gemini-2.0-flash-001", - contents="Why is Boot.dev such a great place to learn backend development? Use one paragraph maximum." + contents=user_prompt ) print(response.text)