From 335bb0f6feec6dbf2aea0d60596d5706e722df64 Mon Sep 17 00:00:00 2001 From: Chandrasekharan M Date: Thu, 14 Mar 2024 14:21:01 +0530 Subject: [PATCH] Fixed import issue with structure tool, updated sample.env to use 0.0.4 --- backend/sample.env | 4 ++-- tools/structure/src/main.py | 3 +-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/backend/sample.env b/backend/sample.env index 2494d77e4..e24aad6cf 100644 --- 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 index 14d6c9e2d..4eabd063c 100644 --- 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: