Skip to content

Commit 40e2b0e

Browse files
fixed integration tests
1 parent 0e446f2 commit 40e2b0e

File tree

1 file changed

+4
-13
lines changed

1 file changed

+4
-13
lines changed

tests/integration_tests/test_tools.py

+4-13
Original file line numberDiff line numberDiff line change
@@ -12,21 +12,12 @@ def tool_constructor(self) -> Type[AIMindTool]:
1212

1313
@property
1414
def tool_constructor_params(self) -> dict:
15-
datasource = AIMindDataSource(
15+
# Re-uses an existing data source and API wrapper.
16+
# Only the name is required to initialize existing objects.
17+
AIMindDataSource(
1618
name="test_datasource",
17-
description="house sales",
18-
engine="postgres",
19-
connection_data={
20-
"user": "demo_user",
21-
"password": "demo_password",
22-
"host": "samples.mindsdb.com",
23-
"port": 5432,
24-
"database": "demo",
25-
"schema": "demo_data",
26-
},
27-
tables=["house_sales"],
2819
)
29-
api_wrapper = AIMindAPIWrapper(name="test_mind", datasources=[datasource])
20+
api_wrapper = AIMindAPIWrapper(name="test_mind")
3021
return {"api_wrapper": api_wrapper}
3122

3223
@property

0 commit comments

Comments
 (0)