Development 2 Test

aixplain/modules/agent/tool/model_tool.py

@@ -58,7 +58,6 @@ def __init__(
         if function is not None:
             if isinstance(function, str):
                 function = Function(function)
-        self.function = function
 
         try:
             if isinstance(supplier, dict):
@@ -68,9 +67,13 @@ def __init__(
 
         if model is not None:
             if isinstance(model, Text) is True:
+                from aixplain.factories.model_factory import ModelFactory
+
                 model = ModelFactory.get(model)
+            function = model.function
             if isinstance(model.supplier, Supplier):
                 supplier = model.supplier
             model = model.id
         self.supplier = supplier
         self.model = model
+        self.function = function

tests/functional/agent/agent_functional_test.py

@@ -37,6 +37,9 @@ def run_input_map(request):
 
 
 def test_end2end(run_input_map):
+    for agent in AgentFactory.list()["results"]:
+        agent.delete()
+
     tools = []
     if "model_tools" in run_input_map:
         for tool in run_input_map["model_tools"]: