feat : 代码梳理 移除所有敏感密钥 通过环境变量方式配置

app/service/chat_robot/script/memory/user_buffer_window.py

@@ -3,13 +3,12 @@ from typing import Any, Dict, List, Tuple
 import json
 
 import redis
+from langchain_classic.memory.chat_memory import BaseChatMemory
+from langchain_classic.memory.utils import get_prompt_input_key
+from langchain_core.messages import messages_from_dict, get_buffer_string, BaseMessage, HumanMessage, AIMessage, message_to_dict
 from redis import Redis
-from langchain.memory.chat_memory import BaseChatMemory
-from langchain.schema.messages import BaseMessage, get_buffer_string, HumanMessage, AIMessage
-from langchain.schema.messages import _message_to_dict, messages_from_dict
-from langchain.memory.utils import get_prompt_input_key
 
-from app.core.config import *
+from app.core.config import settings
 
 
 class UserConversationBufferWindowMemory(BaseChatMemory):
@@ -24,8 +23,8 @@ class UserConversationBufferWindowMemory(BaseChatMemory):
     @classmethod
     def from_redis(
             cls,
-            host: str = REDIS_HOST,
-            port: int = REDIS_PORT,
+            host: str = settings.REDIS_HOST,
+            port: int = settings.REDIS_PORT,
             db: int = 3,
             **kwargs
     ):
@@ -79,7 +78,7 @@ class UserConversationBufferWindowMemory(BaseChatMemory):
         return inputs[prompt_input_key], outputs[output_key]
 
     def add_message(self, key: str, message: BaseMessage) -> None:
-        self.redis_client.lpush(key, json.dumps(_message_to_dict(message)))
+        self.redis_client.lpush(key, json.dumps(message_to_dict(message)))
 
     def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, str], key: str = "") -> None:
         """Save context from this conversation to buffer."""