Fixing dependencies.

Author: monoxgas

docs/api/chat.mdx

@@ -2654,7 +2654,7 @@ def score(
     self.scorers.extend(
         [
             dn.scorers.wrap_chat(
-                scorer if isinstance(scorer, dn.Scorer) else dn.Scorer.from_callable(scorer),
+                scorer if isinstance(scorer, dn.Scorer) else dn.Scorer(scorer),
                 filter=filter,
             )
             for scorer in scorers

docs/api/generator.mdx

@@ -1646,12 +1646,18 @@ Create a new instance of TransformersGenerator from an already loaded model and
   (`Any`)
   –The loaded model for text generation.
 * **`tokenizer`**
+  (`AutoTokenizer`)
   –The tokenizer associated with the model.
 * **`pipeline`**
   (`TextGenerationPipeline | None`, default:
   `None`
   )
   –The text generation pipeline. Defaults to None.
+* **`params`**
+  (`GenerateParams | None`, default:
+  `None`
+  )
+  –Generation parameters. Defaults to None.
 
 **Returns:**
 
@@ -1674,8 +1680,9 @@ def from_obj(
 
     Args:
         model: The loaded model for text generation.
-        tokenizer : The tokenizer associated with the model.
+        tokenizer: The tokenizer associated with the model.
         pipeline: The text generation pipeline. Defaults to None.
+        params: Generation parameters. Defaults to None.
 
     Returns:
         The TransformersGenerator instance.
@@ -2491,12 +2498,18 @@ Create a new instance of TransformersGenerator from an already loaded model and
   (`Any`)
   –The loaded model for text generation.
 * **`tokenizer`**
+  (`AutoTokenizer`)
   –The tokenizer associated with the model.
 * **`pipeline`**
   (`TextGenerationPipeline | None`, default:
   `None`
   )
   –The text generation pipeline. Defaults to None.
+* **`params`**
+  (`GenerateParams | None`, default:
+  `None`
+  )
+  –Generation parameters. Defaults to None.
 
 **Returns:**
 
@@ -2519,8 +2532,9 @@ def from_obj(
 
     Args:
         model: The loaded model for text generation.
-        tokenizer : The tokenizer associated with the model.
+        tokenizer: The tokenizer associated with the model.
         pipeline: The text generation pipeline. Defaults to None.
+        params: Generation parameters. Defaults to None.
 
     Returns:
         The TransformersGenerator instance.

docs/api/model.mdx

@@ -710,7 +710,7 @@ def xml_example(cls) -> str:
         return f"<{cls.__xml_tag__}>{escape_xml(example)}</{cls.__xml_tag__}>"
 
     lines = []
-    attribute_parts = []
+    attribute_parts: list[str] = []
     element_fields = {}
 
     for field_name, field_info in cls.model_fields.items():

docs/api/prompt.mdx

@@ -67,8 +67,8 @@ Prompt(
     max_parsing_rounds: int = DEFAULT_MAX_PARSE_ROUNDS,
     max_tool_rounds: int = DEFAULT_MAX_TOOL_ROUNDS,
     inputs: list[Input] = list(),
-    output: Output = lambda: ChatOutput(
-        id="chat", context=Ctx()
+    output: Output = (
+        lambda: ChatOutput(id="chat", context=(Ctx()))
     )(),
     watch_callbacks: list[WatchChatCallback] = list(),
     then_callbacks: list[ThenChatCallback] = list(),

poetry.lock

@@ -13,7 +13,7 @@ packages = [{ include = "rigging" }]
 [tool.poetry.dependencies]
 python = ">=3.10,<3.14"
 pydantic = "^2.7.3"
-pydantic-xml = "^2.11.0"
+pydantic-xml = "<=2.17.0"
 loguru = "^0.7.2"
 litellm = "^1.67.2"
 xmltodict = "^0.13.0"

pyproject.toml

@@ -59,7 +59,7 @@
 from rigging.util import flatten_list, get_callable_name
 
 if t.TYPE_CHECKING:
-    from dreadnode.metric import Scorer, ScorerCallable
+    from dreadnode.scorers import Scorer, ScorerCallable
     from dreadnode.scorers.rigging import ChatFilterFunction, ChatFilterMode
     from elasticsearch import AsyncElasticsearch  # type: ignore [import-not-found, unused-ignore]
 
@@ -1441,7 +1441,7 @@ def score(
         self.scorers.extend(
             [
                 dn.scorers.wrap_chat(
-                    scorer if isinstance(scorer, dn.Scorer) else dn.Scorer.from_callable(scorer),
+                    scorer if isinstance(scorer, dn.Scorer) else dn.Scorer(scorer),
                     filter=filter,
                 )
                 for scorer in scorers

rigging/chat.py

@@ -116,8 +116,9 @@ def from_obj(
 
         Args:
             model: The loaded model for text generation.
-            tokenizer : The tokenizer associated with the model.
+            tokenizer: The tokenizer associated with the model.
             pipeline: The text generation pipeline. Defaults to None.
+            params: Generation parameters. Defaults to None.
 
         Returns:
             The TransformersGenerator instance.

rigging/generator/transformers_.py

@@ -357,7 +357,7 @@ def xml_example(cls) -> str:
             return f"<{cls.__xml_tag__}>{escape_xml(example)}</{cls.__xml_tag__}>"
 
         lines = []
-        attribute_parts = []
+        attribute_parts: list[str] = []
         element_fields = {}
 
         for field_name, field_info in cls.model_fields.items():