diff --git a/src/seclab_taskflow_agent/__main__.py b/src/seclab_taskflow_agent/__main__.py index ced5c75..5c59466 100644 --- a/src/seclab_taskflow_agent/__main__.py +++ b/src/seclab_taskflow_agent/__main__.py @@ -6,7 +6,7 @@ import argparse import os import sys -from dotenv import load_dotenv +from dotenv import load_dotenv, find_dotenv import logging from logging.handlers import RotatingFileHandler from pprint import pprint, pformat @@ -34,7 +34,7 @@ from .capi import list_tool_call_models from .available_tools import AvailableTools -load_dotenv() +load_dotenv(find_dotenv(usecwd=True)) # only model output or help message should go to stdout, everything else goes to log logging.getLogger('').setLevel(logging.NOTSET) diff --git a/src/seclab_taskflow_agent/agent.py b/src/seclab_taskflow_agent/agent.py index 91b000f..d49c781 100644 --- a/src/seclab_taskflow_agent/agent.py +++ b/src/seclab_taskflow_agent/agent.py @@ -4,7 +4,7 @@ # https://openai.github.io/openai-agents-python/agents/ import os import logging -from dotenv import load_dotenv +from dotenv import load_dotenv, find_dotenv from collections.abc import Callable from typing import Any from urllib.parse import urlparse @@ -18,7 +18,7 @@ from .capi import COPILOT_INTEGRATION_ID, COPILOT_API_ENDPOINT # grab our secrets from .env, this must be in .gitignore -load_dotenv() +load_dotenv(find_dotenv(usecwd=True)) match urlparse(COPILOT_API_ENDPOINT).netloc: case 'api.githubcopilot.com':