fix: output_language setting in ChatAgent (#1165)

Wendong-Fan · web-flow · commit 32f19f59f4ff · 2024-11-10T09:13:42.000+08:00
diff --git a/.github/ISSUE_TEMPLATE/bug_report.yml b/.github/ISSUE_TEMPLATE/bug_report.yml
@@ -26,7 +26,7 @@ body:
     attributes:
       label: What version of camel are you using?
       description: Run command `python3 -c 'print(__import__("camel").__version__)'` in your shell and paste the output here.
-      placeholder: E.g., 0.2.6
+      placeholder: E.g., 0.2.7
     validations:
       required: true
 
diff --git a/README.md b/README.md
@@ -144,7 +144,7 @@ conda create --name camel python=3.10
 conda activate camel
 
 # Clone github repo
-git clone -b v0.2.6 https://github.com/camel-ai/camel.git
+git clone -b v0.2.7 https://github.com/camel-ai/camel.git
 
 # Change directory into project directory
 cd camel
diff --git a/camel/__init__.py b/camel/__init__.py
@@ -12,7 +12,7 @@
 # limitations under the License.
 # =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
 
-__version__ = '0.2.6'
+__version__ = '0.2.7'
 
 __all__ = [
     '__version__',
diff --git a/camel/agents/chat_agent.py b/camel/agents/chat_agent.py
@@ -180,9 +180,6 @@ def __init__(
                 model_type=ModelType.DEFAULT,
             )
         )
-        self.output_language: Optional[str] = output_language
-        if self.output_language is not None:
-            self.set_output_language(self.output_language)
 
         self.model_type = self.model_backend.model_type
 
@@ -218,6 +215,10 @@ def __init__(
             context_creator, window_size=message_window_size
         )
 
+        self.output_language: Optional[str] = output_language
+        if self.output_language is not None:
+            self.set_output_language(self.output_language)
+
         self.terminated: bool = False
         self.response_terminators = response_terminators or []
         self.init_messages()
@@ -371,13 +372,19 @@ def set_output_language(self, output_language: str) -> BaseMessage:
             self._system_message = self.orig_sys_message.create_new_instance(
                 content
             )
-            return self._system_message
         else:
             self._system_message = BaseMessage.make_assistant_message(
                 role_name="Assistant",
                 content=language_prompt,
             )
-            return self._system_message
+
+        system_record = MemoryRecord(
+            message=self._system_message,
+            role_at_backend=OpenAIBackendRole.SYSTEM,
+        )
+        self.memory.clear()
+        self.memory.write_record(system_record)
+        return self._system_message
 
     def get_info(
         self,
diff --git a/docs/conf.py b/docs/conf.py
@@ -27,7 +27,7 @@
 project = 'CAMEL'
 copyright = '2024, CAMEL-AI.org'
 author = 'CAMEL-AI.org'
-release = '0.2.6'
+release = '0.2.7'
 
 html_favicon = (
     'https://raw.githubusercontent.com/camel-ai/camel/master/misc/favicon.png'
diff --git a/docs/cookbooks/agents_message.ipynb b/docs/cookbooks/agents_message.ipynb
@@ -101,7 +101,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install \"camel-ai==0.2.6\""
+        "!pip install \"camel-ai==0.2.7\""
       ]
     },
     {
diff --git a/docs/cookbooks/agents_prompting.ipynb b/docs/cookbooks/agents_prompting.ipynb
@@ -64,7 +64,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install \"camel-ai==0.2.6\""
+        "!pip install \"camel-ai==0.2.7\""
       ]
     },
     {
diff --git a/docs/cookbooks/agents_society.ipynb b/docs/cookbooks/agents_society.ipynb
@@ -192,7 +192,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install \"camel-ai==0.2.6\""
+        "!pip install \"camel-ai==0.2.7\""
       ]
     },
     {
diff --git a/docs/cookbooks/agents_tracking.ipynb b/docs/cookbooks/agents_tracking.ipynb
@@ -64,7 +64,7 @@
       },
       "outputs": [],
       "source": [
-        "%pip install camel-ai[all]==0.2.6\n",
+        "%pip install camel-ai[all]==0.2.7\n",
         "%pip install agentops==0.3.10"
       ]
     },
diff --git a/docs/cookbooks/agents_with_memory.ipynb b/docs/cookbooks/agents_with_memory.ipynb
@@ -73,7 +73,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install \"camel-ai[all]==0.2.6\""
+        "!pip install \"camel-ai[all]==0.2.7\""
       ]
     },
     {
diff --git a/docs/cookbooks/agents_with_rag.ipynb b/docs/cookbooks/agents_with_rag.ipynb
@@ -75,7 +75,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install \"camel-ai[all]==0.2.6\""
+        "!pip install \"camel-ai[all]==0.2.7\""
       ]
     },
     {
diff --git a/docs/cookbooks/agents_with_tools.ipynb b/docs/cookbooks/agents_with_tools.ipynb
@@ -77,7 +77,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install \"camel-ai[all]==0.2.6\""
+        "!pip install \"camel-ai[all]==0.2.7\""
       ]
     },
     {
diff --git a/docs/cookbooks/create_your_first_agent.ipynb b/docs/cookbooks/create_your_first_agent.ipynb
@@ -83,7 +83,7 @@
     {
       "cell_type": "code",
       "source": [
-        "!pip install \"camel-ai[all]==0.2.6\""
+        "!pip install \"camel-ai[all]==0.2.7\""
       ],
       "metadata": {
         "id": "UtcC3c-KVZmU"
diff --git a/docs/cookbooks/create_your_first_agents_society.ipynb b/docs/cookbooks/create_your_first_agents_society.ipynb
@@ -77,7 +77,7 @@
     {
       "cell_type": "code",
       "source": [
-        "!pip install \"camel-ai==0.2.6\""
+        "!pip install \"camel-ai==0.2.7\""
       ],
       "metadata": {
         "id": "RiwfwyyLYYxo"
diff --git a/docs/cookbooks/critic_agents_and_tree_search.ipynb b/docs/cookbooks/critic_agents_and_tree_search.ipynb
@@ -84,7 +84,7 @@
     {
       "cell_type": "code",
       "source": [
-        "%pip install \"camel-ai==0.2.6\""
+        "%pip install \"camel-ai==0.2.7\""
       ],
       "metadata": {
         "id": "UtcC3c-KVZmU"
diff --git a/docs/cookbooks/embodied_agents.ipynb b/docs/cookbooks/embodied_agents.ipynb
@@ -67,7 +67,7 @@
     {
       "cell_type": "code",
       "source": [
-        "%pip install \"camel-ai==0.2.6\""
+        "%pip install \"camel-ai==0.2.7\""
       ],
       "metadata": {
         "id": "UtcC3c-KVZmU"
diff --git a/docs/cookbooks/knowledge_graph.ipynb b/docs/cookbooks/knowledge_graph.ipynb
@@ -73,7 +73,7 @@
       },
       "outputs": [],
       "source": [
-        "pip install \"camel-ai[all]==0.2.6\""
+        "pip install \"camel-ai[all]==0.2.7\""
       ]
     },
     {
diff --git a/docs/cookbooks/model_speed_comparison.ipynb b/docs/cookbooks/model_speed_comparison.ipynb
@@ -63,7 +63,7 @@
     {
       "cell_type": "code",
       "source": [
-        "!pip install \"camel-ai==0.2.6\""
+        "!pip install \"camel-ai==0.2.7\""
       ],
       "metadata": {
         "id": "UtcC3c-KVZmU"
diff --git a/docs/cookbooks/roleplaying_scraper.ipynb b/docs/cookbooks/roleplaying_scraper.ipynb
@@ -89,7 +89,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install \"camel-ai[all]==0.2.6\""
+        "!pip install \"camel-ai[all]==0.2.7\""
       ]
     },
     {
diff --git a/docs/cookbooks/task_generation.ipynb b/docs/cookbooks/task_generation.ipynb
@@ -62,7 +62,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install \"camel-ai==0.2.6\""
+        "!pip install \"camel-ai==0.2.7\""
       ]
     },
     {
diff --git a/docs/cookbooks/video_analysis.ipynb b/docs/cookbooks/video_analysis.ipynb
@@ -28,7 +28,7 @@
    },
    "outputs": [],
    "source": [
-    "%pip install \"camel-ai[all]==0.2.6\""
+    "%pip install \"camel-ai[all]==0.2.7\""
    ]
   },
   {
diff --git a/docs/cookbooks/workforce_judge_committee.ipynb b/docs/cookbooks/workforce_judge_committee.ipynb
@@ -28,7 +28,7 @@
       },
       "outputs": [],
       "source": [
-        "%pip install \"camel-ai[all]==0.2.6\""
+        "%pip install \"camel-ai[all]==0.2.7\""
       ]
     },
     {
diff --git a/docs/get_started/installation.md b/docs/get_started/installation.md
@@ -60,7 +60,7 @@ conda create --name camel python=3.10
 conda activate camel
 
 # Clone github repo
-git clone -b v0.2.6 https://github.com/camel-ai/camel.git
+git clone -b v0.2.7 https://github.com/camel-ai/camel.git
 
 # Change directory into project directory
 cd camel
diff --git a/docs/key_modules/loaders.md b/docs/key_modules/loaders.md
@@ -336,14 +336,14 @@ response = jina_reader.read_content("https://docs.camel-ai.org/")
 print(response)
 ```
 ```markdown
->>>Welcome to CAMEL’s documentation! — CAMEL 0.2.6 documentation
+>>>Welcome to CAMEL’s documentation! — CAMEL 0.2.7 documentation
 ===============
 
 [Skip to main content](https://docs.camel-ai.org/#main-content)
 
 Back to top Ctrl+K
 
- [![Image 1](https://raw.githubusercontent.com/camel-ai/camel/master/misc/logo_light.png) ![Image 2](https://raw.githubusercontent.com/camel-ai/camel/master/misc/logo_light.png)CAMEL 0.2.6](https://docs.camel-ai.org/#)
+ [![Image 1](https://raw.githubusercontent.com/camel-ai/camel/master/misc/logo_light.png) ![Image 2](https://raw.githubusercontent.com/camel-ai/camel/master/misc/logo_light.png)CAMEL 0.2.7](https://docs.camel-ai.org/#)
 
 Search Ctrl+K
 
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "camel-ai"
-version = "0.2.6"
+version = "0.2.7"
 authors = ["CAMEL-AI.org"]
 description = "Communicative Agents for AI Society Study"
 readme = "README.md"
diff --git a/test/agents/test_chat_agent.py b/test/agents/test_chat_agent.py
@@ -414,14 +414,13 @@ def test_set_output_language():
     assert agent.output_language == output_language
 
     # Verify that the system message is updated with the new output language
-    updated_system_message = BaseMessage(
-        role_name="assistant",
-        role_type=RoleType.ASSISTANT,
-        meta_dict=None,
-        content="You are a help assistant."
-        "\nRegardless of the input language, you must output text in Arabic.",
-    )
-    assert agent.system_message.content == updated_system_message.content
+    updated_system_message = {
+        'role': 'system',
+        'content': 'You are a help assistant.\nRegardless of the '
+        'input language, you must output text in Arabic.',
+    }
+    memory_content = agent.memory.get_context()
+    assert memory_content[0][0] == updated_system_message
 
 
 @pytest.mark.model_backend
@@ -444,28 +443,27 @@ def test_set_multiple_output_language():
     agent_without_sys_msg.set_output_language("English")
     agent_without_sys_msg.set_output_language("French")
 
-    updated_system_message_with_content = BaseMessage(
-        role_name="assistant",
-        role_type=RoleType.ASSISTANT,
-        meta_dict=None,
-        content="You are a help assistant."
-        "\nRegardless of the input language, you must output text in French.",
-    )
-    updated_system_message_without_content = BaseMessage(
-        role_name="assistant",
-        role_type=RoleType.ASSISTANT,
-        meta_dict=None,
-        content="\nRegardless of the input language, you must output text "
-        "in French.",
-    )
+    updated_system_message_with_sys_msg = {
+        'role': 'system',
+        'content': 'You are a help assistant.\nRegardless of the '
+        'input language, you must output text in French.',
+    }
+    updated_system_message_without_sys_msg = {
+        'role': 'system',
+        'content': '\nRegardless of the input language, you must output '
+        'text in French.',
+    }
+
+    memory_content_with_sys_msg = agent_with_sys_msg.memory.get_context()
+    memory_content_without_sys_msg = agent_without_sys_msg.memory.get_context()
 
     assert (
-        agent_with_sys_msg.system_message.content
-        == updated_system_message_with_content.content
+        memory_content_with_sys_msg[0][0]
+        == updated_system_message_with_sys_msg
     )
     assert (
-        agent_without_sys_msg.system_message.content
-        == updated_system_message_without_content.content
+        memory_content_without_sys_msg[0][0]
+        == updated_system_message_without_sys_msg
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -101,7 +101,7 @@`
`101`	`101`	`},`
`102`	`102`	`"outputs": [],`
`103`	`103`	`"source": [`
`104`		`- "!pip install \"camel-ai==0.2.6\""`
	`104`	`+ "!pip install \"camel-ai==0.2.7\""`
`105`	`105`	`]`
`106`	`106`	`},`
`107`	`107`	`{`
Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,7 @@`
`64`	`64`	`},`
`65`	`65`	`"outputs": [],`
`66`	`66`	`"source": [`
`67`		`- "!pip install \"camel-ai==0.2.6\""`
	`67`	`+ "!pip install \"camel-ai==0.2.7\""`
`68`	`68`	`]`
`69`	`69`	`},`
`70`	`70`	`{`
Original file line number	Diff line number	Diff line change
`@@ -192,7 +192,7 @@`
`192`	`192`	`},`
`193`	`193`	`"outputs": [],`
`194`	`194`	`"source": [`
`195`		`- "!pip install \"camel-ai==0.2.6\""`
	`195`	`+ "!pip install \"camel-ai==0.2.7\""`
`196`	`196`	`]`
`197`	`197`	`},`
`198`	`198`	`{`
Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@`
`73`	`73`	`},`
`74`	`74`	`"outputs": [],`
`75`	`75`	`"source": [`
`76`		`- "!pip install \"camel-ai[all]==0.2.6\""`
	`76`	`+ "!pip install \"camel-ai[all]==0.2.7\""`
`77`	`77`	`]`
`78`	`78`	`},`
`79`	`79`	`{`