background as variable (sotopia-lab#137)

* background as variable * update docs
petalzx · Jul 16, 2024 · 1bdd155 · 1bdd155
1 parent 1939bc6
commit 1bdd155
Show file tree

Hide file tree

Showing 4 changed files with 441 additions and 352 deletions.
diff --git a/docs/pages/generation.md b/docs/pages/generation.md
@@ -0,0 +1,16 @@
+## Generation functions
+
+The core of generating agent action and environment observation lies in the `agenerate` function:
+
+```python
+@gin.configurable
+@beartype
+async def agenerate(
+    model_name: str,
+    template: str,
+    input_values: dict[str, str],
+    output_parser: BaseOutputParser[OutputType],
+    temperature: float = 0.7,
+) -> OutputType:
+    input_variables = re.findall(r"(?<!{){([^{}]+)}(?!})", template)
+```