unique_internal_search 2026.22.0.dev2__tar.gz → 2026.22.0.dev3__tar.gz

{unique_internal_search-2026.22.0.dev2 → unique_internal_search-2026.22.0.dev3}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.3
 Name: unique-internal-search
-Version: 2026.22.0.dev2
+Version: 2026.22.0.dev3
 Summary: 
 Author: Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri
 Author-email: Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>
 License: Proprietary
 Requires-Dist: pydantic>=2.8.2,<3
 Requires-Dist: typing-extensions>=4.9.0,<5
-Requires-Dist: unique-toolkit>=2026.22.0.dev5,<2026.22.0rc0
+Requires-Dist: unique-toolkit>=2026.22.0.dev7,<2026.22.0rc0
 Requires-Python: >=3.12, <4
 Description-Content-Type: text/markdown
 

{unique_internal_search-2026.22.0.dev2 → unique_internal_search-2026.22.0.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_internal_search"
-version = "2026.22.0.dev2"
+version = "2026.22.0.dev3"
 description = ""
 readme = "README.md"
 license = { text = "Proprietary" }
@@ -14,7 +14,7 @@ requires-python = ">=3.12,<4"
 dependencies = [
     "pydantic>=2.8.2,<3",
     "typing-extensions>=4.9.0,<5",
-    "unique-toolkit>=2026.22.0.dev5,<2026.22.0rc0",
+    "unique-toolkit>=2026.22.0.dev7,<2026.22.0rc0",
 ]
 
 [build-system]

{unique_internal_search-2026.22.0.dev2 → unique_internal_search-2026.22.0.dev3}/unique_internal_search/uploaded_search/config.py

@@ -1,5 +1,7 @@
 from typing import Annotated
 
+from pydantic import Field
+from pydantic.json_schema import SkipJsonSchema
 from unique_toolkit._common.pydantic.rjsf_tags import RJSFMetaTag
 
 from unique_internal_search.config import InternalSearchConfig
@@ -51,3 +53,9 @@ class UploadedSearchConfig(InternalSearchConfig):
         DEFAULT_TOOL_FORMAT_INFORMATION_FOR_SYSTEM_PROMPT,
         description="Tool format information for the system prompt.",
     )
+    # This is a hack to ensure that the system reminder is enabled when the tool is forced,
+    # as we cannot at the moment easily pass extra values to the tool constructor
+    enable_tool_call_system_reminder: SkipJsonSchema[bool] = Field(
+        default=True,
+        description="Whether to attach the uploaded-search system reminder to tool responses.",
+    )

unique_internal_search-2026.22.0.dev2/unique_internal_search/uploaded_search/prompts.py → unique_internal_search-2026.22.0.dev3/unique_internal_search/uploaded_search/prompts/__init__.py

@@ -1,18 +1,11 @@
-DEFAULT_TOOL_DESCRIPTION_FOR_SYSTEM_PROMPT = (
-    "You can use the UploadedSearch tool to access and analyze documents uploaded by users during a chat. This tool is designed to handle a variety of document-related tasks, including summarization, explanation, and detailed information retrieval. "
-    "Use cases for the UploadedSearch tool include:\n"
-    "- Document Analysis: When a user uploads a document and asks for a summary, explanation, or specific details, this tool can extract and provide the requested information.\n"
-    "- Named Document Queries: If a user refers to a previously uploaded document by name (e.g., 'What does the Q2_Report.pdf say about revenue?'), this tool can locate and analyze the document to answer the query.\n"
-    "- Policy and Procedure Verification: Use the tool to find the most current company policies, procedures, or guidelines within uploaded documents.\n"
-    "- Project-Specific Information: Access project documents, reports, or meeting notes uploaded by users to provide precise details.\n"
-    "- Confidential and Proprietary Information: Ensure that sensitive topics requiring proprietary knowledge or confidential data are sourced securely from uploaded documents.\n\n"
-    "**Instruction Query Splitting**\n"
-    "You should split the user question into multiple search strings when the user's question needs to be decomposed / rewritten to find different facts. Perform an individual tool call for each search string. Avoid overly broad queries that may return unrelated results. Ensure the search string is specific and relevant to the uploaded document(s).\n\n"
-    "Examples:\n"
-    'User: "What does the Q2_Report.pdf say about revenue and expenses?" => search strings: ["Q2_Report.pdf revenue", "Q2_Report.pdf expenses"]\n'
-    'User: "Summarize the uploaded document." => search string: ["Summarize the uploaded document"]\n\n'
-    "{system_prompt_valid_documents}"
-    "{system_prompt_expired_documents}"
+from pathlib import Path
+
+from unique_toolkit._common.utils.jinja.helpers import load_template
+
+_PROMPTS_DIR = Path(__file__).parent
+
+DEFAULT_TOOL_DESCRIPTION_FOR_SYSTEM_PROMPT = load_template(
+    _PROMPTS_DIR, "tool_description_for_system_prompt.j2"
 )
 
 DEFAULT_TOOL_FORMAT_INFORMATION_FOR_SYSTEM_PROMPT = (

unique_internal_search-2026.22.0.dev3/unique_internal_search/uploaded_search/prompts/tool_description_for_system_prompt.j2

@@ -0,0 +1,21 @@
+You can use the UploadedSearch tool to access and analyze documents uploaded by users during a chat. This tool is designed to handle a variety of document-related tasks, including summarization, explanation, and detailed information retrieval. Use cases for the UploadedSearch tool include:
+- Document Analysis: When a user uploads a document and asks for a summary, explanation, or specific details, this tool can extract and provide the requested information.
+- Named Document Queries: If a user refers to a previously uploaded document by name (e.g., 'What does the Q2_Report.pdf say about revenue?'), this tool can locate and analyze the document to answer the query.
+- Policy and Procedure Verification: Use the tool to find the most current company policies, procedures, or guidelines within uploaded documents.
+- Project-Specific Information: Access project documents, reports, or meeting notes uploaded by users to provide precise details.
+- Confidential and Proprietary Information: Ensure that sensitive topics requiring proprietary knowledge or confidential data are sourced securely from uploaded documents.
+
+**Instruction Query Splitting**
+You should split the user question into multiple search strings when the user's question needs to be decomposed / rewritten to find different facts. Perform an individual tool call for each search string. Avoid overly broad queries that may return unrelated results. Ensure the search string is specific and relevant to the uploaded document(s).
+
+Examples:
+User: "What does the Q2_Report.pdf say about revenue and expenses?" => search strings: ["Q2_Report.pdf revenue", "Q2_Report.pdf expenses"]
+User: "Summarize the uploaded document." => search string: ["Summarize the uploaded document"]
+
+{% if valid_documents %}
+**The currently uploaded and searchable documents are the following**
+{% for document in valid_documents %}
+- {{ document.name }} (content_id: {{ document.id }})
+{% endfor %}
+
+{% endif %}

{unique_internal_search-2026.22.0.dev2 → unique_internal_search-2026.22.0.dev3}/unique_internal_search/uploaded_search/service.py

@@ -1,6 +1,7 @@
 from pydantic import Field, create_model
 from typing_extensions import override
 from unique_toolkit import ContentService
+from unique_toolkit._common.utils.jinja.render import render_template
 from unique_toolkit.agentic.evaluation.schemas import EvaluationMetricName
 from unique_toolkit.agentic.feature_flags import feature_flags
 from unique_toolkit.agentic.tools.factory import ToolFactory
@@ -82,45 +83,19 @@ class UploadedSearchTool(Tool[UploadedSearchConfig]):
             ]
 
         valid_documents = []
-        expired_documents = []
         for doc in documents:
-            if not doc.is_ingested(default_if_unknown=True):
+            if not doc.is_ingested(default_if_unknown=True) or doc.is_expired():
                 continue
-            if doc.is_expired():
-                expired_documents.append(doc)
-            else:
-                valid_documents.append(doc)
-
-        system_prompt_valid_documents = ""
-        system_prompt_expired_documents = ""
-        if valid_documents:
-            system_prompt_valid_documents = (
-                "**The currently uploaded and valid documents are the following**\n"
-            )
-            system_prompt_valid_documents = (
-                system_prompt_valid_documents
-                + "\n".join(
-                    f"- {doc.title or doc.key} (content_id: {doc.id})"
-                    for doc in valid_documents
-                )
-                + "\n"
+            valid_documents.append(
+                {
+                    "name": doc.title or doc.key,
+                    "id": doc.id,
+                }
             )
 
-        if expired_documents:
-            system_prompt_expired_documents = (
-                "**The currently uploaded and expired documents are the following**\n"
-            )
-            system_prompt_expired_documents = (
-                system_prompt_expired_documents
-                + "\n".join(
-                    f"- {doc.title or doc.key} (content_id: {doc.id})"
-                    for doc in expired_documents
-                )
-            )
-
-        return self._config.tool_description_for_system_prompt.format(
-            system_prompt_valid_documents=system_prompt_valid_documents,
-            system_prompt_expired_documents=system_prompt_expired_documents,
+        return render_template(
+            self._config.tool_description_for_system_prompt,
+            valid_documents=valid_documents,
         )
 
     def tool_format_information_for_system_prompt(self) -> str:
@@ -153,7 +128,10 @@ class UploadedSearchTool(Tool[UploadedSearchConfig]):
                 state=ProgressState.FINISHED,
             )
         tool_response.name = self.name
-        tool_response.system_reminder = self._get_tool_call_response_system_reminder()
+        if self._config.enable_tool_call_system_reminder:
+            tool_response.system_reminder = (
+                self._get_tool_call_response_system_reminder()
+            )
         return tool_response
 
     def _get_tool_call_response_system_reminder(self) -> str: