|
@@ -13,31 +13,42 @@ from typing import Optional
|
|
|
|
|
|
import aiohttp
|
|
|
import requests
|
|
|
-from apps.audio.main import app as audio_app
|
|
|
-from apps.images.main import app as images_app
|
|
|
-from apps.ollama.main import app as ollama_app
|
|
|
-from apps.ollama.main import (
|
|
|
+
|
|
|
+
|
|
|
+from open_webui.apps.audio.main import app as audio_app
|
|
|
+from open_webui.apps.images.main import app as images_app
|
|
|
+from open_webui.apps.ollama.main import app as ollama_app
|
|
|
+from open_webui.apps.ollama.main import (
|
|
|
generate_openai_chat_completion as generate_ollama_chat_completion,
|
|
|
)
|
|
|
-from apps.ollama.main import get_all_models as get_ollama_models
|
|
|
-from apps.openai.main import app as openai_app
|
|
|
-from apps.openai.main import generate_chat_completion as generate_openai_chat_completion
|
|
|
-from apps.openai.main import get_all_models as get_openai_models
|
|
|
-from apps.rag.main import app as rag_app
|
|
|
-from apps.rag.utils import get_rag_context, rag_template
|
|
|
-from apps.socket.main import app as socket_app
|
|
|
-from apps.socket.main import get_event_call, get_event_emitter
|
|
|
-from apps.webui.internal.db import Session
|
|
|
-from apps.webui.main import app as webui_app
|
|
|
-from apps.webui.main import generate_function_chat_completion, get_pipe_models
|
|
|
-from apps.webui.models.auths import Auths
|
|
|
-from apps.webui.models.functions import Functions
|
|
|
-from apps.webui.models.models import Models
|
|
|
-from apps.webui.models.users import UserModel, Users
|
|
|
-from apps.webui.utils import load_function_module_by_id
|
|
|
+from open_webui.apps.ollama.main import get_all_models as get_ollama_models
|
|
|
+from open_webui.apps.openai.main import app as openai_app
|
|
|
+from open_webui.apps.openai.main import (
|
|
|
+ generate_chat_completion as generate_openai_chat_completion,
|
|
|
+)
|
|
|
+from open_webui.apps.openai.main import get_all_models as get_openai_models
|
|
|
+from open_webui.apps.rag.main import app as rag_app
|
|
|
+from open_webui.apps.rag.utils import get_rag_context, rag_template
|
|
|
+from open_webui.apps.socket.main import app as socket_app
|
|
|
+from open_webui.apps.socket.main import get_event_call, get_event_emitter
|
|
|
+from open_webui.apps.webui.internal.db import Session
|
|
|
+from open_webui.apps.webui.main import app as webui_app
|
|
|
+from open_webui.apps.webui.main import (
|
|
|
+ generate_function_chat_completion,
|
|
|
+ get_pipe_models,
|
|
|
+)
|
|
|
+from open_webui.apps.webui.models.auths import Auths
|
|
|
+from open_webui.apps.webui.models.functions import Functions
|
|
|
+from open_webui.apps.webui.models.models import Models
|
|
|
+from open_webui.apps.webui.models.users import UserModel, Users
|
|
|
+from open_webui.apps.webui.utils import load_function_module_by_id
|
|
|
+
|
|
|
+
|
|
|
from authlib.integrations.starlette_client import OAuth
|
|
|
from authlib.oidc.core import UserInfo
|
|
|
-from config import (
|
|
|
+
|
|
|
+
|
|
|
+from open_webui.config import (
|
|
|
CACHE_DIR,
|
|
|
CORS_ALLOW_ORIGIN,
|
|
|
DEFAULT_LOCALE,
|
|
@@ -65,8 +76,8 @@ from config import (
|
|
|
AppConfig,
|
|
|
run_migrations,
|
|
|
)
|
|
|
-from constants import ERROR_MESSAGES, TASKS, WEBHOOK_MESSAGES
|
|
|
-from env import (
|
|
|
+from open_webui.constants import ERROR_MESSAGES, TASKS, WEBHOOK_MESSAGES
|
|
|
+from open_webui.env import (
|
|
|
CHANGELOG,
|
|
|
GLOBAL_LOG_LEVEL,
|
|
|
SAFE_MODE,
|
|
@@ -97,20 +108,23 @@ from starlette.exceptions import HTTPException as StarletteHTTPException
|
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
|
from starlette.middleware.sessions import SessionMiddleware
|
|
|
from starlette.responses import RedirectResponse, Response, StreamingResponse
|
|
|
-from utils.misc import (
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+from open_webui.utils.misc import (
|
|
|
add_or_update_system_message,
|
|
|
get_last_user_message,
|
|
|
parse_duration,
|
|
|
prepend_to_first_user_message_content,
|
|
|
)
|
|
|
-from utils.task import (
|
|
|
+from open_webui.utils.task import (
|
|
|
moa_response_generation_template,
|
|
|
search_query_generation_template,
|
|
|
title_generation_template,
|
|
|
tools_function_calling_generation_template,
|
|
|
)
|
|
|
-from utils.tools import get_tools
|
|
|
-from utils.utils import (
|
|
|
+from open_webui.utils.tools import get_tools
|
|
|
+from open_webui.utils.utils import (
|
|
|
create_token,
|
|
|
decode_token,
|
|
|
get_admin_user,
|
|
@@ -119,7 +133,7 @@ from utils.utils import (
|
|
|
get_password_hash,
|
|
|
get_verified_user,
|
|
|
)
|
|
|
-from utils.webhook import post_webhook
|
|
|
+from open_webui.utils.webhook import post_webhook
|
|
|
|
|
|
if SAFE_MODE:
|
|
|
print("SAFE MODE ENABLED")
|