[FIX] Switch litellm to Zipstack fork after PyPI quarantine

backend/sample.env

@@ -100,9 +100,9 @@ PROMPT_STUDIO_FILE_PATH=/app/prompt-studio-data
 
 # Structure Tool Image (Runs prompt studio exported tools)
 # https://hub.docker.com/r/unstract/tool-structure
-STRUCTURE_TOOL_IMAGE_URL="docker:unstract/tool-structure:0.0.99"
+STRUCTURE_TOOL_IMAGE_URL="docker:unstract/tool-structure:0.0.100"
 STRUCTURE_TOOL_IMAGE_NAME="unstract/tool-structure"
-STRUCTURE_TOOL_IMAGE_TAG="0.0.99"
+STRUCTURE_TOOL_IMAGE_TAG="0.0.100"
 
 # Feature Flags
 EVALUATION_SERVER_IP=unstract-flipt

docker/dockerfiles/worker-unified.Dockerfile

@@ -24,6 +24,7 @@ RUN apt-get update \
        build-essential \
        curl \
        gcc \
+       git \
        libmagic-dev \
        libssl-dev \
        pkg-config \

tools/classifier/Dockerfile

@@ -10,7 +10,7 @@ ENV \
     PIP_DEFAULT_TIMEOUT=120
 
 # Install dependencies for unstructured library's partition
-RUN apt-get update && apt-get --no-install-recommends -y install dumb-init libmagic-dev poppler-utils\
+RUN apt-get update && apt-get --no-install-recommends -y install dumb-init git libmagic-dev poppler-utils\
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 

tools/classifier/src/config/properties.json

@@ -2,7 +2,7 @@
   "schemaVersion": "0.0.1",
   "displayName": "File Classifier",
   "functionName": "classify",
-  "toolVersion": "0.0.78",
+  "toolVersion": "0.0.79",
   "description": "Classifies a file into a bin based on its contents",
   "input": {
     "description": "File to be classified"

tools/structure/Dockerfile

@@ -25,7 +25,7 @@ ENV \
 # Install system dependencies in one layer
 RUN apt-get update && \
     apt-get install -y --no-install-recommends \
-    ffmpeg libsm6 libxext6 libmagic-dev poppler-utils \
+    ffmpeg git libsm6 libxext6 libmagic-dev poppler-utils \
     libreoffice freetds-dev freetds-bin dumb-init && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*

tools/structure/src/config/properties.json

@@ -2,7 +2,7 @@
   "schemaVersion": "0.0.1",
   "displayName": "Structure Tool",
   "functionName": "structure_tool",
-  "toolVersion": "0.0.99",
+  "toolVersion": "0.0.100",
   "description": "This is a template tool which can answer set of input prompts designed in the Prompt Studio",
   "input": {
     "description": "File that needs to be indexed and parsed for answers"

tools/text_extractor/Dockerfile

@@ -10,7 +10,7 @@ ENV \
     PIP_DEFAULT_TIMEOUT=120
 
 # Install dependencies for unstructured library's partition
-RUN apt-get update && apt-get --no-install-recommends -y install dumb-init libmagic-dev poppler-utils\
+RUN apt-get update && apt-get --no-install-recommends -y install dumb-init git libmagic-dev poppler-utils\
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 

tools/text_extractor/src/config/properties.json

@@ -2,7 +2,7 @@
   "schemaVersion": "0.0.1",
   "displayName": "Text Extractor",
   "functionName": "text_extractor",
-  "toolVersion": "0.0.74",
+  "toolVersion": "0.0.75",
   "description": "The Text Extractor is a powerful tool designed to convert documents to its text form or Extract texts from documents",
   "input": {
     "description": "Document"

unstract/sdk1/pyproject.toml

@@ -27,7 +27,7 @@ dependencies = [
     "python-dotenv==1.0.1",
     # # Adapter changes
     "tiktoken~=0.9.0",
-    "litellm==1.82.3",
+    "litellm",
     "llama-index>=0.14.13",
     "llama-index-vector-stores-postgres>=0.7.3",
     "llama-index-vector-stores-milvus>=0.9.6",
@@ -71,7 +71,6 @@ docs = ["lazydocs~=0.4.8"]
 test = [
     "parameterized==0.9.0",
     "pytest==8.3.3",
-    "pytest-asyncio>=0.23.0",
     "pytest-mock==3.14.0",
     "pytest-asyncio>=0.24.0",
     "pytest-cov>=6.0.0",
@@ -80,6 +79,7 @@ test = [
 
 [tool.uv.sources]
 unstract-core = { path = "../core", editable = true }
+litellm = { git = "https://github.com/Zipstack/litellm.git", tag = "v1.82.3" }
 
 [build-system]
 requires = ["hatchling"]