unique_toolkit 1.69.7__tar.gz → 1.70.0__tar.gz

{unique_toolkit-1.69.7 → unique_toolkit-1.70.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.69.7
+Version: 1.70.0
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>
@@ -13,10 +13,10 @@ Requires-Dist: tiktoken>=0.12.0,<0.13
 Requires-Dist: tokenizers>=0.21.0,<0.22
 Requires-Dist: pydantic-settings>=2.10.1,<3
 Requires-Dist: sseclient>=0.0.27,<0.0.28
-Requires-Dist: openai>=1.105.0,<2
+Requires-Dist: openai>=1.105.0,<3
 Requires-Dist: platformdirs>=4.0.0,<5
 Requires-Dist: pillow>=12.1.1
-Requires-Dist: unique-sdk>=0.10.85,<0.11
+Requires-Dist: unique-sdk>=0.10.85,<0.12
 Requires-Dist: jambo>=0.1.2,<0.2
 Requires-Dist: jinja2>=3.1.6,<4
 Requires-Dist: docxtpl>=0.20.1,<0.21
@@ -34,7 +34,7 @@ Requires-Dist: pydantic-core>=2.41.5
 Requires-Dist: fastapi>=0.121.0,<0.122 ; extra == 'fastapi'
 Requires-Dist: starlette>=0.49.1 ; extra == 'fastapi'
 Requires-Dist: uvicorn>=0.37.0,<0.38 ; extra == 'fastapi'
-Requires-Dist: openai>=1.105.0,<2 ; extra == 'langchain'
+Requires-Dist: openai>=1.105.0,<3 ; extra == 'langchain'
 Requires-Dist: langchain>=1.0.0,<2 ; extra == 'langchain'
 Requires-Dist: langchain-openai>=0.3.28 ; extra == 'langchain'
 Requires-Dist: langchain-core>=1.0.0,<2 ; extra == 'langchain'

{unique_toolkit-1.69.7 → unique_toolkit-1.70.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "1.69.7"
+version = "1.70.0"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"
@@ -25,10 +25,10 @@ dependencies = [
     "tokenizers>=0.21.0,<0.22",
     "pydantic-settings>=2.10.1,<3",
     "sseclient>=0.0.27,<0.0.28",
-    "openai>=1.105.0,<2",
+    "openai>=1.105.0,<3",
     "platformdirs>=4.0.0,<5",
     "pillow>=12.1.1",
-    "unique-sdk>=0.10.85,<0.11",
+    "unique-sdk>=0.10.85,<0.12",
     "jambo>=0.1.2,<0.2",
     "jinja2>=3.1.6,<4",
     "docxtpl>=0.20.1,<0.21",
@@ -47,7 +47,7 @@ dependencies = [
 
 [project.optional-dependencies]
 langchain = [
-    "openai>=1.105.0,<2",
+    "openai>=1.105.0,<3",
     "langchain>=1.0.0,<2",
     "langchain-openai>=0.3.28",
     "langchain-core>=1.0.0,<2",

{unique_toolkit-1.69.7 → unique_toolkit-1.70.0}/unique_toolkit/language_model/schemas.py

@@ -1,6 +1,5 @@
 import json
 import math
-from abc import ABC, abstractmethod
 from enum import StrEnum
 from typing import Any, Literal, Self, TypeVar, override
 from uuid import uuid4
@@ -248,7 +247,7 @@ class LanguageModelFunctionCall(BaseModel):
         return assistant_message
 
 
-class LanguageModelMessage(BaseModel, ABC):
+class LanguageModelMessage(BaseModel):
     model_config = model_config
     role: LanguageModelMessageRole
     content: str | list[dict] | None = None
@@ -263,16 +262,13 @@ class LanguageModelMessage(BaseModel, ABC):
         return format_message(self.role.capitalize(), message=message, num_tabs=1)
 
     @overload
-    @abstractmethod
     def to_openai(
         self, mode: Literal["completions"] = "completions"
     ) -> ChatCompletionMessageParam: ...
 
     @overload
-    @abstractmethod
     def to_openai(self, mode: Literal["responses"]) -> ResponseInputItemParam: ...
 
-    @abstractmethod
     def to_openai(
         self, mode: Literal["completions", "responses"] = "completions"
     ) -> ChatCompletionMessageParam | ResponseInputItemParam:
@@ -497,6 +493,31 @@ LanguageModelMessageOptions = (
     | LanguageModelUserMessage
 )
 
+LanguageModelMessageTypes = (
+    LanguageModelAssistantMessage
+    | LanguageModelUserMessage
+    | LanguageModelSystemMessage
+    | LanguageModelToolMessage
+)
+
+
+def _language_model_message_to_subtype(
+    message: LanguageModelMessage,
+) -> LanguageModelMessageTypes:
+    """Narrow a plain ``LanguageModelMessage`` to the concrete subtype for ``role``."""
+    match message.role:
+        case LanguageModelMessageRole.ASSISTANT:
+            return LanguageModelAssistantMessage(content=message.content)
+        case LanguageModelMessageRole.SYSTEM:
+            return LanguageModelSystemMessage(content=message.content)
+        case LanguageModelMessageRole.USER:
+            return LanguageModelUserMessage(content=message.content)
+        case LanguageModelMessageRole.TOOL:
+            raise ValueError(
+                "Cannot convert a base LanguageModelMessage with role tool; "
+                "use LanguageModelToolMessage with name and tool_call_id."
+            )
+
 
 class LanguageModelMessages(RootModel):
     root: list[LanguageModelMessageOptions]
@@ -562,9 +583,17 @@ class LanguageModelMessages(RootModel):
     def to_openai(
         self, mode: Literal["completions", "responses"] = "completions"
     ) -> list[ChatCompletionMessageParam] | list[ResponseInputItemParam]:
+        # Use exact-type check: isinstance would match subclasses and would strip
+        messages = [
+            _language_model_message_to_subtype(m)
+            if type(m) is LanguageModelMessage
+            else m
+            for m in self.root
+        ]
+
         if mode == "responses":
-            return [message.to_openai(mode="responses") for message in self.root]
-        return [message.to_openai(mode="completions") for message in self.root]
+            return [message.to_openai(mode="responses") for message in messages]
+        return [message.to_openai(mode="completions") for message in messages]
 
 
 # This seems similar to