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")
|