Coverage for intelligence_toolkit/AI/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# 

4 

5DEFAULT_ENCODING = "o200k_base" 

6# 

7DEFAULT_LLM_MODEL = "gpt-4.1-mini" 

8DEFAULT_LLM_MAX_TOKENS = 4000 

9DEFAULT_AZ_AUTH_TYPE = "Azure Key" 

10EMBEDDING_BATCHES_NUMBER = 500 

11# 

12# Text Embedding Parameters 

13DEFAULT_EMBEDDING_MODEL = "text-embedding-3-small" 

14DEFAULT_EMBEDDING_MODEL_AZURE = "text-embedding-ada-002" 

15DEFAULT_TEMPERATURE = 0 

16DEFAULT_MAX_INPUT_TOKENS = 128000 

17DEFAULT_OPENAI_VERSION = "2024-08-01-preview" 

18DEFAULT_LOCAL_EMBEDDING_MODEL = "all-distilroberta-v1" 

19 

20API_BASE_REQUIRED_FOR_AZURE = "api_base is required for Azure OpenAI client" 

21 

22CHUNK_SIZE = 500 

23CHUNK_OVERLAP = 0 

24 

25DEFAULT_REPORT_BATCH_SIZE = 100 

26 

27DEFAULT_CONCURRENT_COROUTINES = 50