add email and uri to bindings

dottxt-ai · Dec 20, 2024 · 78eb908 · 78eb908
1 parent 346798c
commit 78eb908
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/python/outlines_core/fsm/json_schema.py b/python/outlines_core/fsm/json_schema.py
@@ -2,12 +2,14 @@
     BOOLEAN,
     DATE,
     DATE_TIME,
+    EMAIL,
     INTEGER,
     NULL,
     NUMBER,
     STRING,
     STRING_INNER,
     TIME,
+    URI,
     UUID,
     WHITESPACE,
     build_regex_from_schema,

diff --git a/python/outlines_core/fsm/outlines_core_rs.pyi b/python/outlines_core/fsm/outlines_core_rs.pyi
@@ -64,6 +64,8 @@ STRING_INNER: str
 TIME: str
 UUID: str
 WHITESPACE: str
+EMAIL: str
+URI: str
 
 class Vocabulary:
     """

diff --git a/src/python_bindings/mod.rs b/src/python_bindings/mod.rs
@@ -319,6 +319,8 @@ fn outlines_core_rs(m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add("TIME", json_schema::TIME)?;
     m.add("UUID", json_schema::UUID)?;
     m.add("WHITESPACE", json_schema::WHITESPACE)?;
+    m.add("EMAIL", json_schema::EMAIL)?;
+    m.add("URI", json_schema::URI)?;
 
     m.add_function(wrap_pyfunction!(build_regex_from_schema_py, m)?)?;
     m.add_function(wrap_pyfunction!(to_regex_py, m)?)?;