andymbryant commited on
Commit
9ae0db9
·
1 Parent(s): 35fa9b8

added debug step for wandb

Browse files
Files changed (1) hide show
  1. src/core.py +4 -3
src/core.py CHANGED
@@ -14,11 +14,12 @@ from src.prompt import (
14
  SPEC_WRITER_PROMPT_STR,
15
  ENGINEER_PROMPT_STR,
16
  )
 
17
 
18
- os.environ["LANGCHAIN_WANDB_TRACING"] = "true"
19
- os.environ["WANDB_PROJECT"] = "llm-data-mapper"
 
20
 
21
- load_dotenv()
22
 
23
  NUM_ROWS_TO_RETURN = 5
24
  DATA_DIR_PATH = os.path.join(os.path.dirname(__file__), "data")
 
14
  SPEC_WRITER_PROMPT_STR,
15
  ENGINEER_PROMPT_STR,
16
  )
17
+ load_dotenv()
18
 
19
+ if os.environ.get('DEBUG') == 'true':
20
+ os.environ["LANGCHAIN_WANDB_TRACING"] = "true"
21
+ os.environ["WANDB_PROJECT"] = "llm-data-mapper"
22
 
 
23
 
24
  NUM_ROWS_TO_RETURN = 5
25
  DATA_DIR_PATH = os.path.join(os.path.dirname(__file__), "data")