Merge pull request #29 from JC1DA/overhaul_loc

Merge main repo
guidance-ai · Dec 15, 2024 · 3bf09b5 · 3bf09b5
2 parents 69f723c + c6b4b3c
commit 3bf09b5
Show file tree

Hide file tree

Showing 11 changed files with 246 additions and 2,028 deletions.
diff --git a/guidance/_grammar.py b/guidance/_grammar.py
diff --git a/guidance/library/_gen.py b/guidance/library/_gen.py
@@ -16,7 +16,7 @@ def gen(
     lm,
     name=None,
     *,
-    max_tokens=1e10,
+    max_tokens=None,
     list_append=False,
     regex=None,
     tools=None,
@@ -174,7 +174,7 @@ def gen(
                     break
         return lm
 
-    pattern = Gen(body_regex=regex, stop_regex=gen_stop, save_stop_text=save_stop_text, name=tagged_name, max_tokens=max_tokens)
+    pattern = Gen(body_regex=regex, stop_regex=gen_stop, save_stop_text=save_stop_text, capture_name=tagged_name, max_tokens=max_tokens)
 
     # define any capture group for non-tool calls
     if name is not None and tools is None: