Fix relative paths and imports to make code work as a package

Author: sylwia-budzynska

__init__.py

@@ -0,0 +1,4 @@
+import runpy
+
+if __name__ == "__main__":
+    runpy.run_module("seclab_taskflow_agent.main", run_name="__main__")

__main__.py

@@ -12,7 +12,7 @@
 from agents.run import RunHooks
 from agents import Agent, Runner, AgentHooks, RunHooks, result, function_tool, Tool, RunContextWrapper, TContext, OpenAIChatCompletionsModel, set_default_openai_client, set_default_openai_api, set_tracing_disabled
 
-from capi import COPILOT_INTEGRATION_ID, COPILOT_API_ENDPOINT
+from .capi import COPILOT_INTEGRATION_ID, COPILOT_API_ENDPOINT
 
 # grab our secrets from .env, this must be in .gitignore
 load_dotenv()

agent.py

@@ -11,7 +11,7 @@
 import json
 import uuid
 
-from agent import DEFAULT_MODEL, TaskRunHooks, TaskAgentHooks
+from .agent import DEFAULT_MODEL, TaskRunHooks, TaskAgentHooks
 #from agents.run import DEFAULT_MAX_TURNS # XXX: this is 10, we need more than that
 from agents.exceptions import MaxTurnsExceeded, AgentsException
 from agents.agent import ModelSettings
@@ -22,21 +22,27 @@
 from openai.types.responses import ResponseTextDeltaEvent
 from typing import Any
 
-from shell_utils import shell_tool_call
-from mcp_utils import DEFAULT_MCP_CLIENT_SESSION_TIMEOUT, ReconnectingMCPServerStdio, AsyncDebugMCPServerStdio, MCPNamespaceWrap, mcp_client_params, mcp_system_prompt, StreamableMCPThread
-from render_utils import render_model_output, flush_async_output
-from env_utils import TmpEnv
-from yaml_parser import YamlParser
-from agent import TaskAgent
-from capi import list_tool_call_models
-from available_tools import AvailableTools
+from .shell_utils import shell_tool_call
+from .mcp_utils import DEFAULT_MCP_CLIENT_SESSION_TIMEOUT, ReconnectingMCPServerStdio, AsyncDebugMCPServerStdio, MCPNamespaceWrap, mcp_client_params, mcp_system_prompt, StreamableMCPThread
+from .render_utils import render_model_output, flush_async_output
+from .env_utils import TmpEnv
+from .yaml_parser import YamlParser
+from .agent import TaskAgent
+from .capi import list_tool_call_models
+from .available_tools import AvailableTools
 
 load_dotenv()
 
 # only model output or help message should go to stdout, everything else goes to log
 logging.getLogger('').setLevel(logging.NOTSET)
+
+# Create logs directory if it doesn't exist
+# This is needed when running the code as a package from any directory
+log_dir = os.path.join(os.getcwd(), 'logs')
+os.makedirs(log_dir, exist_ok=True)
+
 log_file_handler = RotatingFileHandler(
-    'logs/task_agent.log',
+    os.path.join(log_dir, 'task_agent.log'),
     maxBytes=1024*1024*10,
     backupCount=10)
 log_file_handler.setLevel(os.getenv('TASK_AGENT_LOGLEVEL', default='DEBUG'))

main.py

@@ -14,7 +14,7 @@
 from mcp.types import CallToolResult, TextContent
 from agents.mcp import MCPServerStdio
 
-from env_utils import swap_env
+from .env_utils import swap_env
 
 DEFAULT_MCP_CLIENT_SESSION_TIMEOUT = 120