feat: expose partial_lookup

Signed-off-by: Michele Dolfi <[email protected]>
DS4SD · Nov 23, 2023 · a418380 · a418380
1 parent 0073b37
commit a418380
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/deepsearch/chemistry/queries/molecules.py b/deepsearch/chemistry/queries/molecules.py
@@ -94,6 +94,7 @@ def MoleculeQuery(
 def MoleculesInPatentsQuery(
     patents: Union[str, List[str]],
     num_items: int = 10,
+    partial_lookup: bool = False,
 ) -> Query:
     """
     List all molecules contained in a list of patents.
@@ -117,6 +118,7 @@ def MoleculesInPatentsQuery(
                     for v in patents
                 ]
             },
+            "partial_references": partial_lookup,
             "limit": num_items,
         },
         coordinates=KnowledgeDbResource(),