unique_toolkit 0.5.47__py3-none-any.whl → 0.5.49__py3-none-any.whl

unique_toolkit/language_model/schemas.py

@@ -35,7 +35,7 @@ class LanguageModelMessageRole(StrEnum):
 class LanguageModelFunction(BaseModel):
     model_config = model_config
 
-    id: Optional[str] = None
+    id: str | None = None
     name: str
     arguments: Optional[dict[str, Any] | str] = None  # type: ignore
 
@@ -63,8 +63,8 @@ class LanguageModelFunction(BaseModel):
 class LanguageModelFunctionCall(BaseModel):
     model_config = model_config
 
-    id: Optional[str] = None
-    type: Optional[str] = None
+    id: str | None = None
+    type: str | None = None
     function: LanguageModelFunction
 
     @staticmethod
@@ -88,10 +88,17 @@ class LanguageModelFunctionCall(BaseModel):
 class LanguageModelMessage(BaseModel):
     model_config = model_config
     role: LanguageModelMessageRole
-    content: Optional[str | list[dict]] = None
+    content: str | list[dict] | None = None
 
     def __str__(self):
-        return format_message(self.role.capitalize(), message=self.content, num_tabs=1)
+        if not self.content:
+            message = ""
+        if isinstance(self.content, str):
+            message = self.content
+        elif isinstance(self.content, list):
+            message = json.dumps(self.content)
+
+        return format_message(self.role.capitalize(), message=message, num_tabs=1)
 
 
 class LanguageModelSystemMessage(LanguageModelMessage):
@@ -112,7 +119,9 @@ class LanguageModelUserMessage(LanguageModelMessage):
 
 class LanguageModelAssistantMessage(LanguageModelMessage):
     role: LanguageModelMessageRole = LanguageModelMessageRole.ASSISTANT
-    tool_calls: Optional[list[LanguageModelFunctionCall]] = None
+    parsed: dict | None = None
+    refusal: str | None = None
+    tool_calls: list[LanguageModelFunctionCall] | None = None
 
     @field_validator("role", mode="before")
     def set_role(cls, value):
@@ -173,10 +182,12 @@ class LanguageModelStreamResponseMessage(BaseModel):
     model_config = model_config
 
     id: str
-    previous_message_id: str
+    previous_message_id: (
+        str | None
+    )  # Stream response can return a null previous_message_id if an assisstant message is manually added
     role: LanguageModelMessageRole
     text: str
-    original_text: Optional[str] = None
+    original_text: str | None = None
     references: list[dict[str, list | dict | str | int | float | bool]] = []  # type: ignore
 
     # TODO make sdk return role in lowercase
@@ -244,7 +255,9 @@ class LanguageModelTool(BaseModel):
         description="Name must adhere to the pattern ^[a-zA-Z_-]+$",
     )
     description: str
-    parameters: LanguageModelToolParameters
+    parameters: (
+        LanguageModelToolParameters | dict
+    )  # dict represents json schema dumped from pydantic
     returns: LanguageModelToolParameterProperty | LanguageModelToolParameters | None = (
         None
     )

{unique_toolkit-0.5.47.dist-info → unique_toolkit-0.5.49.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.47
+Version: 0.5.49
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -100,6 +100,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.5.49] - 2025-01-24
+- Add `parsed` and `refusal` to `LanguageModelAssistantMessage` to support structured output
+
+## [0.5.48] - 2025-01-19
+- Added the possibility define tool parameters with a json schema (Useful when generating tool parameters from a pydantic object)
+
 ## [0.5.47] - 2025-01-07
 - Added a message builder to build language model messages conveniently without importing all different messages.
 - Move tool_calls to assistant message as not needed anywhere else.

{unique_toolkit-0.5.47.dist-info → unique_toolkit-0.5.49.dist-info}/RECORD

@@ -39,12 +39,12 @@ unique_toolkit/language_model/__init__.py,sha256=hgk5yiFF4SpIcE2QSoki9YknFxmcKnq
 unique_toolkit/language_model/builder.py,sha256=nsRqWO_2dgFehK5CgtqR5aqXgYUU0QL6mR0lALPrQXM,1898
 unique_toolkit/language_model/infos.py,sha256=NgoV05ausVWMqrYqgH6i3s7tYG7mejupROIF_bwEGZo,13050
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
-unique_toolkit/language_model/schemas.py,sha256=3en2aV9I3mD6IzugCRsq33vh9-JPNobszwBTw-LsaJQ,7000
+unique_toolkit/language_model/schemas.py,sha256=87511yupvea-U6sfKWfelETevNMVPevhj7mEqX5FszU,7461
 unique_toolkit/language_model/service.py,sha256=brNCPRA0XxgqHi2rI5i2lyFCkUiw4MNMe1VaR3UgWmY,15500
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
 unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJs8FEZXcgQTNenw,1406
 unique_toolkit/short_term_memory/service.py,sha256=Jd9P72-VvJy7hnqNrjmrmB5BHmsKuOpTiT0Jr-dBbsQ,1682
-unique_toolkit-0.5.47.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.5.47.dist-info/METADATA,sha256=idzuuHNWe1HqAkrEbASKasyAlzWwpsm6_3tI5er7b2c,15544
-unique_toolkit-0.5.47.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.5.47.dist-info/RECORD,,
+unique_toolkit-0.5.49.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.5.49.dist-info/METADATA,sha256=y2wgnOvbhd2PaXa5wshMxjmQ-hNrkaKhqalKAtr5K4E,15819
+unique_toolkit-0.5.49.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.5.49.dist-info/RECORD,,