Skip to content

GitLab

"src/vscode:/vscode.git/clone" did not exist on "78f79ef67062884ef85efeb055b11b5dfc529676"
Commit 9f9122b6 authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

enh: ChatValves

parent 8dcb3d78
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 2 deletions
backend/apps/webui/main.py
View file @ 9f9122b6
......@@ -165,6 +165,10 @@ async def get_pipe_models():
f"{function_module.name}{manifold_pipe_name}"
)
pipe_flag = {"type": pipe.type}
if hasattr(function_module, "ChatValves"):
pipe_flag["valves_spec"] = function_module.ChatValves.schema()
pipe_models.append(
{
"id": manifold_pipe_id,
......@@ -172,10 +176,14 @@ async def get_pipe_models():
"object": "model",
"created": pipe.created_at,
"owned_by": "openai",
"pipe": {"type": pipe.type},
"pipe": pipe_flag,
}
)
else:
pipe_flag = {"type": "pipe"}
if hasattr(function_module, "ChatValves"):
pipe_flag["valves_spec"] = function_module.ChatValves.schema()
pipe_models.append(
{
"id": pipe.id,
......@@ -183,7 +191,7 @@ async def get_pipe_models():
"object": "model",
"created": pipe.created_at,
"owned_by": "openai",
"pipe": {"type": "pipe"},
"pipe": pipe_flag,
}
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment