Merge pull request #94 from codefuse-ai/yuqi_open

同步1220

Author: cloud-waiting-for-wind

.gitignore

@@ -21,6 +21,8 @@ zdatafront*
 *log
 examples/mysql/db
 
+
+
 # frontend
 frontend/node_modules
 frontend/.env.local
@@ -37,4 +39,21 @@ frontend/.mfsu
 frontend/.swc
 frontend/pnpm-lock.yaml
 
-*.jar
+*.jar
+
+
+
+
+.spyproject/
+model_config.py
+model_config_example.py
+config.py
+**/service_onlyant
+**/ekg_test
+**/generalization_reasoning
+ekg.yaml
+*.ipynb
+**/web_operation
+examples/mysql/db
+tests/service/test_*
+tests/service/replacements.py

docker-compose.yaml

@@ -149,7 +149,7 @@ services:
 
 
   mysql:
-    image: mysql:8.0.23
+    image: mysql:8.4.3
     container_name: mysql
     environment:
       MYSQL_ROOT_PASSWORD: 'root'

examples/ekg_examples/who_is_spy_game.py

@@ -5,6 +5,7 @@
 from collections import Counter
 from loguru import logger
 import numpy as np
+import logging
 
 from langchain_community.docstore.document import Document
 
@@ -877,7 +878,7 @@ def get_msg_by_role_name(self, chat_index: str, role_name: str) -> Optional[Mess
                 return msg
         return None
 
-    def get_msg_content_by_role_name(self, chat_index: str, role_name: str) -> Optional[str]:
+    def get_msg_content_by_rule_name(self, chat_index: str, role_name: str) -> Optional[str]:
         message = self.get_msg_by_role_name(chat_index, role_name)
         if message == None:
             return None
@@ -889,16 +890,33 @@ def update_msg_content_by_rule(self, chat_index: str, role_name: str, new_conten
 
         if message == None:
             return False 
-
-        prompt = f"{new_content}\n{role_name}:{message.role_content}\n{update_rule}"
-
+        if update_rule == '':
+            prompt = '任务：请根据游戏内容，更新变量，变量名为：' + role_name + '，变量更新前的内容为：' + message.role_content + '。本节点游戏记录：' + new_content + '。请根据游戏内容，输出更新后的变量内容，不要包含其他信息，不要重复变量名，只输出变量更新后的内容即可。。'
+        else:
+            prompt = '任务：请根据游戏内容，更新变量，变量名为：' + role_name + '，变量更新前的内容为：' + message.role_content + '。本节点游戏记录：' + new_content + '。变量更新规则为：' + update_rule + '。请根据游戏内容和变量更新规则，输出更新后的变量内容，不要包含其他信息，不要重复变量名，只输出变量更新后的内容即可。'
+        logging.info(f'变量更新的prompt:{prompt}')
         model = getChatModelFromConfig(self.llm_config)
 
         new_role_content = model.predict(prompt)
-
+        logging.info(f'变量更新的输出结果:{new_role_content}')
         if new_role_content is not None:
             message.role_content = new_role_content  
             self.append(message)
+            logging.info(f'输出结果：{self.get_msg_content_by_rule_name(chat_index, role_name)}')
+
+            return True
+        else:
+            return False
+        
+    def update_global_msg_content(self, chat_index: str, role_name: str, new_content: str) -> bool:
+        message = self.get_msg_by_role_name(chat_index, role_name)
+        print(f'  message if {message}')
+        if message == None:
+            return False 
+
+        if new_content is not None:
+            message.role_content = new_content  
+            self.append(message)
             return True
         else:
             return False

muagent/connector/memory_manager.py

@@ -63,11 +63,13 @@ def __init__(self,gb_config : GBConfig = None):
             self.nb_pw = '' or 'nebula'
             self.space_name = "client"
         else:
+
             logger.info('NebulaGraph容器启动中，请等待')
 
             if self.nebula_started(gb_config):
                 self.connection_pool.init([(gb_config.extra_kwargs.get("host"), gb_config.extra_kwargs.get("port"))], config)
 
+
             self.username = gb_config.extra_kwargs.get("username")
             self.nb_pw = gb_config.extra_kwargs.get("password")
             self.space_name = gb_config.extra_kwargs.get("space")
@@ -317,16 +319,7 @@ def add_node(self, node: GNode) -> GbaseExecStatus:
                 if prop_name in {'extra', 'description', 'envdescription','updaterule'}:
                     # 转义换行符和双引号
                     value = value.replace("\n", "\\n").replace("\"", "\\\"")
-                    cypher += f'"{value}",'
-                elif prop_name == 'description':
-                    value = value.replace("\n", "\\n").replace("\"", "\\\"")
-                    cypher += f'"{value}",'
-                elif prop_name == 'envdescription':
-                    value = value.replace("\n", "\\n").replace("\"", "\\\"")
-                    cypher += f'"{value}",'
-                else:
-                    cypher += f'"{value}",'
-                #cypher += f'"{value}",'
+                cypher += f'"{value}",'
             else:
                 cypher += f'{value},'
         cypher = cypher.rstrip(',')