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
25987fe3
Commit
25987fe3
authored
Dec 13, 2023
by
Timothy J. Baek
Browse files
feat: better error handling for ollama reverse proxy
parent
346b0df8
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
26 additions
and
21 deletions
+26
-21
backend/apps/ollama/main.py
backend/apps/ollama/main.py
+26
-21
No files found.
backend/apps/ollama/main.py
View file @
25987fe3
...
...
@@ -59,6 +59,7 @@ def proxy(path):
else
:
pass
try
:
# Make a request to the target server
target_response
=
requests
.
request
(
method
=
request
.
method
,
...
...
@@ -68,6 +69,8 @@ def proxy(path):
stream
=
True
,
# Enable streaming for server-sent events
)
target_response
.
raise_for_status
()
# Proxy the target server's response to the client
def
generate
():
for
chunk
in
target_response
.
iter_content
(
chunk_size
=
8192
):
...
...
@@ -80,6 +83,8 @@ def proxy(path):
response
.
headers
[
key
]
=
value
return
response
except
Exception
as
e
:
return
jsonify
({
"detail"
:
"Server Connection Error"
,
"message"
:
str
(
e
)}),
400
if
__name__
==
"__main__"
:
...
...
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