OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-06-25 17:45:56 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac: /v1 prefix support

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-28 21:07:01 -07:00
parent e7da7abb92
commit dc744bf0e0
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.py
View File

@ -175,8 +175,8 @@ async def check_url(request: Request, call_next):
return response return response
@app.get("/models")
@app.get("/v1/models") @app.get("/v1/models")
@app.get("/models")
async def get_models(): async def get_models():
""" """
Returns the available pipelines Returns the available pipelines
@ -209,6 +209,7 @@ async def get_models():
} }
@app.get("/v1/{pipeline_id}/valves")
@app.get("/{pipeline_id}/valves") @app.get("/{pipeline_id}/valves")
async def get_valves(pipeline_id: str): async def get_valves(pipeline_id: str):
if pipeline_id not in app.state.PIPELINES: if pipeline_id not in app.state.PIPELINES:
@ -231,6 +232,7 @@ async def get_valves(pipeline_id: str):
return pipeline_module.valves return pipeline_module.valves
@app.get("/v1/{pipeline_id}/valves/spec")
@app.get("/{pipeline_id}/valves/spec") @app.get("/{pipeline_id}/valves/spec")
async def get_valves_spec(pipeline_id: str): async def get_valves_spec(pipeline_id: str):
@ -255,6 +257,7 @@ async def get_valves_spec(pipeline_id: str):
return pipeline_module.valves.schema() return pipeline_module.valves.schema()
@app.post("/v1/{pipeline_id}/valves/update")
@app.post("/{pipeline_id}/valves/update") @app.post("/{pipeline_id}/valves/update")
async def update_valves(pipeline_id: str, form_data: dict): async def update_valves(pipeline_id: str, form_data: dict):
@ -293,6 +296,7 @@ async def update_valves(pipeline_id: str, form_data: dict):
return pipeline_module.valves return pipeline_module.valves
@app.post("/v1/{pipeline_id}/filter")
@app.post("/{pipeline_id}/filter") @app.post("/{pipeline_id}/filter")
async def filter(pipeline_id: str, form_data: FilterForm): async def filter(pipeline_id: str, form_data: FilterForm):
if ( if (
@ -317,8 +321,8 @@ async def filter(pipeline_id: str, form_data: FilterForm):
) )
@app.post("/chat/completions")
@app.post("/v1/chat/completions") @app.post("/v1/chat/completions")
@app.post("/chat/completions")
async def generate_openai_chat_completion(form_data: OpenAIChatCompletionForm): async def generate_openai_chat_completion(form_data: OpenAIChatCompletionForm):
user_message = get_last_user_message(form_data.messages) user_message = get_last_user_message(form_data.messages)
messages = [message.model_dump() for message in form_data.messages] messages = [message.model_dump() for message in form_data.messages]