Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
open-webui
Commits
b2c9f6df
Commit
b2c9f6df
authored
Jan 06, 2024
by
Timothy J. Baek
Browse files
feat: rag api endpoint
parent
7071716f
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
33 additions
and
4 deletions
+33
-4
backend/apps/rag/main.py
backend/apps/rag/main.py
+23
-0
backend/main.py
backend/main.py
+7
-4
backend/requirements.txt
backend/requirements.txt
+3
-0
No files found.
backend/apps/rag/main.py
0 → 100644
View file @
b2c9f6df
from
fastapi
import
FastAPI
,
Request
,
Depends
,
HTTPException
from
fastapi.middleware.cors
import
CORSMiddleware
from
apps.web.routers
import
auths
,
users
,
chats
,
modelfiles
,
utils
from
config
import
WEBUI_VERSION
,
WEBUI_AUTH
app
=
FastAPI
()
origins
=
[
"*"
]
app
.
add_middleware
(
CORSMiddleware
,
allow_origins
=
origins
,
allow_credentials
=
True
,
allow_methods
=
[
"*"
],
allow_headers
=
[
"*"
],
)
@
app
.
get
(
"/"
)
async
def
get_status
():
return
{
"status"
:
True
}
backend/main.py
View file @
b2c9f6df
...
...
@@ -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"
)
backend/requirements.txt
View file @
b2c9f6df
...
...
@@ -16,7 +16,10 @@ aiohttp
peewee
bcrypt
langchain
chromadb
sentence_transformers
PyJWT
pyjwt[crypto]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment