Spaces:
Runtime error
Runtime error
Commit
·
9ae0db9
1
Parent(s):
35fa9b8
added debug step for wandb
Browse files- 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
|
19 |
-
os.environ["
|
|
|
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")
|