unique_toolkit 0.5.53__py3-none-any.whl → 0.5.55__py3-none-any.whl

unique_toolkit/chat/schemas.py

@@ -1,6 +1,5 @@
 from datetime import datetime
 from enum import StrEnum
-from typing import Optional
 
 from humps import camelize
 from pydantic import (
@@ -39,15 +38,22 @@ class ToolCall(BaseModel):
 
 
 class ChatMessage(BaseModel):
+    # This model should strictly meets https://github.com/Unique-AG/monorepo/blob/master/node/apps/node-chat/src/public-api/2023-12-06/dtos/message/public-message.dto.ts
     model_config = model_config
 
     id: str | None = None
+    chat_id: str
     object: str | None = None
-    content: str = Field(alias="text")
+    content: str | None = Field(default=None, alias="text")
+    original_content: str | None = Field(default=None, alias="originalText")
     role: ChatMessageRole
-    tool_calls: Optional[list[ToolCall]] = None
-    tool_call_id: Optional[str] = None
+    gpt_request: list[dict] | None = None
+    tool_calls: list[ToolCall] | None = None
+    tool_call_id: str | None = None
     debug_info: dict | None = {}
+    created_at: datetime | None = None
+    completed_at: datetime | None = None
+    updated_at: datetime | None = None
 
     # TODO make sdk return role consistently in lowercase
     # Currently needed as sdk returns role in uppercase

unique_toolkit/content/service.py

@@ -64,6 +64,9 @@ class ContentService(BaseService):
         if not scope_ids:
             self.logger.warning("No scope IDs provided for search.")
 
+        if content_ids:
+            self.logger.info("Searching chunks for content IDs: %s", content_ids)
+
         if metadata_filter is None:
             metadata_filter = self.metadata_filter
 
@@ -128,6 +131,9 @@ class ContentService(BaseService):
         if not scope_ids:
             self.logger.warning("No scope IDs provided for search.")
 
+        if content_ids:
+            self.logger.info("Searching chunks for content IDs: %s", content_ids)
+
         if metadata_filter is None:
             metadata_filter = self.metadata_filter
 
@@ -175,6 +181,9 @@ class ContentService(BaseService):
         Returns:
             list[Content]: The search results.
         """
+        if where.get("contentId"):
+            self.logger.info("Searching content for content ID: %s", where["contentId"])
+
         try:
             contents = unique_sdk.Content.search(
                 user_id=self.event.user_id,
@@ -202,6 +211,9 @@ class ContentService(BaseService):
         Returns:
             list[Content]: The search results.
         """
+        if where.get("contentId"):
+            self.logger.info("Searching content for content ID: %s", where["contentId"])
+
         try:
             contents = await unique_sdk.Content.search_async(
                 user_id=self.event.user_id,
@@ -376,6 +388,7 @@ class ContentService(BaseService):
             requests.Response: The response object containing the downloaded content.
 
         """
+        self.logger.info("Requesting content by ID: %s", content_id)
         url = f"{unique_sdk.api_base}/content/{content_id}/file"
         if chat_id:
             url = f"{url}?chatId={chat_id}"
@@ -414,6 +427,7 @@ class ContentService(BaseService):
             Exception: If the download fails or the filename cannot be determined.
         """
 
+        self.logger.info("Requesting content by ID: %s", content_id)
         response = self.request_content_by_id(content_id, chat_id)
         random_dir = tempfile.mkdtemp(dir=tmp_dir_path)
 
@@ -467,6 +481,7 @@ class ContentService(BaseService):
             Exception: If the download fails.
         """
 
+        self.logger.info("Downloading content by ID: %s", content_id)
         response = self.request_content_by_id(content_id, chat_id)
 
         random_dir = tempfile.mkdtemp(dir=dir_path)

{unique_toolkit-0.5.53.dist-info → unique_toolkit-0.5.55.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.53
+Version: 0.5.55
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -100,6 +100,12 @@ 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.5.55] - 2025-02-18
+- Log `contentId` for better debugging
+
+## [0.5.54] - 2025-02-10
+- Add `created_at`, `completed_at`, `updated_at` and `gpt_request` to `ChatMessage` schema.
+
 ## [0.5.53] - 2025-02-01
 - Correct `MessageAssessment` schemas
 

{unique_toolkit-0.5.53.dist-info → unique_toolkit-0.5.55.dist-info}/RECORD

@@ -11,13 +11,13 @@ unique_toolkit/app/performance/async_wrapper.py,sha256=yVVcRDkcdyfjsxro-N29SBvi-
 unique_toolkit/app/schemas.py,sha256=6RY7Ex-B3pOnFILlitHEi9sqJvupbpdxlN9xt33qRsM,1571
 unique_toolkit/app/verification.py,sha256=mffa6wm0i4hJbwzofePrkaia46xumMzECwQ0T3eKAx0,1929
 unique_toolkit/chat/__init__.py,sha256=4xS-Mcv7Oqdhprw1JEqD3nwGFflla4R2o7RNJyA5Wek,537
-unique_toolkit/chat/schemas.py,sha256=LOGO-yyQZ6ifb4lDEOSIQl_6-di8KXEB7AKtkjA3xSU,2253
+unique_toolkit/chat/schemas.py,sha256=PJ9Am-RKxUEo6o1jQ_DV100rt8rZD59GhWTPRgZuDow,2665
 unique_toolkit/chat/service.py,sha256=9sQbfwgb2GoUuDL3xoG-7P9oOEj-TAssZleXwSZ3HiY,28121
 unique_toolkit/chat/state.py,sha256=Cjgwv_2vhDFbV69xxsn7SefhaoIAEqLx3ferdVFCnOg,1445
 unique_toolkit/chat/utils.py,sha256=ihm-wQykBWhB4liR3LnwPVPt_qGW6ETq21Mw4HY0THE,854
 unique_toolkit/content/__init__.py,sha256=MSH2sxjQyKD2Sef92fzE5Dt9SihdzivB6yliSwJfTmQ,890
 unique_toolkit/content/schemas.py,sha256=zks_Pkki2VhxICJJgHZyc-LPmRuj5dLbw3pgcUT7SW8,2362
-unique_toolkit/content/service.py,sha256=rznSUyOuB4nbo6bFworzFv86kzo6TVT4Nc8GMAeAJB4,17711
+unique_toolkit/content/service.py,sha256=83gNLHxLB1hU2XUOpYdTeatpTozPh_GFGBUyUrvrpqQ,18383
 unique_toolkit/content/utils.py,sha256=Lake671plRsqNvO3pN_rmyVcpwbdED_KQpLcCnc4lv4,6902
 unique_toolkit/embedding/__init__.py,sha256=dr8M9jvslQTxPpxgaGwzxY0FildiWf-DidN_cahPAWw,191
 unique_toolkit/embedding/schemas.py,sha256=1GvKCaSk4jixzVQ2PKq8yDqwGEVY_hWclYtoAr6CC2g,96
@@ -44,7 +44,7 @@ unique_toolkit/language_model/service.py,sha256=m4B4YD4wxfU8HNo_stqbfnlKXziYBAwq
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
 unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJs8FEZXcgQTNenw,1406
 unique_toolkit/short_term_memory/service.py,sha256=Jd9P72-VvJy7hnqNrjmrmB5BHmsKuOpTiT0Jr-dBbsQ,1682
-unique_toolkit-0.5.53.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.5.53.dist-info/METADATA,sha256=BUtdCWRnnsrlHOl6XJtWxvXrZbhNPPR00g4R09QGHpI,16357
-unique_toolkit-0.5.53.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.5.53.dist-info/RECORD,,
+unique_toolkit-0.5.55.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.5.55.dist-info/METADATA,sha256=UwzR9G0BkKuD7R10nC7Klz8pjkJ-yL8ikJqt7mLKVxI,16540
+unique_toolkit-0.5.55.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.5.55.dist-info/RECORD,,