ENG-504: Make the agent architecture configurable (#243)

* Make the agent architecture configurable

* Remove One tool requirement in Agent Creation

Author: thiago-aixplain

aixplain/factories/agent_factory/__init__.py

@@ -44,7 +44,7 @@ class AgentFactory:
     def create(
         cls,
         name: Text,
-        llm_id: Text,
+        llm_id: Text = "669a63646eb56306647e1091",
         tools: List[Tool] = [],
         description: Text = "",
         api_key: Text = config.TEAM_API_KEY,
@@ -56,7 +56,7 @@ def create(
 
         Args:
             name (Text): name of the agent
-            llm_id (Text): aiXplain ID of the large language model to be used as agent.
+            llm_id (Text, optional): aiXplain ID of the large language model to be used as agent. Defaults to "669a63646eb56306647e1091" (GPT-4o mini).
             tools (List[Tool], optional): list of tool for the agent. Defaults to [].
             description (Text, optional): description of the agent role. Defaults to "".
             api_key (Text, optional): team/user API key. Defaults to config.TEAM_API_KEY.

tests/functional/agent/agent_functional_test.py

@@ -20,6 +20,7 @@
 
 load_dotenv()
 from aixplain.factories import AgentFactory
+from aixplain.enums.function import Function
 from aixplain.enums.supplier import Supplier
 
 import pytest
@@ -79,5 +80,7 @@ def test_list_agents():
 
 def test_fail_non_existent_llm():
     with pytest.raises(Exception) as exc_info:
-        AgentFactory.create(name="Test Agent", llm_id="non_existent_llm", tools=[])
+        AgentFactory.create(
+            name="Test Agent", llm_id="non_existent_llm", tools=[AgentFactory.create_model_tool(function=Function.TRANSLATION)]
+        )
     assert str(exc_info.value) == "Large Language Model with ID 'non_existent_llm' not found."

tests/unit/agent_test.py

@@ -69,7 +69,10 @@ def test_sucess_query_content():
 def test_invalid_pipelinetool():
     with pytest.raises(Exception) as exc_info:
         AgentFactory.create(
-            name="Test", tools=[PipelineTool(pipeline="309851793", description="Test")], llm_id="6646261c6eb563165658bbb1"
+            name="Test",
+            description="Test Description",
+            tools=[PipelineTool(pipeline="309851793", description="Test")],
+            llm_id="6646261c6eb563165658bbb1",
         )
     assert str(exc_info.value) == "Pipeline Tool Unavailable. Make sure Pipeline '309851793' exists or you have access to it."