unique_toolkit 0.5.18__py3-none-any.whl → 0.5.20__py3-none-any.whl

unique_toolkit/language_model/schemas.py

@@ -43,7 +43,7 @@ class LanguageModelMessage(BaseModel):
     model_config = model_config
 
     role: LanguageModelMessageRole
-    content: Optional[str] = None
+    content: Optional[str | list[dict]] = None
     name: Optional[str] = None
     tool_calls: Optional[list[LanguageModelFunctionCall]] = None
 

unique_toolkit/language_model/service.py

@@ -75,37 +75,47 @@ class LanguageModelService(BaseService):
             self.logger.error(f"Error completing: {e}")
             raise e
 
-    async def complete_async(
-        self,
+    @classmethod
+    async def complete_async_util(
+        cls,
+        company_id: str,
         messages: LanguageModelMessages,
         model_name: LanguageModelName | str,
         temperature: float = DEFAULT_COMPLETE_TEMPERATURE,
         timeout: int = DEFAULT_COMPLETE_TIMEOUT,
         tools: Optional[list[LanguageModelTool]] = None,
-    ):
+        logger: Optional[logging.Logger] = logging.getLogger(__name__),
+    ) -> LanguageModelResponse:
         """
         Calls the completion endpoint asynchronously without streaming the response.
 
+        This method sends a request to the completion endpoint using the provided messages, model name,
+        temperature, timeout, and optional tools. It returns a `LanguageModelResponse` object containing
+        the completed result.
+
         Args:
+            company_id (str): The company ID associated with the request.
             messages (LanguageModelMessages): The messages to complete.
-            model_name (LanguageModelName | str): The model name.
-            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.
+            model_name (LanguageModelName | str): The model name to use for the completion.
+            temperature (float): The temperature setting for the completion. Defaults to 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.
+            logger (Optional[logging.Logger], optional): The logger used to log errors. Defaults to the logger for the current module.
 
         Returns:
-            str: The completed message content.
+            LanguageModelResponse: The response object containing the completed result.
+
+        Raises:
+            Exception: If an error occurs during the request, an exception is raised and logged.
         """
-        options = self._add_tools_to_options({}, tools)
+        options = cls._add_tools_to_options({}, tools)
         messages = messages.model_dump(exclude_none=True, exclude={"tool_calls"})
         model = (
             model_name.name if isinstance(model_name, LanguageModelName) else model_name
         )
-
         try:
             response = await unique_sdk.ChatCompletion.create_async(
-                company_id=self.event.company_id,
-                # TODO change or extend types in unique_sdk
+                company_id=company_id,
                 model=model,
                 messages=cast(
                     list[unique_sdk.Integrated.ChatCompletionRequestMessage],
@@ -117,9 +127,47 @@ class LanguageModelService(BaseService):
             )
             return LanguageModelResponse(**response)
         except Exception as e:
-            self.logger.error(f"Error completing: {e}")
+            logger.error(f"Error completing: {e}")
             raise e
 
+    async def complete_async(
+        self,
+        messages: LanguageModelMessages,
+        model_name: LanguageModelName | str,
+        temperature: float = DEFAULT_COMPLETE_TEMPERATURE,
+        timeout: int = DEFAULT_COMPLETE_TIMEOUT,
+        tools: Optional[list[LanguageModelTool]] = None,
+    ) -> LanguageModelResponse:
+        """
+        Calls the completion endpoint asynchronously without streaming the response.
+
+        This method utilizes the class method `complete_async_util` to perform the asynchronous completion
+        request using the provided messages, model name, temperature, timeout, and optional tools. It
+        returns a `LanguageModelResponse` object containing the result of the completion.
+
+        Args:
+            messages (LanguageModelMessages): The messages to complete.
+            model_name (LanguageModelName | str): The model name to use for the completion.
+            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.
+
+        Returns:
+            LanguageModelResponse: The response object containing the completed result.
+
+        Raises:
+            Exception: If an error occurs during the completion request.
+        """
+        return await self.complete_async_util(
+            company_id=self.event.company_id,
+            messages=messages,
+            model_name=model_name,
+            temperature=temperature,
+            timeout=timeout,
+            tools=tools,
+            logger=self.logger,
+        )
+
     def stream_complete(
         self,
         messages: LanguageModelMessages,

{unique_toolkit-0.5.18.dist-info → unique_toolkit-0.5.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.18
+Version: 0.5.20
 Summary: 
 License: MIT
 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.20] - 2024-09-16
+- `LanguageModelService` now supports complete_util_async that can be called without instantiating the class, currently being used in the Hallucination service and evaluation API
+
+## [0.5.19] - 2024-09-11
+- `LanguageModelMessage` now supports content as a list of dictionary. Useful when adding image_url content along user message. 
+
 ## [0.5.18] - 2024-09-03
 - Adds option to use `metadata_filter` with search.
 - Adds `user_metadata`, `tool_parameters` and `metadata_filter` to `EventPayload`.

{unique_toolkit-0.5.18.dist-info → unique_toolkit-0.5.20.dist-info}/RECORD

@@ -23,10 +23,10 @@ unique_toolkit/embedding/service.py,sha256=Iiw-sbdkjuWlWMfLM9qyC4GNTJOotQAaVjkYv
 unique_toolkit/embedding/utils.py,sha256=v86lo__bCJbxZBQ3OcLu5SuwT6NbFfWlcq8iyk6BuzQ,279
 unique_toolkit/language_model/__init__.py,sha256=QgU_uwpVh1URQyVs6l-6Am4UwmEEhuGXNic3dUZ0FCc,1701
 unique_toolkit/language_model/infos.py,sha256=ETAUV0YTs6BjwuiTdhKz247CtL0W8Jwo3-c0ZQ2HdXs,9962
-unique_toolkit/language_model/schemas.py,sha256=h5zjZNk7O-wLKtRuiNtMCIbp5hEVXrAOviKonQcjFuI,4594
-unique_toolkit/language_model/service.py,sha256=8s2tiGLE5ryKQDOtEbNaFkc73NngANxvFNr-hD-dgps,10948
+unique_toolkit/language_model/schemas.py,sha256=DGZL6j63txkq5rdCn1uuVQIyLOyZt9t8J4f8JzhZENg,4607
+unique_toolkit/language_model/service.py,sha256=CvVo5CBa5Ia_fQD3DtJRsVChybuUfGFV5ml2_78_p1I,13395
 unique_toolkit/language_model/utils.py,sha256=WBPj1XKkDgxy_-T8HCZvsfkkSzj_1w4UZzNmyvdbBLY,1081
-unique_toolkit-0.5.18.dist-info/LICENSE,sha256=bIeCWCYuoUU_MzNdg48-ubJSVm7qxakaRbzTiJ5uxrs,1065
-unique_toolkit-0.5.18.dist-info/METADATA,sha256=Bc1nuWyOLgX0SujW7MCpzcbG8WBlTqPiC3PPLU858WA,10748
-unique_toolkit-0.5.18.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.5.18.dist-info/RECORD,,
+unique_toolkit-0.5.20.dist-info/LICENSE,sha256=bIeCWCYuoUU_MzNdg48-ubJSVm7qxakaRbzTiJ5uxrs,1065
+unique_toolkit-0.5.20.dist-info/METADATA,sha256=RlEnbfzQ12Do8nqVzhdWLfIktxKQbzu7ffdacfkWmTw,11108
+unique_toolkit-0.5.20.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.5.20.dist-info/RECORD,,