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

unique_toolkit/evaluators/config.py

@@ -25,7 +25,7 @@ class EvaluationMetricConfig(BaseModel):
     enabled: bool = False
     name: EvaluationMetricName
     language_model: LanguageModel = LanguageModel(
-        LanguageModelName.AZURE_GPT_35_TURBO_0613
+        LanguageModelName.AZURE_GPT_35_TURBO_0125
     )
     custom_prompts: dict[str, str] = {}
     score_to_emoji: dict[str, str] = {}

unique_toolkit/evaluators/context_relevancy/constants.py

@@ -23,7 +23,7 @@ context_relevancy_required_input_fields = [
 default_config = EvaluationMetricConfig(
     enabled=False,
     name=EvaluationMetricName.CONTEXT_RELEVANCY,
-    language_model=LanguageModel(LanguageModelName.AZURE_GPT_35_TURBO_0613),
+    language_model=LanguageModel(LanguageModelName.AZURE_GPT_35_TURBO_0125),
     score_to_emoji={"LOW": "🟢", "MEDIUM": "🟡", "HIGH": "🔴"},
     custom_prompts={
         SYSTEM_MSG_KEY: CONTEXT_RELEVANCY_METRIC_SYSTEM_MSG,

unique_toolkit/language_model/infos.py

@@ -1,16 +1,14 @@
 from datetime import date
 from enum import StrEnum
-from typing import ClassVar, Optional, Type, cast
+from typing import ClassVar, Optional
 
 from pydantic import BaseModel
+from typing_extensions import deprecated
 
 from unique_toolkit.language_model.schemas import LanguageModelTokenLimits
 
 
 class LanguageModelName(StrEnum):
-    AZURE_GPT_35_TURBO_0613 = "AZURE_GPT_35_TURBO_0613"
-    AZURE_GPT_35_TURBO = "AZURE_GPT_35_TURBO"
-    AZURE_GPT_35_TURBO_16K = "AZURE_GPT_35_TURBO_16K"
     AZURE_GPT_35_TURBO_0125 = "AZURE_GPT_35_TURBO_0125"
     AZURE_GPT_4_0613 = "AZURE_GPT_4_0613"
     AZURE_GPT_4_TURBO_1106 = "AZURE_GPT_4_TURBO_1106"
@@ -30,12 +28,7 @@ class EncoderName(StrEnum):
 def get_encoder_name(model_name: LanguageModelName) -> Optional[EncoderName]:
     LMN = LanguageModelName
     match model_name:
-        case (
-            LMN.AZURE_GPT_35_TURBO
-            | LMN.AZURE_GPT_35_TURBO_16K
-            | LMN.AZURE_GPT_35_TURBO_0613
-            | LMN.AZURE_GPT_35_TURBO_0125
-        ):
+        case LMN.AZURE_GPT_35_TURBO_0125:
             return EncoderName.CL100K_BASE
         case (
             LMN.AZURE_GPT_4_0613
@@ -76,7 +69,147 @@ class LanguageModelInfo(BaseModel):
     deprecated_at: Optional[date] = None
     retirement_text: Optional[str] = None
 
+    @classmethod
+    def from_name(cls, model_name: LanguageModelName):
+        match model_name:
+            case LanguageModelName.AZURE_GPT_35_TURBO_0125:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.AZURE,
+                    version="0125",
+                    encoder_name=get_encoder_name(model_name),
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=16385, token_limit_output=4096
+                    ),
+                    info_cutoff_at=date(2021, 9, 1),
+                    published_at=date(2023, 1, 25),
+                    retirement_at=date(5, 3, 31),
+                )
+            case LanguageModelName.AZURE_GPT_4_0613:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.AZURE,
+                    version="0613",
+                    encoder_name=get_encoder_name(model_name),
+                    token_limits=LanguageModelTokenLimits(token_limit=8192),
+                    info_cutoff_at=date(2021, 9, 1),
+                    published_at=date(2023, 6, 13),
+                    deprecated_at=date(2024, 10, 1),
+                    retirement_at=date(2025, 6, 6),
+                )
+            case LanguageModelName.AZURE_GPT_4_TURBO_1106:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.AZURE,
+                    version="1106-preview",
+                    encoder_name=get_encoder_name(model_name),
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=128000, token_limit_output=4096
+                    ),
+                    info_cutoff_at=date(2023, 4, 1),
+                    published_at=date(2023, 11, 6),
+                )
+            case LanguageModelName.AZURE_GPT_4_VISION_PREVIEW:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.AZURE,
+                    version="vision-preview",
+                    encoder_name=get_encoder_name(model_name),
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=128000, token_limit_output=4096
+                    ),
+                    info_cutoff_at=date(2023, 4, 1),
+                    published_at=date(2023, 11, 6),
+                )
+            case LanguageModelName.AZURE_GPT_4_32K_0613:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.AZURE,
+                    version="1106-preview",
+                    encoder_name=get_encoder_name(model_name),
+                    token_limits=LanguageModelTokenLimits(token_limit=32768),
+                    info_cutoff_at=date(2021, 9, 1),
+                    published_at=date(2023, 6, 13),
+                    deprecated_at=date(2024, 10, 1),
+                    retirement_at=date(2025, 6, 6),
+                )
+            case LanguageModelName.AZURE_GPT_4_TURBO_2024_0409:
+                return cls(
+                    name=model_name,
+                    encoder_name=get_encoder_name(model_name),
+                    provider=LanguageModelProvider.AZURE,
+                    version="turbo-2024-04-09",
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=128000, token_limit_output=4096
+                    ),
+                    info_cutoff_at=date(2023, 12, 1),
+                    published_at=date(2024, 4, 9),
+                )
+            case LanguageModelName.AZURE_GPT_4o_2024_0513:
+                return cls(
+                    name=model_name,
+                    encoder_name=get_encoder_name(model_name),
+                    provider=LanguageModelProvider.AZURE,
+                    version="2024-05-13",
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=128000, token_limit_output=4096
+                    ),
+                    info_cutoff_at=date(2023, 10, 1),
+                    published_at=date(2024, 5, 13),
+                )
+            case LanguageModelName.AZURE_GPT_4o_2024_0806:
+                return cls(
+                    name=model_name,
+                    encoder_name=get_encoder_name(model_name),
+                    provider=LanguageModelProvider.AZURE,
+                    version="2024-08-06",
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=128000, token_limit_output=16384
+                    ),
+                    info_cutoff_at=date(2023, 10, 1),
+                    published_at=date(2024, 8, 6),
+                )
+            case LanguageModelName.AZURE_GPT_4o_MINI_2024_0718:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.AZURE,
+                    version="2024-07-18",
+                    encoder_name=get_encoder_name(model_name),
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=128000, token_limit_output=16384
+                    ),
+                    info_cutoff_at=date(2023, 10, 1),
+                    published_at=date(2024, 7, 18),
+                )
+            case _:
+                if isinstance(model_name, LanguageModelName):
+                    raise ValueError(
+                        f"{model_name} is not supported. Please add model information in toolkit."
+                    )
+
+                return cls(
+                    name=model_name,
+                    version="custom",
+                    provider=LanguageModelProvider.CUSTOM,
+                )
+
+    @property
+    def display_name(self) -> str:
+        """
+        Returns the name of the model as a string.
+        """
+
+        if isinstance(self.name, LanguageModelName):
+            return self.name.value
+        else:
+            return self.name
+
 
+@deprecated(
+    """
+Use `LanguageModelInfo` instead of `LanguageModel`
+"""
+)
 class LanguageModel:
     _info: ClassVar[LanguageModelInfo]
 
@@ -199,17 +332,8 @@ class LanguageModel:
 
     @classmethod
     def get_model_info(cls, model_name: LanguageModelName | str) -> LanguageModelInfo:
-        if not model_name:
-            raise ValueError("Model name must be provided to get the model info.")
-
-        for subclass in cls.__subclasses__():
-            if hasattr(subclass, "info") and subclass._info.name == model_name:
-                # TODO find alternative solution for warning
-                # if subclass._info.retirement_at:
-                #     warning_text = f"WARNING: {subclass._info.name} will be retired on {subclass._info.retirement_at.isoformat()} and from then on not accessible anymore. {subclass._info.retirement_text}"
-                #     print(warning_text)
-                #     warnings.warn(warning_text, DeprecationWarning, stacklevel=2)
-                return subclass._info
+        if isinstance(model_name, LanguageModelName):
+            return LanguageModelInfo.from_name(model_name)
 
         return LanguageModelInfo(
             name=model_name,
@@ -224,190 +348,5 @@ class LanguageModel:
         """
 
         return [
-            cast(LanguageModelInfo, subclass._info)
-            for subclass in cls.__subclasses__()
-            if hasattr(subclass, "_info")
+            LanguageModelInfo.from_name(model_name=name) for name in LanguageModelName
         ]
-
-
-def create_language_model(
-    name: LanguageModelName,
-    version: str,
-    provider: LanguageModelProvider,
-    info_cutoff_at: date,
-    published_at: date,
-    encoder_name: Optional[EncoderName] = None,
-    token_limit: Optional[int] = None,
-    token_limit_input: Optional[int] = None,
-    token_limit_output: Optional[int] = None,
-    retirement_at: Optional[date] = None,
-    deprecated_at: Optional[date] = None,
-    retirement_text: Optional[str] = None,
-) -> Type[LanguageModel]:
-    info = LanguageModelInfo(
-        name=name,
-        version=version,
-        provider=provider,
-        encoder_name=encoder_name,
-        token_limits=LanguageModelTokenLimits(
-            token_limit=token_limit,
-            token_limit_input=token_limit_input,
-            token_limit_output=token_limit_output,
-        ),
-        info_cutoff_at=info_cutoff_at,
-        published_at=published_at,
-        retirement_at=retirement_at,
-        deprecated_at=deprecated_at,
-        retirement_text=retirement_text,
-    )
-
-    class Model(LanguageModel):
-        _info = info
-
-    return Model
-
-
-############################################################################################################
-# Define the models here
-############################################################################################################
-
-
-AzureGpt35Turbo0613 = create_language_model(
-    name=LanguageModelName.AZURE_GPT_35_TURBO_0613,
-    provider=LanguageModelProvider.AZURE,
-    version="0613",
-    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_35_TURBO_0613),
-    token_limit=8192,
-    info_cutoff_at=date(2021, 9, 1),
-    published_at=date(2023, 6, 13),
-    retirement_at=date(2025, 2, 13),
-)
-
-AzureGpt35Turbo = create_language_model(
-    name=LanguageModelName.AZURE_GPT_35_TURBO,
-    provider=LanguageModelProvider.AZURE,
-    version="0301",
-    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_35_TURBO),
-    token_limit=4096,
-    info_cutoff_at=date(2021, 9, 1),
-    published_at=date(2023, 3, 1),
-    retirement_at=date(2025, 2, 13),
-)
-
-
-AzureGpt35Turbo16k = create_language_model(
-    name=LanguageModelName.AZURE_GPT_35_TURBO_16K,
-    provider=LanguageModelProvider.AZURE,
-    version="0613",
-    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_35_TURBO_16K),
-    token_limit=16382,
-    info_cutoff_at=date(2021, 9, 1),
-    published_at=date(2023, 6, 13),
-    retirement_at=date(2025, 2, 13),
-)
-
-
-AzureGpt35Turbo0125 = create_language_model(
-    name=LanguageModelName.AZURE_GPT_35_TURBO_0125,
-    provider=LanguageModelProvider.AZURE,
-    version="0125",
-    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_35_TURBO_0125),
-    token_limit_input=16385,
-    token_limit_output=4096,
-    info_cutoff_at=date(2021, 9, 1),
-    published_at=date(2023, 1, 25),
-    retirement_at=date(5, 3, 31),
-)
-
-
-AzureGpt40613 = create_language_model(
-    name=LanguageModelName.AZURE_GPT_4_0613,
-    provider=LanguageModelProvider.AZURE,
-    version="0613",
-    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_4_0613),
-    token_limit=8192,
-    info_cutoff_at=date(2021, 9, 1),
-    published_at=date(2023, 6, 13),
-    deprecated_at=date(2024, 10, 1),
-    retirement_at=date(2025, 6, 6),
-)
-
-
-AzureGpt4Turbo1106 = create_language_model(
-    name=LanguageModelName.AZURE_GPT_4_TURBO_1106,
-    provider=LanguageModelProvider.AZURE,
-    version="1106-preview",
-    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_4_TURBO_1106),
-    token_limit_input=128000,
-    token_limit_output=4096,
-    info_cutoff_at=date(2023, 4, 1),
-    published_at=date(2023, 11, 6),
-)
-
-
-AzureGpt4VisionPreview = create_language_model(
-    name=LanguageModelName.AZURE_GPT_4_VISION_PREVIEW,
-    provider=LanguageModelProvider.AZURE,
-    version="vision-preview",
-    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_4_VISION_PREVIEW),
-    token_limit_input=128000,
-    token_limit_output=4096,
-    info_cutoff_at=date(2023, 4, 1),
-    published_at=date(2023, 11, 6),
-)
-
-AzureGpt432k0613 = create_language_model(
-    name=LanguageModelName.AZURE_GPT_4_32K_0613,
-    provider=LanguageModelProvider.AZURE,
-    version="1106-preview",
-    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_4_32K_0613),
-    token_limit=32768,
-    info_cutoff_at=date(2021, 9, 1),
-    published_at=date(2023, 6, 13),
-    deprecated_at=date(2024, 10, 1),
-    retirement_at=date(2025, 6, 6),
-)
-
-AzureGpt4Turbo20240409 = create_language_model(
-    name=LanguageModelName.AZURE_GPT_4_TURBO_2024_0409,
-    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_4_TURBO_2024_0409),
-    provider=LanguageModelProvider.AZURE,
-    version="turbo-2024-04-09",
-    token_limit_input=128000,
-    token_limit_output=4096,
-    info_cutoff_at=date(2023, 12, 1),
-    published_at=date(2024, 4, 9),
-)
-
-AzureGpt4o20240513 = create_language_model(
-    name=LanguageModelName.AZURE_GPT_4o_2024_0513,
-    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_4o_2024_0513),
-    provider=LanguageModelProvider.AZURE,
-    version="2024-05-13",
-    token_limit_input=128000,
-    token_limit_output=4096,
-    info_cutoff_at=date(2023, 10, 1),
-    published_at=date(2024, 5, 13),
-)
-
-AzureGpt4o20240806 = create_language_model(
-    name=LanguageModelName.AZURE_GPT_4o_2024_0806,
-    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_4o_2024_0806),
-    provider=LanguageModelProvider.AZURE,
-    version="2024-08-06",
-    token_limit_input=128000,
-    token_limit_output=16384,
-    info_cutoff_at=date(2023, 10, 1),
-    published_at=date(2024, 8, 6),
-)
-
-AzureGpt4oMini20240718 = create_language_model(
-    name=LanguageModelName.AZURE_GPT_4o_MINI_2024_0718,
-    provider=LanguageModelProvider.AZURE,
-    version="2024-07-18",
-    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_4o_MINI_2024_0718),
-    token_limit_input=128000,
-    token_limit_output=16384,
-    info_cutoff_at=date(2023, 10, 1),
-    published_at=date(2024, 7, 18),
-)

unique_toolkit/language_model/schemas.py

@@ -9,11 +9,13 @@ from pydantic import (
     BaseModel,
     ConfigDict,
     Field,
+    PrivateAttr,
     RootModel,
     field_validator,
     model_serializer,
     model_validator,
 )
+from typing_extensions import deprecated
 
 from unique_toolkit.language_model.utils import format_message
 
@@ -205,30 +207,50 @@ class LanguageModelStreamResponse(BaseModel):
 
 
 class LanguageModelTokenLimits(BaseModel):
-    token_limit: Optional[int] = None
-    token_limit_input: Optional[int] = None
-    token_limit_output: Optional[int] = None
-
-    fraction_input: float = Field(default=0.4, le=1, ge=0)
-
-    @model_validator(mode="after")
-    def check_required_fields(self):
-        # Best case input and output is determined
-        if self.token_limit_input and self.token_limit_output:
-            self.token_limit = self.token_limit_input + self.token_limit_output
-            self.fraction_input = self.token_limit_input / self.token_limit
-            return self
-
-        # Deal with case where only token_limit and optional fraction_input is given
-        if self.token_limit:
-            if not self.fraction_input:
-                self.fraction_input = 0.4
-
-            self.token_limit_input = math.floor(self.fraction_input * self.token_limit)
-            self.token_limit_output = math.floor(
-                (1 - self.fraction_input) * self.token_limit
-            )
-            return self
+    token_limit_input: int
+    token_limit_output: int
+
+    _fraction_adaptable = PrivateAttr(default=False)
+
+    @property
+    @deprecated("""
+    Deprecated: Use the more specific `token_limit_input` and `token_limit_output` instead.
+    """)
+    def token_limit(self):
+        return self.token_limit_input + self.token_limit_output
+
+    @token_limit.setter
+    @deprecated("""
+    Deprecated: Token limit can only be reduced
+    """)
+    def token_limit(self, token_limit, fraction_input=0.4):
+        if self.token_limit > token_limit:
+            self.token_limit_input = math.floor(fraction_input * token_limit)
+            self.token_limit_output = math.floor((1 - fraction_input) * token_limit)
+            self._fraction_adaptable = True
+
+    def adapt_fraction(self, fraction_input: float):
+        if self._fraction_adaptable:
+            token_limit = self.token_limit_input + self.token_limit_output
+            self.token_limit_input = math.floor(fraction_input * token_limit)
+            self.token_limit_output = math.floor((1 - fraction_input) * token_limit)
+
+    @model_validator(mode="before")
+    def check_required_fields(cls, data):
+        if isinstance(data, dict):
+            if {"token_limit_input", "token_limit_output"}.issubset(data.keys()):
+                return data
+
+            if {"token_limit"}.issubset(data.keys()):
+                token_limit = data.get("token_limit")
+                fraction_input = data.get("fraction_input", 0.4)
+
+                data["token_limit_input"] = math.floor(fraction_input * token_limit)
+                data["token_limit_output"] = math.floor(
+                    (1 - fraction_input) * token_limit
+                )
+                data["_fraction_adaptpable"] = True
+            return data
 
         raise ValueError(
             'Either "token_limit_input" and "token_limit_output" must be provided together, or "token_limit" must be provided.'

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.6.1
+Version: 0.6.2
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -111,6 +111,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.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`)

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

@@ -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
@@ -48,9 +48,9 @@ unique_toolkit/language_model/__init__.py,sha256=jWko_vQj48wjnpTtlkg8iNdef0SMI3F
 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=I5jHhHsKoq7GwEQyTrM8LXB2n_6dvMAk7UklenjuHSY,7945
-unique_toolkit/language_model/infos.py,sha256=NgoV05ausVWMqrYqgH6i3s7tYG7mejupROIF_bwEGZo,13050
+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/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
@@ -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.1.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.6.1.dist-info/METADATA,sha256=HE5Tg7WWNCSsaU4O0VirvkILqvxRdHtNnfsbbu_S3Z8,19071
-unique_toolkit-0.6.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.6.1.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,,