Optimize code format of record func

camel-ai · Feb 14, 2025 · 4c58f3b · 4c58f3b
1 parent 0b8a132
commit 4c58f3b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/camel/benchmarks/nexus.py b/camel/benchmarks/nexus.py
@@ -312,7 +312,7 @@ def run(  # type: ignore[override, return]
                 try:
                     # Generate response
                     response = agent.step(prompt)
-                    agent_call = response.msgs[0].content
+                    agent_call = response.msgs[0].content.text
 
                     # Evaluate response
                     if agent_call:

diff --git a/camel/memories/records.py b/camel/memories/records.py
@@ -60,16 +60,16 @@ def from_dict(cls, record_dict: Dict[str, Any]) -> "MemoryRecord":
         """
         message_data = record_dict["message"]
         message_class = message_data.get('__class__')
-        
+
         # Extract common data
         role_name = message_data['role_name']
         role_type = message_data['role_type']
         content_data = message_data['content']
-        
+
         # Determine content type
-        if 'image_list' in content_data and content_data['image_list']:
+        if content_data.get('image_list'):
             content = Content.from_dict(content_data)
-        elif 'video_bytes' in content_data and content_data['video_bytes']:
+        elif content_data.get('video_bytes'):
             content = Content.from_dict(content_data)
         else:
             content = Content(text=content_data)