jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Fix imports and initialize services with

Browse source 
fix_migration option
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-01 16:30:35 -03:00
commit cb6228cea1
1 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/backend/langflow/services/utils.py
View file

@ -2,7 +2,8 @@ from langflow.services.auth.utils import create_super_user, verify_password
from langflow.services.database.utils import initialize_database from langflow.services.database.utils import initialize_database
from langflow.services.manager import service_manager from langflow.services.manager import service_manager
from langflow.services.schema import ServiceType from langflow.services.schema import ServiceType
from langflow.services.settings.constants import DEFAULT_SUPERUSER, DEFAULT_SUPERUSER_PASSWORD from langflow.services.settings.constants import (DEFAULT_SUPERUSER,
DEFAULT_SUPERUSER_PASSWORD)
from loguru import logger from loguru import logger
from sqlmodel import Session from sqlmodel import Session
@ -13,12 +14,13 @@ def get_factories_and_deps():
from langflow.services.auth import factory as auth_factory from langflow.services.auth import factory as auth_factory
from langflow.services.cache import factory as cache_factory from langflow.services.cache import factory as cache_factory
from langflow.services.chat import factory as chat_factory from langflow.services.chat import factory as chat_factory
from langflow.services.credentials import factory as credentials_factory
from langflow.services.database import factory as database_factory from langflow.services.database import factory as database_factory
from langflow.services.session import factory as session_service_factory # type: ignore from langflow.services.session import \
factory as session_service_factory # type: ignore
from langflow.services.settings import factory as settings_factory from langflow.services.settings import factory as settings_factory
from langflow.services.store import factory as store_factory from langflow.services.store import factory as store_factory
from langflow.services.task import factory as task_factory from langflow.services.task import factory as task_factory
from langflow.services.credentials import factory as credentials_factory
return [ return [
(settings_factory.SettingsServiceFactory(), []), (settings_factory.SettingsServiceFactory(), []),
@ -171,7 +173,8 @@ def initialize_session_service():
Initialize the session manager. Initialize the session manager.
""" """
from langflow.services.cache import factory as cache_factory from langflow.services.cache import factory as cache_factory
from langflow.services.session import factory as session_service_factory # type: ignore from langflow.services.session import \
factory as session_service_factory # type: ignore
initialize_settings_service() initialize_settings_service()
@ -183,7 +186,7 @@ def initialize_session_service():
) )
def initialize_services(): def initialize_services(fix_migration: bool = False):
""" """
Initialize all the services needed. Initialize all the services needed.
""" """
@ -197,7 +200,7 @@ def initialize_services():
# Test cache connection # Test cache connection
service_manager.get(ServiceType.CACHE_SERVICE) service_manager.get(ServiceType.CACHE_SERVICE)
# Setup the superuser # Setup the superuser
initialize_database() initialize_database(fix_migration=fix_migration)
setup_superuser(service_manager.get(ServiceType.SETTINGS_SERVICE), next(get_session())) setup_superuser(service_manager.get(ServiceType.SETTINGS_SERVICE), next(get_session()))
try: try:
get_db_service().migrate_flows_if_auto_login() get_db_service().migrate_flows_if_auto_login()