palimpzest 0.7.21__tar.gz → 0.8.1__tar.gz

{palimpzest-0.7.21/src/palimpzest.egg-info → palimpzest-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: palimpzest
-Version: 0.7.21
+Version: 0.8.1
 Summary: Palimpzest is a system which enables anyone to process AI-powered analytical queries simply by defining them in a declarative language
 Author-email: MIT DSG Semantic Management Lab <michjc@csail.mit.edu>
 Project-URL: homepage, https://palimpzest.org
@@ -15,12 +15,14 @@ Classifier: Programming Language :: Python :: 3.8
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: anthropic>=0.55.0
 Requires-Dist: beautifulsoup4>=4.13.4
 Requires-Dist: chromadb>=1.0.15
 Requires-Dist: colorama>=0.4.6
 Requires-Dist: datasets>=4.0.0
 Requires-Dist: fastapi~=0.115.0
 Requires-Dist: gradio>=5.26.0
+Requires-Dist: litellm>=1.73.1
 Requires-Dist: numpy==2.0.2
 Requires-Dist: openai>=1.0
 Requires-Dist: pandas>=2.1.1
@@ -37,6 +39,7 @@ Requires-Dist: requests>=2.25
 Requires-Dist: ruff>=0.9.0
 Requires-Dist: sentence-transformers==5.0.0
 Requires-Dist: setuptools>=70.1.1
+Requires-Dist: smolagents[toolkit]
 Requires-Dist: tabulate>=0.9.0
 Requires-Dist: together>=1.5.5
 Requires-Dist: tqdm~=4.66.1
@@ -46,6 +49,8 @@ Requires-Dist: mkdocs>=1.6.1; extra == "docs"
 Requires-Dist: mkdocs-material>=9.6.3; extra == "docs"
 Requires-Dist: mkdocstrings-python>=1.15.0; extra == "docs"
 Requires-Dist: mkdocs-material[imaging]; extra == "docs"
+Provides-Extra: vllm
+Requires-Dist: vllm>=0.10.1.1; extra == "vllm"
 Dynamic: license-file
 
 ![pz-banner](https://palimpzest-workloads.s3.us-east-1.amazonaws.com/palimpzest-cropped.png)

{palimpzest-0.7.21 → palimpzest-0.8.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "palimpzest"
-version = "0.7.21"
+version = "0.8.1"
 description = "Palimpzest is a system which enables anyone to process AI-powered analytical queries simply by defining them in a declarative language"
 readme = "README.md"
 requires-python = ">=3.8"
@@ -9,12 +9,14 @@ authors = [
     {name="MIT DSG Semantic Management Lab", email="michjc@csail.mit.edu"},
 ]
 dependencies = [
+    "anthropic>=0.55.0",
     "beautifulsoup4>=4.13.4",
     "chromadb>=1.0.15",
     "colorama>=0.4.6",
     "datasets>=4.0.0",
     "fastapi~=0.115.0",
     "gradio>=5.26.0",
+    "litellm>=1.73.1",
     "numpy==2.0.2",
     "openai>=1.0",
     "pandas>=2.1.1",
@@ -31,6 +33,7 @@ dependencies = [
     "ruff>=0.9.0",
     "sentence-transformers==5.0.0",
     "setuptools>=70.1.1",
+    "smolagents[toolkit]",
     "tabulate>=0.9.0",
     "together>=1.5.5",
     "tqdm~=4.66.1",
@@ -52,6 +55,9 @@ docs = [
     "mkdocstrings-python>=1.15.0",
     "mkdocs-material[imaging]",
 ]
+vllm = [
+    "vllm>=0.10.1.1",
+]
 
 [tool.setuptools]
 package-dir = {"" = "src"}

palimpzest-0.8.1/src/palimpzest/__init__.py

@@ -0,0 +1,72 @@
+import logging
+
+from palimpzest.constants import Cardinality, Model
+from palimpzest.core.data.context import Context, TextFileContext
+from palimpzest.core.data.dataset import Dataset
+from palimpzest.core.data.iter_dataset import (
+    AudioFileDataset,
+    HTMLFileDataset,
+    ImageFileDataset,
+    IterDataset,
+    MemoryDataset,
+    PDFFileDataset,
+    TextFileDataset,
+    XLSFileDataset,
+)
+from palimpzest.core.elements.groupbysig import GroupBySig
+from palimpzest.core.lib.schemas import AudioBase64, AudioFilepath, ImageBase64, ImageFilepath, ImageURL
+from palimpzest.policy import (
+    MaxQuality,
+    MaxQualityAtFixedCost,
+    MaxQualityAtFixedTime,
+    MinCost,
+    MinCostAtFixedQuality,
+    MinTime,
+    MinTimeAtFixedQuality,
+    PlanCost,
+    Policy,
+)
+from palimpzest.query.processor.config import QueryProcessorConfig
+from palimpzest.validator.validator import Validator
+
+# Initialize the root logger
+logging.getLogger(__name__).addHandler(logging.NullHandler())
+
+__all__ = [
+    # constants
+    "Cardinality",
+    "Model",
+    # core
+    "GroupBySig",
+    "Context",
+    "TextFileContext",
+    "Dataset",
+    "IterDataset",
+    "AudioFileDataset",
+    "MemoryDataset",
+    "HTMLFileDataset",
+    "ImageFileDataset",
+    "PDFFileDataset",
+    "TextFileDataset",
+    "XLSFileDataset",
+    # schemas
+    "AudioBase64",
+    "AudioFilepath",
+    "ImageBase64",
+    "ImageFilepath",
+    "ImageURL",
+    # policy
+    "MaxQuality",
+    "MaxQualityAtFixedCost",
+    "MaxQualityAtFixedTime",
+    "MinCost",
+    "MinCostAtFixedQuality",
+    "MinTime",
+    "MinTimeAtFixedQuality",
+    "PlanCost",
+    "Policy",
+    # query
+    "QueryProcessorConfig",
+    # validator
+    "Validator",
+]