default settings

aixplain/factories/team_agent_factory/__init__.py

@@ -45,8 +45,6 @@ def create(
         api_key: Text = config.TEAM_API_KEY,
         supplier: Union[Dict, Text, Supplier, int] = "aiXplain",
         version: Optional[Text] = None,
-        use_mentalist: bool = True,
-        use_inspector: bool = True,
         use_mentalist_and_inspector: bool = True,
     ) -> TeamAgent:
         """Create a new team agent in the platform."""
@@ -64,20 +62,9 @@ def create(
 
                 assert isinstance(agent, Agent), "TeamAgent Onboarding Error: Agents must be instances of Agent class"
 
-        mentalist_llm_id = None
-        if use_mentalist is True:
-            mentalist_llm_id = llm_id
-
-        inspector_llm_id = None
-        if use_inspector is True: 
-            inspector_llm_id = llm_id
-
-        if use_inspector and not use_mentalist:
-            raise Exception("TeamAgent Onboarding Error: To use the Inspector agent, you must enable Mentalist.")
-
+        mentalist_and_inspector_llm_id = None
         if use_mentalist_and_inspector is True:
-            inspector_llm_id = llm_id
-            mentalist_llm_id = llm_id
+            mentalist_and_inspector_llm_id = llm_id
 
         team_agent = None
         url = urljoin(config.BACKEND_URL, "sdk/agent-communities")
@@ -99,8 +86,7 @@ def create(
             "description": description,
             "llmId": llm_id,
             "supervisorId": llm_id,
-            "plannerId": mentalist_llm_id,
-            "inspectorId": inspector_llm_id,
+            "plannerId": mentalist_and_inspector_llm_id,
             "supplier": supplier,
             "version": version,
             "status": "draft",