Skip to content

GitLab

"3rdParty/InstallRBuild.sh" did not exist on "b8fc3c491c990a648b238e41f944c114249e71cc"
Commit 9f9122b6 authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

enh: ChatValves

parent 8dcb3d78
Show 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(): ...@@ -165,6 +165,10 @@ async def get_pipe_models():
f"{function_module.name}{manifold_pipe_name}" 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( pipe_models.append(
{ {
"id": manifold_pipe_id, "id": manifold_pipe_id,
...@@ -172,10 +176,14 @@ async def get_pipe_models(): ...@@ -172,10 +176,14 @@ async def get_pipe_models():
"object": "model", "object": "model",
"created": pipe.created_at, "created": pipe.created_at,
"owned_by": "openai", "owned_by": "openai",
"pipe": {"type": pipe.type}, "pipe": pipe_flag,
} }
) )
else: else:
pipe_flag = {"type": "pipe"}
if hasattr(function_module, "ChatValves"):
pipe_flag["valves_spec"] = function_module.ChatValves.schema()
pipe_models.append( pipe_models.append(
{ {
"id": pipe.id, "id": pipe.id,
...@@ -183,7 +191,7 @@ async def get_pipe_models(): ...@@ -183,7 +191,7 @@ async def get_pipe_models():
"object": "model", "object": "model",
"created": pipe.created_at, "created": pipe.created_at,
"owned_by": "openai", "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