unique_toolkit 0.8.19__tar.gz → 0.8.20__tar.gz

{unique_toolkit-0.8.19 → unique_toolkit-0.8.20}/CHANGELOG.md

@@ -5,6 +5,8 @@ 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.8.20] - 2025-08-24
+- Fixed forced-tool-calls
 
 ## [0.8.19] - 2025-08-24
 - Enforce usage of ruff using pipeline

{unique_toolkit-0.8.19 → unique_toolkit-0.8.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.8.19
+Version: 0.8.20
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -114,6 +114,8 @@ 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.8.20] - 2025-08-24
+- Fixed forced-tool-calls
 
 ## [0.8.19] - 2025-08-24
 - Enforce usage of ruff using pipeline

{unique_toolkit-0.8.19 → unique_toolkit-0.8.20}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.8.19"
+version = "0.8.20"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",

{unique_toolkit-0.8.19 → unique_toolkit-0.8.20}/unique_toolkit/content/service.py

@@ -1,4 +1,6 @@
 import logging
+import mimetypes
+from enum import StrEnum
 from pathlib import Path
 from typing import Any, overload
 
@@ -33,6 +35,29 @@ from unique_toolkit.content.schemas import (
 logger = logging.getLogger(f"toolkit.{DOMAIN_NAME}.{__name__}")
 
 
+class FileMimeType(StrEnum):
+    PDF = "application/pdf"
+    DOCX = "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+    DOC = "application/msword"
+    XLSX = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+    XLS = "application/vnd.ms-excel"
+    PPTX = "application/vnd.openxmlformats-officedocument.presentationml.presentation"
+    CSV = "text/csv"
+    HTML = "text/html"
+    MD = "text/markdown"
+    TXT = "text/plain"
+
+
+class ImageMimeType(StrEnum):
+    JPEG = "image/jpeg"
+    PNG = "image/png"
+    GIF = "image/gif"
+    BMP = "image/bmp"
+    WEBP = "image/webp"
+    TIFF = "image/tiff"
+    SVG = "image/svg+xml"
+
+
 class ContentService:
     """
     Provides methods for searching, downloading and uploading content in the knowledge base.
@@ -622,3 +647,51 @@ class ContentService:
             content_id=content_id,
             chat_id=chat_id,
         )
+
+    def get_documents_uploaded_to_chat(self) -> list[Content]:
+        chat_contents = self.search_contents(
+            where={
+                "ownerId": {
+                    "equals": self._chat_id,
+                },
+            },
+        )
+
+        content: list[Content] = []
+        for c in chat_contents:
+            if self.is_file_content(c.key):
+                content.append(c)
+
+        return content
+
+    def get_images_uploaded_to_chat(self) -> list[Content]:
+        chat_contents = self.search_contents(
+            where={
+                "ownerId": {
+                    "equals": self._chat_id,
+                },
+            },
+        )
+
+        content: list[Content] = []
+        for c in chat_contents:
+            if self.is_image_content(c.key):
+                content.append(c)
+
+        return content
+
+    def is_file_content(self, filename: str) -> bool:
+        mimetype, _ = mimetypes.guess_type(filename)
+
+        if not mimetype:
+            return False
+
+        return mimetype in FileMimeType.__members__.values()
+
+    def is_image_content(self, filename: str) -> bool:
+        mimetype, _ = mimetypes.guess_type(filename)
+
+        if not mimetype:
+            return False
+
+        return mimetype in ImageMimeType.__members__.values()

{unique_toolkit-0.8.19 → unique_toolkit-0.8.20}/unique_toolkit/tools/schemas.py

@@ -132,6 +132,7 @@ class Source(BaseModel):
 class ToolPrompts(BaseModel):
     name: str
     display_name: str
+    tool_system_prompt: str
     tool_description: str
     tool_format_information_for_system_prompt: str
     input_model: dict[str, Any]

{unique_toolkit-0.8.19 → unique_toolkit-0.8.20}/unique_toolkit/tools/tool.py

@@ -109,6 +109,7 @@ class Tool(ABC, Generic[ConfigType]):
             name=self.name,
             display_name=self.display_name(),
             tool_description=self.tool_description().description,
+            tool_system_prompt=self.tool_description_for_system_prompt(),
             tool_format_information_for_system_prompt=self.tool_format_information_for_system_prompt(),
             input_model=self.tool_description_as_json(),
         )

{unique_toolkit-0.8.19 → unique_toolkit-0.8.20}/unique_toolkit/tools/tool_manager.py

@@ -1,5 +1,6 @@
 import asyncio
 from logging import Logger, getLogger
+from typing import Any
 
 from pydantic import BaseModel, Field
 
@@ -18,13 +19,6 @@ from unique_toolkit.tools.tool_progress_reporter import ToolProgressReporter
 from unique_toolkit.tools.utils.execution.execution import Result, SafeTaskExecutor
 
 
-class ForcedToolOption:
-    type: str = "function"
-
-    def __init__(self, name: str):
-        self.name = name
-
-
 class ToolManagerConfig(BaseModel):
     tools: list[ToolBuildConfig] = Field(
         default=[],
@@ -122,13 +116,20 @@ class ToolManager:
                 return tool
         return None
 
-    def get_forced_tools(self) -> list[ForcedToolOption]:
+    def get_forced_tools(self) -> list[dict[str, Any]]:
         return [
-            ForcedToolOption(t.name)
+            self._convert_to_forced_tool(t.name)
             for t in self._tools
             if t.name in self._tool_choices
         ]
 
+    def add_forced_tool(self, name):
+        tool = self.get_tool_by_name(name)
+        if not tool:
+            raise ValueError(f"Tool {name} not found")
+        self._tools.append(tool)
+        self._tool_choices.append(tool.name)
+
     def get_tool_definitions(
         self,
     ) -> list[LanguageModelTool | LanguageModelToolDescription]:
@@ -198,7 +199,9 @@ class ToolManager:
     ) -> ToolCallResponse:
         self._logger.info(f"Processing tool call: {tool_call.name}")
 
-        tool_instance = self.get_tool_by_name(tool_call.name)
+        tool_instance = self.get_tool_by_name(
+            tool_call.name
+        )  # we need to copy this as it will have problematic interference on multi calls.
 
         if tool_instance:
             # Execute the tool
@@ -254,3 +257,11 @@ class ToolManager:
                 f"Filtered out {len(tool_calls) - len(unique_tool_calls)} duplicate tool calls."
             )
         return unique_tool_calls
+
+    from typing import Any
+
+    def _convert_to_forced_tool(self, tool_name: str) -> dict[str, Any]:
+        return {
+            "type": "function",
+            "function": {"name": tool_name},
+        }