giving user an option to pass in the prompt

This commit is contained in:
2025-09-08 14:07:14 +03:00
parent e9a1552bab
commit b2d99d4de2

View File

@@ -1,4 +1,5 @@
import os import os
import sys
from dotenv import load_dotenv from dotenv import load_dotenv
from google import genai from google import genai
@@ -8,11 +9,15 @@ api_key = os.environ.get("GEMINI_API_KEY")
client = genai.Client(api_key=api_key) client = genai.Client(api_key=api_key)
def main(): def main():
print("Hello from boot-dev-ai-agent!") if len(sys.argv) < 2:
print("Error: No prompt provided.\nUsage: uv run main.py \"<your prompt here>\"")
sys.exit(1)
user_prompt = sys.argv[1]
response = client.models.generate_content( response = client.models.generate_content(
model="gemini-2.0-flash-001", 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) print(response.text)