chore: upgrade distilabel version to 1.5.0 (#682)

Author: sbetlewski

examples/evaluate/dataset-generator/config/generate.yaml

@@ -11,21 +11,21 @@ tasks:
       prompt_class: ragbits.evaluate.dataset_generator.prompts.corpus_generation:BasicCorpusGenerationPrompt
   - type: ragbits.evaluate.dataset_generator.tasks.text_generation.qa:QueryGenTask
     llm:
-      provider_type: distilabel.llms:OpenAILLM
+      provider_type: distilabel.models:OpenAILLM
       kwargs:
         model: gpt-4o
     kwargs:
       prompt_class: ragbits.evaluate.dataset_generator.prompts.qa:QueryGenPrompt
   - type: ragbits.evaluate.dataset_generator.tasks.text_generation.qa:AnswerGenTask
     llm:
-      provider_type: distilabel.llms:OpenAILLM
+      provider_type: distilabel.models:OpenAILLM
       kwargs:
         model: gpt-4o
     kwargs:
       prompt_class: ragbits.evaluate.dataset_generator.prompts.qa:BasicAnswerGenPrompt
   - type: ragbits.evaluate.dataset_generator.tasks.text_generation.qa:PassagesGenTask
     llm:
-      provider_type: distilabel.llms:OpenAILLM
+      provider_type: distilabel.models:OpenAILLM
       kwargs:
         model: gpt-4o
     kwargs:

packages/ragbits-evaluate/CHANGELOG.md

@@ -4,6 +4,7 @@
 
 - Update qa data loader docstring (#565)
 - Fix deadlock on qa metrics compute (#609)
+- Upgrade distilabel version to 1.5.0 (#682)
 
 ## 1.0.0 (2025-06-04)
 

packages/ragbits-evaluate/pyproject.toml

@@ -32,7 +32,7 @@ classifiers = [
     "Topic :: Scientific/Engineering :: Artificial Intelligence",
     "Topic :: Software Development :: Libraries :: Python Modules",
 ]
-dependencies = ["hydra-core>=1.3.2,<2.0.0", "neptune[optuna]>=1.12.0,<2.0.0", "optuna>=4.0.0,<5.0.0", "distilabel>=1.4.1,<1.5.0", "datasets>=3.0.1,<4.0.0", "ragbits-core==1.0.0"]
+dependencies = ["hydra-core>=1.3.2,<2.0.0", "neptune[optuna]>=1.12.0,<2.0.0", "optuna>=4.0.0,<5.0.0", "distilabel>=1.5.0,<2.0.0", "datasets>=3.0.1,<4.0.0", "ragbits-core==1.0.0"]
 
 [project.urls]
 "Homepage" = "https://github.com/deepsense-ai/ragbits"

packages/ragbits-evaluate/src/ragbits/evaluate/dataset_generator/tasks/text_generation/base.py

@@ -2,7 +2,7 @@
 from abc import ABC, abstractmethod
 from typing import Any
 
-from distilabel.llms.base import LLM
+from distilabel.models import LLM
 from distilabel.steps.tasks import TextGeneration
 
 from ragbits.core.prompt import ChatFormat, Prompt

packages/ragbits-evaluate/src/ragbits/evaluate/dataset_generator/tasks/text_generation/qa.py

@@ -1,6 +1,6 @@
 from typing import Any
 
-from distilabel.llms.base import LLM
+from distilabel.models import LLM
 
 from ragbits.evaluate.dataset_generator.tasks.text_generation.base import BaseDistilabelTask
 from ragbits.evaluate.dataset_generator.utils import get_closest_substring, get_passages_list