[dev] foramt code

This commit is contained in:
jinli.yl 2024-06-20 22:20:27 +08:00
parent bccfb4c122
commit cd331b693d
7 changed files with 4 additions and 12 deletions

View file

@ -17,4 +17,3 @@ class MemoryNode(BaseModel):
status: str = Field("active", description="active or expired")
vector: List[float] = Field([], description="content embedding result, return empty")

View file

@ -6,4 +6,4 @@ class Message(BaseModel):
content: str = Field(..., description="The body of the message")
time_created: int = Field("", description="Timestamp when the message was created")
time_created: int = Field("", description="Timestamp when the message was created")

View file

@ -92,4 +92,4 @@ class Logger(logging.Logger):
if name not in LOGGER_DICT:
LOGGER_DICT[name] = Logger(name=name, **kwargs)
return LOGGER_DICT[name]
return LOGGER_DICT[name]

View file

@ -2,7 +2,6 @@
Registry for different modules.
Init class according to the class name and verify the input parameters.
"""
import inspect
from typing import Dict, Any, List
@ -26,7 +25,3 @@ class Registry(object):
def __getitem__(self, module_name: str):
assert module_name in self.module_dict, f'{module_name} not found in {self.name}'
return self.module_dict[module_name]

View file

@ -1,5 +1,3 @@
import time
from memory_scope.utils.logger import Logger

View file

@ -1,5 +1,6 @@
from importlib import import_module
import re
from importlib import import_module
from memory_scope.enumeration.message_role_enum import MessageRoleEnum

View file

@ -68,4 +68,3 @@ class BaseWorker(object):
if not self._name_simple:
self._name_simple = self.__class__.__name__.replace("Worker", "")
return self._name_simple