[MRG] refactor the github login

mle/cli.py

@@ -208,12 +208,13 @@ def new(name):
         if not api_key:
             console.log("API key is required. Aborted.")
             return
-    
+
     elif platform == 'MistralAI':
         api_key = questionary.password("What is your MistralAI API key?").ask()
         if not api_key:
             console.log("API key is required. Aborted.")
             return
+
     elif platform == 'DeepSeek':
         api_key = questionary.password("What is your DeepSeek API key?").ask()
         if not api_key:
@@ -232,8 +233,10 @@ def new(name):
         yaml.dump({
             'platform': platform,
             'api_key': api_key,
-            'search_key': search_api_key
+            'search_key': search_api_key,
+            'integration': {},
         }, outfile, default_flow_style=False)
+
     # init the memory
     Memory(project_dir)
 
@@ -257,13 +260,11 @@ def integrate(reset):
     ).ask()
 
     if platform == "GitHub":
+        from mle.integration.github import github_login
         if not reset and config.get("integration").get("github"):
             print("GitHub is already integrated.")
         else:
-            token = questionary.password(
-                "What is your GitHub token? (https://github.com/settings/tokens)"
-            ).ask()
-
+            token = github_login()
             config["integration"]["github"] = {
                 "token": token
             }

mle/integration/__init__.py

@@ -1,3 +1,3 @@
-from .github import GitHubIntegration
+from .github import GitHubIntegration, github_login
 from .google_calendar import GoogleCalendarIntegration, google_calendar_login
 from .kaggle import KaggleIntegration, kaggle_login

mle/integration/github.py

@@ -1,10 +1,21 @@
 import os
 import base64
 import requests
+import questionary
 from fnmatch import fnmatch
 from datetime import datetime, timezone, timedelta
 
 
+def github_login():
+    """
+    GitHub login by API token.
+    :returns: API key.
+    """
+    token = questionary.password(
+        "What is your GitHub token? (https://github.com/settings/tokens)").ask()
+    return token
+
+
 class GitHubIntegration:
     BASE_URL = "https://api.github.com"
 

mle/workflow/report.py

@@ -8,7 +8,7 @@
 from mle.model import load_model
 from mle.agents import SummaryAgent, ReportAgent
 from mle.utils.system import get_config, write_config, check_config
-from mle.integration import GoogleCalendarIntegration
+from mle.integration import GoogleCalendarIntegration, github_login
 
 
 def ask_data(data_str: str):
@@ -24,26 +24,6 @@ def ask_data(data_str: str):
         return f"[green]Dataset:[/green] {data_str}"
 
 
-def ask_github_token():
-    """
-    Ask the user to integrate GitHub.
-    :return: the GitHub token.
-    """
-    config = get_config() or {}
-    if "integration" not in config.keys():
-        config["integration"] = {}
-
-    if "github" not in config["integration"].keys():
-        token = questionary.password(
-            "What is your GitHub token? (https://github.com/settings/tokens)"
-        ).ask()
-
-        config["integration"]["github"] = {"token": token}
-        write_config(config)
-
-    return config["integration"]["github"]["token"]
-
-
 def report(
         work_dir: str,
         github_repo: str,
@@ -68,6 +48,14 @@ def report(
     events = None
     if check_config(console):
         config = get_config()
+        if github_token is None:
+            if "github" in config.get("integration", {}).keys():
+                github_token = config["integration"]["github"].get("token")
+            else:
+                github_token = github_login()
+                config["integration"]["github"] = {"token": github_token}
+                write_config(config)
+
         if "google_calendar" in config.get("integration", {}).keys():
             google_token = pickle.loads(config["integration"]["google_calendar"].get("token"))
             google_calendar = GoogleCalendarIntegration(google_token)
@@ -77,7 +65,7 @@ def report(
         model,
         github_repo=github_repo,
         username=github_username,
-        github_token=github_token or ask_github_token(),
+        github_token=github_token,
     )
     reporter = ReportAgent(model, console)