Fixed import issue with structure tool, updated sample.env to use 0.0.4

Zipstack · Mar 14, 2024 · 335bb0f · 335bb0f
1 parent f26634a
commit 335bb0f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/backend/sample.env b/backend/sample.env
@@ -86,9 +86,9 @@ PROMPT_PORT=3003
 PROMPT_STUDIO_FILE_PATH=/app/prompt-studio-data
 
 # Structure Tool
-STRUCTURE_TOOL_IMAGE_URL="docker:unstract/tool-structure:0.0.3"
+STRUCTURE_TOOL_IMAGE_URL="docker:unstract/tool-structure:0.0.4"
 STRUCTURE_TOOL_IMAGE_NAME="unstract/tool-structure"
-STRUCTURE_TOOL_IMAGE_TAG="0.0.3"
+STRUCTURE_TOOL_IMAGE_TAG="0.0.4"
 
 # Feature Flags
 EVALUATION_SERVER_IP=localhost

diff --git a/tools/structure/src/main.py b/tools/structure/src/main.py
@@ -3,14 +3,13 @@
 from pathlib import Path
 from typing import Any
 
+from constants import SettingsKeys  # type: ignore [attr-defined]
 from unstract.sdk.constants import LogState, MetadataKey
 from unstract.sdk.index import ToolIndex
 from unstract.sdk.prompt import PromptTool
 from unstract.sdk.tool.base import BaseTool
 from unstract.sdk.tool.entrypoint import ToolEntrypoint
 
-from .constants import SettingsKeys
-
 
 class StructureTool(BaseTool):
     def validate(self, input_file: str, settings: dict[str, Any]) -> None: