diff --git a/aixplain/enums/function_type.py b/aixplain/enums/function_type.py index 465e0e31..3652737a 100644 --- a/aixplain/enums/function_type.py +++ b/aixplain/enums/function_type.py @@ -32,5 +32,6 @@ class FunctionType(Enum): METRIC = "metric" SEARCH = "search" INTEGRATION = "connector" + CONNECTION = "connection" MCP_CONNECTION = "mcpconnection" MCPSERVER = "mcpserver" diff --git a/aixplain/modules/model/index_model.py b/aixplain/modules/model/index_model.py index 81560f1f..31c91531 100644 --- a/aixplain/modules/model/index_model.py +++ b/aixplain/modules/model/index_model.py @@ -55,6 +55,7 @@ def __init__( self.split_length = split_length self.split_overlap = split_overlap + class IndexModel(Model): def __init__( self, diff --git a/aixplain/modules/model/integration.py b/aixplain/modules/model/integration.py index 081c8cbc..0e4c5ee5 100644 --- a/aixplain/modules/model/integration.py +++ b/aixplain/modules/model/integration.py @@ -138,7 +138,6 @@ def connect(self, authentication_schema: AuthenticationSchema, args: Optional[Ba raise ValueError(f"Invalid authentication schema: {authentication_schema}") - def __repr__(self): try: return f"Integration: {self.name} by {self.supplier['name']} (id={self.id})" diff --git a/tests/functional/agent/agent_functional_test.py b/tests/functional/agent/agent_functional_test.py index db11222a..db1f88db 100644 --- a/tests/functional/agent/agent_functional_test.py +++ b/tests/functional/agent/agent_functional_test.py @@ -790,4 +790,3 @@ def test_agent_with_mcp_tool(): assert "helsinki" in response.data.output.lower() assert action_name in [step["tool"] for step in response.data.intermediate_steps[0]["tool_steps"]] connection.delete() -