|
|
|
@ -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"])
|
|
|
|
|