Demonstrate generate function

jncraton · Jan 4, 2024 · 1a065a3 · 1a065a3
1 parent 3eecebb
commit 1a065a3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/languagemodels/inference.py b/languagemodels/inference.py
@@ -124,6 +124,9 @@ def generate(
 
     This may use a local model, or it may make an API call to an external
     model if API keys are available.
+
+    >>> generate(["What is the capital of France?"])
+    ['...Paris...']
     """
     if os.environ.get("LANGUAGEMODELS_TS_KEY") or os.environ.get(
         "LANGUAGEMODELS_TS_SERVER"