CLI 0.0.13, Configurable Template Demo (#12796)

libs/cli/langchain_cli/dev_scripts.py

@@ -2,6 +2,8 @@
 Development Scripts for template packages
 """
 
+from typing import Sequence
+
 from fastapi import FastAPI
 from langserve import add_routes
 from langserve.packages import get_langserve_export
@@ -9,7 +11,10 @@ from langserve.packages import get_langserve_export
 from langchain_cli.utils.packages import get_package_root
 
 
-def create_demo_server():
+def create_demo_server(
+    *,
+    config_keys: Sequence[str] = (),
+):
     """
     Creates a demo server for the current template.
     """
@@ -22,10 +27,18 @@ def create_demo_server():
         mod = __import__(package["module"], fromlist=[package["attr"]])
 
         chain = getattr(mod, package["attr"])
-        add_routes(app, chain)
+        add_routes(
+            app,
+            chain,
+            config_keys=config_keys,
+        )
     except KeyError as e:
         raise KeyError("Missing fields from pyproject.toml") from e
     except ImportError as e:
         raise ImportError("Could not import module defined in pyproject.toml") from e
 
     return app
+
+
+def create_demo_server_configurable():
+    return create_demo_server(config_keys=["configurable"])