Coverage for intelligence_toolkit/tests/unit/AI/test_defaults.py: 100%

1# Copyright (c) 2024 Microsoft Corporation. All rights reserved. 

2# Licensed under the MIT license. See LICENSE file in the project. 

3# 

4from intelligence_toolkit.AI import defaults 

5 

6 

7def test_default_encoding(): 

8 assert defaults.DEFAULT_ENCODING == "o200k_base" 

9 

10 

11def test_default_llm_model(): 

12 assert defaults.DEFAULT_LLM_MODEL == "gpt-4.1-mini" 

13 

14 

15def test_default_llm_max_tokens(): 

16 assert defaults.DEFAULT_LLM_MAX_TOKENS == 4000 

17 

18 

19def test_default_az_auth_type(): 

20 assert defaults.DEFAULT_AZ_AUTH_TYPE == "Azure Key" 

21 

22 

23def test_embedding_batches_number(): 

24 assert defaults.EMBEDDING_BATCHES_NUMBER == 500 

25 

26 

27def test_default_embedding_model(): 

28 assert defaults.DEFAULT_EMBEDDING_MODEL == "text-embedding-3-small" 

29 

30 

31def test_default_embedding_model_azure(): 

32 assert defaults.DEFAULT_EMBEDDING_MODEL_AZURE == "text-embedding-ada-002" 

33 

34 

35def test_default_temperature(): 

36 assert defaults.DEFAULT_TEMPERATURE == 0 

37 

38 

39def test_default_max_input_tokens(): 

40 assert defaults.DEFAULT_MAX_INPUT_TOKENS == 128000 

41 

42 

43def test_default_openai_version(): 

44 assert defaults.DEFAULT_OPENAI_VERSION == "2024-08-01-preview" 

45 

46 

47def test_default_local_embedding_model(): 

48 assert defaults.DEFAULT_LOCAL_EMBEDDING_MODEL == "all-distilroberta-v1" 

49 

50 

51def test_api_base_required_for_azure(): 

52 assert defaults.API_BASE_REQUIRED_FOR_AZURE == "api_base is required for Azure OpenAI client" 

53 

54 

55def test_chunk_size(): 

56 assert defaults.CHUNK_SIZE == 500 

57 

58 

59def test_chunk_overlap(): 

60 assert defaults.CHUNK_OVERLAP == 0 

61 

62 

63def test_default_report_batch_size(): 

64 assert defaults.DEFAULT_REPORT_BATCH_SIZE == 100 

65 

66 

67def test_default_concurrent_coroutines(): 

68 assert defaults.DEFAULT_CONCURRENT_COROUTINES == 50