AI editor

.github/workflows/ci.yml

@@ -38,6 +38,18 @@ jobs:
           google_api_key: ${{ secrets.GOOGLE_API_KEY }}
           install_qt: true
 
+      - name: Create ankiaddon file
+        run: |
+          bash ./scripts/release.sh
+        env:
+          GOOGLE_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
+
+      - name: Archive .ankiaddon
+        uses: actions/upload-artifact@v4
+        with:
+          name: ankihub_addon_${{ github.sha }}
+          path: ankihub.ankiaddon
+
       - name: Replace Anki version
         if: ${{ matrix.replace_anki_version }}
         run: pip install "${{ matrix.anki_package_version }}"
@@ -159,18 +171,6 @@ jobs:
         with:
           fail_ci_if_error: false
 
-      - name: Create ankiaddon file
-        run: |
-          bash ./scripts/release.sh
-        env:
-          GOOGLE_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
-
-      - name: Archive .ankiaddon
-        uses: actions/upload-artifact@v4
-        with:
-          name: ankihub_addon_${{ github.sha }}
-          path: ankihub.ankiaddon
-
       - name: Run pre-commit
         run: pre-commit run --all
 

.gitignore

@@ -2,10 +2,9 @@
 ankihub/meta.json
 
 # Anki base directory
-anki_base
-
+anki_base/
 # AnkiHub base directory
-ankihub_base
+ankihub_base/
 
 # Ignore everything except the README.
 /ankihub/user_files/*

.pre-commit-config.yaml

@@ -27,6 +27,12 @@ repos:
       hooks:
           - id: pycln
 
+    - repo: https://github.com/PyCQA/autoflake
+      rev: v2.3.1
+      hooks:
+          - id: autoflake
+            args: [--remove-all-unused-imports, --in-place]
+
     - repo: local
       hooks:
           - id: no_absolute_imports_from_ankihub

ankihub/entry_point.py

@@ -237,6 +237,23 @@ def _general_setup():
     LOGGER.info(
         "Set up feature flag fetching (flags will be fetched in the background)."
     )
+    # Make all .sh files executable
+    for sh_file in Path(__file__).parent.rglob("*.sh"):
+        current_mode = sh_file.stat().st_mode
+        executable_mode = (
+            current_mode | 0o111
+        )  # Add execute permission for user, group and others
+        sh_file.chmod(executable_mode)
+    LOGGER.info("Made all .sh files executable.")
+
+    if config.labs_enabled:
+        from .labs.llm import llm
+
+        try:
+            llm.setup()
+            LOGGER.info("Set up LLM prompt functionality.")
+        except Exception as e:
+            LOGGER.exception("Failed to set up LLM prompt functionality", error=e)
 
 
 def _copy_web_media_to_media_folder():

ankihub/gui/menu.py

@@ -24,6 +24,7 @@
 from ..addon_ankihub_client import AddonAnkiHubClient as AnkiHubClient
 from ..ankihub_client import AnkiHubHTTPError
 from ..db import ankihub_db
+from ..labs.gui.menu import setup_labs_menu
 from ..media_import.ui import open_import_dialog
 from ..settings import ADDON_VERSION, config
 from .config_dialog import get_config_dialog_manager
@@ -74,6 +75,8 @@ def refresh_ankihub_menu() -> None:
     _config_setup(parent=menu_state.ankihub_menu)
     _ankihub_terms_and_policy_setup(parent=menu_state.ankihub_menu)
     _ankihub_help_setup(parent=menu_state.ankihub_menu)
+    if config.labs_enabled:
+        setup_labs_menu(menu_state.ankihub_menu)
 
 
 class AnkiHubLogin(QWidget):

ankihub/labs/__init__.py

@@ -0,0 +1 @@
+"""AnkiHub Labs module for experimental features."""

ankihub/labs/gui/__init__.py

@@ -0,0 +1 @@
+"""GUI components for AnkiHub Labs features."""

ankihub/labs/gui/menu.py

@@ -0,0 +1,18 @@
+"""AnkiHub Labs menu setup."""
+
+from aqt.qt import QAction, QMenu, qconnect
+
+from ..secrets import open_secrets_dialog
+
+
+def setup_labs_menu(parent_menu: QMenu) -> None:
+    """Set up the AnkiHub Labs menu if labs are enabled."""
+    # Create Labs submenu
+    labs_menu = QMenu("🧪 AnkiHub Labs", parent_menu)
+
+    # Add Secrets submenu
+    secrets_action = QAction("🔑 Secrets", labs_menu)
+    qconnect(secrets_action.triggered, open_secrets_dialog)
+    labs_menu.addAction(secrets_action)
+
+    parent_menu.addMenu(labs_menu)

ankihub/labs/llm/__init__.py

@@ -0,0 +1 @@
+"""LLM-related functionality for AnkiHub."""

ankihub/labs/llm/llm.ps1

@@ -0,0 +1,68 @@
+# Ensure ~/.local/bin is in PATH
+$env:PATH = "$HOME\.local\bin;$env:PATH"
+
+# Define UV executable path
+$UV_PATH = "$HOME\.local\bin\uv"
+
+function Check-Uv {
+    & "$UV_PATH" version
+}
+
+function Install-Uv {
+    # Install uv using the official installer
+    Invoke-RestMethod https://astral.sh/uv/install.ps1 | Invoke-Expression
+
+    # Add uv to the current PATH
+    $env:PATH = "$HOME\.local\bin;$env:PATH"
+}
+
+function Check-Llm {
+    & llm --version
+}
+
+function Install-Llm {
+    # Install llm using uv
+    & "$UV_PATH" tool install llm
+}
+
+function Install-Provider {
+    param (
+        [string]$provider
+    )
+    Write-Output "Installing provider: '$provider'"
+    & "$UV_PATH" run --no-project llm install -U $provider
+}
+
+function Get-Templates-Path {
+    & "$UV_PATH" run --no-project llm templates path
+}
+
+function Get-Keys-Path {
+    & "$UV_PATH" run --no-project llm keys path
+}
+
+function Execute-Prompt {
+    param (
+        [string]$templateName,
+        [string]$noteContent
+    )
+    & "$UV_PATH" run --no-project llm -m gpt-4o --no-stream -t "$templateName" "$noteContent"
+}
+
+$cmd = $args[0]
+$args = $args[1..$args.Length]
+
+switch ($cmd) {
+    "check_uv" { Check-Uv }
+    "install_uv" { Install-Uv }
+    "check_llm" { Check-Llm }
+    "install_llm" { Install-Llm }
+    "install_provider" { Install-Provider -provider $args[0] }
+    "get_templates_path" { Get-Templates-Path }
+    "get_keys_path" { Get-Keys-Path }
+    "execute_prompt" { Execute-Prompt -templateName $args[0] -noteContent $args[1] }
+    default {
+        Write-Output "Unknown command: $cmd"
+        exit 1
+    }
+}