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

unique_toolkit/language_model/service.py

@@ -6,7 +6,6 @@ from typing_extensions import deprecated
 
 from unique_toolkit._common.validate_required_values import validate_required_values
 from unique_toolkit.app.schemas import BaseEvent, ChatEvent, Event
-from unique_toolkit.content.schemas import ContentChunk
 from unique_toolkit.language_model.constants import (
     DEFAULT_COMPLETE_TEMPERATURE,
     DEFAULT_COMPLETE_TIMEOUT,
@@ -15,14 +14,11 @@ from unique_toolkit.language_model.constants import (
 from unique_toolkit.language_model.functions import (
     complete,
     complete_async,
-    stream_complete_to_chat,
-    stream_complete_to_chat_async,
 )
 from unique_toolkit.language_model.infos import LanguageModelName
 from unique_toolkit.language_model.schemas import (
     LanguageModelMessages,
     LanguageModelResponse,
-    LanguageModelStreamResponse,
     LanguageModelTool,
 )
 
@@ -36,10 +32,8 @@ class LanguageModelService:
     Args:
         company_id (str | None, optional): The company identifier. Defaults to None.
         user_id (str | None, optional): The user identifier. Defaults to None.
-        assistant_message_id (str | None, optional): The assistant message identifier. Defaults to None.
         chat_id (str | None, optional): The chat identifier. Defaults to None.
         assistant_id (str | None, optional): The assistant identifier. Defaults to None.
-        user_message_id (str | None, optional): The user message identifier. Defaults to None.
     """
 
     def __init__(
@@ -47,16 +41,12 @@ class LanguageModelService:
         event: Event | BaseEvent | None = None,
         company_id: str | None = None,
         user_id: str | None = None,
-        assistant_message_id: str | None = None,
         chat_id: str | None = None,
         assistant_id: str | None = None,
-        user_message_id: str | None = None,
     ):
         self._event = event
         self.company_id = company_id
         self.user_id = user_id
-        self.assistant_message_id = assistant_message_id
-        self.user_message_id = user_message_id
         self.chat_id = chat_id
         self.assistant_id = assistant_id
 
@@ -64,8 +54,6 @@ class LanguageModelService:
             self.company_id = event.company_id
             self.user_id = event.user_id
             if isinstance(event, (ChatEvent, Event)):
-                self.assistant_message_id = event.payload.assistant_message.id
-                self.user_message_id = event.payload.user_message.id
                 self.chat_id = event.payload.chat_id
                 self.assistant_id = event.payload.assistant_id
 
@@ -167,211 +155,3 @@ class LanguageModelService:
             structured_output_model=structured_output_model,
             structured_output_enforce_schema=structured_output_enforce_schema,
         )
-
-    @deprecated("Use stream_complete_to_chat instead")
-    def stream_complete(
-        self,
-        messages: LanguageModelMessages,
-        model_name: LanguageModelName | str,
-        content_chunks: list[ContentChunk] = [],
-        debug_info: dict = {},
-        temperature: float = DEFAULT_COMPLETE_TEMPERATURE,
-        timeout: int = DEFAULT_COMPLETE_TIMEOUT,
-        tools: Optional[list[LanguageModelTool]] = None,
-        start_text: Optional[str] = None,
-        other_options: Optional[dict] = None,
-    ) -> LanguageModelStreamResponse:
-        """
-        Streams a completion in the chat session synchronously.
-        """
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            assistant_id,
-        ] = validate_required_values(
-            [
-                self.company_id,
-                self.user_id,
-                self.assistant_message_id,
-                self.user_message_id,
-                self.chat_id,
-                self.assistant_id,
-            ]
-        )
-
-        return stream_complete_to_chat(
-            company_id=company_id,
-            user_id=user_id,
-            assistant_message_id=assistant_message_id,
-            user_message_id=user_message_id,
-            chat_id=chat_id,
-            assistant_id=assistant_id,
-            messages=messages,
-            model_name=model_name,
-            content_chunks=content_chunks,
-            debug_info=debug_info,
-            temperature=temperature,
-            timeout=timeout,
-            tools=tools,
-            start_text=start_text,
-            other_options=other_options,
-        )
-
-    def stream_complete_to_chat(
-        self,
-        messages: LanguageModelMessages,
-        model_name: LanguageModelName | str,
-        content_chunks: list[ContentChunk] = [],
-        debug_info: dict = {},
-        temperature: float = DEFAULT_COMPLETE_TEMPERATURE,
-        timeout: int = DEFAULT_COMPLETE_TIMEOUT,
-        tools: Optional[list[LanguageModelTool]] = None,
-        start_text: Optional[str] = None,
-        other_options: Optional[dict] = None,
-    ) -> LanguageModelStreamResponse:
-        """
-        Streams a completion in the chat session synchronously.
-        """
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            assistant_id,
-        ] = validate_required_values(
-            [
-                self.company_id,
-                self.user_id,
-                self.assistant_message_id,
-                self.user_message_id,
-                self.chat_id,
-                self.assistant_id,
-            ]
-        )
-
-        return stream_complete_to_chat(
-            company_id=company_id,
-            user_id=user_id,
-            assistant_message_id=assistant_message_id,
-            user_message_id=user_message_id,
-            chat_id=chat_id,
-            assistant_id=assistant_id,
-            messages=messages,
-            model_name=model_name,
-            content_chunks=content_chunks,
-            debug_info=debug_info,
-            temperature=temperature,
-            timeout=timeout,
-            tools=tools,
-            start_text=start_text,
-            other_options=other_options,
-        )
-
-    @deprecated("Use stream_complete_to_chat_async instead")
-    async def stream_complete_async(
-        self,
-        messages: LanguageModelMessages,
-        model_name: LanguageModelName | str,
-        content_chunks: list[ContentChunk] = [],
-        debug_info: dict = {},
-        temperature: float = DEFAULT_COMPLETE_TEMPERATURE,
-        timeout: int = DEFAULT_COMPLETE_TIMEOUT,
-        tools: Optional[list[LanguageModelTool]] = None,
-        start_text: Optional[str] = None,
-        other_options: Optional[dict] = None,
-    ) -> LanguageModelStreamResponse:
-        """
-        Streams a completion in the chat session asynchronously.
-        """
-
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            assistant_id,
-        ] = validate_required_values(
-            [
-                self.company_id,
-                self.user_id,
-                self.assistant_message_id,
-                self.user_message_id,
-                self.chat_id,
-                self.assistant_id,
-            ]
-        )
-
-        return await stream_complete_to_chat_async(
-            company_id=company_id,
-            user_id=user_id,
-            assistant_message_id=assistant_message_id,
-            user_message_id=user_message_id,
-            chat_id=chat_id,
-            assistant_id=assistant_id,
-            messages=messages,
-            model_name=model_name,
-            content_chunks=content_chunks,
-            debug_info=debug_info,
-            temperature=temperature,
-            timeout=timeout,
-            tools=tools,
-            start_text=start_text,
-            other_options=other_options,
-        )
-
-    async def stream_complete_to_chat_async(
-        self,
-        messages: LanguageModelMessages,
-        model_name: LanguageModelName | str,
-        content_chunks: list[ContentChunk] = [],
-        debug_info: dict = {},
-        temperature: float = DEFAULT_COMPLETE_TEMPERATURE,
-        timeout: int = DEFAULT_COMPLETE_TIMEOUT,
-        tools: Optional[list[LanguageModelTool]] = None,
-        start_text: Optional[str] = None,
-        other_options: Optional[dict] = None,
-    ) -> LanguageModelStreamResponse:
-        """
-        Streams a completion in the chat session asynchronously.
-        """
-
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            assistant_id,
-        ] = validate_required_values(
-            [
-                self.company_id,
-                self.user_id,
-                self.assistant_message_id,
-                self.user_message_id,
-                self.chat_id,
-                self.assistant_id,
-            ]
-        )
-
-        return await stream_complete_to_chat_async(
-            company_id=company_id,
-            user_id=user_id,
-            assistant_message_id=assistant_message_id,
-            user_message_id=user_message_id,
-            chat_id=chat_id,
-            assistant_id=assistant_id,
-            messages=messages,
-            model_name=model_name,
-            content_chunks=content_chunks,
-            debug_info=debug_info,
-            temperature=temperature,
-            timeout=timeout,
-            tools=tools,
-            start_text=start_text,
-            other_options=other_options,
-        )

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.6.0
+Version: 0.6.2
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -55,7 +55,7 @@ The `unique_toolkit.app` module encompasses functions for initializing and secur
 The `unique_toolkit.chat` module encompasses all chat related functionality.
 
 - `functions.py` comprises the functions to manage and load the chat history and interact with the chat ui, e.g., creating a new assistant message.
-- `service.py` comprises the ChatService and provides an interface to manage and load the chat history and interact with the chat ui, e.g., creating a new assistant message.
+- `service.py` comprises the ChatService and provides an interface to manage and load the chat history and interact with the chat ui, e.g., creating a new assistant message and stream complete.
 - `schemas.py` comprises all relevant schemas, e.g., ChatMessage, used in the ChatService.
 - `utils.py` comprises utility functions to use and convert ChatMessage objects in assistants, e.g., convert_chat_history_to_injectable_string converts the chat history to a string that can be injected into a prompt. 
 
@@ -83,7 +83,7 @@ Unique platform.
 
 - `infos.py` comprises the information on all language models deployed through the Unique platform. We recommend to use the LanguageModel class, initialized with the LanguageModelName, e.g., LanguageModel(LanguageModelName.AZURE_GPT_35_TURBO_0125) to get the information on the specific language model like the name, version, token limits or retirement date.
 - `functions.py` comprises the functions to complete and stream complete to chat.
-- `service.py` comprises the LanguageModelService and provides an interface to interact with the language models, e.g., complete or stream_complete. 
+- `service.py` comprises the LanguageModelService and provides an interface to interact with the language models, e.g., complete. 
 - `schemas.py` comprises all relevant schemas, e.g., LanguageModelResponse, used in the LanguageModelService.
 - `utils.py` comprises utility functions to parse the output of the language model, e.g., convert_string_to_json finds and parses the last json object in a string.
 
@@ -111,6 +111,16 @@ 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.2] - 2025-02-25
+- Deprecate `LanguageModel` in favor of `LanguageModelInfo`
+- `LanguageModelTokenLimits` properties become mandatory, initialization allows 
+  - init with `token_limit` and `fraction_input` or `input_token_limit` and `output_token_limit`
+  - only `input_token_limit` and `output_token_limit` are members of model
+
+## [0.6.1] - 2025-02-25
+- [BREAKING] `LanguageModelService.stream_complete` and `LanguageModelService.stream_complete_async` are now moved to `ChatService.stream_complete` and `ChatService.stream_complete_async`. Correspondingly `assistant_message_id` and `user_message_id` are removed from `LanguageModelService`.
+- Add `create_user_message` and `create_user_message_async` to `ChatService` (similar to `create_assistant_message` and `create_assistant_message_async`)
+
 ## [0.6.0] - 2025-02-21
 - make for each domain, its base functionality accessible from `functions.py`
 - make it possible to instantiate the domain services directly from different event types, inhereted from common `BaseEvent`

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

@@ -13,9 +13,9 @@ unique_toolkit/app/schemas.py,sha256=hPOh5xLNNWgWVIkdrj6ZHYaGz0cTV-5Kv7OQHOaUgV8
 unique_toolkit/app/verification.py,sha256=mffa6wm0i4hJbwzofePrkaia46xumMzECwQ0T3eKAx0,1929
 unique_toolkit/chat/__init__.py,sha256=LRs2G-JTVuci4lbtHTkVUiNcZcSR6uqqfnAyo7af6nY,619
 unique_toolkit/chat/constants.py,sha256=05kq6zjqUVB2d6_P7s-90nbljpB3ryxwCI-CAz0r2O4,83
-unique_toolkit/chat/functions.py,sha256=CtlvrwYWUQ7_wSQyAXf8jnARv0P-Ffh1gx2JJzISvvI,22231
+unique_toolkit/chat/functions.py,sha256=J9Cmgkhj9bBxZja3ggkSp48af_LPU4Dfi9Sbc_WhhNY,27204
 unique_toolkit/chat/schemas.py,sha256=MNcGAXjK1K8zOODeMFz3FHVQL5sIBQXRwkr_2hFkG8k,2672
-unique_toolkit/chat/service.py,sha256=abBHpPonQYA408c_Z-WNmnROzMmYNUL55D6ek3iDubI,22048
+unique_toolkit/chat/service.py,sha256=SoFeaOi0BTexhiyX6busui_7JjhlRu30YNiKdgwV3JQ,29127
 unique_toolkit/chat/state.py,sha256=Cjgwv_2vhDFbV69xxsn7SefhaoIAEqLx3ferdVFCnOg,1445
 unique_toolkit/chat/utils.py,sha256=ihm-wQykBWhB4liR3LnwPVPt_qGW6ETq21Mw4HY0THE,854
 unique_toolkit/content/__init__.py,sha256=EdJg_A_7loEtCQf4cah3QARQreJx6pdz89Rm96YbMVg,940
@@ -31,9 +31,9 @@ unique_toolkit/embedding/schemas.py,sha256=1GvKCaSk4jixzVQ2PKq8yDqwGEVY_hWclYtoA
 unique_toolkit/embedding/service.py,sha256=sCMKeFjwNrWYQic1UUW2c1jnhjRQLcDYfsBgxmR70sY,2697
 unique_toolkit/embedding/utils.py,sha256=v86lo__bCJbxZBQ3OcLu5SuwT6NbFfWlcq8iyk6BuzQ,279
 unique_toolkit/evaluators/__init__.py,sha256=3Rfpnowm7MUXHWmeU4UV4s_3Hk-sw3V20oBwQCYlejQ,50
-unique_toolkit/evaluators/config.py,sha256=JRSHJvIjioXDMgd9hodK10J-52j3LMgJFvG0Vy7ePa8,1056
+unique_toolkit/evaluators/config.py,sha256=iYiBi7M6u5MG9nVgpxl9dKfoS4j72stA6Hl-MQHmYp8,1056
 unique_toolkit/evaluators/constants.py,sha256=1oI93jsh0R_TjX_8OenliiiywVe3vTooSnaMqtq6R18,27
-unique_toolkit/evaluators/context_relevancy/constants.py,sha256=jqrXqGZevlgWFnU4fqGKxjCT5oQXYVMAO4Uu44AAKmo,1126
+unique_toolkit/evaluators/context_relevancy/constants.py,sha256=9mAGc23e5XjTYOBfeuZVbaqOyYrvRoXYjfUnsBOVShU,1126
 unique_toolkit/evaluators/context_relevancy/prompts.py,sha256=gTlWP7fDuxhrXhCYNCqXMbCey_DalZMdi5l-a6RHgk0,713
 unique_toolkit/evaluators/context_relevancy/service.py,sha256=9hzdMuF4A4T97-3X3zcXgrDISLn1bleZ6tTL1bHa9dQ,1722
 unique_toolkit/evaluators/context_relevancy/utils.py,sha256=E9ljdRNbwYlx04fQDLvgF4SwxvlTJT0vE328PlUF6KA,5191
@@ -47,18 +47,18 @@ unique_toolkit/evaluators/schemas.py,sha256=Jaue6Uhx75X1CyHKWj8sT3RE1JZXTqoLtfLt
 unique_toolkit/language_model/__init__.py,sha256=jWko_vQj48wjnpTtlkg8iNdef0SMI3FN2kGywXRTMzg,1880
 unique_toolkit/language_model/builder.py,sha256=nsRqWO_2dgFehK5CgtqR5aqXgYUU0QL6mR0lALPrQXM,1898
 unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
-unique_toolkit/language_model/functions.py,sha256=Gvf7nYwfNvkER59KVGursOvvZj5uYBFgxQyG9fHRBJQ,12677
-unique_toolkit/language_model/infos.py,sha256=NgoV05ausVWMqrYqgH6i3s7tYG7mejupROIF_bwEGZo,13050
+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=87511yupvea-U6sfKWfelETevNMVPevhj7mEqX5FszU,7461
-unique_toolkit/language_model/service.py,sha256=IQamCLVL-k4-YifgGpwhvlPwZ81EQRpBI6eMK176ciU,12967
+unique_toolkit/language_model/schemas.py,sha256=76FtgWc0qtk9fpEwoecA59WxG__aR7_kYyWJooHIaF8,8297
+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
 unique_toolkit/short_term_memory/constants.py,sha256=698CL6-wjup2MvU19RxSmQk3gX7aqW_OOpZB7sbz_Xg,34
 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.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.6.0.dist-info/METADATA,sha256=glSKhyn4h1sfvzWEDmJ1WBYYtXVwCs823B5HACP_ZJY,18600
-unique_toolkit-0.6.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.6.0.dist-info/RECORD,,
+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,,