unique_toolkit 0.6.2__py3-none-any.whl → 0.6.4__py3-none-any.whl

unique_toolkit/__init__.py

@@ -0,0 +1,17 @@
+# Re-export commonly used classes for easier imports
+from unique_toolkit.chat import ChatService
+from unique_toolkit.content import ContentService
+from unique_toolkit.embedding import EmbeddingService
+from unique_toolkit.language_model import LanguageModelMessages, LanguageModelService
+from unique_toolkit.short_term_memory import ShortTermMemoryService
+
+# You can add other classes you frequently use here as well
+
+__all__ = [
+    "LanguageModelService",
+    "LanguageModelMessages",
+    "ChatService",
+    "ContentService",
+    "EmbeddingService",
+    "ShortTermMemoryService",
+]

unique_toolkit/language_model/builder.py

@@ -1,3 +1,5 @@
+from typing_extensions import Self
+
 from unique_toolkit.language_model import (
     LanguageModelAssistantMessage,
     LanguageModelMessage,
@@ -12,25 +14,38 @@ class MessagesBuilder:
     def __init__(self):
         self.messages: list[LanguageModelMessage] = []
 
-    def system_message_append(self, content):
-        """Appends a user message to the messages list."""
+    def system_message_append(self, content: str) -> Self:
+        """Appends a system message to the messages list."""
         message = LanguageModelSystemMessage(content=content)
         self.messages.append(message)
         return self  # Return self to allow method chaining
 
-    def user_message_append(self, content):
+    def user_message_append(self, content: str) -> Self:
         """Appends a user message to the messages list."""
         message = LanguageModelUserMessage(content=content)
         self.messages.append(message)
         return self  # Return self to allow method chaining
 
-    def assistant_message_append(self, content):
+    def image_message_append(self, content: str, images: list[str]) -> Self:
+        message = LanguageModelUserMessage(
+            content=[
+                {"type": "text", "text": content},
+                *[
+                    {"type": "image_url", "imageUrl": {"url": image}}
+                    for image in images
+                ],
+            ]
+        )
+        self.messages.append(message)
+        return self
+
+    def assistant_message_append(self, content: str) -> Self:
         """Appends an assistant message to the messages list."""
         message = LanguageModelAssistantMessage(content=content)
         self.messages.append(message)
         return self  # Return self to allow method chaining
 
-    def tool_message_append(self, name: str, tool_call_id: str, content: str):
+    def tool_message_append(self, name: str, tool_call_id: str, content: str) -> Self:
         """Appends a tool message to the messages list."""
         message = LanguageModelToolMessage(
             name=name, tool_call_id=tool_call_id, content=content

unique_toolkit/language_model/schemas.py

@@ -165,6 +165,14 @@ class LanguageModelMessages(RootModel):
     def __getitem__(self, item):
         return self.root[item]
 
+    def builder(self):
+        """Returns a MessagesBuilder instance pre-populated with existing messages."""
+        from unique_toolkit.language_model.builder import MessagesBuilder
+
+        builder = MessagesBuilder()
+        builder.messages = self.root.copy()  # Start with existing messages
+        return builder
+
 
 class LanguageModelCompletionChoice(BaseModel):
     model_config = model_config

{unique_toolkit-0.6.2.dist-info → unique_toolkit-0.6.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.6.2
+Version: 0.6.4
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -111,6 +111,13 @@ 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.6.3] - 2025-02-27
+- Simplified imports for services. `from unique_toolkit.language_model import LanguageModelService` -> `from unique_toolkit import LanguageModelService` to reduce number of import lines.
+
+## [0.6.3] - 2025-02-26
+- Add `builder` method to `LanguageModelMessages` class
+
 ## [0.6.2] - 2025-02-25
 - Deprecate `LanguageModel` in favor of `LanguageModelInfo`
 - `LanguageModelTokenLimits` properties become mandatory, initialization allows 

{unique_toolkit-0.6.2.dist-info → unique_toolkit-0.6.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-unique_toolkit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+unique_toolkit/__init__.py,sha256=waK7W0EK3v2RJ26hawccwVz1i3yHGvHIIu5qgGjEGHQ,583
 unique_toolkit/_common/_base_service.py,sha256=S8H0rAebx7GsOldA7xInLp3aQJt9yEPDQdsGSFRJsGg,276
 unique_toolkit/_common/_time_utils.py,sha256=ztmTovTvr-3w71Ns2VwXC65OKUUh-sQlzbHdKTQWm-w,135
 unique_toolkit/_common/exception.py,sha256=caQIE1btsQnpKCHqL2cgWUSbHup06enQu_Pt7uGUTTE,727
@@ -45,12 +45,12 @@ unique_toolkit/evaluators/hallucination/utils.py,sha256=4KTJH8low_fBzOcuVlcHB2FR
 unique_toolkit/evaluators/output_parser.py,sha256=eI72qkzK1dZyUvnfP2SOAQCGBj_-PwX5wy_aLPMsJMY,883
 unique_toolkit/evaluators/schemas.py,sha256=Jaue6Uhx75X1CyHKWj8sT3RE1JZXTqoLtfLt2xQNCX8,2507
 unique_toolkit/language_model/__init__.py,sha256=jWko_vQj48wjnpTtlkg8iNdef0SMI3FN2kGywXRTMzg,1880
-unique_toolkit/language_model/builder.py,sha256=nsRqWO_2dgFehK5CgtqR5aqXgYUU0QL6mR0lALPrQXM,1898
+unique_toolkit/language_model/builder.py,sha256=qP1SlUnYJHLqT-fpXs4lgUixnekhx8IIfuoXnMHvRKE,2408
 unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
 unique_toolkit/language_model/functions.py,sha256=I5jHhHsKoq7GwEQyTrM8LXB2n_6dvMAk7UklenjuHSY,7945
 unique_toolkit/language_model/infos.py,sha256=-axWHj55mp6tZfX_3i-FSkfh8e9fwORXWMfi9xQ_UjA,12232
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
-unique_toolkit/language_model/schemas.py,sha256=76FtgWc0qtk9fpEwoecA59WxG__aR7_kYyWJooHIaF8,8297
+unique_toolkit/language_model/schemas.py,sha256=rrwzUgKANFOrdehCULW8Hh03uRW3tsE5dXpWqxmClfg,8618
 unique_toolkit/language_model/service.py,sha256=GupYD4uDZjy1TfVQW3jichmgQwiSgQCj350FtL4O0W4,5569
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
 unique_toolkit/short_term_memory/__init__.py,sha256=2mI3AUrffgH7Yt-xS57EGqnHf7jnn6xquoKEhJqk3Wg,185
@@ -58,7 +58,7 @@ unique_toolkit/short_term_memory/constants.py,sha256=698CL6-wjup2MvU19RxSmQk3gX7
 unique_toolkit/short_term_memory/functions.py,sha256=3WiK-xatY5nh4Dr5zlDUye1k3E6kr41RiscwtTplw5k,4484
 unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJs8FEZXcgQTNenw,1406
 unique_toolkit/short_term_memory/service.py,sha256=gdsVzoNqTXmLoBR_-p_lJlZDBo8L7Cr5EKchTNVJg1Q,5233
-unique_toolkit-0.6.2.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.6.2.dist-info/METADATA,sha256=X3vk5KZ3npmcio6kiyCcOKhw2WE7g4zLCkA-urhZsSA,19409
-unique_toolkit-0.6.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.6.2.dist-info/RECORD,,
+unique_toolkit-0.6.4.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.6.4.dist-info/METADATA,sha256=dSeB-ilRjS8PXAPrssjGar0aCsxTqBqqju2hPiDCYNw,19703
+unique_toolkit-0.6.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.6.4.dist-info/RECORD,,