jerpint commited on
Commit
c60c929
1 Parent(s): 6498684

Update cfg.py

Browse files
Files changed (1) hide show
  1. cfg.py +6 -0
cfg.py CHANGED
@@ -7,6 +7,12 @@ from buster.tokenizers import GPTTokenizer
7
  from buster.validators import Validator
8
  from buster.llm_utils import get_openai_embedding_constructor
9
 
 
 
 
 
 
 
10
  embedding_fn = get_openai_embedding_constructor(client_kwargs=client_kwargs)
11
 
12
  buster_cfg = BusterConfig(
 
7
  from buster.validators import Validator
8
  from buster.llm_utils import get_openai_embedding_constructor
9
 
10
+ # kwargs to pass to OpenAI client
11
+ client_kwargs = {
12
+ "timeout": 20,
13
+ "max_retries": 3,
14
+ }
15
+
16
  embedding_fn = get_openai_embedding_constructor(client_kwargs=client_kwargs)
17
 
18
  buster_cfg = BusterConfig(