apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-14 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: +code

Browse source
This commit is contained in:
莘权 马 2023-09-04 14:38:35 +08:00
parent 41e90b4f48
commit 26c4ed6e22
2 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

8
metagpt/memory/brain_memory.py
View file

@ -93,8 +93,8 @@ class BrainMemory(pydantic.BaseModel):
return hashlib.md5(text.encode()).hexdigest()
@staticmethod
async def loads(redis_key: str) -> "BrainMemory":
redis = Redis()
async def loads(redis_key: str, redis_conf: Dict = None) -> "BrainMemory":
redis = Redis(conf=redis_conf)
if not redis.is_valid() or not redis_key:
return BrainMemory()
v = await redis.get(key=redis_key)
@ -103,8 +103,8 @@ class BrainMemory(pydantic.BaseModel):
return BrainMemory(**data)
return BrainMemory()
async def dumps(self, redis_key: str, timeout_sec: int = 30 * 60):
redis = Redis()
async def dumps(self, redis_key: str, timeout_sec: int = 30 * 60, redis_conf: Dict = None):
redis = Redis(conf=redis_conf)
if not redis.is_valid() or not redis_key:
return False
v = self.json()

6
metagpt/utils/redis.py
View file

@ -6,7 +6,7 @@
import json
from datetime import timedelta
from enum import Enum
from typing import Awaitable, Callable, Optional, Union
from typing import Awaitable, Callable, Dict, Optional, Union
from redis import asyncio as aioredis
@ -184,8 +184,8 @@ class RedisManager:
class Redis:
def __init__(self):
self._config = CONFIG.REDIS
def __init__(self, conf: Dict = None):
self._config = conf or CONFIG.REDIS
if not self._config:
return
try: