Handle NAGLCharges

devtools/conda-envs/dev_env.yaml

@@ -7,19 +7,18 @@ dependencies:
   - python
   - pip
   - versioningit
-  - pip
   - numpy <2.3
   - pydantic =2
   # OpenFF stack
-  - openff-toolkit-base ~=0.16.6
+  - openff-toolkit-base =0.17
   - openff-units
   - ambertools =23
   # Optional features
   - openmm =8.2
   # smirnoff-plugins =2024
   # de-forcefields  # add back after smirnoff-plugins update
-  - openff-nagl ~=0.5
-  - openff-nagl-models ~=0.3
+  - openff-nagl
+  - openff-nagl-models
   - mbuild ~=0.18
   - foyer =1
   - gmso ~=0.12

devtools/conda-envs/docs_env.yaml

@@ -10,7 +10,7 @@ dependencies:
   - pip
   - numpy =1
   - pydantic =2
-  - openff-toolkit-base ~=0.16.6
+  - openff-toolkit-base =0.17
   - openmm =8.2
   - mbuild
   - foyer =1

devtools/conda-envs/examples_env.yaml

@@ -9,12 +9,12 @@ dependencies:
   - numpy <2.3
   - pydantic =2
   # OpenFF stack
-  - openff-toolkit-base ~=0.16.6
+  - openff-toolkit-base =0.17
   - openff-units
   - ambertools =23
   # Optional features
-  - openff-nagl ~=0.5
-  - openff-nagl-models ~=0.3
+  - openff-nagl
+  - openff-nagl-models
   - mbuild
   - foyer
   - nglview

devtools/conda-envs/test_env.yaml

@@ -8,7 +8,7 @@ dependencies:
   - numpy <2.3
   - pydantic =2
   # OpenFF stack
-  - openff-toolkit-base ~=0.16.8
+  - openff-toolkit-base =0.17
   - openff-units =0.3
   - ambertools =24
   # Needs to be explicitly listed to not be dropped when AmberTools is removed

devtools/conda-envs/test_not_py313.yaml

@@ -8,7 +8,7 @@ dependencies:
   - numpy <2.3
   - pydantic =2
   # OpenFF stack
-  - openff-toolkit-base ~=0.16.8
+  - openff-toolkit-base =0.17
   - openff-units =0.3
   - ambertools
   # Needs to be explicitly listed to not be dropped when AmberTools is removed

docs/releasehistory.md

@@ -11,6 +11,12 @@ Dates are given in YYYY-MM-DD format.
 
 Please note that all releases prior to a version 1.0.0 are considered pre-releases and many API changes will come before a stable release.
 
+## 0.4.5 - 2025-08-19
+
+### New features
+
+* #1206 Support `<NAGLCharges>` tags in SMIRNOFF force fields.
+
 ## 0.4.4 - 2025-07-29
 
 ### Behavior changes

examples/host-guest/host_guest.ipynb

@@ -89,7 +89,7 @@
    "source": [
     "NAGLToolkitWrapper().assign_partial_charges(\n",
     "    molecule=host,\n",
-    "    partial_charge_method=\"openff-gnn-am1bcc-0.1.0-rc.1.pt\",\n",
+    "    partial_charge_method=\"openff-gnn-am1bcc-0.1.0-rc.3.pt\",\n",
     ")\n",
     "\n",
     "host.partial_charges.round(3)"
@@ -110,7 +110,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "sage = ForceField(\"openff-2.1.0.offxml\")\n",
+    "sage = ForceField(\"openff-2.2.1.offxml\")\n",
     "\n",
     "out = sage.create_interchange(topology=docked_topology, charge_from_molecules=[host])"
    ]
@@ -192,7 +192,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.7"
+   "version": "3.11.13"
   }
  },
  "nbformat": 4,

openff/interchange/_tests/conftest.py

@@ -53,7 +53,7 @@ def sage_with_bond_charge(sage):
             type="BondCharge",
             match="all_permutations",
             distance="0.8 * angstrom ** 1",
-            charge_increment1="0.0 * elementary_charge ** 1",
+            charge_increment1="0.123 * elementary_charge ** 1",
             charge_increment2="0.0 * elementary_charge ** 1",
         ),
     )
@@ -187,6 +187,19 @@ def sage_with_off_center_hydrogen(sage):
     return sage
 
 
+@pytest.fixture
+def sage_nagl(sage):
+    sage.get_parameter_handler(
+        "NAGLCharges",
+        {
+            "model_file": "openff-gnn-am1bcc-0.1.0-rc.3.pt",
+            "version": "0.3",
+        },
+    )
+    sage.deregister_parameter_handler("ToolkitAM1BCC")
+    return sage
+
+
 @pytest.fixture
 def _simple_force_field():
     # TODO: Create a minimal force field for faster tests
@@ -590,7 +603,6 @@ def hydrogen_cyanide_reversed():
 def hexane_diol():
     molecule = Molecule.from_smiles("OCCCCCCO")
     molecule.assign_partial_charges(partial_charge_method="gasteiger")
-    molecule.partial_charges.m
     return molecule
 
 

openff/interchange/_tests/unit_tests/smirnoff/test_charge_assignment_logging.py

@@ -12,10 +12,12 @@
 
 """
 Hierarchy is
-1. Match molceules with preset charges
+1. Match molecules with preset charges
 2. Match chemical environment against library charges
-3. Match chemical environment against charge increments
-4. Run AM1-BCC (or a variant) on remaining molecules
+3. Run NAGLCharges (if present in the FF)
+3. Run charge method in ChargeIncrementModel section (if present in the FF) and then
+   match chemical environment against charge increments
+4. Run AM1-BCC (or a variant) on remaining molecules (if present in the FF)
 
 Test cases
 ----------
@@ -86,6 +88,19 @@
 * Ions get library charges
 * Ligand gets charge increments
 
+11. Sage with NAGL and ligand in vacuum
+* Ligand gets NAGL charges
+
+12. Sage with NAGL and water molecules (library precedence)
+* Water gets library charges (precedence over NAGL)
+
+13. Sage with NAGL mixed topology (ligand and water)
+* Ligand gets NAGL charges
+* Water gets library charges
+
+14. Sage with NAGL and preset charges on ligand
+* Ligand gets preset charges (precedence over NAGL)
+
 Other details
 * Specifics of charge method (AM1-BCC, AM1-BCC-ELF10, AM1-BCC via NAGL)
 * Molecules with preset charges can be similar but not exact enough
@@ -118,6 +133,9 @@ def map_methods_to_atom_indices(caplog: pytest.LogCaptureFixture) -> dict[str, l
         elif message.startswith("Charge section ToolkitAM1BCC"):
             info[message.split(", ")[1].split(" ")[-1]].append(int(message.split("atom index ")[-1]))
 
+        elif message.startswith("Charge section NAGLCharges"):
+            info["NAGLChargesHandler"].append(int(message.split("atom index ")[-1]))
+
         # without also pulling the virtual site - particle mapping (which is different for each engine)
         # it's hard to store more information than the orientation atoms that are affected by each
         # virtual site's charges
@@ -149,7 +167,7 @@ def map_methods_to_atom_indices(caplog: pytest.LogCaptureFixture) -> dict[str, l
 
 
 @pytest.fixture
-def sage_with_nagl(sage):
+def sage_with_nagl_chargeincrements(sage):
     from openff.toolkit.typing.engines.smirnoff.parameters import ChargeIncrementModelHandler, ChargeIncrementType
 
     sage.register_parameter_handler(
@@ -241,6 +259,10 @@ def ligand_and_water_and_ions(ligand, water_and_ions) -> Topology:
 8.xSage with preset charges on water
 9.xSage with (ligand) virtual site parameters
 10.xAM1-with-custom-BCCs Sage with ligand and ions water
+11.xSage with NAGL and ligand in vacuum
+12.xSage with NAGL and water molecules (library precedence)
+13.xSage with NAGL mixed topology (ligand and water)
+14.xSage with NAGL and preset charges on ligand
 """
 
 
@@ -416,7 +438,7 @@ def test_case9(caplog, sage_with_bond_charge):
         assert info["orientation"] == [0, 1]
 
 
-def test_case10(caplog, sage_with_nagl, ligand):
+def test_case10(caplog, sage_with_nagl_chargeincrements, ligand):
     from openff.toolkit.utils.nagl_wrapper import NAGLToolkitWrapper
     from openff.toolkit.utils.rdkit_wrapper import RDKitToolkitWrapper
     from openff.toolkit.utils.toolkit_registry import ToolkitRegistry, toolkit_registry_manager
@@ -430,7 +452,7 @@ def test_case10(caplog, sage_with_nagl, ligand):
                 toolkit_precedence=[NAGLToolkitWrapper, RDKitToolkitWrapper],
             ),
         ):
-            sage_with_nagl.create_interchange(ligand.to_topology())
+            sage_with_nagl_chargeincrements.create_interchange(ligand.to_topology())
 
         info = map_methods_to_atom_indices(caplog)
 
@@ -445,3 +467,70 @@ def test_case10(caplog, sage_with_nagl, ligand):
 
         # the standard AM1-BCC should not have ran
         assert AM1BCC_KEY not in info
+
+
+def test_case11(caplog, sage_nagl, ligand):
+    """Test that NAGL charge assignment is properly logged."""
+    pytest.importorskip("openff.nagl")
+
+    with caplog.at_level(logging.INFO):
+        sage_nagl.create_interchange(ligand.to_topology())
+
+        info = map_methods_to_atom_indices(caplog)
+
+        # Should log NAGL charges for all atoms
+        assert "NAGLChargesHandler" in info
+        assert info["NAGLChargesHandler"] == [*range(0, ligand.n_atoms)]
+
+
+def test_case12(caplog, sage_nagl, water):
+    """Test logging when LibraryCharges takes precedence over NAGL."""
+    pytest.importorskip("openff.nagl")
+
+    topology = Topology.from_molecules([water, water])
+
+    with caplog.at_level(logging.INFO):
+        sage_nagl.create_interchange(topology)
+
+        info = map_methods_to_atom_indices(caplog)
+
+        # Water should get library charges, not NAGL
+        assert info["library"] == [*range(0, 6)]  # 2 water molecules, 3 atoms each
+        assert "NAGLChargesHandler" not in info
+
+
+def test_case13(caplog, sage_nagl, ligand, water):
+    """Test logging with mixed molecule types (some library, some NAGL)."""
+    pytest.importorskip("openff.nagl")
+
+    topology = Topology.from_molecules([ligand, water])
+
+    with caplog.at_level(logging.INFO):
+        sage_nagl.create_interchange(topology)
+
+        info = map_methods_to_atom_indices(caplog)
+
+        # Ligand should get NAGL charges
+        assert info["NAGLChargesHandler"] == [*range(0, ligand.n_atoms)]
+
+        # Water should get library charges
+        assert info["library"] == [*range(ligand.n_atoms, ligand.n_atoms + water.n_atoms)]
+
+
+def test_case14(caplog, sage_nagl, ligand):
+    """Test logging when preset charges are used instead of NAGL."""
+    pytest.importorskip("openff.nagl")
+
+    ligand.assign_partial_charges("gasteiger")
+
+    with caplog.at_level(logging.INFO):
+        sage_nagl.create_interchange(
+            ligand.to_topology(),
+            charge_from_molecules=[ligand],
+        )
+
+        info = map_methods_to_atom_indices(caplog)
+
+        # Should use preset charges, not NAGL
+        assert info["preset"] == [*range(0, ligand.n_atoms)]
+        assert "NAGLChargesHandler" not in info