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

🔧 chore(main.py): refactor imports and remove unused code in main.py for better code organization and maintainability

Browse source 
🔧 chore(manager.py): remove unused functions and refactor code in manager.py for better code organization and maintainability
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-18 17:43:05 -03:00
commit bc53800b1c
2 changed files with 5 additions and 65 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/main.py
View file

@ -10,9 +10,11 @@ from langflow.api import router
from langflow.interface.utils import setup_llm_caching from langflow.interface.utils import setup_llm_caching
from langflow.services.database.utils import initialize_database from langflow.services.database.utils import initialize_database
from langflow.services.manager import initialize_services, teardown_services from langflow.services.utils import initialize_services
from langflow.services.plugins.langfuse import LangfuseInstance from langflow.services.plugins.langfuse import LangfuseInstance
from langflow.services.utils import setup_superuser from langflow.services.utils import (
teardown_services,
)
from langflow.utils.logger import configure from langflow.utils.logger import configure
@ -41,12 +43,12 @@ def create_app():
app.on_event("startup")(initialize_services) app.on_event("startup")(initialize_services)
app.on_event("startup")(initialize_database) app.on_event("startup")(initialize_database)
app.on_event("startup")(setup_superuser)
app.on_event("startup")(setup_llm_caching) app.on_event("startup")(setup_llm_caching)
app.on_event("startup")(LangfuseInstance.update) app.on_event("startup")(LangfuseInstance.update)
app.on_event("shutdown")(teardown_services) app.on_event("shutdown")(teardown_services)
app.on_event("shutdown")(LangfuseInstance.teardown) app.on_event("shutdown")(LangfuseInstance.teardown)
return app return app

62
src/backend/langflow/services/manager.py
View file

@ -93,65 +93,3 @@ class ServiceManager:
service_manager = ServiceManager() service_manager = ServiceManager()
def initialize_services():
"""
Initialize all the services needed.
"""
from langflow.services.database import factory as database_factory
from langflow.services.cache import factory as cache_factory
from langflow.services.chat import factory as chat_factory
from langflow.services.settings import factory as settings_factory
from langflow.services.auth import factory as auth_factory
service_manager.register_factory(settings_factory.SettingsManagerFactory())
service_manager.register_factory(
auth_factory.AuthManagerFactory(), dependencies=[ServiceType.SETTINGS_MANAGER]
)
service_manager.register_factory(
database_factory.DatabaseManagerFactory(),
dependencies=[ServiceType.SETTINGS_MANAGER],
)
service_manager.register_factory(cache_factory.CacheManagerFactory())
service_manager.register_factory(chat_factory.ChatManagerFactory())
# Test cache connection
service_manager.get(ServiceType.CACHE_MANAGER)
# Test database connection
service_manager.get(ServiceType.DATABASE_MANAGER)
def initialize_settings_manager():
"""
Initialize the settings manager.
"""
from langflow.services.settings import factory as settings_factory
service_manager.register_factory(settings_factory.SettingsManagerFactory())
def initialize_session_manager():
"""
Initialize the session manager.
"""
from langflow.services.session import factory as session_manager_factory # type: ignore
from langflow.services.cache import factory as cache_factory
initialize_settings_manager()
service_manager.register_factory(
cache_factory.CacheManagerFactory(), dependencies=[ServiceType.SETTINGS_MANAGER]
)
service_manager.register_factory(
session_manager_factory.SessionManagerFactory(),
dependencies=[ServiceType.CACHE_MANAGER],
)
def teardown_services():
"""
Teardown all the services.
"""
service_manager.teardown()