ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-15 05:45:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-09-11 21:56:59 +04:00
parent edc056351d
commit 6c0a5fa91c
4 changed files with 20 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
backend/open_webui/functions.py
View file

@ -221,10 +221,11 @@ async def generate_function_chat_completion(
oauth_token = None oauth_token = None
try: try:
oauth_token = request.app.state.oauth_manager.get_oauth_token( if request.cookies.get("oauth_session_id", None):
user.id, oauth_token = request.app.state.oauth_manager.get_oauth_token(
request.cookies.get("oauth_session_id", None), user.id,
) request.cookies.get("oauth_session_id", None),
)
except Exception as e: except Exception as e:
log.error(f"Error getting OAuth token: {e}") log.error(f"Error getting OAuth token: {e}")

8
backend/open_webui/main.py
View file

@ -1418,14 +1418,6 @@ async def chat_completion(
model_item = form_data.pop("model_item", {}) model_item = form_data.pop("model_item", {})
tasks = form_data.pop("background_tasks", None) tasks = form_data.pop("background_tasks", None)
oauth_token = None
try:
oauth_token = request.app.state.oauth_manager.get_oauth_token(
user.id, request.cookies.get("oauth_session_id", None)
)
except Exception as e:
log.error(f"Error getting OAuth token: {e}")
metadata = {} metadata = {}
try: try:
if not model_item.get("direct", False): if not model_item.get("direct", False):

9
backend/open_webui/routers/openai.py
View file

@ -171,10 +171,11 @@ def get_headers_and_cookies(
oauth_token = None oauth_token = None
try: try:
oauth_token = request.app.state.oauth_manager.get_oauth_token( if request.cookies.get("oauth_session_id", None):
user.id, oauth_token = request.app.state.oauth_manager.get_oauth_token(
request.cookies.get("oauth_session_id", None), user.id,
) request.cookies.get("oauth_session_id", None),
)
except Exception as e: except Exception as e:
log.error(f"Error getting OAuth token: {e}") log.error(f"Error getting OAuth token: {e}")

18
backend/open_webui/utils/middleware.py
View file

@ -817,10 +817,11 @@ async def process_chat_payload(request, form_data, user, metadata, model):
oauth_token = None oauth_token = None
try: try:
oauth_token = request.app.state.oauth_manager.get_oauth_token( if request.cookies.get("oauth_session_id", None):
user.id, oauth_token = request.app.state.oauth_manager.get_oauth_token(
request.cookies.get("oauth_session_id", None), user.id,
) request.cookies.get("oauth_session_id", None),
)
except Exception as e: except Exception as e:
log.error(f"Error getting OAuth token: {e}") log.error(f"Error getting OAuth token: {e}")
@ -1496,10 +1497,11 @@ async def process_chat_response(
oauth_token = None oauth_token = None
try: try:
oauth_token = request.app.state.oauth_manager.get_oauth_token( if request.cookies.get("oauth_session_id", None):
user.id, oauth_token = request.app.state.oauth_manager.get_oauth_token(
request.cookies.get("oauth_session_id", None), user.id,
) request.cookies.get("oauth_session_id", None),
)
except Exception as e: except Exception as e:
log.error(f"Error getting OAuth token: {e}") log.error(f"Error getting OAuth token: {e}")