tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

feat: rag api endpoint

Browse source
This commit is contained in:
Timothy J. Baek 2024-01-06 22:07:20 -08:00
parent 7071716f54
commit b2c9f6dff8
3 changed files with 33 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
backend/main.py
View file

@ -5,16 +5,18 @@ from fastapi.middleware.wsgi import WSGIMiddleware
from fastapi.middleware.cors import CORSMiddleware
from starlette.exceptions import HTTPException as StarletteHTTPException
from apps.ollama.main import app as ollama_app
from apps.openai.main import app as openai_app
from apps.web.main import app as webui_app
from apps.rag.main import app as rag_app
import time
class SPAStaticFiles(StaticFiles):
async def get_response(self, path: str, scope):
try:
return await super().get_response(path, scope)
@ -49,9 +51,10 @@ async def check_url(request: Request, call_next):
app.mount("/api/v1", webui_app)
app.mount("/ollama/api", ollama_app)
app.mount("/openai/api", openai_app)
app.mount("/rag/api/v1", rag_app)
app.mount("/",
SPAStaticFiles(directory="../build", html=True),
name="spa-static-files")
app.mount("/", SPAStaticFiles(directory="../build", html=True), name="spa-static-files")