fabricatio 0.2.13.dev3__cp312-cp312-win_amd64.whl → 0.3.13__cp312-cp312-win_amd64.whl

fabricatio/models/generic.py

@@ -6,13 +6,7 @@ from pathlib import Path
 from typing import Any, Callable, Dict, Iterable, List, Mapping, Optional, Self, Type, Union, final, overload
 
 import ujson
-from fabricatio.config import configs
-from fabricatio.fs.readers import safe_text_read
-from fabricatio.journal import logger
-from fabricatio.parser import JsonCapture
-from fabricatio.rust import blake3_hash, detect_language
-from fabricatio.rust_instances import TEMPLATE_MANAGER
-from fabricatio.utils import ok
+from fabricatio.rust import CONFIG, TEMPLATE_MANAGER, blake3_hash, detect_language
 from litellm.utils import token_counter
 from pydantic import (
     BaseModel,
@@ -27,6 +21,12 @@ from pydantic import (
 )
 from pydantic.json_schema import GenerateJsonSchema, JsonSchemaValue
 
+from fabricatio.fs import dump_text
+from fabricatio.fs.readers import safe_text_read
+from fabricatio.journal import logger
+from fabricatio.parser import JsonCapture
+from fabricatio.utils import ok
+
 
 class Base(BaseModel):
     """Base class for all models with Pydantic configuration.
@@ -74,9 +74,9 @@ class Display(Base):
             str: Combined display output with boundary markers
         """
         return (
-            "--- Start of Extra Info Sequence ---"
-            + "\n".join(d.compact() if compact else d.display() for d in seq)
-            + "--- End of Extra Info Sequence ---"
+                "--- Start of Extra Info Sequence ---"
+                + "\n".join(d.compact() if compact else d.display() for d in seq)
+                + "--- End of Extra Info Sequence ---"
         )
 
 
@@ -117,7 +117,7 @@ class WordCount(Base):
     """Expected word count of this research component."""
 
 
-class FromMapping(Base):
+class FromMapping:
     """Class that provides a method to generate a list of objects from a mapping."""
 
     @classmethod
@@ -126,7 +126,7 @@ class FromMapping(Base):
         """Generate a list of objects from a mapping."""
 
 
-class AsPrompt(Base):
+class AsPrompt:
     """Class that provides a method to generate a prompt from the model.
 
     This class includes a method to generate a prompt based on the model's attributes.
@@ -140,7 +140,7 @@ class AsPrompt(Base):
             str: The generated prompt.
         """
         return TEMPLATE_MANAGER.render_template(
-            configs.templates.as_prompt_template,
+            CONFIG.templates.as_prompt_template,
             self._as_prompt_inner(),
         )
 
@@ -178,13 +178,16 @@ class WithRef[T](Base):
         )
 
     @overload
-    def update_ref[S: WithRef](self: S, reference: T) -> S: ...
+    def update_ref[S: WithRef](self: S, reference: T) -> S:
+        ...
 
     @overload
-    def update_ref[S: WithRef](self: S, reference: "WithRef[T]") -> S: ...
+    def update_ref[S: WithRef](self: S, reference: "WithRef[T]") -> S:
+        ...
 
     @overload
-    def update_ref[S: WithRef](self: S, reference: None = None) -> S: ...
+    def update_ref[S: WithRef](self: S, reference: None = None) -> S:
+        ...
 
     def update_ref[S: WithRef](self: S, reference: Union[T, "WithRef[T]", None] = None) -> S:
         """Update the reference of the object.
@@ -201,19 +204,6 @@ class WithRef[T](Base):
             self._reference = reference  # pyright: ignore [reportAttributeAccessIssue]
         return self
 
-    def derive[S: WithRef](self: S, reference: Any) -> S:
-        """Derive a new object from the current object.
-
-        Args:
-            reference (Any): The reference for the new object.
-
-        Returns:
-            S: A new instance derived from the current object with the provided reference.
-        """
-        new = self.model_copy()
-        new._reference = reference
-        return new
-
 
 class PersistentAble(Base):
     """Class providing file persistence capabilities.
@@ -306,7 +296,7 @@ class PersistentAble(Base):
         return cls.model_validate_json(safe_text_read(path))
 
 
-class Language(Base):
+class Language:
     """Class that provides a language attribute."""
 
     @property
@@ -318,8 +308,7 @@ class Language(Base):
             return detect_language(self.title)
         if isinstance(self, Named) and self.name:
             return detect_language(self.name)
-
-        return detect_language(self.model_dump_json(by_alias=True))
+        raise RuntimeError(f"Cannot determine language! class that not support language: {self.__class__.__name__}")
 
 
 class ModelHash(Base):
@@ -337,7 +326,7 @@ class ModelHash(Base):
         return hash(self.model_dump_json())
 
 
-class UpdateFrom(Base):
+class UpdateFrom:
     """Class that provides a method to update the object from another object.
 
     This class includes methods to update the current object with the attributes of another object.
@@ -386,25 +375,7 @@ class UpdateFrom(Base):
         return self.update_pre_check(other).update_from_inner(other)
 
 
-class ResolveUpdateConflict(Base):
-    """Class that provides a method to update the object from another object.
-
-    This class includes a method to resolve conflicts when updating the object from another object.
-    """
-
-    @abstractmethod
-    def resolve_update_conflict(self, other: Self) -> str:
-        """Resolve the update conflict between two objects.
-
-        Args:
-            other (Self): The other object to resolve the update conflict with.
-
-        Returns:
-            str: The resolved update conflict.
-        """
-
-
-class Introspect(Base):
+class Introspect:
     """Class that provides a method to introspect the object.
 
     This class includes a method to perform internal introspection of the object.
@@ -498,12 +469,12 @@ class CreateJsonObjPrompt(WithFormatedJsonSchema):
         """
         if isinstance(requirement, str):
             return TEMPLATE_MANAGER.render_template(
-                configs.templates.create_json_obj_template,
+                CONFIG.templates.create_json_obj_template,
                 {"requirement": requirement, "json_schema": cls.formated_json_schema()},
             )
         return [
             TEMPLATE_MANAGER.render_template(
-                configs.templates.create_json_obj_template,
+                CONFIG.templates.create_json_obj_template,
                 {"requirement": r, "json_schema": cls.formated_json_schema()},
             )
             for r in requirement
@@ -575,7 +546,7 @@ class FinalizedDumpAble(Base):
         Returns:
             Self: The current instance of the object.
         """
-        Path(path).write_text(self.finalized_dump(), encoding="utf-8")
+        dump_text(path, self.finalized_dump())
         return self
 
 
@@ -655,7 +626,7 @@ class WithDependency(Base):
         from fabricatio.fs import MAGIKA
 
         return TEMPLATE_MANAGER.render_template(
-            configs.templates.dependencies_template,
+            CONFIG.templates.dependencies_template,
             {
                 (pth := Path(p)).name: {
                     "path": pth.as_posix(),
@@ -672,7 +643,7 @@ class WithDependency(Base):
         )
 
 
-class Vectorizable(Base):
+class Vectorizable:
     """Class that prepares the vectorization of the model.
 
     This class includes methods to prepare the model for vectorization, ensuring it fits within a specified token length.
@@ -695,7 +666,7 @@ class Vectorizable(Base):
         Raises:
             ValueError: If the chunk exceeds the maximum sequence length.
         """
-        max_length = max_length or configs.embedding.max_sequence_length
+        max_length = max_length or CONFIG.embedding.max_sequence_length
         chunk = self._prepare_vectorization_inner()
         if max_length and (length := token_counter(text=chunk)) > max_length:
             raise ValueError(f"Chunk exceeds maximum sequence length {max_length}, got {length}, see \n{chunk}")
@@ -883,7 +854,8 @@ class Patch[T](ProposedAble):
             # copy the desc info of each corresponding fields from `ref_cls`
             for field_name in [f for f in cls.model_fields if f in ref_cls.model_fields]:
                 my_schema["properties"][field_name]["description"] = (
-                    ref_cls.model_fields[field_name].description or my_schema["properties"][field_name]["description"]
+                        ref_cls.model_fields[field_name].description or my_schema["properties"][field_name][
+                    "description"]
                 )
             my_schema["description"] = ref_cls.__doc__
 

fabricatio/models/role.py

@@ -2,15 +2,16 @@
 
 from typing import Any, Self, Set
 
+from fabricatio.rust import Event
+from pydantic import Field, ConfigDict
+
 from fabricatio.capabilities.propose import Propose
 from fabricatio.core import env
 from fabricatio.journal import logger
 from fabricatio.models.action import WorkFlow
-from fabricatio.models.events import Event
 from fabricatio.models.generic import WithBriefing
 from fabricatio.models.tool import ToolBox
 from fabricatio.models.usages import ToolBoxUsage
-from pydantic import Field
 
 
 class Role(WithBriefing, Propose, ToolBoxUsage):
@@ -23,7 +24,8 @@ class Role(WithBriefing, Propose, ToolBoxUsage):
         registry: Mapping of events to workflows that handle them
         toolboxes: Set of toolboxes available to this role and its workflows
     """
-
+    # fixme: not use arbitrary_types_allowed
+    model_config = ConfigDict(use_attribute_docstrings=True, arbitrary_types_allowed=True)
     description: str = ""
     """A brief description of the role's responsibilities and capabilities."""
 

fabricatio/models/task.py

@@ -4,16 +4,16 @@ It includes methods to manage the task's lifecycle, such as starting, finishing,
 """
 
 from asyncio import Queue
-from typing import Any, Dict, List, Optional, Self
+from typing import Any, Dict, List, Optional, Self, Union
+
+from fabricatio.rust import CONFIG, TEMPLATE_MANAGER, Event, TaskStatus
+from pydantic import Field, PrivateAttr
 
-from fabricatio.config import configs
-from fabricatio.constants import TaskStatus
 from fabricatio.core import env
 from fabricatio.journal import logger
-from fabricatio.models.events import Event, EventLike
 from fabricatio.models.generic import ProposedAble, WithBriefing, WithDependency
-from fabricatio.rust_instances import TEMPLATE_MANAGER
-from pydantic import Field, PrivateAttr
+
+type EventLike = Union[str, Event, List[str]]
 
 
 class Task[T](WithBriefing, ProposedAble, WithDependency):
@@ -33,7 +33,7 @@ class Task[T](WithBriefing, ProposedAble, WithDependency):
     description: str = Field(default="")
     """A detailed explanation of the task that includes all necessary information. Should be clear and answer what, why, when, where, who, and how questions."""
 
-    goals: List[str] = Field(default=[])
+    goals: List[str] = Field(default_factory=list)
     """A list of objectives that the task aims to accomplish. Each goal should be clear and specific. Complex tasks should be broken into multiple smaller goals."""
 
     namespace: List[str] = Field(default_factory=list)
@@ -65,7 +65,7 @@ class Task[T](WithBriefing, ProposedAble, WithDependency):
 
     def model_post_init(self, __context: Any) -> None:
         """Initialize the task with a namespace event."""
-        self._namespace.segments.extend(self.namespace)
+        self._namespace.concat(self.namespace)
 
     def move_to(self, new_namespace: EventLike) -> Self:
         """Move the task to a new namespace.
@@ -266,7 +266,7 @@ class Task[T](WithBriefing, ProposedAble, WithDependency):
             str: The briefing of the task.
         """
         return TEMPLATE_MANAGER.render_template(
-            configs.templates.task_briefing_template,
+            CONFIG.templates.task_briefing_template,
             self.model_dump(),
         )
 

fabricatio/models/tool.py

@@ -10,11 +10,12 @@ from inspect import iscoroutinefunction, signature
 from types import CodeType, ModuleType
 from typing import Any, Callable, Dict, List, Optional, Self, cast, overload
 
-from fabricatio.config import configs
+from fabricatio.rust import CONFIG
+from pydantic import BaseModel, ConfigDict, Field
+
 from fabricatio.decorators import logging_execution_info, use_temp_module
 from fabricatio.journal import logger
 from fabricatio.models.generic import WithBriefing
-from pydantic import BaseModel, ConfigDict, Field
 
 
 class Tool[**P, R](WithBriefing):
@@ -210,7 +211,7 @@ class ToolExecutor(BaseModel):
             M: The module with injected tools.
         """
         module = module or cast(
-            "M", module_from_spec(spec=ModuleSpec(name=configs.toolbox.tool_module_name, loader=None))
+            "M", module_from_spec(spec=ModuleSpec(name=CONFIG.toolbox.tool_module_name, loader=None))
         )
         for tool in self.candidates:
             logger.debug(f"Injecting tool: {tool.name}")
@@ -229,7 +230,7 @@ class ToolExecutor(BaseModel):
             M: The module with injected data.
         """
         module = module or cast(
-            'M', module_from_spec(spec=ModuleSpec(name=configs.toolbox.data_module_name, loader=None))
+            'M', module_from_spec(spec=ModuleSpec(name=CONFIG.toolbox.data_module_name, loader=None))
         )
         for key, value in self.data.items():
             logger.debug(f"Injecting data: {key}")