Skip to content

Commit 130f8a6

Browse files
Draw defaults from config file
1 parent 4013d70 commit 130f8a6

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

edsl/language_models/key_management/KeyLookupBuilder.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,12 @@ class KeyLookupBuilder:
5454
('openai', 'rpm')
5555
"""
5656

57-
DEFAULT_RPM = 10
58-
DEFAULT_TPM = 2000000
57+
# DEFAULT_RPM = 10
58+
# DEFAULT_TPM = 2000000
59+
from edsl.config import CONFIG
60+
61+
DEFAULT_RPM = CONFIG.get("EDSL_SERVICE_RPM_BASELINE")
62+
DEFAULT_TPM = CONFIG.get("EDSL_SERVICE_TPM_BASELINE")
5963

6064
def __init__(self, fetch_order: Optional[tuple[str]] = None):
6165
if fetch_order is None:

0 commit comments

Comments
 (0)