unique_toolkit 1.69.4__tar.gz → 1.69.6__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.69.4
+Version: 1.69.6
 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.6}/pyproject.toml

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

{unique_toolkit-1.69.4 → unique_toolkit-1.69.6}/unique_toolkit/agentic/tools/a2a/tool/config.py

@@ -125,7 +125,7 @@ class SubAgentToolConfig(BaseToolConfig):
         return v if v is not None else []
 
     tool_description_for_system_prompt: Annotated[
-        str, RJSFMetaTag.StringWidget.textarea(rows=5)
+        str, RJSFMetaTag.StringWidget.textarea(rows=3)
     ] = Field(
         default="",
         description="Description of the tool that will be included in the system prompt.",
@@ -134,30 +134,31 @@ class SubAgentToolConfig(BaseToolConfig):
         default="",
         description="Description of the tool that will be included in the tools sent to the model.",
     )
-    param_description_sub_agent_user_message: Annotated[
-        str, RJSFMetaTag.StringWidget.textarea(rows=5)
-    ] = Field(
-        default=DEFAULT_PARAM_DESCRIPTION_SUB_AGENT_USER_MESSAGE,
-        description="Description of the user message parameter that will be sent to the model.",
-    )
     tool_format_information_for_system_prompt: Annotated[
-        str, RJSFMetaTag.StringWidget.textarea(rows=5)
+        str, RJSFMetaTag.StringWidget.textarea(rows=2)
     ] = Field(
         default="",
         description="Format information that will be included in the system prompt to guide response formatting.",
     )
     tool_description_for_user_prompt: Annotated[
-        str, RJSFMetaTag.StringWidget.textarea(rows=5)
+        str, RJSFMetaTag.SpecialWidget.hidden()
     ] = Field(
         default="",
         description="Description of the tool that will be included in the user prompt.",
     )
     tool_format_information_for_user_prompt: Annotated[
-        str, RJSFMetaTag.StringWidget.textarea(rows=5)
+        str, RJSFMetaTag.SpecialWidget.hidden()
     ] = Field(
         default="",
         description="Format information that will be included in the user prompt to guide response formatting.",
     )
+    param_description_sub_agent_user_message: Annotated[
+        str, RJSFMetaTag.StringWidget.textarea(rows=1)
+    ] = Field(
+        default=DEFAULT_PARAM_DESCRIPTION_SUB_AGENT_USER_MESSAGE,
+        title="Parameter Description Sub Agent Message",
+        description="Description of the message parameter extracted by the orchestrator and sent as the input message to the sub-agent.",
+    )
 
     poll_interval: float = Field(
         default=1.0,

{unique_toolkit-1.69.4 → unique_toolkit-1.69.6}/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.6}/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