unique_toolkit 0.7.42__tar.gz → 0.8.0__tar.gz

{unique_toolkit-0.7.42 → unique_toolkit-0.8.0}/CHANGELOG.md

@@ -5,6 +5,9 @@ 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.0] - 2025-08-04
+- Add MCP support
+
 ## [0.7.42] - 2025-08-01
 - Added tool definitions
 

{unique_toolkit-0.7.42 → unique_toolkit-0.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.7.42
+Version: 0.8.0
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -19,7 +19,7 @@ Requires-Dist: regex (>=2024.5.15,<2025.0.0)
 Requires-Dist: sseclient (>=0.0.27,<0.0.28)
 Requires-Dist: tiktoken (>=0.7.0,<0.8.0)
 Requires-Dist: typing-extensions (>=4.9.0,<5.0.0)
-Requires-Dist: unique-sdk (>=0.9.40,<0.10.0)
+Requires-Dist: unique-sdk (>=0.10.0,<0.11.0)
 Description-Content-Type: text/markdown
 
 # Unique Toolkit
@@ -113,6 +113,9 @@ 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.0] - 2025-08-04
+- Add MCP support
+
 ## [0.7.42] - 2025-08-01
 - Added tool definitions
 

{unique_toolkit-0.7.42 → unique_toolkit-0.8.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.7.42"
+version = "0.8.0"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",
@@ -20,7 +20,7 @@ numpy = "^1.26.4"
 python-dotenv = "^1.0.1"
 regex = "^2024.5.15"
 tiktoken = "^0.7.0"
-unique-sdk = "^0.9.40"
+unique-sdk = "^0.10.0"
 pydantic-settings = "^2.10.1"
 sseclient = "^0.0.27"
 

{unique_toolkit-0.7.42 → unique_toolkit-0.8.0}/unique_toolkit/app/__init__.py

@@ -41,5 +41,11 @@ from .schemas import (
 from .verification import (
     verify_signature_and_construct_event as verify_signature_and_construct_event,
 )
+from .schemas import (
+    McpServer as McpServer,
+)
+from .schemas import (
+    McpTool as McpTool,
+)
 
 DOMAIN_NAME = "app"

{unique_toolkit-0.7.42 → unique_toolkit-0.8.0}/unique_toolkit/app/schemas.py

@@ -38,6 +38,53 @@ class BaseEvent(BaseModel):
         return cls.model_validate(data)
 
 
+###
+# MCP schemas
+###
+
+class McpTool(BaseModel):
+    model_config = model_config
+
+    name: str
+    description: Optional[str] = None
+    input_schema: dict[str, Any]
+    output_schema: Optional[dict[str, Any]] = None
+    annotations: Optional[dict[str, Any]] = None
+    title: Optional[str] = Field(
+        default=None,
+        description="The display title for a tool. This is a Unique specific field.",
+    )
+    icon: Optional[str] = Field(
+        default=None,
+        description="An icon name from the Lucide icon set for the tool. This is a Unique specific field.",
+    )
+    system_prompt: Optional[str] = Field(
+        default=None,
+        description="An optional system prompt for the tool. This is a Unique specific field.",
+    )
+    user_prompt: Optional[str] = Field(
+        default=None,
+        description="An optional user prompt for the tool. This is a Unique specific field.",
+    )
+    is_connected: bool = Field(
+        description="Whether the tool is connected to the MCP server. This is a Unique specific field.",
+    )
+
+class McpServer(BaseModel):
+    model_config = model_config
+
+    id: str
+    name: str
+    system_prompt: Optional[str] = Field(
+        default=None,
+        description="An optional system prompt for the MCP server.",
+    )
+    user_prompt: Optional[str] = Field(
+        default=None,
+        description="An optional user prompt for the MCP server.",
+    )
+    tools: list[McpTool] = []
+
 ###
 # ChatEvent schemas
 ###
@@ -133,6 +180,10 @@ class ChatEventPayload(BaseModel):
         default=None,
         description="Raw UniqueQL rule that can be compiled to a metadata filter.",
     )
+    mcp_servers: list[McpServer] = Field(
+        default_factory=list,
+        description="A list of MCP servers with tools available for the chat session.",
+    )
 
     @field_validator("raw_scope_rules", mode="before")
     def validate_scope_rules(cls, value: dict[str, Any] | None) -> UniqueQL | None: