camel-ai 0.2.0__py3-none-any.whl → 0.2.3a0__py3-none-any.whl

{camel_ai-0.2.0.dist-info → camel_ai-0.2.3a0.dist-info}/RECORD

@@ -1,7 +1,7 @@
-camel/__init__.py,sha256=zzLXt4OmIHyizUAU7n1idW4aN99Vcbjpo9JHPUUprko,778
+camel/__init__.py,sha256=lQSKaE7GqJm4K6yZtPgrzu5SimTTJ16AVlL93Pk2DuI,778
 camel/agents/__init__.py,sha256=SSU1wbhZXWwQnE0rRxkpyN57kEu72KklsZNcdLkXfTs,1551
 camel/agents/base.py,sha256=X39qWSiT1WnDqaJ9k3gQrTpOQSwUKzNEVpp5AY6fDH8,1130
-camel/agents/chat_agent.py,sha256=cIAGXDTbZVrVsq8UsZRv8Et9eAx-lFIT73DsYTVopo0,36276
+camel/agents/chat_agent.py,sha256=KwbKYfNzYSgNDHeUeBWOFUTlqVn8P8h2eHTVpeBNK8Q,43735
 camel/agents/critic_agent.py,sha256=To-istnO-9Eb0iabdeIDrgfvkxYYfsdX9xIZiSrc3oM,7493
 camel/agents/deductive_reasoner_agent.py,sha256=49vwglWYHgXf-VRftdMN9OFGOwqdsXyTt45PP6z-pbg,13473
 camel/agents/embodied_agent.py,sha256=3ABuiRQXBpplKbuhPY5KNLJyKc6Z8SgXgzIges3ZwVs,7542
@@ -12,10 +12,13 @@ camel/agents/task_agent.py,sha256=n9xIU3QtcptRPSuHZJ4ntQ_M_a8AvJ6U9ZRV8VaxV5A,14
 camel/agents/tool_agents/__init__.py,sha256=ulTNWU2qoFGe3pvVmCq_sdfeSX3NKZ0due66TYvsL-M,862
 camel/agents/tool_agents/base.py,sha256=nQAhfWi8a_bCgzlf5-G-tmj1fKm6AjpRc89NQkWwpnc,1399
 camel/agents/tool_agents/hugging_face_tool_agent.py,sha256=1Z5tG6f_86eL0vmtRZ-BJvoLDFFLhoHt8JtDvgat1xU,8723
+camel/bots/__init__.py,sha256=DGgIhb7Gr_ShnAldq-4wAXucOdSvKtiV3m6e9G10rbA,834
+camel/bots/discord_bot.py,sha256=VplQA162NqSohMzpCexSsuCCx_GdhAXOnL3040Ba1uE,8138
+camel/bots/telegram_bot.py,sha256=MM91afWKUtEZLOoybFY7LGisQJFypc_p_P8kQOUlXwg,2747
 camel/configs/__init__.py,sha256=3SCXrA7ML8CNoOWQ1poJY6dE2e07Qxf7ACD_mdo2aJ8,2341
 camel/configs/anthropic_config.py,sha256=DGQoPyYrayhYQ7aSjkYYGHOZ5VdQ9qahtaS0p_GpU0Q,3294
 camel/configs/base_config.py,sha256=gjsDACMCk-hXDBk7qkeHcpbQrWy6jbp4iyzfqgghJEk,2485
-camel/configs/gemini_config.py,sha256=YHJSNEAIxBxPX1NAj2rWvM4OOR7vmIANH88pZO-aOsY,6880
+camel/configs/gemini_config.py,sha256=qhGdWMRP87E99ApQdRrf9UyFpp2R0k6Y2n9ZV27qpzs,6879
 camel/configs/groq_config.py,sha256=-ihiO5h4N_kUSlPNeBqIlnIkLkhC7oXsC2FYyAqtKp0,5755
 camel/configs/litellm_config.py,sha256=77k7HT-0s9Sq_g4KeDjL_MZId0Tx5TB8oupIyGQHx08,4692
 camel/configs/mistral_config.py,sha256=G9LuY0-3S6az-8j8kpqB-4asgoaxTOsZVYeZBYJl6LI,3634
@@ -43,9 +46,9 @@ camel/interpreters/ipython_interpreter.py,sha256=B0v1DCiq6PmOYQXXQAQBX1oOYjgJ0ge
 camel/interpreters/subprocess_interpreter.py,sha256=nKxFXZJ9zGYlKdNlz6Ln7bvg65ejKZ8yAHgIFuR2WzM,6835
 camel/loaders/__init__.py,sha256=ClE516UbyUes6Zut8CCiH0zWqFwwpgEcP9ur3dte8iU,949
 camel/loaders/base_io.py,sha256=xzK67fqx66eYaM6fMXRJiSZfwKhFVNQmzuKURPtTzhk,10339
-camel/loaders/firecrawl_reader.py,sha256=6CfuSA3LrjnDhApoPnZfmTHbauN2TmseBTWpx0sutzo,7943
+camel/loaders/firecrawl_reader.py,sha256=OmnT4XdoL-naTg_nvUnN_sjiP6Xf4gdDxFUOqyQbKJo,6691
 camel/loaders/jina_url_reader.py,sha256=ur_2Z3NFrz5AbPFi5REyZh5fISkJ9H_UZV4gtmOSO04,3607
-camel/loaders/unstructured_io.py,sha256=cY-5mqViE5Sobr3dnbK-KYcXjalUfeDfb8Yk6yOsu_g,16330
+camel/loaders/unstructured_io.py,sha256=c2-vBhgVAWD3bJ3Gqo4nOcM5-xGc8CIMXogkc5Li9dw,17502
 camel/memories/__init__.py,sha256=ml1Uj4Y_1Q2LfrTXOY38niF0x1H-N-u_zoN_VvR939U,1364
 camel/memories/agent_memories.py,sha256=pzmjztFXPyNpabMWLi_-oJljMkYQvDP_s6Yq5U0hVEs,6097
 camel/memories/base.py,sha256=kbyAmKkOfFdOKfHxwao8bIAbRSuOEXyzxPFd0NlvUCE,5003
@@ -57,7 +60,7 @@ camel/memories/context_creators/score_based.py,sha256=UrioLcsj4jB4Rz5mg5hYmR8UYZ
 camel/memories/records.py,sha256=kcXOATDTRRo-SCAcDpsV8Ttfie7p1GcXYzuXgXKJB0E,3686
 camel/messages/__init__.py,sha256=djLvpz6AmjeLzuUSQl7J6T2O4x8MwSdcH0l9fbj_3yg,1468
 camel/messages/base.py,sha256=694Zz19D4u-j8mmpRXwCVJ8cd2Wll6h7acbyNRofNTI,13722
-camel/messages/func_message.py,sha256=CCVkbz-2pdxXV0vBETI0xt7d7uiN8zACpRI7lCnfTFQ,3841
+camel/messages/func_message.py,sha256=wqVhTC81g5Fb2b_2EPsOBP3rHZBTP6o9wMiIRkN_HAE,3807
 camel/models/__init__.py,sha256=CxrRisXkHgJPv7HdC6pblHuhSpDjUbUjrTdZeZ-0v7w,1971
 camel/models/anthropic_model.py,sha256=_xhnbrMsuumB2jkuv2pVv3MFYxNE5EL5kVlZbYYBo5E,5751
 camel/models/azure_openai_model.py,sha256=r5diPZp4XmCcZClkCqvTHB8frzRNou559j89dryKLKw,6078
@@ -65,13 +68,13 @@ camel/models/base_model.py,sha256=UHyAgo6GzYZNLTZD1T0C3_WmHUPoh9Qoe_SfvdI7HrU,43
 camel/models/gemini_model.py,sha256=h_kyD8LSpXCn2dQ4OEer5HwwEUwuTD65yRIRV4LD3Vs,7700
 camel/models/groq_model.py,sha256=Lm1br_2FBdqNQ3pCgMNf3VnjykYzttUKnHWExEXshLo,4753
 camel/models/litellm_model.py,sha256=5sTOzI07FsxDEW3jSK-XXBx91Yo8z9voahyCsK36U6U,5748
-camel/models/mistral_model.py,sha256=39rHJ-z_6Z-UbtUqZPEAbCdFYY1Ft0Drs42jG5hHaho,9517
+camel/models/mistral_model.py,sha256=7K0hwk4QEGtCO_Btq9v7_iLQRgg6COpdgvXxRB8rd3U,9517
 camel/models/model_factory.py,sha256=_dQOx_MYxXih6uQOjkKR7uoIrhDcWRsMTKHbO3NLYN0,5974
 camel/models/nemotron_model.py,sha256=2Idf4wrZervxvfu6av42EKjefFtDnBb6cKnWCJUkqI4,2682
 camel/models/ollama_model.py,sha256=FSMwH2-856Zhxusm4B773xBBHdlD3UOw9OAuH5eTJTw,5686
 camel/models/open_source_model.py,sha256=p5a2sCeZl5SyrgkygClndOrHEjpJxmyhE1CqKE2fZSw,6363
 camel/models/openai_audio_models.py,sha256=_ddOxqzFZCVZaK6h33Z0THU6HXk2XlJTxVWquZ3oOaQ,10042
-camel/models/openai_compatibility_model.py,sha256=7h1zSFBgg_mQojFvtSqC54tcZOZY0NFsZ7ZNlns5CWk,4229
+camel/models/openai_compatibility_model.py,sha256=jeYBqB3xjV8jx8CJGFShkISO_TrgwubCU1WfmSiDhBY,4758
 camel/models/openai_model.py,sha256=27NbN0bU_cAPmWjwsWceNO4zW8WS2j3P_YWNTXeg1O8,5464
 camel/models/reka_model.py,sha256=_ERZvtkK0Gd7GUx3f4VVqqtH093clVMoJfa896t9f2M,8043
 camel/models/samba_model.py,sha256=CgAYMIVJFAEoyCOsYS7qD_bvWhzOkvA6SD5nGBClbzE,17699
@@ -102,14 +105,15 @@ camel/retrievers/auto_retriever.py,sha256=XPyECIXO7lFbHoLMPasFurYmC9Me6wa7JPKHZW
 camel/retrievers/base.py,sha256=sgqaJDwIkWluEgPBlukFN7RYZJnrp0imCAOEWm6bZ40,2646
 camel/retrievers/bm25_retriever.py,sha256=Dr7Yfkjw45sovI1EVNByGIMj7KERWrr8JHlh8csSF1s,5155
 camel/retrievers/cohere_rerank_retriever.py,sha256=HvnFqXpsX9EdBOab0kFLDyxxJnknPFMVxyQJQDlHbOA,4100
-camel/retrievers/vector_retriever.py,sha256=4tzqsUIW29Pq8uvQw_2iXhcJyTmIhvq7CinhCrr7Sps,8256
+camel/retrievers/vector_retriever.py,sha256=zVsxWrv7-69GBRzC90Bt_JuMy8rQLsUYXXn5glZgdgo,9019
 camel/societies/__init__.py,sha256=JhGwUHjht4CewzC3shKuxmgB3oS7FIxIxmiKyhNsfIs,832
 camel/societies/babyagi_playing.py,sha256=tZTcfQrD9ECUhkqwdthsM2yFPfYGKsoAGMfTlrxt5as,11675
-camel/societies/role_playing.py,sha256=ZiiT1RtS3cQtV3XqyAXOH_Up7-b7WRDznC2qiX4PDGg,22982
-camel/storages/__init__.py,sha256=ghlDZ1cF3O_QxwX9xIIZ__bnHcUjh7dbAF1X_ivkofc,1551
-camel/storages/graph_storages/__init__.py,sha256=vsJZkedaCS-cLQ-KgMqio8cxXvbousBWVqzZJvlimT8,897
+camel/societies/role_playing.py,sha256=tu6AyUPEe0j3rGNwzCCTiGqHw9RWdyTStB5JbPNvBs8,23403
+camel/storages/__init__.py,sha256=erzsXX2rPVodgWxlV6hSjwmc4UCguzVb6m3ahBoyWu0,1623
+camel/storages/graph_storages/__init__.py,sha256=YFwNjNYaxpKT2j5dxWZur62T8J0ZPER3DHaWGEAXgo8,954
 camel/storages/graph_storages/base.py,sha256=-Ys1BIuz4H5FvYMZTBIjg8Cfv40CPQ-OsovwMzygEgU,2858
 camel/storages/graph_storages/graph_element.py,sha256=FGYJZpS3UeSd9yyzz5KelMqbpJ6m5177PspZ8ueoQQI,2580
+camel/storages/graph_storages/nebula_graph.py,sha256=PFLI--kgC93Nt5kPFWH2DVqwvk2GrC--dr2bzwvmeXw,18863
 camel/storages/graph_storages/neo4j_graph.py,sha256=YAT7u2jPs5pNSO_tSwARHJyt3HqYXpbpE4ZBtv935Kg,22138
 camel/storages/key_value_storages/__init__.py,sha256=v3Wy3CAJNgrPyBV4miOC6TxQDL-PYdGW8HbqiYl7k00,968
 camel/storages/key_value_storages/base.py,sha256=knxni8WiyTXJ2emZQO-JIsbxw6Ei7EO6dj-bU2YCoSY,2183
@@ -126,8 +130,8 @@ camel/storages/vectordb_storages/base.py,sha256=XNRv6CFg1rb9G_hZhUmN4t9XCU9rCakH
 camel/storages/vectordb_storages/milvus.py,sha256=EkwaG3bq7djektSI2jdSTccE7GjBpZruqXQoPTxtyoo,13489
 camel/storages/vectordb_storages/qdrant.py,sha256=lG_tOdjfTybSUNluQ1TEqYX96dSMrVSPg6HjO6Uzw7I,13439
 camel/tasks/__init__.py,sha256=g8_UAa4WmhaesYtzAUMW7m3mg-3ILEUvTaB7eOU9-3I,912
-camel/tasks/task.py,sha256=1klroUleHhnEJmWqnFHILqMryVBqL0bUXyvzoabuga4,12614
-camel/tasks/task_prompt.py,sha256=eDmjB0vv2InmPjKAMzy8wU5PzTPy3WFk5F7zKW9d_ZQ,2093
+camel/tasks/task.py,sha256=ptVnC1cbY6einAPf7_p8AV5d04IcTsvv7gay_qmKJ48,12817
+camel/tasks/task_prompt.py,sha256=iQRzCe6YoOH29ktjp5iOsE_qbWCNljV0HyaWmNsHH2s,2156
 camel/terminators/__init__.py,sha256=pE7fcfDUNngdbm1BhzSQPRMXNbdd28rl9YbF4gKWwXE,997
 camel/terminators/base.py,sha256=TSkl3maNEsdjyAniJaSgFfD4UF8RQ1LwNIiGw0dN8Gg,1396
 camel/terminators/response_terminator.py,sha256=zcXuigbvlclUoBv4xcVbfU36ZohUT1RhI-rSnukloUY,4951
@@ -174,23 +178,22 @@ camel/toolkits/slack_toolkit.py,sha256=gblCbN_RCsOdgo1GGUF-R8YJneNRjezJMHhYoRFjC
 camel/toolkits/twitter_toolkit.py,sha256=0QYLlsg4hUVev2Z0hPJHksDNHG_54IiVHsB3IxCqrXs,19741
 camel/toolkits/weather_toolkit.py,sha256=n4YrUI_jTIH7oqH918IdHbXLgfQ2BPGIWWK8Jp8G1Uw,7054
 camel/types/__init__.py,sha256=ArKXATj3z_Vv4ISmROVeo6Mv3tj5kE1dTkqfgwyxVY4,1975
-camel/types/enums.py,sha256=APqZIKtR2tVbUe-1JZWi7a50gjjDC7QHw9A0eKJuefA,17642
+camel/types/enums.py,sha256=OfxiXL7VRcIQgfEEu0gdFeHPWGQpiuXHOC86jJE47DA,17773
 camel/types/openai_types.py,sha256=BNQ6iCzKTjSvgcXFsAFIgrUS_YUFZBU6bDoyAp387hI,2045
 camel/utils/__init__.py,sha256=IdI9v0FetNR-nx-Hg4bmNHoYto6Xfcs_uaomksdewmo,2303
 camel/utils/async_func.py,sha256=SLo8KPkrNKdsONvFf3KBb33EgFn4gH2EKSX1aI_LKes,1578
-camel/utils/commons.py,sha256=y7eng5QF5Hkt5tuNhtEOJycTIq9hXymrUuwIS5nRad4,16481
+camel/utils/commons.py,sha256=kJ6-9pnt6HgERZh3DXofMUzT0FvDujvybuZG4rzjlso,16611
 camel/utils/constants.py,sha256=8n4F8Y-DZy4z2F0hRvAq6f-d9SbS59kK5FyLrnJ3mkY,1360
 camel/utils/token_counting.py,sha256=AVGml8X_qq3rPdzw2tc9I3n-oEkGyNH_vPsedhVtew0,21318
-camel/workforce/__init__.py,sha256=6jwJWDlESEqcnWCm61WCyjzFUF6KLzXA_fGI86rHfiE,878
-camel/workforce/base.py,sha256=lEHqgOV1tmsy7y4wuuKClcDkoPCRvXVdMrBngsM_6yY,1722
-camel/workforce/manager_node.py,sha256=eMmsOAoy0Wtk92b_06GhGnwKDgrTo0w-UgQorkh-az0,11529
-camel/workforce/role_playing_node.py,sha256=OrHI0WbN8-jW7UPnl0hzmRkNSjQERhvc4QaoNSl_3Iw,7008
-camel/workforce/single_agent_node.py,sha256=AzbsAX58q_2VpVN1iArauLKu76VwpTMqvlyrHTyDWjw,2928
-camel/workforce/task_channel.py,sha256=jYwhF3Ppc7839Kv0kWXDXkWtjRwPvoP_uF0zTwC_-jg,6752
-camel/workforce/utils.py,sha256=Z-kODz5PMPtfeKKVqpcQq-b-B8oqC7XSwi_F3__Ijhs,3526
-camel/workforce/worker_node.py,sha256=wsRqk2rugCvvkcmCzvn-y-gQuyuJGAG8PIr1KtgqJFw,3878
-camel/workforce/workforce.py,sha256=SVJJgSSkYvk05RgL9oaJzHwzziH7u51KLINRuzLB8BI,1773
-camel/workforce/workforce_prompt.py,sha256=cAWYEIA0rau5itEekSoUIFttBzpKM9RzB6x-mfukGSU,4665
-camel_ai-0.2.0.dist-info/METADATA,sha256=6mTZQestY-Rvn1pkyLO2-DycdgKFKy6YO7yDf37Vmq8,24677
-camel_ai-0.2.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-camel_ai-0.2.0.dist-info/RECORD,,
+camel/workforce/__init__.py,sha256=m2KbPItARjIyKiHc_Z34joWTJ-1XnIJfQqAZEeonz2U,926
+camel/workforce/base.py,sha256=ScUPdtMLJgvvJkGEku6rrr4gBRw3GFt3V8HlYOY5Y8A,1809
+camel/workforce/prompts.py,sha256=0Wcu7HpiyaD_387urc2P1uY9-DN8JAdk00HSlhNWz34,6184
+camel/workforce/role_playing_worker.py,sha256=pKKT6MdVCPeag5s45yOzjT4Gn1u_aTbERSVw53ISBUY,7107
+camel/workforce/single_agent_worker.py,sha256=DsibU4VkdFWuzi6mhMnT-YYm3FcVGcKfF7r1WtGii3E,3561
+camel/workforce/task_channel.py,sha256=oQJ-qzXWBqZljwV3EdZ7Kpo7RExZfI_tqg0bop14clM,6741
+camel/workforce/utils.py,sha256=kBrjA_k9BblJBT13kW0eQYcBDjetyjOUHMbRuh_IfPM,2270
+camel/workforce/worker.py,sha256=oZtl3PhxgpizKy7W_wB94Qm4Z2Tw4k1SQsqAGAi_XoI,3844
+camel/workforce/workforce.py,sha256=lF0ZeGtol91N3kAlAFwDxnO6YYVKmZPohxNb5ezHEUc,18185
+camel_ai-0.2.3a0.dist-info/METADATA,sha256=usnLO_QSIxLsoD6iyL09pp0ED-Z5foxzcpvclWoKWpo,25040
+camel_ai-0.2.3a0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+camel_ai-0.2.3a0.dist-info/RECORD,,

camel/workforce/manager_node.py

@@ -1,299 +0,0 @@
-# =========== 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
-
-import asyncio
-from collections import deque
-from typing import Deque, Dict, List, Optional
-
-from colorama import Fore
-
-from camel.agents import ChatAgent
-from camel.messages.base import BaseMessage
-from camel.tasks.task import Task, TaskState
-from camel.workforce.base import BaseNode
-from camel.workforce.single_agent_node import SingleAgentNode
-from camel.workforce.task_channel import TaskChannel
-from camel.workforce.utils import (
-    check_if_running,
-    parse_assign_task_resp,
-    parse_create_node_resp,
-)
-from camel.workforce.worker_node import WorkerNode
-from camel.workforce.workforce_prompt import (
-    ASSIGN_TASK_PROMPT,
-    CREATE_NODE_PROMPT,
-)
-
-
-class ManagerNode(BaseNode):
-    r"""A node that manages multiple nodes. It will split the task it
-    receives into subtasks and assign them to the child nodes under
-    it, and also handles the situation when the task fails.
-
-    Args:
-        description (str): Description of the node.
-        coordinator_agent_kwargs (Optional[Dict]): Keyword arguments for the
-            coordinator agent, e.g. `model`, `api_key`, `tools`, etc.
-        task_agent_kwargs (Optional[Dict]): Keyword arguments for the task
-            agent, e.g. `model`, `api_key`, `tools`, etc.
-    """
-
-    def __init__(
-        self,
-        description: str,
-        children: List[BaseNode],
-        coordinator_agent_kwargs: Optional[Dict] = None,
-        task_agent_kwargs: Optional[Dict] = None,
-    ) -> None:
-        super().__init__(description)
-        self._child_listening_tasks: Deque[asyncio.Task] = deque()
-        self._children = children
-
-        coord_agent_sysmsg = BaseMessage.make_assistant_message(
-            role_name="Workforce Manager",
-            content="You are coordinating a group of workers. A worker can be "
-            "a group of agents or a single agent. Each worker is created to"
-            " solve a specific kind of task. Your job includes assigning "
-            "tasks to a existing worker, creating a new worker for a task, "
-            "etc.",
-        )
-        self.coordinator_agent = ChatAgent(
-            coord_agent_sysmsg, **(coordinator_agent_kwargs or {})
-        )
-
-        task_sys_msg = BaseMessage.make_assistant_message(
-            role_name="Task Planner",
-            content="You are going to compose and decompose tasks.",
-        )
-        self.task_agent = ChatAgent(task_sys_msg, **(task_agent_kwargs or {}))
-
-        # if there is one, will set by the workforce class wrapping this
-        self._task: Optional[Task] = None
-        self._pending_tasks: Deque[Task] = deque()
-
-    @check_if_running(False)
-    def set_main_task(self, task: Task) -> None:
-        r"""Set the main task for the node."""
-        self._task = task
-
-    def _get_child_nodes_info(self) -> str:
-        r"""Get the information of all the child nodes under this node."""
-        return '\n'.join(
-            f'{child.node_id}: {child.description}' for child in self._children
-        )
-
-    def _find_assignee(
-        self,
-        task: Task,
-        failed_log: Optional[str] = None,
-    ) -> str:
-        r"""Assigns a task to a child node if capable, otherwise create a
-        new worker node.
-
-        Parameters:
-            task (Task): The task to be assigned.
-            failed_log (Optional[str]): Optional log of a previous failed
-                attempt.
-
-        Returns:
-            str: ID of the assigned node.
-        """
-        prompt = ASSIGN_TASK_PROMPT.format(
-            content=task.content,
-            child_nodes_info=self._get_child_nodes_info(),
-        )
-        req = BaseMessage.make_user_message(
-            role_name="User",
-            content=prompt,
-        )
-        response = self.coordinator_agent.step(req)
-        try:
-            print(f"{Fore.YELLOW}{response.msg.content}{Fore.RESET}")
-            assignee_id = parse_assign_task_resp(response.msg.content)
-        except ValueError:
-            assignee_id = self._create_worker_node_for_task(task).node_id
-        return assignee_id
-
-    async def _post_task(self, task: Task, assignee_id: str) -> None:
-        await self._channel.post_task(task, self.node_id, assignee_id)
-
-    async def _post_dependency(self, dependency: Task) -> None:
-        await self._channel.post_dependency(dependency, self.node_id)
-
-    def _create_worker_node_for_task(self, task: Task) -> WorkerNode:
-        r"""Creates a new worker node for a given task and add it to the
-        children list of this node. This is one of the actions that
-        the coordinator can take when a task has failed.
-
-        Args:
-            task (Task): The task for which the worker node is created.
-
-        Returns:
-            WorkerNode: The created worker node.
-        """
-        prompt = CREATE_NODE_PROMPT.format(
-            content=task.content,
-            child_nodes_info=self._get_child_nodes_info(),
-        )
-        req = BaseMessage.make_user_message(
-            role_name="User",
-            content=prompt,
-        )
-        response = self.coordinator_agent.step(req)
-        new_node_conf = parse_create_node_resp(response.msg.content)
-
-        worker_sysmsg = BaseMessage.make_assistant_message(
-            role_name=new_node_conf.role,
-            content=new_node_conf.system,
-        )
-
-        # TODO: add a default selection of tools for the worker
-        worker = ChatAgent(worker_sysmsg)
-
-        new_node = SingleAgentNode(
-            description=new_node_conf.description,
-            worker=worker,
-        )
-        new_node.set_channel(self._channel)
-
-        print(
-            f"{Fore.GREEN}New worker node {new_node.node_id} created."
-            f"{Fore.RESET}"
-        )
-
-        self._children.append(new_node)
-        self._child_listening_tasks.append(
-            asyncio.create_task(new_node.start())
-        )
-        return new_node
-
-    async def _get_returned_task(self) -> Task:
-        r"""Get the task that's published by this node and just get returned
-        from the assignee.
-        """
-        return await self._channel.get_returned_task_by_publisher(self.node_id)
-
-    async def _post_ready_tasks(self) -> None:
-        r"""Send all the pending tasks that have all the dependencies met to
-        the channel, or directly return if there is none. For now, we will
-        directly send the first task in the pending list because all the tasks
-        are linearly dependent."""
-
-        if not self._pending_tasks:
-            return
-
-        ready_task = self._pending_tasks[0]
-
-        # if the task has failed previously, just compose and send the task
-        # to the channel as a dependency
-        if ready_task.state == TaskState.FAILED:
-            # TODO: the composing of tasks seems not work very well
-            ready_task.compose(self.task_agent)
-            # remove the subtasks from the channel
-            for subtask in ready_task.subtasks:
-                await self._channel.remove_task(subtask.id)
-            # send the task to the channel as a dependency
-            await self._post_dependency(ready_task)
-            self._pending_tasks.popleft()
-            # try to send the next task in the pending list
-            await self._post_ready_tasks()
-        else:
-            # directly post the task to the channel if it's a new one
-            # find a node to assign the task
-            assignee_id = self._find_assignee(task=ready_task)
-            await self._post_task(ready_task, assignee_id)
-
-    async def _handle_failed_task(self, task: Task) -> None:
-        # remove the failed task from the channel
-        await self._channel.remove_task(task.id)
-        if task.get_depth() >= 3:
-            # create a new WF and reassign
-            # TODO: add a state for reassign?
-            assignee = self._create_worker_node_for_task(task)
-            # print('create_new_assignee:', assignee)
-            await self._post_task(task, assignee.node_id)
-        else:
-            subtasks = task.decompose(self.task_agent)
-            # Insert packets at the head of the queue
-            self._pending_tasks.extendleft(reversed(subtasks))
-            await self._post_ready_tasks()
-
-    async def _handle_completed_task(self, task: Task) -> None:
-        # archive the packet, making it into a dependency
-        self._pending_tasks.popleft()
-        await self._channel.archive_task(task.id)
-        await self._post_ready_tasks()
-
-    @check_if_running(False)
-    def set_channel(self, channel: TaskChannel):
-        r"""Set the channel for the node and all the child nodes under it."""
-        self._channel = channel
-        for child in self._children:
-            child.set_channel(channel)
-
-    @check_if_running(False)
-    async def _listen_to_channel(self) -> None:
-        r"""Continuously listen to the channel, post task to the channel and
-        track the status of posted tasks.
-        """
-
-        self._running = True
-        print(f"{Fore.GREEN}Manager node {self.node_id} started.{Fore.RESET}")
-
-        # if this node is at the top level, it will have an initial task
-        # the initial task must be decomposed into subtasks first
-        if self._task is not None:
-            subtasks = self._task.decompose(self.task_agent)
-            self._pending_tasks.extend(subtasks)
-            self._task.state = TaskState.FAILED
-            self._pending_tasks.append(self._task)
-
-        # before starting the loop, send ready pending tasks to the channel
-        await self._post_ready_tasks()
-
-        await self._channel.print_channel()
-
-        while self._task is None or self._pending_tasks:
-            returned_task = await self._get_returned_task()
-            if returned_task.state == TaskState.DONE:
-                await self._handle_completed_task(returned_task)
-            elif returned_task.state == TaskState.FAILED:
-                await self._handle_failed_task(returned_task)
-            else:
-                raise ValueError(
-                    f"Task {returned_task.id} has an unexpected state."
-                )
-
-        # shut down the whole workforce tree
-        self.stop()
-
-    @check_if_running(False)
-    async def start(self) -> None:
-        r"""Start itself and all the child nodes under it."""
-        for child in self._children:
-            child_listening_task = asyncio.create_task(child.start())
-            self._child_listening_tasks.append(child_listening_task)
-        await self._listen_to_channel()
-
-    @check_if_running(True)
-    def stop(self) -> None:
-        r"""Stop all the child nodes under it. The node itself will be stopped
-        by its parent node.
-        """
-        for child in self._children:
-            child.stop()
-        for child_task in self._child_listening_tasks:
-            child_task.cancel()
-        self._running = False

camel/workforce/role_playing_node.py

@@ -1,168 +0,0 @@
-# =========== 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 Dict, List, Optional
-
-from colorama import Fore, Style
-
-from camel.agents.chat_agent import ChatAgent, FunctionCallingRecord
-from camel.messages.base import BaseMessage
-from camel.societies import RolePlaying
-from camel.tasks.task import Task, TaskState
-from camel.utils import print_text_animated
-from camel.workforce.utils import parse_task_result_resp
-from camel.workforce.worker_node import WorkerNode
-from camel.workforce.workforce_prompt import (
-    ROLEPLAY_PROCESS_TASK_PROMPT,
-    ROLEPLAY_SUMMERIZE_PROMPT,
-)
-
-
-class RolePlayingNode(WorkerNode):
-    r"""A worker node that contains a role playing.
-
-    Args:
-        description (str): Description of the node.
-        assistant_role_name (str): The role name of the assistant agent.
-        user_role_name (str): The role name of the user agent.
-        assistant_agent_kwargs (Optional[Dict], optional): The keyword
-            arguments to initialize the assistant agent in the role playing,
-            like the model name, etc. Defaults to None.
-        user_agent_kwargs (Optional[Dict], optional): The keyword arguments to
-            initialize the user agent in the role playing, like the model name,
-            etc. Defaults to None.
-        chat_turn_limit (int, optional): The maximum number of chat turns in
-            the role playing. Defaults to 3.
-    """
-
-    def __init__(
-        self,
-        description: str,
-        assistant_role_name: str,
-        user_role_name: str,
-        assistant_agent_kwargs: Optional[Dict] = None,
-        user_agent_kwargs: Optional[Dict] = None,
-        chat_turn_limit: int = 3,
-    ) -> None:
-        super().__init__(description)
-        sys_message = BaseMessage.make_assistant_message(
-            role_name="Summarizer",
-            content="Good at summarizing the results of the chat logs",
-        )
-        self.summerize_agent = ChatAgent(sys_message)
-        self.chat_turn_limit = chat_turn_limit
-        self.assistant_role_name = assistant_role_name
-        self.user_role_name = user_role_name
-        self.assistant_agent_kwargs = assistant_agent_kwargs
-        self.user_agent_kwargs = user_agent_kwargs
-
-    async def _process_task(
-        self, task: Task, dependencies: List[Task]
-    ) -> TaskState:
-        r"""Processes a task leveraging its dependencies through role-playing.
-
-        This method orchestrates a role-playing session between an AI
-        assistant and an AI user to process a given task. It initiates with a
-        generated prompt based on the task and its dependencies, conducts a
-        dialogue up to a specified chat turn limit, and then summarizes the
-        dialogue to determine the task's outcome.
-
-        Args:
-            task (Task): The task object to be processed, containing necessary
-                details like content and type.
-            dependencies (List[Task]): A list of task objects that the current
-                task depends on.
-
-        Returns:
-            TaskState: `TaskState.DONE` if processed successfully, otherwise
-                `TaskState.FAILED`.
-        """
-        try:
-            dependency_tasks_info = self._get_dep_tasks_info(dependencies)
-            prompt = ROLEPLAY_PROCESS_TASK_PROMPT.format(
-                content=task.content,
-                type=task.type,
-                dependency_task_info=dependency_tasks_info,
-            )
-            role_play_session = RolePlaying(
-                assistant_role_name=self.assistant_role_name,
-                user_role_name=self.user_role_name,
-                assistant_agent_kwargs=self.assistant_agent_kwargs,
-                user_agent_kwargs=self.user_agent_kwargs,
-                task_prompt=prompt,
-                with_task_specify=False,
-            )
-            n = 0
-            input_msg = role_play_session.init_chat()
-            chat_history = []
-            while n < self.chat_turn_limit:
-                n += 1
-                assistant_response, user_response = role_play_session.step(
-                    input_msg
-                )
-
-                if assistant_response.terminated:
-                    reason = assistant_response.info['termination_reasons']
-                    print(
-                        Fore.GREEN + f"AI Assistant terminated. Reason: "
-                        f"{reason}. "
-                    )
-                    break
-                if user_response.terminated:
-                    reason = user_response.info['termination_reasons']
-                    print(
-                        Fore.GREEN + f"AI User terminated. Reason: {reason}."
-                    )
-                    break
-                print_text_animated(
-                    Fore.BLUE + f"AI User:\n\n{user_response.msg.content}\n",
-                    delay=0.005,
-                )
-                chat_history.append(f"AI User: {user_response.msg.content}")
-
-                print_text_animated(Fore.GREEN + "AI Assistant:", delay=0.005)
-                tool_calls: List[FunctionCallingRecord] = (
-                    assistant_response.info['tool_calls']
-                )
-                for func_record in tool_calls:
-                    print_text_animated(f"{func_record}", delay=0.005)
-                print_text_animated(
-                    f"{assistant_response.msg.content}\n", delay=0.005
-                )
-                chat_history.append(
-                    f"AI Assistant: {assistant_response.msg.content}"
-                )
-
-                if "CAMEL_TASK_DONE" in user_response.msg.content:
-                    break
-
-                input_msg = assistant_response.msg
-
-            chat_history_str = "\n".join(chat_history)
-            prompt = ROLEPLAY_SUMMERIZE_PROMPT.format(
-                content=task.content,
-                type=task.type,
-                chat_history=chat_history_str,
-            )
-            req = BaseMessage.make_user_message(
-                role_name="User",
-                content=prompt,
-            )
-            response = self.summerize_agent.step(req)
-            task.result = parse_task_result_resp(response.msg.content)
-            print(Style.RESET_ALL + 'Task result:', task.result, '\n')
-            return TaskState.DONE
-        except Exception:
-            return TaskState.FAILED