Update manifold_pipeline.py

This commit is contained in:
Timothy J. Baek 2024-05-26 16:05:03 -07:00
parent eff0a968c7
commit ad5846eb6a

View File

@ -4,18 +4,20 @@ from schemas import OpenAIChatMessage
class Pipeline: class Pipeline:
def __init__(self): def __init__(self):
# Optionally, you can set the id and name of the pipeline. self.id = "manifold_pipeline"
self.id = "pipeline_example" self.name = "Manifold Pipeline"
self.name = "Pipeline Example"
# You can also set the pipelines that are available in this 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 = [ self.pipelines = [
{ {
"id": "pipeline-1", "id": "pipeline-1", # This will turn into `manifold_pipeline.pipeline-1`
"name": "Pipeline 1", "name": "Manifold: Pipeline 1",
}, },
{ {
"id": "pipeline-2", "id": "pipeline-2",
"name": "Pipeline 2", "name": "Manifold: Pipeline 2",
}, },
] ]
pass pass
@ -40,4 +42,4 @@ class Pipeline:
print(user_message) print(user_message)
print(body) print(body)
return f"{__name__} response to: {user_message}" return f"{model_id} response to: {user_message}"