diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..4c49bd7 --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +.env diff --git a/main.py b/main.py index af734f9..bbf3dd8 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,26 @@ +import os +from dotenv import load_dotenv +from google import genai + +load_dotenv() +api_key = os.environ.get("GEMINI_API_KEY") + +client = genai.Client(api_key=api_key) + def main(): print("Hello from boot-dev-ai-agent!") + 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." + ) + + print(response.text) + + print(f"Prompt tokens: {response.usage_metadata.prompt_token_count}") + print(f"Response tokens: {response.usage_metadata.candidates_token_count}") + + if __name__ == "__main__": main()