Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(sqlalchemy): remove session reference from router

Browse Source
This commit is contained in:
Jonathan Rohde
2024-06-27 07:48:08 +02:00
parent df09d0830a
commit bee835cb65
34 changed files with 1227 additions and 1207 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/ollama/main.py
+2 -5
View File
@@ -31,7 +31,6 @@ from typing import Optional, List, Union
from starlette.background import BackgroundTask
from apps.webui.internal.db import get_db
from apps.webui.models.models import Models
from apps.webui.models.users import Users
from constants import ERROR_MESSAGES
@@ -712,7 +711,6 @@ async def generate_chat_completion(
form_data: GenerateChatCompletionForm,
url_idx: Optional[int] = None,
user=Depends(get_verified_user),
db=Depends(get_db),
):
log.debug(
@@ -726,7 +724,7 @@ async def generate_chat_completion(
}
model_id = form_data.model
model_info = Models.get_model_by_id(db, model_id)
model_info = Models.get_model_by_id(model_id)
if model_info:
if model_info.base_model_id:
@@ -885,7 +883,6 @@ async def generate_openai_chat_completion(
form_data: dict,
url_idx: Optional[int] = None,
user=Depends(get_verified_user),
db=Depends(get_db),
):
form_data = OpenAIChatCompletionForm(**form_data)
@@ -894,7 +891,7 @@ async def generate_openai_chat_completion(
}
model_id = form_data.model
model_info = Models.get_model_by_id(db, model_id)
model_info = Models.get_model_by_id(model_id)
if model_info:
if model_info.base_model_id: