format code

Author: Nyakult

src/memos/llms/openai.py

@@ -65,7 +65,7 @@ def generate(self, messages: MessageList, **kwargs) -> str:
             temperature=self.config.temperature,
             max_tokens=self.config.max_tokens,
             top_p=self.config.top_p,
-            **kwargs
+            **kwargs,
         )
         logger.info(f"Response from OpenAI: {response.model_dump_json()}")
         response_content = response.choices[0].message.content

src/memos/llms/vllm.py

@@ -106,7 +106,7 @@ def _generate_with_api_client(self, messages: list[MessageDict], **kwargs) -> st
                 "max_tokens": int(getattr(self.config, "max_tokens", 1024)),
                 "top_p": float(getattr(self.config, "top_p", 0.9)),
                 "extra_body": {"chat_template_kwargs": {"enable_thinking": False}},
-                **kwargs
+                **kwargs,
             }
 
             response = self.client.chat.completions.create(**completion_kwargs)

src/memos/mem_reader/simple_struct.py

@@ -27,7 +27,7 @@
     SIMPLE_STRUCT_MEM_READER_EXAMPLE_ZH,
     SIMPLE_STRUCT_MEM_READER_PROMPT,
     SIMPLE_STRUCT_MEM_READER_PROMPT_ZH,
-    reader_output_schema
+    reader_output_schema,
 )
 from memos.utils import timed
 
@@ -201,9 +201,9 @@ def _get_llm_response(self, mem_str: str) -> dict:
             prompt = prompt.replace(examples, "")
         messages = [{"role": "user", "content": prompt}]
         try:
-            response_text = self.llm.generate(messages,
-                                              response_format={"type": "json_object",
-                                                               "schema": reader_output_schema})
+            response_text = self.llm.generate(
+                messages, response_format={"type": "json_object", "schema": reader_output_schema}
+            )
             response_json = self.parse_json_result(response_text)
         except Exception as e:
             logger.error(f"[LLM] Exception during chat generation: {e}")

src/memos/templates/mem_reader_prompts.py

@@ -419,46 +419,44 @@
 """
 
 reader_output_schema = {
-  "$schema": "https://json-schema.org/draft/2020-12/schema",
-  "type": "object",
-  "properties": {
-    "memory list": {
-      "type": "array",
-      "items": {
-        "type": "object",
-        "properties": {
-          "key": {
-            "type": "string",
-            "description": "A brief title or identifier for the memory."
-          },
-          "memory_type": {
-            "type": "string",
-            "enum": ["LongTermMemory", "ShortTermMemory", "WorkingMemory"],
-            "description": "The type of memory, expected to be 'LongTermMemory' in this context."
-          },
-          "value": {
-            "type": "string",
-            "description": "Detailed description of the memory, including viewpoint, time, and content."
-          },
-          "tags": {
+    "$schema": "https://json-schema.org/draft/2020-12/schema",
+    "type": "object",
+    "properties": {
+        "memory list": {
             "type": "array",
             "items": {
-              "type": "string"
+                "type": "object",
+                "properties": {
+                    "key": {
+                        "type": "string",
+                        "description": "A brief title or identifier for the memory.",
+                    },
+                    "memory_type": {
+                        "type": "string",
+                        "enum": ["LongTermMemory", "ShortTermMemory", "WorkingMemory"],
+                        "description": "The type of memory, expected to be 'LongTermMemory' in this context.",
+                    },
+                    "value": {
+                        "type": "string",
+                        "description": "Detailed description of the memory, including viewpoint, time, and content.",
+                    },
+                    "tags": {
+                        "type": "array",
+                        "items": {"type": "string"},
+                        "description": "List of keywords or categories associated with the memory.",
+                    },
+                },
+                "required": ["key", "memory_type", "value", "tags"],
+                "additionalProperties": False,
             },
-            "description": "List of keywords or categories associated with the memory."
-          }
+            "description": "List of memory entries.",
+        },
+        "summary": {
+            "type": "string",
+            "description": "A synthesized summary of the overall situation based on all memories.",
         },
-        "required": ["key", "memory_type", "value", "tags"],
-        "additionalProperties": False
-      },
-      "description": "List of memory entries."
     },
-    "summary": {
-      "type": "string",
-      "description": "A synthesized summary of the overall situation based on all memories."
-    }
-  },
-  "required": ["memory list", "summary"],
-  "additionalProperties": False,
-  "description": "Structured output containing a list of memories and a summary."
-}
+    "required": ["memory list", "summary"],
+    "additionalProperties": False,
+    "description": "Structured output containing a list of memories and a summary.",
+}