camel-ai 0.1.6.7__tar.gz → 0.1.6.8__tar.gz

{camel_ai-0.1.6.7 → camel_ai-0.1.6.8}/PKG-INFO

@@ -1,15 +1,14 @@
 Metadata-Version: 2.1
 Name: camel-ai
-Version: 0.1.6.7
+Version: 0.1.6.8
 Summary: Communicative Agents for AI Society Study
 Home-page: https://www.camel-ai.org/
 License: Apache-2.0
 Keywords: communicative-ai,ai-societies,artificial-intelligence,deep-learning,multi-agent-systems,cooperative-ai,natural-language-processing,large-language-models
 Author: CAMEL-AI.org
-Requires-Python: >=3.9.0,<3.12
+Requires-Python: >=3.10.0,<3.12
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Provides-Extra: all
@@ -29,7 +28,7 @@ Requires-Dist: agentops (>=0.3.6,<0.4.0) ; extra == "tools" or extra == "all"
 Requires-Dist: anthropic (>=0.29.0,<0.30.0)
 Requires-Dist: azure-storage-blob (>=12.21.0,<13.0.0) ; extra == "object-storages" or extra == "all"
 Requires-Dist: beautifulsoup4 (>=4,<5) ; extra == "tools" or extra == "all"
-Requires-Dist: boto3 (>=1.34.149,<2.0.0) ; extra == "object-storages" or extra == "all"
+Requires-Dist: botocore (>=1.35.3,<2.0.0) ; extra == "object-storages" or extra == "all"
 Requires-Dist: cohere (>=4.56,<5.0) ; extra == "retrievers" or extra == "all"
 Requires-Dist: colorama (>=0,<1)
 Requires-Dist: curl_cffi (==0.6.2)
@@ -75,6 +74,7 @@ Requires-Dist: pytest-asyncio (>=0.23.0,<0.24.0) ; extra == "test"
 Requires-Dist: qdrant-client (>=1.9.0,<2.0.0) ; extra == "vector-databases" or extra == "all"
 Requires-Dist: rank-bm25 (>=0.2.2,<0.3.0) ; extra == "retrievers" or extra == "all"
 Requires-Dist: redis (>=5.0.6,<6.0.0) ; extra == "kv-stroages" or extra == "all"
+Requires-Dist: reka-api (>=3.0.8,<4.0.0) ; extra == "model-platforms" or extra == "all"
 Requires-Dist: requests_oauthlib (>=1.3.1,<2.0.0) ; extra == "tools" or extra == "all"
 Requires-Dist: sentence-transformers (>=3.0.1,<4.0.0) ; extra == "encoders" or extra == "all"
 Requires-Dist: sentencepiece (>=0,<1) ; extra == "huggingface-agent" or extra == "all"
@@ -164,7 +164,7 @@ Some features require extra dependencies:
 
 Install `CAMEL` from source with poetry (Recommended):
 ```sh
-# Make sure your python version is later than 3.9
+# Make sure your python version is later than 3.10
 # You can use pyenv to manage multiple python verisons in your sytstem
 
 # Clone github repo
@@ -187,22 +187,31 @@ poetry shell
 poetry install
 
 # Install CAMEL with all dependencies
-poetry install -E all  # (Optional)
+poetry install -E all # (Optional)
 
 # Exit the virtual environment
 exit
 ```
 
+> [!TIP]
+> If you encounter errors when running `poetry install`, it may be due to a cache-related problem. You can try running:
+> ```sh
+> poetry install --no-cache
+> ```
+
+
+
+
 Install `CAMEL` from source with conda and pip:
 ```sh
 # Create a conda virtual environment
-conda create --name camel python=3.9
+conda create --name camel python=3.10
 
 # Activate CAMEL conda environment
 conda activate camel
 
 # Clone github repo
-git clone -b v0.1.6.7 https://github.com/camel-ai/camel.git
+git clone -b v0.1.6.8 https://github.com/camel-ai/camel.git
 
 # Change directory into project directory
 cd camel
@@ -415,8 +424,8 @@ We appreciate your interest in contributing to our open-source initiative. We pr
 ## Contact
 For more information please contact camel.ai.team@gmail.com.
 
-[python-image]: https://img.shields.io/badge/Python-3.9%2B-brightgreen.svg
-[python-url]: https://docs.python.org/3.9/
+[python-image]: https://img.shields.io/badge/Python-3.10%2B-brightgreen.svg
+[python-url]: https://docs.python.org/3.10/
 [pytest-image]: https://github.com/camel-ai/camel/actions/workflows/pytest_package.yml/badge.svg
 [pytest-url]: https://github.com/camel-ai/camel/actions/workflows/pytest_package.yml
 [docs-image]: https://img.shields.io/badge/Documentation-grey.svg?logo=github

{camel_ai-0.1.6.7 → camel_ai-0.1.6.8}/README.md

@@ -72,7 +72,7 @@ Some features require extra dependencies:
 
 Install `CAMEL` from source with poetry (Recommended):
 ```sh
-# Make sure your python version is later than 3.9
+# Make sure your python version is later than 3.10
 # You can use pyenv to manage multiple python verisons in your sytstem
 
 # Clone github repo
@@ -95,22 +95,31 @@ poetry shell
 poetry install
 
 # Install CAMEL with all dependencies
-poetry install -E all  # (Optional)
+poetry install -E all # (Optional)
 
 # Exit the virtual environment
 exit
 ```
 
+> [!TIP]
+> If you encounter errors when running `poetry install`, it may be due to a cache-related problem. You can try running:
+> ```sh
+> poetry install --no-cache
+> ```
+
+
+
+
 Install `CAMEL` from source with conda and pip:
 ```sh
 # Create a conda virtual environment
-conda create --name camel python=3.9
+conda create --name camel python=3.10
 
 # Activate CAMEL conda environment
 conda activate camel
 
 # Clone github repo
-git clone -b v0.1.6.7 https://github.com/camel-ai/camel.git
+git clone -b v0.1.6.8 https://github.com/camel-ai/camel.git
 
 # Change directory into project directory
 cd camel
@@ -323,8 +332,8 @@ We appreciate your interest in contributing to our open-source initiative. We pr
 ## Contact
 For more information please contact camel.ai.team@gmail.com.
 
-[python-image]: https://img.shields.io/badge/Python-3.9%2B-brightgreen.svg
-[python-url]: https://docs.python.org/3.9/
+[python-image]: https://img.shields.io/badge/Python-3.10%2B-brightgreen.svg
+[python-url]: https://docs.python.org/3.10/
 [pytest-image]: https://github.com/camel-ai/camel/actions/workflows/pytest_package.yml/badge.svg
 [pytest-url]: https://github.com/camel-ai/camel/actions/workflows/pytest_package.yml
 [docs-image]: https://img.shields.io/badge/Documentation-grey.svg?logo=github

{camel_ai-0.1.6.7 → camel_ai-0.1.6.8}/camel/__init__.py

@@ -12,7 +12,7 @@
 # limitations under the License.
 # =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
 
-__version__ = '0.1.6.7'
+__version__ = '0.1.6.8'
 
 __all__ = [
     '__version__',

{camel_ai-0.1.6.7 → camel_ai-0.1.6.8}/camel/agents/chat_agent.py

@@ -14,6 +14,7 @@
 from __future__ import annotations
 
 import json
+import logging
 from collections import defaultdict
 from typing import (
     TYPE_CHECKING,
@@ -61,6 +62,9 @@ if TYPE_CHECKING:
     from camel.terminators import ResponseTerminator
     from camel.toolkits import OpenAIFunction
 
+
+logger = logging.getLogger(__name__)
+
 # AgentOps decorator setting
 try:
     import os
@@ -437,10 +441,22 @@ class ChatAgent(BaseAgent):
             for base_message_item in output_messages:
                 base_message_item.content = str(info['tool_calls'][-1].result)
 
-        return ChatAgentResponse(
+        chat_agent_response = ChatAgentResponse(
             msgs=output_messages, terminated=self.terminated, info=info
         )
 
+        # If the output result is single message, it will be
+        # automatically added to the memory.
+        if len(chat_agent_response.msgs) == 1:
+            self.record_message(chat_agent_response.msg)
+        else:
+            logger.warning(
+                "Multiple messages are available in `ChatAgentResponse`. "
+                "Please manually run the `record_message` function to "
+                "record the selected message."
+            )
+        return chat_agent_response
+
     async def step_async(
         self,
         input_message: BaseMessage,
@@ -563,10 +579,23 @@ class ChatAgent(BaseAgent):
             for base_message_item in output_messages:
                 base_message_item.content = str(info['tool_calls'][0].result)
 
-        return ChatAgentResponse(
+        chat_agent_response = ChatAgentResponse(
             msgs=output_messages, terminated=self.terminated, info=info
         )
 
+        # If the output result is single message, it will be
+        # automatically added to the memory.
+        if len(chat_agent_response.msgs) == 1:
+            self.record_message(chat_agent_response.msg)
+        else:
+            logger.warning(
+                "Multiple messages are presented in `chat_agent_response`. "
+                "Please manually call the `record_message` function to "
+                "record the chosen message."
+            )
+
+        return chat_agent_response
+
     def _add_tools_for_func_call(
         self,
         response: ChatCompletion,
@@ -736,7 +765,9 @@ class ChatAgent(BaseAgent):
             str(choice.finish_reason) for choice in response.choices
         ]
         usage = (
-            response.usage.model_dump() if response.usage is not None else {}
+            self._safe_model_dump(response.usage)
+            if response.usage is not None
+            else {}
         )
         return (
             output_messages,
@@ -745,6 +776,16 @@ class ChatAgent(BaseAgent):
             response.id,
         )
 
+    def _safe_model_dump(self, obj):
+        # Check if the `model_dump` method exists (Pydantic v2)
+        if hasattr(obj, 'model_dump'):
+            return obj.model_dump()
+        # Fallback to `dict()` method (Pydantic v1)
+        elif hasattr(obj, 'dict'):
+            return obj.dict()
+        else:
+            raise TypeError("The object is not a Pydantic model")
+
     def handle_stream_response(
         self,
         response: Stream[ChatCompletionChunk],

{camel_ai-0.1.6.7 → camel_ai-0.1.6.8}/camel/agents/critic_agent.py

@@ -123,7 +123,6 @@ class CriticAgent(ChatAgent):
                 raise RuntimeError("Critic step failed.")
 
             critic_msg = critic_response.msg
-            self.record_message(critic_msg)
             if self.verbose:
                 print_text_animated(
                     self.logger_color + "\n> Critic response: "

{camel_ai-0.1.6.7 → camel_ai-0.1.6.8}/camel/configs/__init__.py

@@ -19,6 +19,7 @@ from .litellm_config import LITELLM_API_PARAMS, LiteLLMConfig
 from .mistral_config import MISTRAL_API_PARAMS, MistralConfig
 from .ollama_config import OLLAMA_API_PARAMS, OllamaConfig
 from .openai_config import OPENAI_API_PARAMS, ChatGPTConfig, OpenSourceConfig
+from .reka_config import REKA_API_PARAMS, RekaConfig
 from .samba_config import SAMBA_API_PARAMS, SambaConfig
 from .togetherai_config import TOGETHERAI_API_PARAMS, TogetherAIConfig
 from .vllm_config import VLLM_API_PARAMS, VLLMConfig
@@ -45,6 +46,8 @@ __all__ = [
     'VLLM_API_PARAMS',
     'MistralConfig',
     'MISTRAL_API_PARAMS',
+    'RekaConfig',
+    'REKA_API_PARAMS',
     'SambaConfig',
     'SAMBA_API_PARAMS',
     'TogetherAIConfig',

camel_ai-0.1.6.8/camel/configs/reka_config.py

@@ -0,0 +1,74 @@
+# =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
+# Licensed under the Apache License, Version 2.0 (the “License”);
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an “AS IS” BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
+from __future__ import annotations
+
+from typing import Any, Optional, Union
+
+from camel.configs.base_config import BaseConfig
+
+
+class RekaConfig(BaseConfig):
+    r"""Defines the parameters for generating chat completions using the
+    Reka API.
+
+    Reference: https://docs.reka.ai/api-reference/chat/create
+
+    Args:
+        temperature (Optional[float], optional): temperature the temperature
+            to use for sampling, e.g. 0.5.
+        top_p (Optional[float], optional): the cumulative probability of
+            tokens to generate, e.g. 0.9. Defaults to None.
+        top_k (Optional[int], optional): Parameter which forces the model to
+            only consider the tokens with the `top_k` highest probabilities at
+            the next step. Defaults to 1024.
+        max_tokens (Optional[int], optional): the maximum number of tokens to
+            generate, e.g. 100. Defaults to None.
+        stop (Optional[Union[str,list[str]]]): Stop generation if this token
+            is detected. Or if one of these tokens is detected when providing
+            a string list.
+        seed (Optional[int], optional): the random seed to use for sampling, e.
+            g. 42. Defaults to None.
+        presence_penalty (float, optional): Number between :obj:`-2.0` and
+            :obj:`2.0`. Positive values penalize new tokens based on whether
+            they appear in the text so far, increasing the model's likelihood
+            to talk about new topics. See more information about frequency and
+            presence penalties. (default: :obj:`0.0`)
+        frequency_penalty (float, optional): Number between :obj:`-2.0` and
+            :obj:`2.0`. Positive values penalize new tokens based on their
+            existing frequency in the text so far, decreasing the model's
+            likelihood to repeat the same line verbatim. See more information
+            about frequency and presence penalties. (default: :obj:`0.0`)
+        use_search_engine (Optional[bool]): Whether to consider using search
+            engine to complete the request. Note that even if this is set to
+            `True`, the model might decide to not use search.
+    """
+
+    temperature: Optional[float] = None
+    top_p: Optional[float] = None
+    top_k: Optional[int] = None
+    max_tokens: Optional[int] = None
+    stop: Optional[Union[str, list[str]]] = None
+    seed: Optional[int] = None
+    frequency_penalty: float = 0.0
+    presence_penalty: float = 0.0
+    use_search_engine: Optional[bool] = False
+
+    def as_dict(self) -> dict[str, Any]:
+        config_dict = super().as_dict()
+        if "tools" in config_dict:
+            del config_dict["tools"]  # Reka does not support tool calling
+        return config_dict
+
+
+REKA_API_PARAMS = {param for param in RekaConfig().model_fields.keys()}

{camel_ai-0.1.6.7 → camel_ai-0.1.6.8}/camel/models/__init__.py

@@ -25,6 +25,7 @@ from .open_source_model import OpenSourceModel
 from .openai_audio_models import OpenAIAudioModels
 from .openai_compatibility_model import OpenAICompatibilityModel
 from .openai_model import OpenAIModel
+from .reka_model import RekaModel
 from .samba_model import SambaModel
 from .stub_model import StubModel
 from .togetherai_model import TogetherAIModel
@@ -49,6 +50,7 @@ __all__ = [
     'VLLMModel',
     'GeminiModel',
     'OpenAICompatibilityModel',
+    'RekaModel',
     'SambaModel',
     'TogetherAIModel',
 ]

{camel_ai-0.1.6.7 → camel_ai-0.1.6.8}/camel/models/model_factory.py

@@ -24,6 +24,7 @@ from camel.models.ollama_model import OllamaModel
 from camel.models.open_source_model import OpenSourceModel
 from camel.models.openai_compatibility_model import OpenAICompatibilityModel
 from camel.models.openai_model import OpenAIModel
+from camel.models.reka_model import RekaModel
 from camel.models.samba_model import SambaModel
 from camel.models.stub_model import StubModel
 from camel.models.togetherai_model import TogetherAIModel
@@ -93,6 +94,8 @@ class ModelFactory:
                 model_class = GeminiModel
             elif model_platform.is_mistral and model_type.is_mistral:
                 model_class = MistralModel
+            elif model_platform.is_reka and model_type.is_reka:
+                model_class = RekaModel
             elif model_platform.is_samba and model_type.is_samba:
                 model_class = SambaModel
             elif model_type == ModelType.STUB:

camel_ai-0.1.6.8/camel/models/reka_model.py

@@ -0,0 +1,232 @@
+# =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
+# Licensed under the Apache License, Version 2.0 (the “License”);
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an “AS IS” BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
+import os
+from typing import TYPE_CHECKING, Any, Dict, List, Optional
+
+from camel.configs import REKA_API_PARAMS
+from camel.messages import OpenAIMessage
+from camel.models import BaseModelBackend
+from camel.types import ChatCompletion, ModelType
+from camel.utils import (
+    BaseTokenCounter,
+    OpenAITokenCounter,
+    api_keys_required,
+)
+
+if TYPE_CHECKING:
+    from reka.types import ChatMessage, ChatResponse
+
+try:
+    import os
+
+    if os.getenv("AGENTOPS_API_KEY") is not None:
+        from agentops import LLMEvent, record
+    else:
+        raise ImportError
+except (ImportError, AttributeError):
+    LLMEvent = None
+
+
+class RekaModel(BaseModelBackend):
+    r"""Reka API in a unified BaseModelBackend interface."""
+
+    def __init__(
+        self,
+        model_type: ModelType,
+        model_config_dict: Dict[str, Any],
+        api_key: Optional[str] = None,
+        url: Optional[str] = None,
+        token_counter: Optional[BaseTokenCounter] = None,
+    ) -> None:
+        r"""Constructor for Reka backend.
+
+        Args:
+            model_type (ModelType): Model for which a backend is created,
+                one of REKA_* series.
+            model_config_dict (Dict[str, Any]): A dictionary that will
+                be fed into `Reka.chat.create`.
+            api_key (Optional[str]): The API key for authenticating with the
+                Reka service. (default: :obj:`None`)
+            url (Optional[str]): The url to the Reka service.
+            token_counter (Optional[BaseTokenCounter]): Token counter to use
+                for the model. If not provided, `OpenAITokenCounter` will be
+                used.
+        """
+        super().__init__(
+            model_type, model_config_dict, api_key, url, token_counter
+        )
+        self._api_key = api_key or os.environ.get("REKA_API_KEY")
+        self._url = url or os.environ.get("REKA_SERVER_URL")
+
+        from reka.client import Reka
+
+        self._client = Reka(api_key=self._api_key, base_url=self._url)
+        self._token_counter: Optional[BaseTokenCounter] = None
+
+    def _convert_reka_to_openai_response(
+        self, response: 'ChatResponse'
+    ) -> ChatCompletion:
+        r"""Converts a Reka `ChatResponse` to an OpenAI-style `ChatCompletion`
+        response.
+
+        Args:
+            response (ChatResponse): The response object from the Reka API.
+
+        Returns:
+            ChatCompletion: An OpenAI-compatible chat completion response.
+        """
+        openai_response = ChatCompletion.construct(
+            id=response.id,
+            choices=[
+                dict(
+                    message={
+                        "role": response.responses[0].message.role,
+                        "content": response.responses[0].message.content,
+                    },
+                    finish_reason=response.responses[0].finish_reason
+                    if response.responses[0].finish_reason
+                    else None,
+                )
+            ],
+            created=None,
+            model=response.model,
+            object="chat.completion",
+            usage=response.usage,
+        )
+
+        return openai_response
+
+    def _convert_openai_to_reka_messages(
+        self,
+        messages: List[OpenAIMessage],
+    ) -> List["ChatMessage"]:
+        r"""Converts OpenAI API messages to Reka API messages.
+
+        Args:
+            messages (List[OpenAIMessage]): A list of messages in OpenAI
+                format.
+
+        Returns:
+            List[ChatMessage]: A list of messages converted to Reka's format.
+        """
+        from reka.types import ChatMessage
+
+        reka_messages = []
+        for msg in messages:
+            role = msg.get("role")
+            content = str(msg.get("content"))
+
+            if role == "user":
+                reka_messages.append(ChatMessage(role="user", content=content))
+            elif role == "assistant":
+                reka_messages.append(
+                    ChatMessage(role="assistant", content=content)
+                )
+            elif role == "system":
+                reka_messages.append(ChatMessage(role="user", content=content))
+
+                # Add one more assistant msg since Reka requires conversation
+                # history must alternate between 'user' and 'assistant',
+                # starting and ending with 'user'.
+                reka_messages.append(
+                    ChatMessage(
+                        role="assistant",
+                        content="",
+                    )
+                )
+            else:
+                raise ValueError(f"Unsupported message role: {role}")
+
+        return reka_messages
+
+    @property
+    def token_counter(self) -> BaseTokenCounter:
+        r"""Initialize the token counter for the model backend.
+
+        # NOTE: Temporarily using `OpenAITokenCounter`
+
+        Returns:
+            BaseTokenCounter: The token counter following the model's
+                tokenization style.
+        """
+        if not self._token_counter:
+            self._token_counter = OpenAITokenCounter(
+                model=ModelType.GPT_4O_MINI
+            )
+        return self._token_counter
+
+    @api_keys_required("REKA_API_KEY")
+    def run(
+        self,
+        messages: List[OpenAIMessage],
+    ) -> ChatCompletion:
+        r"""Runs inference of Mistral chat completion.
+
+        Args:
+            messages (List[OpenAIMessage]): Message list with the chat history
+                in OpenAI API format.
+
+        Returns:
+            ChatCompletion.
+        """
+        reka_messages = self._convert_openai_to_reka_messages(messages)
+
+        response = self._client.chat.create(
+            messages=reka_messages,
+            model=self.model_type.value,
+            **self.model_config_dict,
+        )
+
+        openai_response = self._convert_reka_to_openai_response(response)
+
+        # Add AgentOps LLM Event tracking
+        if LLMEvent:
+            llm_event = LLMEvent(
+                thread_id=openai_response.id,
+                prompt=" ".join(
+                    [message.get("content") for message in messages]  # type: ignore[misc]
+                ),
+                prompt_tokens=openai_response.usage.input_tokens,  # type: ignore[union-attr]
+                completion=openai_response.choices[0].message.content,
+                completion_tokens=openai_response.usage.output_tokens,  # type: ignore[union-attr]
+                model=self.model_type.value,
+            )
+            record(llm_event)
+
+        return openai_response
+
+    def check_model_config(self):
+        r"""Check whether the model configuration contains any
+        unexpected arguments to Reka API.
+
+        Raises:
+            ValueError: If the model configuration dictionary contains any
+                unexpected arguments to Reka API.
+        """
+        for param in self.model_config_dict:
+            if param not in REKA_API_PARAMS:
+                raise ValueError(
+                    f"Unexpected argument `{param}` is "
+                    "input into Reka model backend."
+                )
+
+    @property
+    def stream(self) -> bool:
+        r"""Returns whether the model is in stream mode, which sends partial
+        results each time.
+
+        Returns:
+            bool: Whether the model is in stream mode.
+        """
+        return self.model_config_dict.get('stream', False)

{camel_ai-0.1.6.7 → camel_ai-0.1.6.8}/camel/societies/babyagi_playing.py

@@ -243,9 +243,6 @@ class BabyAGI:
 
         assistant_response = self.assistant_agent.step(assistant_msg_msg)
         assistant_msg = assistant_response.msgs[0]
-        self.assistant_agent.record_message(assistant_msg)
-        self.task_creation_agent.record_message(assistant_msg)
-        self.task_prioritization_agent.record_message(assistant_msg)
 
         self.solved_subtasks.append(task_name)
         past_tasks = self.solved_subtasks + list(self.subtasks)

{camel_ai-0.1.6.7 → camel_ai-0.1.6.8}/camel/societies/role_playing.py

@@ -486,7 +486,15 @@ class RolePlaying:
                 ),
             )
         user_msg = self._reduce_message_options(user_response.msgs)
-        self.user_agent.record_message(user_msg)
+
+        # To prevent recording the same memory more than once (once in chat
+        # step and once in role play), and the model generates only one
+        # response when multi-response support is enabled.
+        if (
+            'n' in self.user_agent.model_config_dict.keys()
+            and self.user_agent.model_config_dict['n'] > 1
+        ):
+            self.user_agent.record_message(user_msg)
 
         assistant_response = self.assistant_agent.step(user_msg)
         if assistant_response.terminated or assistant_response.msgs is None:
@@ -501,7 +509,15 @@ class RolePlaying:
                 ),
             )
         assistant_msg = self._reduce_message_options(assistant_response.msgs)
-        self.assistant_agent.record_message(assistant_msg)
+
+        # To prevent recording the same memory more than once (once in chat
+        # step and once in role play), and the model generates only one
+        # response when multi-response support is enabled.
+        if (
+            'n' in self.assistant_agent.model_config_dict.keys()
+            and self.assistant_agent.model_config_dict['n'] > 1
+        ):
+            self.assistant_agent.record_message(assistant_msg)
 
         return (
             ChatAgentResponse(