Coverage for mindsdb / integrations / handlers / xata_handler / connection_args.py: 0%

1from collections import OrderedDict 

2 

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

4 

5 

6connection_args = OrderedDict( 

7 db_url={ 

8 "type": ARG_TYPE.STR, 

9 "description": "Xata database url with region, database and, optionally the branch information", 

10 "required": True, 

11 }, 

12 api_key={ 

13 "type": ARG_TYPE.STR, 

14 "description": "personal Xata API key", 

15 "required": True, 

16 "secret": True 

17 }, 

18 dimension={ 

19 "type": ARG_TYPE.INT, 

20 "description": "default dimension of embeddings vector used to create table when using create (default=8)", 

21 "required": False, 

22 }, 

23 similarity_function={ 

24 "type": ARG_TYPE.STR, 

25 "description": "similarity function to use for vector searches (default=cosineSimilarity)", 

26 "required": False, 

27 } 

28) 

29 

30connection_args_example = OrderedDict( 

31 db_url="https://...", 

32 api_key="abc_def...", 

33 dimension=8, 

34 similarity_function="l1" 

35)