fixed import path (#128)

Author: sauravpanda

workflows/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "workflow-use"
-version = "0.2.3"
+version = "0.2.4"
 authors = [{ name = "Gregor Zunic" }]
 description = "Create, edit, run deterministic workflows"
 readme = "README.md"

workflows/workflow_use/healing/prompts.py

@@ -1 +1,6 @@
-HEALING_AGENT_SYSTEM_PROMPT = open('workflow_use/healing/_agent/agent_prompt.md').read()
+import os
+from pathlib import Path
+
+# Use __file__ to get absolute path relative to this file's location
+_PROMPTS_DIR = Path(__file__).parent / '_agent'
+HEALING_AGENT_SYSTEM_PROMPT = (_PROMPTS_DIR / 'agent_prompt.md').read_text()

workflows/workflow_use/healing/service.py

@@ -1,5 +1,6 @@
 import hashlib
 import json
+from pathlib import Path
 from typing import Any, Dict, List, Sequence, Union
 
 import aiofiles
@@ -16,6 +17,9 @@
 from workflow_use.healing.views import ParsedAgentStep, SimpleDomElement, SimpleResult
 from workflow_use.schema.views import SelectorWorkflowSteps, WorkflowDefinitionSchema
 
+# Get the absolute path to the prompts directory
+_PROMPTS_DIR = Path(__file__).parent / 'prompts'
+
 
 class HealingService:
 	def __init__(
@@ -141,7 +145,9 @@ def _populate_selector_fields(self, workflow_definition: WorkflowDefinitionSchem
 	async def create_workflow_definition(
 		self, task: str, history_list: AgentHistoryList, extract_variables: bool = True
 	) -> WorkflowDefinitionSchema:
-		async with aiofiles.open('workflow_use/healing/prompts/workflow_creation_prompt.md', mode='r') as f:
+		# Load prompt using absolute path
+		prompt_file = _PROMPTS_DIR / 'workflow_creation_prompt.md'
+		async with aiofiles.open(prompt_file, mode='r') as f:
 			prompt_content = await f.read()
 
 		prompt_content = prompt_content.format(goal=task, actions=BuilderService._get_available_actions_markdown())

workflows/workflow_use/healing/tests/test_exploration_agent.py

@@ -1,6 +1,7 @@
 import asyncio
 import logging
 import os
+from pathlib import Path
 
 from browser_use import Agent, Browser
 from browser_use.llm import ChatBrowserUse
@@ -26,7 +27,9 @@
 	temperature=0.0,
 )
 
-system_prompt = open('workflow_use/healing/_agent/agent_prompt.md').read()
+# Use absolute path for prompt file
+_PROMPT_FILE = Path(__file__).parent.parent / '_agent' / 'agent_prompt.md'
+system_prompt = _PROMPT_FILE.read_text()
 
 
 async def explore_page():

workflows/workflow_use/healing/validator.py

@@ -8,6 +8,7 @@
 """
 
 import json
+from pathlib import Path
 from typing import List, Optional
 
 import aiofiles
@@ -17,6 +18,9 @@
 
 from workflow_use.schema.views import WorkflowDefinitionSchema
 
+# Get the absolute path to the prompts directory
+_PROMPTS_DIR = Path(__file__).parent / 'prompts'
+
 
 class WorkflowIssue(BaseModel):
 	"""Represents an issue found in a workflow."""
@@ -56,8 +60,9 @@ async def validate_workflow(
 		Returns:
 		    WorkflowValidationResult with identified issues and optional corrections
 		"""
-		# Load validation prompt
-		async with aiofiles.open('workflow_use/healing/prompts/workflow_validation_prompt.md', mode='r') as f:
+		# Load validation prompt using absolute path
+		prompt_file = _PROMPTS_DIR / 'workflow_validation_prompt.md'
+		async with aiofiles.open(prompt_file, mode='r') as f:
 			prompt_content = await f.read()
 
 		# Prepare workflow JSON for review