Update langgraph_pr_review_bot.py

langgraph_pr_review_bot.py

@@ -40,6 +40,19 @@ git_hub_token = user_secrets.get_secret("GITHUB_token_ID")
 google_api_key = user_secrets.get_secret("GOOGLE_API_KEY")
 '''
 
+google_api_key = os.getenv("GOOGLE_API_KEY")
+if not google_api_key:
+    print("Google API key not found in environment variables.")
+    return "Google API key not found. Please set GOOGLE_API_KEY environment variable.", None
+    print(f"Using Google API key: {google_api_key[:4]}... (truncated for security)")
+
+git_hub_token = os.getenv("GITHUB_token_ID")
+if not git_hub_token:
+    print("git_hub_token not found in environment variables.")
+    return "git_hub_token not found. Please set git_hub_token environment variable.", None
+    print(f"git_hub_token : {git_hub_token[:4]}... (truncated for security)")
+
+
 import os
 from typing import Dict, Any, List, Optional, Tuple # Ensure Tuple is imported
 from github import Github, PullRequest
@@ -78,8 +91,8 @@ def fetch_pr_code_changes(repo_name: str, pr_id: int) -> Tuple[Optional[str], Op
             - head_commit_sha (str or None): The SHA of the head commit of the PR.
             - error_message (str or None): An error message if something went wrong.
     """
-    #github_token = os.getenv("GITHUB_TOKEN")
-    github_token = git_hub_token
+    github_token = os.getenv("GITHUB_token_ID")
+    #github_token = git_hub_token
 
     if not github_token:
         print("Error: GITHUB_TOKEN environment variable not set.")
@@ -626,7 +639,7 @@ def post_review_comments_on_github(
                         Raises an exception on failure.
     """
     if github_token is None:
-        github_token = os.getenv("GITHUB_TOKEN")
+        github_token = os.getenv("GITHUB_token_ID")
         if github_token is None:
             logging.error("GitHub token not provided and GITHUB_TOKEN environment variable not set.")
             raise ValueError("GitHub token not provided and GITHUB_TOKEN environment variable not set.")