shared.py 728 B

12345678910111213141516171819202122232425262728
  1. import os
  2. import socketio
  3. from dotenv import load_dotenv
  4. from openhands.core.config import load_app_config
  5. from openhands.server.session import SessionManager
  6. from openhands.storage import get_file_store
  7. load_dotenv()
  8. config = load_app_config()
  9. file_store = get_file_store(config.file_store, config.file_store_path)
  10. client_manager = None
  11. redis_host = os.environ.get('REDIS_HOST')
  12. if redis_host:
  13. client_manager = socketio.AsyncRedisManager(
  14. f'redis://{redis_host}',
  15. redis_options={'password': os.environ.get('REDIS_PASSWORD')},
  16. )
  17. sio = socketio.AsyncServer(
  18. async_mode='asgi', cors_allowed_origins='*', client_manager=client_manager
  19. )
  20. session_manager = SessionManager(sio, config, file_store)