unique_toolkit 1.69.4__tar.gz → 1.69.5__tar.gz

{unique_toolkit-1.69.4 → unique_toolkit-1.69.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.69.4
+Version: 1.69.5
 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>

{unique_toolkit-1.69.4 → unique_toolkit-1.69.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "1.69.4"
+version = "1.69.5"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"

{unique_toolkit-1.69.4 → unique_toolkit-1.69.5}/unique_toolkit/language_model/functions.py

@@ -26,7 +26,6 @@ from unique_toolkit.language_model.reference import (
 )
 from unique_toolkit.language_model.schemas import (
     LanguageModelFunction,
-    LanguageModelMessageRole,
     LanguageModelMessages,
     LanguageModelResponse,
     LanguageModelStreamResponse,
@@ -530,7 +529,7 @@ def _create_language_model_stream_response_with_references(
     stream_response_message = ChatMessage(
         id="stream_unknown",
         previous_message_id=None,
-        role=LanguageModelMessageRole.ASSISTANT,
+        role=ChatMessageRole.ASSISTANT,
         text=message.content or "",
         original_text=content,
         chat_id="stream_unknown",
@@ -661,7 +660,7 @@ async def stream_complete_with_references_openai(
                 id=message_id or uuid4().hex,
                 chat_id="",
                 previous_message_id=None,
-                role=LanguageModelMessageRole.ASSISTANT,
+                role=ChatMessageRole.ASSISTANT,
                 text=text_,
                 original_text=full_text,
                 references=references_,

{unique_toolkit-1.69.4 → unique_toolkit-1.69.5}/unique_toolkit/language_model/schemas.py

@@ -262,6 +262,16 @@ 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"
@@ -541,10 +551,20 @@ class LanguageModelMessages(RootModel):
         builder.messages = self.root.copy()  # Start with existing messages
         return builder
 
+    @overload
+    def to_openai(self, mode: Literal["responses"]) -> list[ResponseInputItemParam]: ...
+
+    @overload
+    def to_openai(
+        self, mode: Literal["completions"] = "completions"
+    ) -> list[ChatCompletionMessageParam]: ...
+
     def to_openai(
         self, mode: Literal["completions", "responses"] = "completions"
-    ) -> list[ChatCompletionMessageParam | ResponseInputItemParam]:
-        return [message.to_openai(mode=mode) for message in self.root]
+    ) -> list[ChatCompletionMessageParam] | list[ResponseInputItemParam]:
+        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]
 
 
 # This seems similar to