unique_toolkit 0.5.50__tar.gz → 0.5.51__tar.gz

{unique_toolkit-0.5.50 → unique_toolkit-0.5.51}/CHANGELOG.md

@@ -5,10 +5,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.51] - 2025-01-30
+- Add missing structured output arguments in complete_async
+
 ## [0.5.50] - 2025-01-30
 - Add the possibility to define completion output structure through a pydantic class
 
-
 ## [0.5.49] - 2025-01-24
 - Add `parsed` and `refusal` to `LanguageModelAssistantMessage` to support structured output
 

{unique_toolkit-0.5.50 → unique_toolkit-0.5.51}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.50
+Version: 0.5.51
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -100,10 +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.51] - 2025-01-30
+- Add missing structured output arguments in complete_async
+
 ## [0.5.50] - 2025-01-30
 - Add the possibility to define completion output structure through a pydantic class
 
-
 ## [0.5.49] - 2025-01-24
 - Add `parsed` and `refusal` to `LanguageModelAssistantMessage` to support structured output
 

{unique_toolkit-0.5.50 → unique_toolkit-0.5.51}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.5.50"
+version = "0.5.51"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",

{unique_toolkit-0.5.50 → unique_toolkit-0.5.51}/unique_toolkit/language_model/service.py

@@ -51,6 +51,8 @@ class LanguageModelService(BaseService):
             temperature (float): The temperature value. Defaults to 0.
             timeout (int): The timeout value in milliseconds. Defaults to 240_000.
             tools (Optional[list[LanguageModelTool]]): The tools to use. Defaults to None.
+            structured_output_model (Optional[Type[BaseModel]]): The structured output model. Defaults to None.
+            structured_output_enforce_schema (bool): Whether to enforce the schema. Defaults to False.
             other_options (Optional[dict]): The other options to use. Defaults to None.
 
         Returns:
@@ -110,6 +112,8 @@ class LanguageModelService(BaseService):
             timeout (int): The timeout value in milliseconds for the request. Defaults to 240_000.
             tools (Optional[list[LanguageModelTool]]): Optional list of tools to include in the request.
             other_options (Optional[dict]): The other options to use. Defaults to None.
+            structured_output_model (Optional[Type[BaseModel]]): The structured output model. Defaults to None.
+            structured_output_enforce_schema (bool): Whether to enforce the schema. Defaults to False.
             logger (Optional[logging.Logger], optional): The logger used to log errors. Defaults to the logger for the current module.
 
         Returns:
@@ -151,6 +155,8 @@ class LanguageModelService(BaseService):
         temperature: float = DEFAULT_COMPLETE_TEMPERATURE,
         timeout: int = DEFAULT_COMPLETE_TIMEOUT,
         tools: Optional[list[LanguageModelTool]] = None,
+        structured_output_model: Optional[Type[BaseModel]] = None,
+        structured_output_enforce_schema: bool = False,
         other_options: Optional[dict] = None,
     ) -> LanguageModelResponse:
         """
@@ -166,6 +172,8 @@ class LanguageModelService(BaseService):
             temperature (float): The temperature setting for the completion. Defaults to 0.0.
             timeout (int): The timeout value in milliseconds for the request. Defaults to 240,000.
             tools (Optional[list[LanguageModelTool]]): Optional list of tools to include in the request.
+            structured_output_model (Optional[Type[BaseModel]]): The structured output model. Defaults to None.
+            structured_output_enforce_schema (bool): Whether to enforce the schema. Defaults to False.
             other_options (Optional[dict]): The other options to use. Defaults to None.
         Returns:
             LanguageModelResponse: The response object containing the completed result.
@@ -182,6 +190,8 @@ class LanguageModelService(BaseService):
             tools=tools,
             other_options=other_options,
             logger=self.logger,
+            structured_output_model=structured_output_model,
+            structured_output_enforce_schema=structured_output_enforce_schema,
         )
 
     def stream_complete(