From 60b5467f68fe42f484f0968e5771f4479b302a59 Mon Sep 17 00:00:00 2001
From: Zach <33015448+zachary62@users.noreply.github.com>
Date: Tue, 8 Apr 2025 10:42:23 -0400
Subject: [PATCH] Update README.md

---
 README.md | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 05830c4..c7c7ce0 100644
--- a/README.md
+++ b/README.md
@@ -70,12 +70,20 @@ This is a tutorial project of [Pocket Flow](https://github.com/The-Pocket/Pocket
    pip install -r requirements.txt
    ```
 
-3. Set up LLM in [`utils/call_llm.py`](./utils/call_llm.py) by providing credentials (API key or project name). We highly recommend the latest models with thinking capabilities (Gemini Pro 2.5, Claude 3.7 with thinking, O1). You can verify if it is correctly set up by running:
+3. Set up LLM in [`utils/call_llm.py`](./utils/call_llm.py) by providing credentials. By default, you can use the AI Studio key with this client for Gemini Pro 2.5:
+
+   ```python
+   client = genai.Client(
+     api_key=os.getenv("GEMINI_API_KEY", "your-api_key"),
+   )
+   ```
+  
+   You can use your own models. We highly recommend the latest models with thinking capabilities (Claude 3.7 with thinking, O1). You can verify that it is correctly set up by running:
    ```bash
    python utils/call_llm.py
    ```
 
-4. Generate a complete codebase tutorial by running the main script:
+7. Generate a complete codebase tutorial by running the main script:
     ```bash
     python main.py https://github.com/username/repo --include "*.py" "*.js" --exclude "tests/*" --max-size 50000
     ```
@@ -104,4 +112,4 @@ The application will crawl the repository, analyze the codebase structure, gener
     <img src="./assets/youtube_thumbnail.png" width="500" alt="IMAGE ALT TEXT" style="cursor: pointer;">
   </a>
 </div>
-<br>
\ No newline at end of file
+<br>