OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-05-13 09:00:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: custom id, name support

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-21 20:01:08 -07:00
parent a22133b191
commit 97528e2724
3 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@ -43,8 +43,8 @@ for loaded_module in load_modules_from_directory("./pipelines"):
PIPELINES[loaded_module.__name__] = { PIPELINES[loaded_module.__name__] = {
"module": pipeline, "module": pipeline,
"id": loaded_module.__name__, "id": pipeline.id if hasattr(pipeline, "id") else loaded_module.__name__,
"name": loaded_module.__name__, "name": pipeline.name if hasattr(pipeline, "name") else loaded_module.__name__,
} }

3
pipelines/examples/openai_pipeline.py
View File

@ -5,6 +5,9 @@ import requests
class Pipeline: class Pipeline:
def __init__(self): def __init__(self):
# Optionally, you can set the id and name of the pipeline.
self.id = "openai_pipeline"
self.name = "OpenAI Pipeline"
pass pass
async def on_startup(self): async def on_startup(self):

3
pipelines/examples/pipeline_example.py
View File

@ -4,6 +4,9 @@ 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 = "pipeline_example"
self.name = "Pipeline Example"
pass pass
async def on_startup(self): async def on_startup(self):