From 89aa09bc35ffb91f2d60622ab61c1ab5a49ead53 Mon Sep 17 00:00:00 2001
From: sugarkwork <sugarkwork@sugar-knight.com>
Date: Mon, 14 Apr 2025 12:06:51 +0900
Subject: [PATCH] update

---
 nodes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index 8f30dba..e32a526 100644
--- a/nodes.py
+++ b/nodes.py
@@ -73,7 +73,7 @@ class SimpleCohereNode:
         key = f"cohere: system={system} , text={text}"
         response = load_memory(key)
         if response is not None:
-            return (response.message.content[0].text, )
+            return (response, )
         
         if not response:
             response = cohere_client.chat(
@@ -84,7 +84,7 @@ class SimpleCohereNode:
                 model="command-r-plus"
             )
         
-        save_memory(key, response)
+        save_memory(key, response.message.content[0].text)
 
 
         return (response.message.content[0].text, )