diff --git a/unstract/tool-registry/src/unstract/tool_registry/tool_registry.py b/unstract/tool-registry/src/unstract/tool_registry/tool_registry.py index 3de029420..4a060a3a0 100644 --- a/unstract/tool-registry/src/unstract/tool_registry/tool_registry.py +++ b/unstract/tool-registry/src/unstract/tool_registry/tool_registry.py @@ -26,12 +26,8 @@ class ToolRegistry: REGISTRY_FILE = "registry.yaml" - if check_feature_flag_status(FeatureFlag.REMOTE_FILE_STORAGE): - PRIVATE_TOOL_CONFIG_FILE = "private_tools_remote_storage.json" - PUBLIC_TOOL_CONFIG_FILE = "public_tools_remote_storage.json" - else: - PRIVATE_TOOL_CONFIG_FILE = "private_tools.json" - PUBLIC_TOOL_CONFIG_FILE = "public_tools.json" + PRIVATE_TOOL_CONFIG_FILE = "private_tools.json" + PUBLIC_TOOL_CONFIG_FILE = "public_tools.json" def __init__( self, diff --git a/unstract/tool-registry/tool_registry_config/public_tools_remote_storage.json b/unstract/tool-registry/tool_registry_config/public_tools_remote_storage.json deleted file mode 100644 index 412549e8d..000000000 --- a/unstract/tool-registry/tool_registry_config/public_tools_remote_storage.json +++ /dev/null @@ -1,198 +0,0 @@ -{ - "classify": { - "tool_uid": "classify", - "properties": { - "schemaVersion": "0.0.1", - "displayName": "File Classifier Remote Storage", - "functionName": "classify", - "toolVersion": "0.0.40", - "description": "Classifies a file into a bin based on its contents", - "input": { - "description": "File to be classified" - }, - "output": { - "description": "Places the file into a folder (bin) that its classified into." - }, - "result": { - "type": "JSON", - "description": "JSON response containing the bin to which the file was classified into", - "schema": {} - }, - "adapter": { - "languageModels": [ - { - "isEnabled": true, - "title": "Classifier LLM", - "isRequired": true, - "description": "LLM to use for classification" - } - ], - "embeddingServices": [ - { - "isEnabled": false - } - ], - "vectorStores": [ - { - "isEnabled": false - } - ], - "textExtractors": [ - { - "isEnabled": true, - "adapterId": "textExtractorId", - "title": "Text Extraction Adapter", - "isRequired": true, - "description": "Choose extractor adapter to extract text" - } - ] - }, - "ioCompatibility": { - "api": { - "sourceSupport": true, - "destinationSupport": true, - "additionalArgs": { - "sync": true - } - }, - "file": { - "sourceSupport": true, - "destinationSupport": true, - "additionalArgs": {} - }, - "db": { - "destinationSupport": true, - "additionalArgs": {} - } - }, - "restrictions": { - "maxFileSize": "50MB", - "allowedFileTypes": [ - "*" - ] - } - }, - "spec": { - "title": "Document classifier tool settings", - "description": "Classify documents based on their content", - "type": "object", - "required": [ - "classificationBins" - ], - "properties": { - "classificationBins": { - "type": "array", - "title": "Classification bins", - "description": "Specify at least two unique classification bins. 'unknown' and '__unstract_failed' are reserved bins. 'unknown' indicates the LLM can't determine the classification and `__unstract_failed` indicates a tool run failure for the given file.", - "items": { - "type": "string" - }, - "minItems": 2, - "uniqueItems": true - }, - "useCache": { - "type": "boolean", - "title": "Cache and use cached results", - "default": true, - "description": "Use cached results" - } - } - }, - "variables": { - "title": "Runtime Variables", - "description": "Runtime Variables for classifier", - "type": "object", - "required": [], - "properties": {} - }, - "icon": "\n\n \n \n \n \n \n \n \n \n \n \n \n\n", - "image_url": "docker:unstract/tool-classifier:0.0.40", - "image_name": "unstract/tool-classifier", - "image_tag": "0.0.40" - }, - "text_extractor": { - "tool_uid": "text_extractor", - "properties": { - "schemaVersion": "0.0.1", - "displayName": "Text Extractor Remote Storage", - "functionName": "text_extractor", - "toolVersion": "0.0.39", - "description": "The Text Extractor is a powerful tool designed to convert documents to its text form or Extract texts from documents", - "input": { - "description": "Document" - }, - "output": { - "description": "structured json output" - }, - "result": { - "type": "TXT", - "description": "Text format of input", - "schema": {} - }, - "adapter": { - "textExtractors": [ - { - "isEnabled": true, - "adapterId": "extractorId", - "title": "Text Extraction Adapter", - "isRequired": true, - "description": "Adapter to extract" - } - ] - }, - "ioCompatibility": { - "api": { - "sourceSupport": true, - "destinationSupport": true, - "additionalArgs": { - "sync": true - } - }, - "file": { - "sourceSupport": true, - "destinationSupport": true, - "additionalArgs": {} - }, - "db": { - "destinationSupport": true, - "additionalArgs": {} - } - }, - "restrictions": { - "maxFileSize": "10MB", - "allowedFileTypes": [ - "*" - ] - } - }, - "spec": { - "title": "Text Extractor tool settings", - "description": "Text extraction from documents", - "type": "object", - "required": [], - "properties": {} - }, - "variables": { - "title": "Runtime Variables", - "description": "Runtime Variables for text extractor", - "type": "object", - "required": [], - "properties": { - "X2TEXT_HOST": { - "type": "string", - "title": "X2Text service host", - "description": "The host where the x2text service is running" - }, - "X2TEXT_PORT": { - "type": "string", - "title": "X2Text service port", - "description": "The port where the x2text service is running" - } - } - }, - "icon": "\n\n \n \n \n \n \n \n \n \n \n \n \n\n", - "image_url": "docker:unstract/tool-text-extractor:0.0.39", - "image_name": "unstract/tool-text-extractor", - "image_tag": "0.0.39" - } -}