loreguard-cli 0.14.1__tar.gz → 0.14.3__tar.gz

{loreguard_cli-0.14.1 → loreguard_cli-0.14.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: loreguard-cli
-Version: 0.14.1
+Version: 0.14.3
 Summary: Local inference client for Loreguard NPCs
 Project-URL: Homepage, https://loreguard.com
 Project-URL: Documentation, https://github.com/beyond-logic-labs/loreguard-cli#readme

{loreguard_cli-0.14.1 → loreguard_cli-0.14.3}/loreguard_entry.py

@@ -3,6 +3,9 @@
 
 Imports src as a proper package so relative imports inside src/__main__.py work.
 """
+import multiprocessing
+multiprocessing.freeze_support()  # Required for PyInstaller on macOS/Windows
+
 from src.__main__ import main
 
 main()

{loreguard_cli-0.14.1 → loreguard_cli-0.14.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "loreguard-cli"
-version = "0.14.1"
+version = "0.14.3"
 description = "Local inference client for Loreguard NPCs"
 readme = "README.md"
 license = "MIT"

{loreguard_cli-0.14.1 → loreguard_cli-0.14.3}/src/nli.py

@@ -123,6 +123,10 @@ class NLIService:
                     self._model_path,
                     trust_remote_code=True,
                 )
+                # HHEMv2 custom class may lack all_tied_weights_keys (needed by
+                # newer transformers for .to() / .eval()). Patch if missing.
+                if not hasattr(self._model, "_tied_weights_keys"):
+                    self._model._tied_weights_keys = []
                 self._model.to(self._device)
                 self._model.eval()
 

{loreguard_cli-0.14.1 → loreguard_cli-0.14.3}/src/tui/screens/main.py

@@ -546,11 +546,17 @@ class MainScreen(Screen):
                 intent_classifier = None
 
             # Load dialogue act classifier (filler selection) - run in thread pool
-            self._update_status("Loading dialogue act model...", log=False)
-            self._log("Loading dialogue act classifier...")
             dialogue_act_classifier = None
+            enable_dialogue_act = os.getenv("LOREGUARD_DIALOGUE_ACT_ENABLED", "true").lower() == "true"
+            if not enable_dialogue_act:
+                self._log("Dialogue act classifier disabled via LOREGUARD_DIALOGUE_ACT_ENABLED")
+            else:
+                self._update_status("Loading dialogue act model...", log=False)
+                self._log("Loading dialogue act classifier...")
             try:
-                if is_dialogue_act_model_available():
+                if not enable_dialogue_act:
+                    pass  # Skip loading
+                elif is_dialogue_act_model_available():
                     dialogue_act_classifier = DialogueActClassifier()
                     loop = asyncio.get_event_loop()
                     with concurrent.futures.ThreadPoolExecutor() as pool:

{loreguard_cli-0.14.1 → loreguard_cli-0.14.3}/src/tui/screens/running.py

@@ -2,6 +2,7 @@
 
 import asyncio
 import concurrent.futures
+import os
 from typing import TYPE_CHECKING, Optional
 
 from textual.app import ComposeResult
@@ -247,10 +248,17 @@ class RunningScreen(Screen):
 
                 # Load Dialogue Act Classifier
                 dialogue_act_classifier = None
-                self._update_status("dialogue_act", "Dialogue Act", "Loading...", "info")
-                self._log("Loading Dialogue Act classifier...", "info")
+                enable_dialogue_act = os.getenv("LOREGUARD_DIALOGUE_ACT_ENABLED", "true").lower() == "true"
+                if not enable_dialogue_act:
+                    self._update_status("dialogue_act", "Dialogue Act", "Disabled", "info")
+                    self._log("Dialogue act classifier disabled via LOREGUARD_DIALOGUE_ACT_ENABLED", "info")
+                else:
+                    self._update_status("dialogue_act", "Dialogue Act", "Loading...", "info")
+                    self._log("Loading Dialogue Act classifier...", "info")
                 try:
-                    if is_dialogue_act_model_available():
+                    if not enable_dialogue_act:
+                        pass  # Skip loading
+                    elif is_dialogue_act_model_available():
                         dialogue_act_classifier = DialogueActClassifier()
                         loop = asyncio.get_event_loop()
                         with concurrent.futures.ThreadPoolExecutor() as pool: