Coverage for mindsdb / integrations / handlers / chromadb_handler / connection_args.py: 100%

1from collections import OrderedDict 

2 

3from mindsdb.integrations.libs.const import HANDLER_CONNECTION_ARG_TYPE as ARG_TYPE 

4 

5 

6connection_args = OrderedDict( 

7 host={ 

8 "type": ARG_TYPE.STR, 

9 "description": "chromadb server host", 

10 "required": False, 

11 }, 

12 port={ 

13 "type": ARG_TYPE.STR, 

14 "description": "chromadb server port", 

15 "required": False, 

16 }, 

17 persist_directory={ 

18 "type": ARG_TYPE.STR, 

19 "description": "persistence directory for ChromaDB", 

20 "required": False, 

21 }, 

22) 

23 

24connection_args_example = OrderedDict( 

25 host="localhost", 

26 port="8000", 

27 persist_directory="chromadb", 

28)