From ad5846eb6a5dddcf94aba4bd9cd19575e7bee779 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Sun, 26 May 2024 16:05:03 -0700
Subject: [PATCH] Update manifold_pipeline.py
---
pipelines/examples/manifold_pipeline.py | 16 +++++++++-------
1 file changed, 9 insertions(+), 7 deletions(-)
diff --git a/pipelines/examples/manifold_pipeline.py b/pipelines/examples/manifold_pipeline.py
index c0eda65..b5d141c 100644
--- a/pipelines/examples/manifold_pipeline.py
+++ b/pipelines/examples/manifold_pipeline.py
@@ -4,18 +4,20 @@ from schemas import OpenAIChatMessage
class Pipeline:
def __init__(self):
- # Optionally, you can set the id and name of the pipeline.
- self.id = "pipeline_example"
- self.name = "Pipeline Example"
+ self.id = "manifold_pipeline"
+ self.name = "Manifold Pipeline"
# You can also set the pipelines that are available in this pipeline.
+ # Set manifold to True if you want to use this pipeline as a manifold.
+ # Manifold pipelines can have multiple pipelines.
+ self.manifold = True
self.pipelines = [
{
- "id": "pipeline-1",
- "name": "Pipeline 1",
+ "id": "pipeline-1", # This will turn into `manifold_pipeline.pipeline-1`
+ "name": "Manifold: Pipeline 1",
},
{
"id": "pipeline-2",
- "name": "Pipeline 2",
+ "name": "Manifold: Pipeline 2",
},
]
pass
@@ -40,4 +42,4 @@ class Pipeline:
print(user_message)
print(body)
- return f"{__name__} response to: {user_message}"
+ return f"{model_id} response to: {user_message}"