mirror of
https://github.com/magicseek/nblm.git
synced 2026-09-14 15:39:27 +08:00
Deheng Huang
71acf76aeb
Add GOOGLE_AUTH_DIR, GOOGLE_AUTH_INDEX for new multi-account structure. Keep GOOGLE_AUTH_FILE_LEGACY for migration detection. Co-Authored-By: Claude <noreply@anthropic.com>
65 lines
2.0 KiB
Python
Executable File
65 lines
2.0 KiB
Python
Executable File
"""
|
|
Configuration for nblm
|
|
Centralizes constants, selectors, and paths
|
|
"""
|
|
|
|
from pathlib import Path
|
|
import os
|
|
import tempfile
|
|
|
|
# Paths
|
|
SKILL_DIR = Path(__file__).parent.parent
|
|
DATA_DIR = SKILL_DIR / "data"
|
|
AUTH_INFO_FILE = DATA_DIR / "auth_info.json"
|
|
AUTH_DIR = DATA_DIR / "auth"
|
|
GOOGLE_AUTH_FILE = AUTH_DIR / "google.json"
|
|
ZLIBRARY_AUTH_FILE = AUTH_DIR / "zlibrary.json"
|
|
LIBRARY_FILE = DATA_DIR / "library.json"
|
|
|
|
# Multi-account Google auth structure
|
|
GOOGLE_AUTH_DIR = AUTH_DIR / "google"
|
|
GOOGLE_AUTH_INDEX = GOOGLE_AUTH_DIR / "index.json"
|
|
|
|
# Legacy path (for migration detection)
|
|
GOOGLE_AUTH_FILE_LEGACY = AUTH_DIR / "google.json"
|
|
|
|
# Set NOTEBOOKLM_HOME to use our auth directory for notebooklm-py
|
|
# This ensures download methods find our storage_state.json
|
|
os.environ.setdefault("NOTEBOOKLM_HOME", str(AUTH_DIR))
|
|
|
|
# Agent-browser configuration
|
|
AGENT_BROWSER_PROFILE_DIR = DATA_DIR / "agent_browser" / "profile"
|
|
AGENT_BROWSER_SESSION_FILE = DATA_DIR / "agent_browser" / "session_id"
|
|
AGENT_BROWSER_SOCKET_DIR = Path(tempfile.gettempdir())
|
|
DEFAULT_SESSION_ID = "notebooklm"
|
|
AGENT_BROWSER_ACTIVITY_FILE = DATA_DIR / "agent_browser" / "last_activity.json"
|
|
AGENT_BROWSER_WATCHDOG_PID_FILE = DATA_DIR / "agent_browser" / "watchdog.pid"
|
|
AGENT_BROWSER_STATE_FILE = DATA_DIR / "agent_browser" / "storage_state.json"
|
|
AGENT_BROWSER_IDLE_TIMEOUT_SECONDS = int(
|
|
os.environ.get("AGENT_BROWSER_IDLE_TIMEOUT_SECONDS", "600")
|
|
)
|
|
AGENT_BROWSER_WATCHDOG_INTERVAL_SECONDS = int(
|
|
os.environ.get("AGENT_BROWSER_WATCHDOG_INTERVAL_SECONDS", "30")
|
|
)
|
|
|
|
# NotebookLM token staleness threshold
|
|
NOTEBOOKLM_TOKEN_STALENESS_DAYS = 7
|
|
|
|
# NotebookLM Selectors
|
|
QUERY_INPUT_SELECTORS = [
|
|
"textarea.query-box-input", # Primary
|
|
'textarea[aria-label="Feld für Anfragen"]', # Fallback German
|
|
'textarea[aria-label="Input for queries"]', # Fallback English
|
|
]
|
|
|
|
RESPONSE_SELECTORS = [
|
|
".to-user-container .message-text-content", # Primary
|
|
"[data-message-author='bot']",
|
|
"[data-message-author='assistant']",
|
|
]
|
|
|
|
# Timeouts
|
|
LOGIN_TIMEOUT_MINUTES = 10
|
|
QUERY_TIMEOUT_SECONDS = 120
|
|
PAGE_LOAD_TIMEOUT = 30000
|