Small fixes in Argilla steps

Fix `ImportError` for `argilla` in `TextGenerationToArgilla`

src/distilabel/steps/argilla/base.py

@@ -61,11 +61,17 @@ class Argilla(Step, ABC):
         - dynamic, based on the `inputs` value provided
     """
 
-    dataset_name: str
-    dataset_workspace: Optional[str] = None
+    dataset_name: RuntimeParameter[str] = Field(
+        default=None, description="The name of the dataset in Argilla."
+    )
+    dataset_workspace: Optional[RuntimeParameter[str]] = Field(
+        default=None,
+        description="The workspace where the dataset will be created in Argilla. Defaults"
+        "to `None` which means it will be created in the default workspace.",
+    )
 
     api_url: Optional[RuntimeParameter[str]] = Field(
-        default_factory=lambda: os.getenv("ARGILLA_BASE_URL"),
+        default_factory=lambda: os.getenv("ARGILLA_API_URL"),
         description="The base URL to use for the Argilla API requests.",
     )
     api_key: Optional[RuntimeParameter[SecretStr]] = Field(
@@ -122,6 +128,8 @@ def load(self) -> None:
         """
         super().load()
 
+        self._rg_init()
+
     @property
     @abstractmethod
     def inputs(self) -> List[str]:

src/distilabel/steps/argilla/preference.py

@@ -88,8 +88,6 @@ def load(self) -> None:
         """
         super().load()
 
-        self._rg_init()
-
         # Both `instruction` and `generations` will be used as the fields of the dataset
         self._instruction = self.input_mappings.get("instruction", "instruction")
         self._generations = self.input_mappings.get("generations", "generations")

src/distilabel/steps/argilla/text_generation.py

@@ -20,10 +20,8 @@
 
 try:
     import argilla as rg
-except ImportError as ie:
-    raise ImportError(
-        "Argilla is not installed. Please install it using `pip install argilla`."
-    ) from ie
+except ImportError:
+    pass
 
 from distilabel.steps.argilla.base import Argilla
 from distilabel.steps.base import StepInput
@@ -71,8 +69,6 @@ def load(self) -> None:
         """
         super().load()
 
-        self._rg_init()
-
         self._instruction = self.input_mappings.get("instruction", "instruction")
         self._generation = self.input_mappings.get("generation", "generation")