unique_toolkit 0.7.28__py3-none-any.whl → 0.7.29__py3-none-any.whl

unique_toolkit/app/schemas.py

@@ -131,8 +131,9 @@ class ChatEventPayload(BaseModel):
     )
 
     @field_validator("raw_scope_rules", mode="before")
-    def validate_scope_rules(cls, value: dict[str, Any]) -> UniqueQL:
-        return parse_uniqueql(value)
+    def validate_scope_rules(cls, value: dict[str, Any] | None) -> UniqueQL | None:
+        if value:
+            return parse_uniqueql(value)
 
 
 @deprecated("""Use `ChatEventPayload` instead.

unique_toolkit/chat/service.py

@@ -3,7 +3,6 @@ from typing import Optional
 
 from typing_extensions import deprecated
 
-from unique_toolkit._common.validate_required_values import validate_required_values
 from unique_toolkit.app.schemas import ChatEvent, Event
 from unique_toolkit.chat.constants import (
     DEFAULT_MAX_MESSAGES,
@@ -71,13 +70,13 @@ class ChatService:
 
     def __init__(self, event: ChatEvent | Event):
         self._event = event
-        self._company_id = event.company_id
-        self._user_id = event.user_id
-        self._assistant_message_id = event.payload.assistant_message.id
-        self._user_message_id = event.payload.user_message.id
-        self._chat_id = event.payload.chat_id
-        self._assistant_id = event.payload.assistant_id
-        self._user_message_text = event.payload.user_message.text
+        self._company_id: str = event.company_id
+        self._user_id: str = event.user_id
+        self._assistant_message_id: str = event.payload.assistant_message.id
+        self._user_message_id: str = event.payload.user_message.id
+        self._chat_id: str = event.payload.chat_id
+        self._assistant_id: str = event.payload.assistant_id
+        self._user_message_text: str = event.payload.user_message.text
 
     @property
     @deprecated(
@@ -96,7 +95,7 @@ class ChatService:
     @deprecated(
         "The company_id property is deprecated and will be removed in a future version."
     )
-    def company_id(self) -> str | None:
+    def company_id(self) -> str:
         """
         Get the company identifier (deprecated).
 
@@ -109,7 +108,7 @@ class ChatService:
     @deprecated(
         "The company_id setter is deprecated and will be removed in a future version."
     )
-    def company_id(self, value: str | None) -> None:
+    def company_id(self, value: str) -> None:
         """
         Set the company identifier (deprecated).
 
@@ -122,7 +121,7 @@ class ChatService:
     @deprecated(
         "The user_id property is deprecated and will be removed in a future version."
     )
-    def user_id(self) -> str | None:
+    def user_id(self) -> str:
         """
         Get the user identifier (deprecated).
 
@@ -135,7 +134,7 @@ class ChatService:
     @deprecated(
         "The user_id setter is deprecated and will be removed in a future version."
     )
-    def user_id(self, value: str | None) -> None:
+    def user_id(self, value: str) -> None:
         """
         Set the user identifier (deprecated).
 
@@ -148,7 +147,7 @@ class ChatService:
     @deprecated(
         "The assistant_message_id property is deprecated and will be removed in a future version."
     )
-    def assistant_message_id(self) -> str | None:
+    def assistant_message_id(self) -> str:
         """
         Get the assistant message identifier (deprecated).
 
@@ -161,7 +160,7 @@ class ChatService:
     @deprecated(
         "The assistant_message_id setter is deprecated and will be removed in a future version."
     )
-    def assistant_message_id(self, value: str | None) -> None:
+    def assistant_message_id(self, value: str) -> None:
         """
         Set the assistant message identifier (deprecated).
 
@@ -174,7 +173,7 @@ class ChatService:
     @deprecated(
         "The user_message_id property is deprecated and will be removed in a future version."
     )
-    def user_message_id(self) -> str | None:
+    def user_message_id(self) -> str:
         """
         Get the user message identifier (deprecated).
 
@@ -187,7 +186,7 @@ class ChatService:
     @deprecated(
         "The user_message_id setter is deprecated and will be removed in a future version."
     )
-    def user_message_id(self, value: str | None) -> None:
+    def user_message_id(self, value: str) -> None:
         """
         Set the user message identifier (deprecated).
 
@@ -200,7 +199,7 @@ class ChatService:
     @deprecated(
         "The chat_id property is deprecated and will be removed in a future version."
     )
-    def chat_id(self) -> str | None:
+    def chat_id(self) -> str:
         """
         Get the chat identifier (deprecated).
 
@@ -213,7 +212,7 @@ class ChatService:
     @deprecated(
         "The chat_id setter is deprecated and will be removed in a future version."
     )
-    def chat_id(self, value: str | None) -> None:
+    def chat_id(self, value: str) -> None:
         """
         Set the chat identifier (deprecated).
 
@@ -226,7 +225,7 @@ class ChatService:
     @deprecated(
         "The assistant_id property is deprecated and will be removed in a future version."
     )
-    def assistant_id(self) -> str | None:
+    def assistant_id(self) -> str:
         """
         Get the assistant identifier (deprecated).
 
@@ -239,7 +238,7 @@ class ChatService:
     @deprecated(
         "The assistant_id setter is deprecated and will be removed in a future version."
     )
-    def assistant_id(self, value: str | None) -> None:
+    def assistant_id(self, value: str) -> None:
         """
         Set the assistant identifier (deprecated).
 
@@ -252,7 +251,7 @@ class ChatService:
     @deprecated(
         "The user_message_text property is deprecated and will be removed in a future version."
     )
-    def user_message_text(self) -> str | None:
+    def user_message_text(self) -> str:
         """
         Get the user message text (deprecated).
 
@@ -265,7 +264,7 @@ class ChatService:
     @deprecated(
         "The user_message_text setter is deprecated and will be removed in a future version."
     )
-    def user_message_text(self, value: str | None) -> None:
+    def user_message_text(self, value: str) -> None:
         """
         Set the user message text (deprecated).
 
@@ -281,30 +280,13 @@ class ChatService:
         Args:
             debug_info (dict): The new debug information.
         """
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            user_message_text,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-                self._assistant_message_id,
-                self._user_message_id,
-                self._chat_id,
-                self._user_message_text,
-            ]
-        )
         return await modify_message_async(
-            user_id=user_id,
-            company_id=company_id,
-            assistant_message_id=assistant_message_id,
-            chat_id=chat_id,
-            user_message_id=user_message_id,
-            user_message_text=user_message_text,
+            user_id=self._user_id,
+            company_id=self._company_id,
+            assistant_message_id=self._assistant_message_id,
+            chat_id=self._chat_id,
+            user_message_id=self._user_message_id,
+            user_message_text=self._user_message_text,
             assistant=False,
             debug_info=debug_info,
         )
@@ -316,31 +298,14 @@ class ChatService:
         Args:
             debug_info (dict): The new debug information.
         """
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            user_message_text,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-                self._assistant_message_id,
-                self._user_message_id,
-                self._chat_id,
-                self._user_message_text,
-            ]
-        )
 
         return modify_message(
-            user_id=user_id,
-            company_id=company_id,
-            assistant_message_id=assistant_message_id,
-            chat_id=chat_id,
-            user_message_id=user_message_id,
-            user_message_text=user_message_text,
+            user_id=self._user_id,
+            company_id=self._company_id,
+            assistant_message_id=self._assistant_message_id,
+            chat_id=self._chat_id,
+            user_message_id=self._user_message_id,
+            user_message_text=self._user_message_text,
             assistant=False,
             debug_info=debug_info,
         )
@@ -369,31 +334,14 @@ class ChatService:
         Raises:
             Exception: If the modification fails.
         """
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            user_message_text,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-                self._assistant_message_id,
-                self._user_message_id,
-                self._chat_id,
-                self._user_message_text,
-            ]
-        )
 
         return modify_message(
-            user_id=user_id,
-            company_id=company_id,
-            assistant_message_id=assistant_message_id,
-            chat_id=chat_id,
-            user_message_id=user_message_id,
-            user_message_text=user_message_text,
+            user_id=self._user_id,
+            company_id=self._company_id,
+            assistant_message_id=self._assistant_message_id,
+            chat_id=self._chat_id,
+            user_message_id=self._user_message_id,
+            user_message_text=self._user_message_text,
             assistant=False,
             content=content,
             references=references,
@@ -427,31 +375,13 @@ class ChatService:
             Exception: If the modification fails.
         """
 
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            user_message_text,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-                self._assistant_message_id,
-                self._user_message_id,
-                self._chat_id,
-                self._user_message_text,
-            ]
-        )
-
         return await modify_message_async(
-            user_id=user_id,
-            company_id=company_id,
-            assistant_message_id=assistant_message_id,
-            chat_id=chat_id,
-            user_message_id=user_message_id,
-            user_message_text=user_message_text,
+            user_id=self._user_id,
+            company_id=self._company_id,
+            assistant_message_id=self._assistant_message_id,
+            chat_id=self._chat_id,
+            user_message_id=self._user_message_id,
+            user_message_text=self._user_message_text,
             assistant=False,
             content=content,
             references=references,
@@ -486,31 +416,14 @@ class ChatService:
         Raises:
             Exception: If the modification fails.
         """
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            user_message_text,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-                self._assistant_message_id,
-                self._user_message_id,
-                self._chat_id,
-                self._user_message_text,
-            ]
-        )
 
         return modify_message(
-            user_id=user_id,
-            company_id=company_id,
-            assistant_message_id=assistant_message_id,
-            chat_id=chat_id,
-            user_message_id=user_message_id,
-            user_message_text=user_message_text,
+            user_id=self._user_id,
+            company_id=self._company_id,
+            assistant_message_id=self._assistant_message_id,
+            chat_id=self._chat_id,
+            user_message_id=self._user_message_id,
+            user_message_text=self._user_message_text,
             assistant=True,
             content=content,
             original_content=original_content,
@@ -546,30 +459,14 @@ class ChatService:
         Raises:
             Exception: If the modification fails.
         """
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            user_message_text,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-                self._assistant_message_id,
-                self._user_message_id,
-                self._chat_id,
-                self._user_message_text,
-            ]
-        )
+
         return await modify_message_async(
-            user_id=user_id,
-            company_id=company_id,
-            assistant_message_id=assistant_message_id,
-            chat_id=chat_id,
-            user_message_id=user_message_id,
-            user_message_text=user_message_text,
+            user_id=self._user_id,
+            company_id=self._company_id,
+            assistant_message_id=self._assistant_message_id,
+            chat_id=self._chat_id,
+            user_message_id=self._user_message_id,
+            user_message_text=self._user_message_text,
             assistant=True,
             content=content,
             original_content=original_content,
@@ -701,25 +598,12 @@ class ChatService:
         Raises:
             Exception: If the creation fails.
         """
-        [
-            company_id,
-            user_id,
-            assistant_id,
-            chat_id,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-                self._assistant_id,
-                self._chat_id,
-            ]
-        )
 
         chat_message = create_message(
-            user_id=user_id,
-            company_id=company_id,
-            chat_id=chat_id,
-            assistant_id=assistant_id,
+            user_id=self._user_id,
+            company_id=self._company_id,
+            chat_id=self._chat_id,
+            assistant_id=self._assistant_id,
             role=ChatMessageRole.ASSISTANT,
             content=content,
             original_content=original_content,
@@ -728,7 +612,7 @@ class ChatService:
             set_completed_at=set_completed_at,
         )
         # Update the assistant message id
-        self._assistant_message_id = chat_message.id
+        self._assistant_message_id = chat_message.id or "unknown"
         return chat_message
 
     async def create_assistant_message_async(
@@ -755,24 +639,12 @@ class ChatService:
         Raises:
             Exception: If the creation fails.
         """
-        [
-            company_id,
-            user_id,
-            assistant_id,
-            chat_id,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-                self._assistant_id,
-                self._chat_id,
-            ]
-        )
+
         chat_message = await create_message_async(
-            user_id=user_id,
-            company_id=company_id,
-            chat_id=chat_id,
-            assistant_id=assistant_id,
+            user_id=self._user_id,
+            company_id=self._company_id,
+            chat_id=self._chat_id,
+            assistant_id=self._assistant_id,
             role=ChatMessageRole.ASSISTANT,
             content=content,
             original_content=original_content,
@@ -781,7 +653,7 @@ class ChatService:
             set_completed_at=set_completed_at,
         )
         # Update the assistant message id
-        self._assistant_message_id = chat_message.id
+        self._assistant_message_id = chat_message.id or "unknown"
         return chat_message
 
     def create_user_message(
@@ -808,24 +680,12 @@ class ChatService:
         Raises:
             Exception: If the creation fails.
         """
-        [
-            company_id,
-            user_id,
-            assistant_id,
-            chat_id,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-                self._assistant_id,
-                self._chat_id,
-            ]
-        )
+
         chat_message = create_message(
-            user_id=user_id,
-            company_id=company_id,
-            chat_id=chat_id,
-            assistant_id=assistant_id,
+            user_id=self._user_id,
+            company_id=self._company_id,
+            chat_id=self._chat_id,
+            assistant_id=self._assistant_id,
             role=ChatMessageRole.USER,
             content=content,
             original_content=original_content,
@@ -834,7 +694,7 @@ class ChatService:
             set_completed_at=set_completed_at,
         )
         # Update the user message id
-        self._user_message_id = chat_message.id
+        self._user_message_id = chat_message.id or "unknown"
         return chat_message
 
     async def create_user_message_async(
@@ -861,24 +721,12 @@ class ChatService:
         Raises:
             Exception: If the creation fails.
         """
-        [
-            company_id,
-            user_id,
-            assistant_id,
-            chat_id,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-                self._assistant_id,
-                self._chat_id,
-            ]
-        )
+
         chat_message = await create_message_async(
-            user_id=user_id,
-            company_id=company_id,
-            chat_id=chat_id,
-            assistant_id=assistant_id,
+            user_id=self._user_id,
+            company_id=self._company_id,
+            chat_id=self._chat_id,
+            assistant_id=self._assistant_id,
             role=ChatMessageRole.USER,
             content=content,
             original_content=original_content,
@@ -887,7 +735,7 @@ class ChatService:
             set_completed_at=set_completed_at,
         )
         # Update the user message id
-        self._user_message_id = chat_message.id
+        self._user_message_id = chat_message.id or "unknown"
         return chat_message
 
     def create_message_assessment(
@@ -918,19 +766,10 @@ class ChatService:
         Raises:
             Exception: If the creation fails
         """
-        [
-            company_id,
-            user_id,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-            ]
-        )
 
         return create_message_assessment(
-            user_id=user_id,
-            company_id=company_id,
+            user_id=self._user_id,
+            company_id=self._company_id,
             assistant_message_id=assistant_message_id,
             status=status,
             type=type,
@@ -968,19 +807,10 @@ class ChatService:
         Raises:
             Exception: If the creation fails
         """
-        [
-            company_id,
-            user_id,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-            ]
-        )
 
         return await create_message_assessment_async(
-            user_id=user_id,
-            company_id=company_id,
+            user_id=self._user_id,
+            company_id=self._company_id,
             assistant_message_id=assistant_message_id,
             status=status,
             type=type,
@@ -1016,19 +846,10 @@ class ChatService:
         Raises:
             Exception: If the modification fails
         """
-        [
-            company_id,
-            user_id,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-            ]
-        )
 
         return modify_message_assessment(
-            user_id=user_id,
-            company_id=company_id,
+            user_id=self._user_id,
+            company_id=self._company_id,
             assistant_message_id=assistant_message_id,
             status=status,
             type=type,
@@ -1063,19 +884,9 @@ class ChatService:
         Raises:
             Exception: If the modification fails
         """
-        [
-            company_id,
-            user_id,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-            ]
-        )
-
         return await modify_message_assessment_async(
-            user_id=user_id,
-            company_id=company_id,
+            user_id=self._user_id,
+            company_id=self._company_id,
             assistant_message_id=assistant_message_id,
             status=status,
             type=type,
@@ -1124,31 +935,14 @@ class ChatService:
         """
         Streams a completion in the chat session synchronously.
         """
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            assistant_id,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-                self._assistant_message_id,
-                self._user_message_id,
-                self._chat_id,
-                self._assistant_id,
-            ]
-        )
 
         return stream_complete_with_references(
-            company_id=company_id,
-            user_id=user_id,
-            assistant_message_id=assistant_message_id,
-            user_message_id=user_message_id,
-            chat_id=chat_id,
-            assistant_id=assistant_id,
+            company_id=self._company_id,
+            user_id=self._user_id,
+            assistant_message_id=self._assistant_message_id,
+            user_message_id=self._user_message_id,
+            chat_id=self._chat_id,
+            assistant_id=self._assistant_id,
             messages=messages,
             model_name=model_name,
             content_chunks=content_chunks,
@@ -1224,31 +1018,13 @@ class ChatService:
         start_text: Optional[str] = None,
         other_options: Optional[dict] = None,
     ) -> LanguageModelStreamResponse:
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            assistant_id,
-        ] = validate_required_values(
-            [
-                self._company_id,
-                self._user_id,
-                self._assistant_message_id,
-                self._user_message_id,
-                self._chat_id,
-                self._assistant_id,
-            ]
-        )
-
         return await stream_complete_with_references_async(
-            company_id=company_id,
-            user_id=user_id,
-            assistant_message_id=assistant_message_id,
-            user_message_id=user_message_id,
-            chat_id=chat_id,
-            assistant_id=assistant_id,
+            company_id=self._company_id,
+            user_id=self._user_id,
+            assistant_message_id=self._assistant_message_id,
+            user_message_id=self._user_message_id,
+            chat_id=self._chat_id,
+            assistant_id=self._assistant_id,
             messages=messages,
             model_name=model_name,
             content_chunks=content_chunks,

unique_toolkit/content/service.py

@@ -54,16 +54,16 @@ class ContentService:
         self._event = event  # Changed to protected attribute
         self._metadata_filter = None
         if event:
-            self._company_id = event.company_id
-            self._user_id = event.user_id
+            self._company_id: str = event.company_id
+            self._user_id: str = event.user_id
             if isinstance(event, (ChatEvent, Event)):
                 self._metadata_filter = event.payload.metadata_filter
-                self._chat_id = event.payload.chat_id
+                self._chat_id: str | None = event.payload.chat_id
         else:
             [company_id, user_id] = validate_required_values([company_id, user_id])
-            self._company_id = company_id
-            self._user_id = user_id
-            self._chat_id = chat_id
+            self._company_id: str = company_id
+            self._user_id: str = user_id
+            self._chat_id: str | None = chat_id
 
     @property
     @deprecated(
@@ -95,7 +95,7 @@ class ContentService:
     @deprecated(
         "The company_id setter is deprecated and will be removed in a future version."
     )
-    def company_id(self, value: str | None) -> None:
+    def company_id(self, value: str) -> None:
         """
         Set the company identifier (deprecated).
 
@@ -121,7 +121,7 @@ class ContentService:
     @deprecated(
         "The user_id setter is deprecated and will be removed in a future version."
     )
-    def user_id(self, value: str | None) -> None:
+    def user_id(self, value: str) -> None:
         """
         Set the user identifier (deprecated).
 

unique_toolkit/embedding/service.py

@@ -25,12 +25,12 @@ class EmbeddingService(BaseService):
     ):
         self._event = event
         if event:
-            self._company_id = event.company_id
-            self._user_id = event.user_id
+            self._company_id: str = event.company_id
+            self._user_id: str = event.user_id
         else:
             [company_id, user_id] = validate_required_values([company_id, user_id])
-            self._company_id = company_id
-            self._user_id = user_id
+            self._company_id: str = company_id
+            self._user_id: str = user_id
 
     @property
     @deprecated(
@@ -62,7 +62,7 @@ class EmbeddingService(BaseService):
     @deprecated(
         "The company_id setter is deprecated and will be removed in a future version."
     )
-    def company_id(self, value: str | None) -> None:
+    def company_id(self, value: str) -> None:
         """
         Set the company identifier (deprecated).
 
@@ -88,7 +88,7 @@ class EmbeddingService(BaseService):
     @deprecated(
         "The user_id setter is deprecated and will be removed in a future version."
     )
-    def user_id(self, value: str | None) -> None:
+    def user_id(self, value: str) -> None:
         """
         Set the user identifier (deprecated).
 

unique_toolkit/language_model/service.py

@@ -50,10 +50,8 @@ class LanguageModelService:
         assistant_id: str | None = None,
     ):
         self._event = event
-        self._company_id = company_id
-        self._user_id = user_id
-        self._chat_id = chat_id
-        self._assistant_id = assistant_id
+        self._chat_id: str | None = chat_id
+        self._assistant_id: str | None = assistant_id
 
         if event:
             self._company_id = event.company_id
@@ -61,6 +59,10 @@ class LanguageModelService:
             if isinstance(event, (ChatEvent, Event)):
                 self._chat_id = event.payload.chat_id
                 self._assistant_id = event.payload.assistant_id
+        else:
+            [company_id, user_id] = validate_required_values([company_id, user_id])
+            self._company_id: str = company_id
+            self._user_id: str = user_id
 
     @property
     @deprecated(
@@ -92,7 +94,7 @@ class LanguageModelService:
     @deprecated(
         "The company_id setter is deprecated and will be removed in a future version."
     )
-    def company_id(self, value: str | None) -> None:
+    def company_id(self, value: str) -> None:
         """
         Set the company identifier (deprecated).
 
@@ -118,7 +120,7 @@ class LanguageModelService:
     @deprecated(
         "The user_id setter is deprecated and will be removed in a future version."
     )
-    def user_id(self, value: str | None) -> None:
+    def user_id(self, value: str) -> None:
         """
         Set the user identifier (deprecated).
 
@@ -193,10 +195,9 @@ class LanguageModelService:
         """
         Calls the completion endpoint synchronously without streaming the response.
         """
-        [company_id] = validate_required_values([self._company_id])
 
         return complete(
-            company_id=company_id,
+            company_id=self._company_id,
             messages=messages,
             model_name=model_name,
             temperature=temperature,
@@ -221,10 +222,9 @@ class LanguageModelService:
         """
         Calls the completion endpoint asynchronously without streaming the response.
         """
-        [company_id] = validate_required_values([self._company_id])
 
         return await complete_async(
-            company_id=company_id,
+            company_id=self._company_id,
             messages=messages,
             model_name=model_name,
             temperature=temperature,
@@ -277,10 +277,8 @@ class LanguageModelService:
         start_text: str | None = None,
         other_options: dict[str, Any] | None = None,
     ) -> LanguageModelStreamResponse:
-        [company_id] = validate_required_values([self._company_id])
-
         return complete_with_references(
-            company_id=company_id,
+            company_id=self._company_id,
             messages=messages,
             model_name=model_name,
             content_chunks=content_chunks,
@@ -303,10 +301,8 @@ class LanguageModelService:
         start_text: str | None = None,
         other_options: dict[str, Any] | None = None,
     ) -> LanguageModelStreamResponse:
-        [company_id] = validate_required_values([self._company_id])
-
         return await complete_with_references_async(
-            company_id=company_id,
+            company_id=self._company_id,
             messages=messages,
             model_name=model_name,
             content_chunks=content_chunks,

unique_toolkit/short_term_memory/service.py

@@ -34,8 +34,8 @@ class ShortTermMemoryService:
     ):
         self._event = event
         if event:
-            self._company_id = event.company_id
-            self._user_id = event.user_id
+            self._company_id: str = event.company_id
+            self._user_id: str = event.user_id
             if isinstance(event, (ChatEvent, Event)):
                 self._chat_id = event.payload.chat_id
                 self._message_id = event.payload.user_message.id
@@ -44,10 +44,11 @@ class ShortTermMemoryService:
             assert (
                 chat_id or message_id
             ), "Either chat_id or message_id must be provided"
-            self._company_id = company_id
-            self._user_id = user_id
-            self._chat_id = chat_id
-            self._message_id = message_id
+
+            self._company_id: str = company_id
+            self._user_id: str = user_id
+            self._chat_id: str | None = chat_id
+            self._message_id: str | None = message_id
 
     @property
     @deprecated(
@@ -79,7 +80,7 @@ class ShortTermMemoryService:
     @deprecated(
         "The company_id setter is deprecated and will be removed in a future version."
     )
-    def company_id(self, value: str | None) -> None:
+    def company_id(self, value: str) -> None:
         """
         Set the company identifier (deprecated).
 
@@ -105,7 +106,7 @@ class ShortTermMemoryService:
     @deprecated(
         "The user_id setter is deprecated and will be removed in a future version."
     )
-    def user_id(self, value: str | None) -> None:
+    def user_id(self, value: str) -> None:
         """
         Set the user identifier (deprecated).
 

unique_toolkit/smart_rules/compile.py

@@ -26,7 +26,7 @@ class Operator(str, Enum):
 
 
 class BaseStatement(BaseModel):
-    model_config: ConfigDict = {"serialize_by_alias": True}
+    model_config = ConfigDict(serialize_by_alias=True)
 
     def with_variables(
         self,

{unique_toolkit-0.7.28.dist-info → unique_toolkit-0.7.29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.7.28
+Version: 0.7.29
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -17,7 +17,7 @@ Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: regex (>=2024.5.15,<2025.0.0)
 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.31,<0.10.0)
+Requires-Dist: unique-sdk (>=0.9.33,<0.10.0)
 Description-Content-Type: text/markdown
 
 # Unique Toolkit
@@ -111,6 +111,14 @@ 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.7.29] - 2025-06-19
+- Fix typehintin in services
+- Error on invalid initialization
+
+#
+
 ## [0.7.28] - 2025-06-17
 - Revert default factory change on `ChatEventPayload` for attribute `metadata_filter` due to error in `backend-ingestion` on empty dict
 

{unique_toolkit-0.7.28.dist-info → unique_toolkit-0.7.29.dist-info}/RECORD

@@ -9,26 +9,26 @@ unique_toolkit/app/init_logging.py,sha256=Sh26SRxOj8i8dzobKhYha2lLrkrMTHfB1V4jR3
 unique_toolkit/app/init_sdk.py,sha256=Nv4Now4pMfM0AgRhbtatLpm_39rKxn0WmRLwmPhRl-8,1285
 unique_toolkit/app/performance/async_tasks.py,sha256=H0l3OAcosLwNHZ8d2pd-Di4wHIXfclEvagi5kfqLFPA,1941
 unique_toolkit/app/performance/async_wrapper.py,sha256=yVVcRDkcdyfjsxro-N29SBvi-7773wnfDplef6-y8xw,1077
-unique_toolkit/app/schemas.py,sha256=2IeOGnjMc9M_3u4ZvPWegMmQVWlW_teTbNFxjjJA_E4,5008
+unique_toolkit/app/schemas.py,sha256=f1WHnEqSMX2C_c9vtOOVNSdlSH--jK22GDbtRAKPVjc,5044
 unique_toolkit/app/verification.py,sha256=GxFFwcJMy25fCA_Xe89wKW7bgqOu8PAs5y8QpHF0GSc,3861
 unique_toolkit/chat/__init__.py,sha256=LRs2G-JTVuci4lbtHTkVUiNcZcSR6uqqfnAyo7af6nY,619
 unique_toolkit/chat/constants.py,sha256=05kq6zjqUVB2d6_P7s-90nbljpB3ryxwCI-CAz0r2O4,83
 unique_toolkit/chat/functions.py,sha256=QsJVhBXgK6jDWRYpEAt-22jy5NKGsYs4fmHrOEdHAyc,29865
 unique_toolkit/chat/schemas.py,sha256=ct3BbvdQmpcNeDmJdc-Iz33LX1_3O-SGVgrsuT0B99k,2881
-unique_toolkit/chat/service.py,sha256=jAgInqq7wU-erezl-BOS9BQ3DOR1nrbVFlMfACn26XI,42827
+unique_toolkit/chat/service.py,sha256=dRydaABJspU1KlFUQ_Q-rXl9lDkhgeh99SxAguY4s_M,37432
 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=EdJg_A_7loEtCQf4cah3QARQreJx6pdz89Rm96YbMVg,940
 unique_toolkit/content/constants.py,sha256=1iy4Y67xobl5VTnJB6SxSyuoBWbdLl9244xfVMUZi5o,60
 unique_toolkit/content/functions.py,sha256=0ELepm3_sl0SD_SYzvQVQ-jTdrcUqK5mVJZv0nQBuAw,18367
 unique_toolkit/content/schemas.py,sha256=iednldGojmoy9iPZwuIW23BfhWLE9G-fG_qAhxSgr7k,2451
-unique_toolkit/content/service.py,sha256=i7Cf6eQUwBjJntzBfQUvO8JiIjGsNLnsSl9FAwij8xw,19461
+unique_toolkit/content/service.py,sha256=jGRTIt0JQKL6qRyZ9-_njWJejKKvIvDOzEQrf8CdMaU,19491
 unique_toolkit/content/utils.py,sha256=GUVPrkZfMoAj4MRoBs5BD_7vSuLZTZx69hyWzYFrI50,7747
 unique_toolkit/embedding/__init__.py,sha256=uUyzjonPvuDCYsvXCIt7ErQXopLggpzX-MEQd3_e2kE,250
 unique_toolkit/embedding/constants.py,sha256=Lj8-Lcy1FvuC31PM9Exq7vaFuxQV4pEI1huUMFX-J2M,52
 unique_toolkit/embedding/functions.py,sha256=3qp-BfuMAbnp8YB04rh3xH8vsJuCBPizoy-JeaBFtoQ,1944
 unique_toolkit/embedding/schemas.py,sha256=1GvKCaSk4jixzVQ2PKq8yDqwGEVY_hWclYtoAr6CC2g,96
-unique_toolkit/embedding/service.py,sha256=ptwNNe2ji7FGqAb5VayedrB9T5b1T00XABwYtgvlGO8,4076
+unique_toolkit/embedding/service.py,sha256=xdR28rqxCicXN9qGPm0oaw88NAbRXJNIBwTZSARLn2g,4082
 unique_toolkit/embedding/utils.py,sha256=v86lo__bCJbxZBQ3OcLu5SuwT6NbFfWlcq8iyk6BuzQ,279
 unique_toolkit/evaluators/__init__.py,sha256=3Rfpnowm7MUXHWmeU4UV4s_3Hk-sw3V20oBwQCYlejQ,50
 unique_toolkit/evaluators/config.py,sha256=_DIXToJ-hGNpDAdWa7Q6GMjAsxiC_DquLF-SS5s9rTE,717
@@ -52,17 +52,17 @@ unique_toolkit/language_model/infos.py,sha256=peJ4cSJC__jGLWZoOZGRhoersmkwFmclsX
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
 unique_toolkit/language_model/reference.py,sha256=TrRfnxd-cjpdijmZ1dKII87aEACIbW7iU_cE9gtGuH4,8314
 unique_toolkit/language_model/schemas.py,sha256=DJD2aoMfs2Irnc4rzOrVuV4Fbt84LQAiDGG5rse1dgk,12770
-unique_toolkit/language_model/service.py,sha256=PJ0vOGPxzqsQXTQma8blLT4DfGP4gVZVcX2v1GVYozI,10610
+unique_toolkit/language_model/service.py,sha256=VRkUk2XbijqGlnTTvqU7uCue6qtT7lpLd_Y8f3bWv1I,10486
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
 unique_toolkit/protocols/support.py,sha256=V15WEIFKVMyF1QCnR8vIi4GrJy4dfTCB6d6JlqPZ58o,2341
 unique_toolkit/short_term_memory/__init__.py,sha256=2mI3AUrffgH7Yt-xS57EGqnHf7jnn6xquoKEhJqk3Wg,185
 unique_toolkit/short_term_memory/constants.py,sha256=698CL6-wjup2MvU19RxSmQk3gX7aqW_OOpZB7sbz_Xg,34
 unique_toolkit/short_term_memory/functions.py,sha256=3WiK-xatY5nh4Dr5zlDUye1k3E6kr41RiscwtTplw5k,4484
 unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJs8FEZXcgQTNenw,1406
-unique_toolkit/short_term_memory/service.py,sha256=vEKFxP1SScPrFniso492fVthWR1sosdFibhiNF3zRvI,8081
+unique_toolkit/short_term_memory/service.py,sha256=cqpXA0nMbi4PhFweg-Cql3u0RvaTi5c8Xjv0uHMiSGc,8112
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-unique_toolkit/smart_rules/compile.py,sha256=44qDrrKD-bKCjjyUep9qa1IwNkneXoQezfFoVm1QToM,9558
-unique_toolkit-0.7.28.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.7.28.dist-info/METADATA,sha256=CR6v3EWcXcTcDcL05cVP7LjRk9wE1Bwe6qPii5sKFjc,24161
-unique_toolkit-0.7.28.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.7.28.dist-info/RECORD,,
+unique_toolkit/smart_rules/compile.py,sha256=cxWjb2dxEI2HGsakKdVCkSNi7VK9mr08w5sDcFCQyWI,9553
+unique_toolkit-0.7.29.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.7.29.dist-info/METADATA,sha256=9-GvYOZhYI_KjxC9SVHK2lU5cniY1ZPAPoY9CGKkdrw,24255
+unique_toolkit-0.7.29.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.7.29.dist-info/RECORD,,