qanswer_sdk 3.1213.0__py3-none-any.whl → 3.1245.0__py3-none-any.whl

qanswer_sdk/models/{report_copilot_template.py → report_template.py}

@@ -18,23 +18,27 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, StrictInt, StrictStr
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
 from qanswer_sdk.models.slot import Slot
+from qanswer_sdk.models.template_clip_structure_output import TemplateClipStructureOutput
 from typing import Optional, Set
 from typing_extensions import Self
 
-class ReportCopilotTemplate(BaseModel):
+class ReportTemplate(BaseModel):
     """
-    ReportCopilotTemplate
+    ReportTemplate
     """ # noqa: E501
     id: StrictInt
     username: StrictStr
     template_name: StrictStr
     creation_date: StrictStr
     last_updated: StrictStr
+    shared_to: Optional[StrictStr] = None
+    conversation_id: Optional[StrictStr] = None
     template_with_slots: StrictStr
     slots: List[Slot]
-    __properties: ClassVar[List[str]] = ["id", "username", "template_name", "creation_date", "last_updated", "template_with_slots", "slots"]
+    clip_structure: Optional[TemplateClipStructureOutput] = None
+    __properties: ClassVar[List[str]] = ["id", "username", "template_name", "creation_date", "last_updated", "shared_to", "conversation_id", "template_with_slots", "slots", "clip_structure"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -54,7 +58,7 @@ class ReportCopilotTemplate(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of ReportCopilotTemplate from a JSON string"""
+        """Create an instance of ReportTemplate from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -82,11 +86,14 @@ class ReportCopilotTemplate(BaseModel):
                 if _item_slots:
                     _items.append(_item_slots.to_dict())
             _dict['slots'] = _items
+        # override the default output from pydantic by calling `to_dict()` of clip_structure
+        if self.clip_structure:
+            _dict['clip_structure'] = self.clip_structure.to_dict()
         return _dict
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of ReportCopilotTemplate from a dict"""
+        """Create an instance of ReportTemplate from a dict"""
         if obj is None:
             return None
 
@@ -99,8 +106,11 @@ class ReportCopilotTemplate(BaseModel):
             "template_name": obj.get("template_name"),
             "creation_date": obj.get("creation_date"),
             "last_updated": obj.get("last_updated"),
+            "shared_to": obj.get("shared_to"),
+            "conversation_id": obj.get("conversation_id"),
             "template_with_slots": obj.get("template_with_slots"),
-            "slots": [Slot.from_dict(_item) for _item in obj["slots"]] if obj.get("slots") is not None else None
+            "slots": [Slot.from_dict(_item) for _item in obj["slots"]] if obj.get("slots") is not None else None,
+            "clip_structure": TemplateClipStructureOutput.from_dict(obj["clip_structure"]) if obj.get("clip_structure") is not None else None
         })
         return _obj
 

qanswer_sdk/models/{report_copilot_create_payload.py → report_template_create_payload.py}

@@ -23,9 +23,9 @@ from qanswer_sdk.models.slot_update import SlotUpdate
 from typing import Optional, Set
 from typing_extensions import Self
 
-class ReportCopilotCreatePayload(BaseModel):
+class ReportTemplateCreatePayload(BaseModel):
     """
-    ReportCopilotCreatePayload
+    ReportTemplateCreatePayload
     """ # noqa: E501
     username: StrictStr
     dataset: StrictStr
@@ -52,7 +52,7 @@ class ReportCopilotCreatePayload(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of ReportCopilotCreatePayload from a JSON string"""
+        """Create an instance of ReportTemplateCreatePayload from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -84,7 +84,7 @@ class ReportCopilotCreatePayload(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of ReportCopilotCreatePayload from a dict"""
+        """Create an instance of ReportTemplateCreatePayload from a dict"""
         if obj is None:
             return None
 

qanswer_sdk/models/{report_copilot_template_simplified.py → report_template_simplified.py}

@@ -18,20 +18,22 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, StrictInt, StrictStr
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
-class ReportCopilotTemplateSimplified(BaseModel):
+class ReportTemplateSimplified(BaseModel):
     """
-    ReportCopilotTemplateSimplified
+    ReportTemplateSimplified
     """ # noqa: E501
     id: StrictInt
     username: StrictStr
     template_name: StrictStr
     creation_date: StrictStr
     last_updated: StrictStr
-    __properties: ClassVar[List[str]] = ["id", "username", "template_name", "creation_date", "last_updated"]
+    shared_to: Optional[StrictStr] = None
+    conversation_id: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["id", "username", "template_name", "creation_date", "last_updated", "shared_to", "conversation_id"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -51,7 +53,7 @@ class ReportCopilotTemplateSimplified(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of ReportCopilotTemplateSimplified from a JSON string"""
+        """Create an instance of ReportTemplateSimplified from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -76,7 +78,7 @@ class ReportCopilotTemplateSimplified(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of ReportCopilotTemplateSimplified from a dict"""
+        """Create an instance of ReportTemplateSimplified from a dict"""
         if obj is None:
             return None
 
@@ -88,7 +90,9 @@ class ReportCopilotTemplateSimplified(BaseModel):
             "username": obj.get("username"),
             "template_name": obj.get("template_name"),
             "creation_date": obj.get("creation_date"),
-            "last_updated": obj.get("last_updated")
+            "last_updated": obj.get("last_updated"),
+            "shared_to": obj.get("shared_to"),
+            "conversation_id": obj.get("conversation_id")
         })
         return _obj
 

qanswer_sdk/models/{report_copilot_export_template_as_docx_payload_simplified.py → report_template_simplified_payload.py}

@@ -22,9 +22,9 @@ from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
-class ReportCopilotExportTemplateAsDocxPayloadSimplified(BaseModel):
+class ReportTemplateSimplifiedPayload(BaseModel):
     """
-    ReportCopilotExportTemplateAsDocxPayloadSimplified
+    ReportTemplateSimplifiedPayload
     """ # noqa: E501
     username: Optional[StrictStr] = None
     dataset: Optional[StrictStr] = None
@@ -49,7 +49,7 @@ class ReportCopilotExportTemplateAsDocxPayloadSimplified(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of ReportCopilotExportTemplateAsDocxPayloadSimplified from a JSON string"""
+        """Create an instance of ReportTemplateSimplifiedPayload from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -74,7 +74,7 @@ class ReportCopilotExportTemplateAsDocxPayloadSimplified(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of ReportCopilotExportTemplateAsDocxPayloadSimplified from a dict"""
+        """Create an instance of ReportTemplateSimplifiedPayload from a dict"""
         if obj is None:
             return None
 

qanswer_sdk/models/{report_copilot_update_payload.py → report_template_update_payload.py}

@@ -18,20 +18,22 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, StrictStr
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
 from qanswer_sdk.models.slot_update import SlotUpdate
+from qanswer_sdk.models.template_clip_structure_input import TemplateClipStructureInput
 from typing import Optional, Set
 from typing_extensions import Self
 
-class ReportCopilotUpdatePayload(BaseModel):
+class ReportTemplateUpdatePayload(BaseModel):
     """
-    ReportCopilotUpdatePayload
+    ReportTemplateUpdatePayload
     """ # noqa: E501
     username: StrictStr
     dataset: StrictStr
-    template_with_slots: StrictStr
-    slots: List[SlotUpdate]
-    __properties: ClassVar[List[str]] = ["username", "dataset", "template_with_slots", "slots"]
+    template_with_slots: Optional[StrictStr] = None
+    slots: Optional[List[SlotUpdate]] = None
+    clip_structure: Optional[TemplateClipStructureInput] = None
+    __properties: ClassVar[List[str]] = ["username", "dataset", "template_with_slots", "slots", "clip_structure"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -51,7 +53,7 @@ class ReportCopilotUpdatePayload(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of ReportCopilotUpdatePayload from a JSON string"""
+        """Create an instance of ReportTemplateUpdatePayload from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -79,11 +81,14 @@ class ReportCopilotUpdatePayload(BaseModel):
                 if _item_slots:
                     _items.append(_item_slots.to_dict())
             _dict['slots'] = _items
+        # override the default output from pydantic by calling `to_dict()` of clip_structure
+        if self.clip_structure:
+            _dict['clip_structure'] = self.clip_structure.to_dict()
         return _dict
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of ReportCopilotUpdatePayload from a dict"""
+        """Create an instance of ReportTemplateUpdatePayload from a dict"""
         if obj is None:
             return None
 
@@ -94,7 +99,8 @@ class ReportCopilotUpdatePayload(BaseModel):
             "username": obj.get("username"),
             "dataset": obj.get("dataset"),
             "template_with_slots": obj.get("template_with_slots"),
-            "slots": [SlotUpdate.from_dict(_item) for _item in obj["slots"]] if obj.get("slots") is not None else None
+            "slots": [SlotUpdate.from_dict(_item) for _item in obj["slots"]] if obj.get("slots") is not None else None,
+            "clip_structure": TemplateClipStructureInput.from_dict(obj["clip_structure"]) if obj.get("clip_structure") is not None else None
         })
         return _obj
 

qanswer_sdk/models/search_task_settings.py

@@ -17,9 +17,9 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, StrictBool, StrictFloat, StrictInt, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictFloat, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional, Union
-from qanswer_sdk.models.llm_details import LLMDetails
+from qanswer_sdk.models.llm_endpoint_read_output import LLMEndpointReadOutput
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -28,7 +28,6 @@ class SearchTaskSettings(BaseModel):
     SearchTaskSettings
     """ # noqa: E501
     prompt: Optional[StrictStr] = None
-    llm_choice: Optional[StrictStr] = None
     bot_seed: Optional[StrictInt] = None
     bot_temperature: Optional[Union[StrictFloat, StrictInt]] = None
     bot_answer_length: Optional[StrictStr] = None
@@ -36,14 +35,15 @@ class SearchTaskSettings(BaseModel):
     stream_speed: Optional[Union[StrictFloat, StrictInt]] = None
     context_window: Optional[StrictInt] = None
     max_tokens: Optional[StrictInt] = None
+    slots_values: Optional[Dict[str, StrictStr]] = None
     bot_name: Optional[StrictStr] = None
     bot_description: Optional[StrictStr] = None
-    llm_details: LLMDetails
+    llm_id: Optional[StrictInt] = Field(default=None, description="The LLM ID. This field is populated based on the llm_choice.")
+    llm_endpoint: Optional[LLMEndpointReadOutput] = None
     query_expansion_prompt: Optional[StrictStr] = None
     show_date_in_sources: Optional[StrictBool] = None
-    slots_values: Optional[Dict[str, StrictStr]] = None
     selected_aggregations: Optional[List[StrictStr]] = None
-    __properties: ClassVar[List[str]] = ["prompt", "llm_choice", "bot_seed", "bot_temperature", "bot_answer_length", "number_of_references", "stream_speed", "context_window", "max_tokens", "bot_name", "bot_description", "llm_details", "query_expansion_prompt", "show_date_in_sources", "slots_values", "selected_aggregations"]
+    __properties: ClassVar[List[str]] = ["prompt", "bot_seed", "bot_temperature", "bot_answer_length", "number_of_references", "stream_speed", "context_window", "max_tokens", "slots_values", "bot_name", "bot_description", "llm_id", "llm_endpoint", "query_expansion_prompt", "show_date_in_sources", "selected_aggregations"]
 
     @field_validator('bot_answer_length')
     def bot_answer_length_validate_enum(cls, value):
@@ -94,9 +94,9 @@ class SearchTaskSettings(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of llm_details
-        if self.llm_details:
-            _dict['llm_details'] = self.llm_details.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of llm_endpoint
+        if self.llm_endpoint:
+            _dict['llm_endpoint'] = self.llm_endpoint.to_dict()
         return _dict
 
     @classmethod
@@ -110,7 +110,6 @@ class SearchTaskSettings(BaseModel):
 
         _obj = cls.model_validate({
             "prompt": obj.get("prompt"),
-            "llm_choice": obj.get("llm_choice"),
             "bot_seed": obj.get("bot_seed"),
             "bot_temperature": obj.get("bot_temperature"),
             "bot_answer_length": obj.get("bot_answer_length"),
@@ -118,12 +117,13 @@ class SearchTaskSettings(BaseModel):
             "stream_speed": obj.get("stream_speed"),
             "context_window": obj.get("context_window"),
             "max_tokens": obj.get("max_tokens"),
+            "slots_values": obj.get("slots_values"),
             "bot_name": obj.get("bot_name"),
             "bot_description": obj.get("bot_description"),
-            "llm_details": LLMDetails.from_dict(obj["llm_details"]) if obj.get("llm_details") is not None else None,
+            "llm_id": obj.get("llm_id"),
+            "llm_endpoint": LLMEndpointReadOutput.from_dict(obj["llm_endpoint"]) if obj.get("llm_endpoint") is not None else None,
             "query_expansion_prompt": obj.get("query_expansion_prompt"),
             "show_date_in_sources": obj.get("show_date_in_sources"),
-            "slots_values": obj.get("slots_values"),
             "selected_aggregations": obj.get("selected_aggregations")
         })
         return _obj

qanswer_sdk/models/search_task_update.py

@@ -17,8 +17,9 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, StrictBool, StrictFloat, StrictInt, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictFloat, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional, Union
+from qanswer_sdk.models.llm_endpoint_read_input import LLMEndpointReadInput
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -27,7 +28,7 @@ class SearchTaskUpdate(BaseModel):
     SearchTaskUpdate
     """ # noqa: E501
     prompt: Optional[StrictStr] = None
-    llm_choice: Optional[StrictStr] = None
+    llm_choice: Optional[StrictStr] = Field(default=None, description="The LLM choice. If not provided, the system default will be used.")
     bot_seed: Optional[StrictInt] = None
     bot_temperature: Optional[Union[StrictFloat, StrictInt]] = None
     bot_answer_length: Optional[StrictStr] = None
@@ -35,12 +36,15 @@ class SearchTaskUpdate(BaseModel):
     stream_speed: Optional[Union[StrictFloat, StrictInt]] = None
     context_window: Optional[StrictInt] = None
     max_tokens: Optional[StrictInt] = None
+    slots_values: Optional[Dict[str, StrictStr]] = None
     bot_name: Optional[StrictStr] = None
     bot_description: Optional[StrictStr] = None
+    llm_id: Optional[StrictInt] = Field(default=None, description="The LLM ID. This field is populated based on the llm_choice.")
+    llm_endpoint: Optional[LLMEndpointReadInput] = None
     query_expansion_prompt: Optional[StrictStr] = None
     show_date_in_sources: Optional[StrictBool] = None
     selected_aggregations: Optional[List[StrictStr]] = None
-    __properties: ClassVar[List[str]] = ["prompt", "llm_choice", "bot_seed", "bot_temperature", "bot_answer_length", "number_of_references", "stream_speed", "context_window", "max_tokens", "bot_name", "bot_description", "query_expansion_prompt", "show_date_in_sources", "selected_aggregations"]
+    __properties: ClassVar[List[str]] = ["prompt", "llm_choice", "bot_seed", "bot_temperature", "bot_answer_length", "number_of_references", "stream_speed", "context_window", "max_tokens", "slots_values", "bot_name", "bot_description", "llm_id", "llm_endpoint", "query_expansion_prompt", "show_date_in_sources", "selected_aggregations"]
 
     @field_validator('bot_answer_length')
     def bot_answer_length_validate_enum(cls, value):
@@ -91,6 +95,9 @@ class SearchTaskUpdate(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of llm_endpoint
+        if self.llm_endpoint:
+            _dict['llm_endpoint'] = self.llm_endpoint.to_dict()
         return _dict
 
     @classmethod
@@ -112,8 +119,11 @@ class SearchTaskUpdate(BaseModel):
             "stream_speed": obj.get("stream_speed"),
             "context_window": obj.get("context_window"),
             "max_tokens": obj.get("max_tokens"),
+            "slots_values": obj.get("slots_values"),
             "bot_name": obj.get("bot_name"),
             "bot_description": obj.get("bot_description"),
+            "llm_id": obj.get("llm_id"),
+            "llm_endpoint": LLMEndpointReadInput.from_dict(obj["llm_endpoint"]) if obj.get("llm_endpoint") is not None else None,
             "query_expansion_prompt": obj.get("query_expansion_prompt"),
             "show_date_in_sources": obj.get("show_date_in_sources"),
             "selected_aggregations": obj.get("selected_aggregations")

qanswer_sdk/models/{set_logo1_request.py → set_logo2_request.py}

@@ -22,9 +22,9 @@ from typing import Any, ClassVar, Dict, List, Tuple, Union
 from typing import Optional, Set
 from typing_extensions import Self
 
-class SetLogo1Request(BaseModel):
+class SetLogo2Request(BaseModel):
     """
-    SetLogo1Request
+    SetLogo2Request
     """ # noqa: E501
     file: Union[StrictBytes, StrictStr, Tuple[StrictStr, StrictBytes]]
     __properties: ClassVar[List[str]] = ["file"]
@@ -47,7 +47,7 @@ class SetLogo1Request(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of SetLogo1Request from a JSON string"""
+        """Create an instance of SetLogo2Request from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -72,7 +72,7 @@ class SetLogo1Request(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of SetLogo1Request from a dict"""
+        """Create an instance of SetLogo2Request from a dict"""
         if obj is None:
             return None
 

qanswer_sdk/models/shared_organization_access.py

@@ -19,6 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
+from qanswer_sdk.models.organization import Organization
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -27,10 +28,10 @@ class SharedOrganizationAccess(BaseModel):
     SharedOrganizationAccess
     """ # noqa: E501
     id: Optional[StrictInt] = None
-    organization_id: Optional[StrictInt] = Field(default=None, alias="organizationId")
+    organization: Optional[Organization] = None
     shared_access_type: Optional[StrictStr] = Field(default=None, alias="sharedAccessType")
     ai_assistant: Optional[UserDataset] = Field(default=None, alias="aiAssistant")
-    __properties: ClassVar[List[str]] = ["id", "organizationId", "sharedAccessType", "aiAssistant"]
+    __properties: ClassVar[List[str]] = ["id", "organization", "sharedAccessType", "aiAssistant"]
 
     @field_validator('shared_access_type')
     def shared_access_type_validate_enum(cls, value):
@@ -81,6 +82,9 @@ class SharedOrganizationAccess(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of organization
+        if self.organization:
+            _dict['organization'] = self.organization.to_dict()
         # override the default output from pydantic by calling `to_dict()` of ai_assistant
         if self.ai_assistant:
             _dict['aiAssistant'] = self.ai_assistant.to_dict()
@@ -97,7 +101,7 @@ class SharedOrganizationAccess(BaseModel):
 
         _obj = cls.model_validate({
             "id": obj.get("id"),
-            "organizationId": obj.get("organizationId"),
+            "organization": Organization.from_dict(obj["organization"]) if obj.get("organization") is not None else None,
             "sharedAccessType": obj.get("sharedAccessType"),
             "aiAssistant": UserDataset.from_dict(obj["aiAssistant"]) if obj.get("aiAssistant") is not None else None
         })

qanswer_sdk/models/shared_team_access.py

@@ -19,7 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
-from qanswer_sdk.models.user_dataset import UserDataset
+from qanswer_sdk.models.team import Team
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -28,10 +28,10 @@ class SharedTeamAccess(BaseModel):
     SharedTeamAccess
     """ # noqa: E501
     id: Optional[StrictInt] = None
-    team_id: Optional[StrictInt] = Field(default=None, alias="teamId")
+    team: Optional[Team] = None
     shared_access_type: Optional[StrictStr] = Field(default=None, alias="sharedAccessType")
     ai_assistant: Optional[UserDataset] = Field(default=None, alias="aiAssistant")
-    __properties: ClassVar[List[str]] = ["id", "teamId", "sharedAccessType", "aiAssistant"]
+    __properties: ClassVar[List[str]] = ["id", "team", "sharedAccessType", "aiAssistant"]
 
     @field_validator('shared_access_type')
     def shared_access_type_validate_enum(cls, value):
@@ -82,6 +82,9 @@ class SharedTeamAccess(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of team
+        if self.team:
+            _dict['team'] = self.team.to_dict()
         # override the default output from pydantic by calling `to_dict()` of ai_assistant
         if self.ai_assistant:
             _dict['aiAssistant'] = self.ai_assistant.to_dict()
@@ -98,10 +101,13 @@ class SharedTeamAccess(BaseModel):
 
         _obj = cls.model_validate({
             "id": obj.get("id"),
-            "teamId": obj.get("teamId"),
+            "team": Team.from_dict(obj["team"]) if obj.get("team") is not None else None,
             "sharedAccessType": obj.get("sharedAccessType"),
             "aiAssistant": UserDataset.from_dict(obj["aiAssistant"]) if obj.get("aiAssistant") is not None else None
         })
         return _obj
 
+from qanswer_sdk.models.user_dataset import UserDataset
+# TODO: Rewrite to not use raise_errors
+SharedTeamAccess.model_rebuild(raise_errors=False)
 

qanswer_sdk/models/slot.py

@@ -34,11 +34,13 @@ class Slot(BaseModel):
     prompt: StrictStr
     completion: StrictStr
     sources: List[Source]
+    prompt_used: Optional[StrictStr] = None
+    llm_choice: Optional[StrictStr] = None
     metadata_filters: Optional[List[List[SearchMetadataFilter]]] = None
     mode: Optional[StrictStr] = Field(default=None, description="The mode of the response generation.")
     is_input_data_cropped: Optional[StrictBool] = Field(default=None, description="Indicates whether the input data was cropped during prompting.")
     task_settings: Optional[ReportCopilotSlotTaskSettings] = None
-    __properties: ClassVar[List[str]] = ["id", "node_id", "prompt", "completion", "sources", "metadata_filters", "mode", "is_input_data_cropped", "task_settings"]
+    __properties: ClassVar[List[str]] = ["id", "node_id", "prompt", "completion", "sources", "prompt_used", "llm_choice", "metadata_filters", "mode", "is_input_data_cropped", "task_settings"]
 
     @field_validator('mode')
     def mode_validate_enum(cls, value):
@@ -125,6 +127,8 @@ class Slot(BaseModel):
             "prompt": obj.get("prompt"),
             "completion": obj.get("completion"),
             "sources": [Source.from_dict(_item) for _item in obj["sources"]] if obj.get("sources") is not None else None,
+            "prompt_used": obj.get("prompt_used"),
+            "llm_choice": obj.get("llm_choice"),
             "metadata_filters": [
                     [SearchMetadataFilter.from_dict(_inner_item) for _inner_item in _item]
                     for _item in obj["metadata_filters"]

qanswer_sdk/models/slot_update.py

@@ -34,11 +34,13 @@ class SlotUpdate(BaseModel):
     prompt: StrictStr
     completion: StrictStr
     sources: List[Source]
+    prompt_used: Optional[StrictStr] = None
+    llm_choice: Optional[StrictStr] = None
     metadata_filters: Optional[List[List[SearchMetadataFilter]]] = None
     mode: Optional[StrictStr] = Field(default=None, description="The mode of the response generation.")
     is_input_data_cropped: Optional[StrictBool] = Field(default=None, description="Indicates whether the input data was cropped during prompting.")
     task_settings: Optional[ReportCopilotSlotTaskUpdate] = None
-    __properties: ClassVar[List[str]] = ["id", "node_id", "prompt", "completion", "sources", "metadata_filters", "mode", "is_input_data_cropped", "task_settings"]
+    __properties: ClassVar[List[str]] = ["id", "node_id", "prompt", "completion", "sources", "prompt_used", "llm_choice", "metadata_filters", "mode", "is_input_data_cropped", "task_settings"]
 
     @field_validator('mode')
     def mode_validate_enum(cls, value):
@@ -125,6 +127,8 @@ class SlotUpdate(BaseModel):
             "prompt": obj.get("prompt"),
             "completion": obj.get("completion"),
             "sources": [Source.from_dict(_item) for _item in obj["sources"]] if obj.get("sources") is not None else None,
+            "prompt_used": obj.get("prompt_used"),
+            "llm_choice": obj.get("llm_choice"),
             "metadata_filters": [
                     [SearchMetadataFilter.from_dict(_inner_item) for _inner_item in _item]
                     for _item in obj["metadata_filters"]

qanswer_sdk/models/socket_conversation_message.py

@@ -21,7 +21,6 @@ from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictInt, Strict
 from typing import Any, ClassVar, Dict, List, Optional
 from qanswer_sdk.models.aggregations_group import AggregationsGroup
 from qanswer_sdk.models.content_item import ContentItem
-from qanswer_sdk.models.question_completion import QuestionCompletion
 from qanswer_sdk.models.source import Source
 from typing import Optional, Set
 from typing_extensions import Self
@@ -49,14 +48,14 @@ class SocketConversationMessage(BaseModel):
     interface_origin: Optional[StrictStr] = None
     comment: Optional[StrictStr] = None
     llm_choice: Optional[StrictStr] = None
-    training_examples: Optional[List[QuestionCompletion]] = None
     prompt_used: Optional[StrictStr] = None
     mode: Optional[StrictStr] = None
     is_input_data_cropped: Optional[StrictBool] = None
     aggs: Optional[List[AggregationsGroup]] = None
     expanded_question: Optional[StrictStr] = None
+    prompt_type: Optional[StrictStr] = None
     status: StrictStr
-    __properties: ClassVar[List[str]] = ["username", "dataset", "jwt", "api_key", "conversation_id", "user", "role", "contents", "message_id", "parent_message_id", "timestamp", "feedback", "origin", "sources", "all_sources", "task_category", "interface_origin", "comment", "llm_choice", "training_examples", "prompt_used", "mode", "is_input_data_cropped", "aggs", "expanded_question", "status"]
+    __properties: ClassVar[List[str]] = ["username", "dataset", "jwt", "api_key", "conversation_id", "user", "role", "contents", "message_id", "parent_message_id", "timestamp", "feedback", "origin", "sources", "all_sources", "task_category", "interface_origin", "comment", "llm_choice", "prompt_used", "mode", "is_input_data_cropped", "aggs", "expanded_question", "prompt_type", "status"]
 
     @field_validator('role')
     def role_validate_enum(cls, value):
@@ -149,13 +148,6 @@ class SocketConversationMessage(BaseModel):
                 if _item_all_sources:
                     _items.append(_item_all_sources.to_dict())
             _dict['all_sources'] = _items
-        # override the default output from pydantic by calling `to_dict()` of each item in training_examples (list)
-        _items = []
-        if self.training_examples:
-            for _item_training_examples in self.training_examples:
-                if _item_training_examples:
-                    _items.append(_item_training_examples.to_dict())
-            _dict['training_examples'] = _items
         # override the default output from pydantic by calling `to_dict()` of each item in aggs (list)
         _items = []
         if self.aggs:
@@ -194,12 +186,12 @@ class SocketConversationMessage(BaseModel):
             "interface_origin": obj.get("interface_origin"),
             "comment": obj.get("comment"),
             "llm_choice": obj.get("llm_choice"),
-            "training_examples": [QuestionCompletion.from_dict(_item) for _item in obj["training_examples"]] if obj.get("training_examples") is not None else None,
             "prompt_used": obj.get("prompt_used"),
             "mode": obj.get("mode"),
             "is_input_data_cropped": obj.get("is_input_data_cropped"),
             "aggs": [AggregationsGroup.from_dict(_item) for _item in obj["aggs"]] if obj.get("aggs") is not None else None,
             "expanded_question": obj.get("expanded_question"),
+            "prompt_type": obj.get("prompt_type"),
             "status": obj.get("status")
         })
         return _obj