mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-26 16:53:36 +08:00
27 lines
824 B
Python
27 lines
824 B
Python
import redis
|
|
from redis.connection import Connection, SSLConnection
|
|
|
|
redis_client = redis.Redis()
|
|
|
|
|
|
def init_app(app):
|
|
connection_class = Connection
|
|
if app.config.get("REDIS_USE_SSL", False):
|
|
connection_class = SSLConnection
|
|
|
|
redis_client.connection_pool = redis.ConnectionPool(
|
|
**{
|
|
"host": app.config.get("REDIS_HOST", "localhost"),
|
|
"port": app.config.get("REDIS_PORT", 6379),
|
|
"username": app.config.get("REDIS_USERNAME", None),
|
|
"password": app.config.get("REDIS_PASSWORD", None),
|
|
"db": app.config.get("REDIS_DB", 0),
|
|
"encoding": "utf-8",
|
|
"encoding_errors": "strict",
|
|
"decode_responses": False,
|
|
},
|
|
connection_class=connection_class,
|
|
)
|
|
|
|
app.extensions["redis"] = redis_client
|