qanswer_sdk 3.1509.0__py3-none-any.whl → 3.1510.0__py3-none-any.whl

qanswer_sdk/models/socket_conversation_message.py

@@ -22,6 +22,9 @@ 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.source import Source
+from qanswer_sdk.models.thinking_block import ThinkingBlock
+from qanswer_sdk.models.tool_selection import ToolSelection
+from qanswer_sdk.models.websearch_scraping_site import WebsearchScrapingSite
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -56,12 +59,15 @@ class SocketConversationMessage(BaseModel):
     expanded_question: Optional[StrictStr] = None
     prompt_type: Optional[StrictStr] = None
     tagged_assistant_name: Optional[StrictStr] = None
-    synonyms_used: Optional[List[Dict[str, Any]]] = None
-    tool_calls: Optional[List[Dict[str, Any]]] = None
+    synonyms_used: Optional[List[StrictStr]] = None
+    tool_calls: Optional[List[Dict[str, Dict[str, Any]]]] = None
     tool_call_id: Optional[StrictStr] = None
-    tool_selections: Optional[List[Dict[str, Any]]] = None
+    tool_selections: Optional[List[ToolSelection]] = None
+    server_uids: Optional[List[StrictStr]] = None
+    thinking_blocks: Optional[List[ThinkingBlock]] = None
+    websearch_sites: Optional[List[WebsearchScrapingSite]] = 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", "error_key", "prompt_used", "mode", "is_input_data_cropped", "aggs", "expanded_question", "prompt_type", "tagged_assistant_name", "synonyms_used", "tool_calls", "tool_call_id", "tool_selections", "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", "error_key", "prompt_used", "mode", "is_input_data_cropped", "aggs", "expanded_question", "prompt_type", "tagged_assistant_name", "synonyms_used", "tool_calls", "tool_call_id", "tool_selections", "server_uids", "thinking_blocks", "websearch_sites", "status"]
 
     @field_validator('role')
     def role_validate_enum(cls, value):
@@ -73,8 +79,8 @@ class SocketConversationMessage(BaseModel):
     @field_validator('task_category')
     def task_category_validate_enum(cls, value):
         """Validates the enum"""
-        if value not in set(['text', 'sparql', 'sql', 'sparql2text', 'sparqlvisualize', 'undefined', 'typing', 'user_joined', 'user_left', 'human_takeover_proposed', 'human_takeover_accepted', 'human_takeover_rejected', 'thinking', 'something_went_wrong', 'mcp_tool_calls', 'mcp_tool_choice_accept', 'mcp_tool_choice_reject', 'mcp_ask_for_connection', 'mcp_tool_choice_ask_confirmation', 'mcp_tool_call_response']):
-            raise ValueError("must be one of enum values ('text', 'sparql', 'sql', 'sparql2text', 'sparqlvisualize', 'undefined', 'typing', 'user_joined', 'user_left', 'human_takeover_proposed', 'human_takeover_accepted', 'human_takeover_rejected', 'thinking', 'something_went_wrong', 'mcp_tool_calls', 'mcp_tool_choice_accept', 'mcp_tool_choice_reject', 'mcp_ask_for_connection', 'mcp_tool_choice_ask_confirmation', 'mcp_tool_call_response')")
+        if value not in set(['text', 'sparql', 'sql', 'sparql2text', 'sparqlvisualize', 'undefined', 'typing', 'user_joined', 'user_left', 'human_takeover_proposed', 'human_takeover_accepted', 'human_takeover_rejected', 'websearch', 'thinking', 'something_went_wrong', 'mcp_tool_calls', 'mcp_tool_choice_accept', 'mcp_tool_choice_reject', 'mcp_ask_for_connection', 'mcp_tool_choice_ask_confirmation', 'mcp_tool_call_response']):
+            raise ValueError("must be one of enum values ('text', 'sparql', 'sql', 'sparql2text', 'sparqlvisualize', 'undefined', 'typing', 'user_joined', 'user_left', 'human_takeover_proposed', 'human_takeover_accepted', 'human_takeover_rejected', 'websearch', 'thinking', 'something_went_wrong', 'mcp_tool_calls', 'mcp_tool_choice_accept', 'mcp_tool_choice_reject', 'mcp_ask_for_connection', 'mcp_tool_choice_ask_confirmation', 'mcp_tool_call_response')")
         return value
 
     @field_validator('error_key')
@@ -171,6 +177,27 @@ class SocketConversationMessage(BaseModel):
                 if _item_aggs:
                     _items.append(_item_aggs.to_dict())
             _dict['aggs'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in tool_selections (list)
+        _items = []
+        if self.tool_selections:
+            for _item_tool_selections in self.tool_selections:
+                if _item_tool_selections:
+                    _items.append(_item_tool_selections.to_dict())
+            _dict['tool_selections'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in thinking_blocks (list)
+        _items = []
+        if self.thinking_blocks:
+            for _item_thinking_blocks in self.thinking_blocks:
+                if _item_thinking_blocks:
+                    _items.append(_item_thinking_blocks.to_dict())
+            _dict['thinking_blocks'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in websearch_sites (list)
+        _items = []
+        if self.websearch_sites:
+            for _item_websearch_sites in self.websearch_sites:
+                if _item_websearch_sites:
+                    _items.append(_item_websearch_sites.to_dict())
+            _dict['websearch_sites'] = _items
         return _dict
 
     @classmethod
@@ -213,7 +240,10 @@ class SocketConversationMessage(BaseModel):
             "synonyms_used": obj.get("synonyms_used"),
             "tool_calls": obj.get("tool_calls"),
             "tool_call_id": obj.get("tool_call_id"),
-            "tool_selections": obj.get("tool_selections"),
+            "tool_selections": [ToolSelection.from_dict(_item) for _item in obj["tool_selections"]] if obj.get("tool_selections") is not None else None,
+            "server_uids": obj.get("server_uids"),
+            "thinking_blocks": [ThinkingBlock.from_dict(_item) for _item in obj["thinking_blocks"]] if obj.get("thinking_blocks") is not None else None,
+            "websearch_sites": [WebsearchScrapingSite.from_dict(_item) for _item in obj["websearch_sites"]] if obj.get("websearch_sites") is not None else None,
             "status": obj.get("status")
         })
         return _obj

qanswer_sdk/models/template_toggle_request.py

@@ -26,11 +26,11 @@ class TemplateToggleRequest(BaseModel):
     """
     TemplateToggleRequest
     """ # noqa: E501
-    username: StrictStr = Field(description="Username associated with the dataset")
-    dataset: StrictStr = Field(description="Dataset identifier")
-    dataset_id: Optional[StrictInt] = Field(default=None, description="Dataset ID (optional, for additional verification)")
+    username: Optional[StrictStr] = Field(default=None, description="Username associated with the dataset")
+    dataset: Optional[StrictStr] = Field(default=None, description="Dataset identifier")
+    assistant_id: StrictInt = Field(description="Dataset ID (optional, for additional verification)")
     publish: StrictBool = Field(description="Whether the assistant should be published as a template")
-    __properties: ClassVar[List[str]] = ["username", "dataset", "dataset_id", "publish"]
+    __properties: ClassVar[List[str]] = ["username", "dataset", "assistant_id", "publish"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -85,7 +85,7 @@ class TemplateToggleRequest(BaseModel):
         _obj = cls.model_validate({
             "username": obj.get("username"),
             "dataset": obj.get("dataset"),
-            "dataset_id": obj.get("dataset_id"),
+            "assistant_id": obj.get("assistant_id"),
             "publish": obj.get("publish")
         })
         return _obj

qanswer_sdk/models/text2_sparql_payload.py

@@ -52,10 +52,12 @@ class Text2SparqlPayload(BaseModel):
     index_hashes: Optional[List[Dict[str, Any]]] = Field(default=None, description="The hashes of the indexes to be used for the response.")
     apply_map_reduce: Optional[StrictBool] = Field(default=None, description="Determines whether to apply map-reduce strategy for the response.")
     conversation_has_clip: Optional[StrictBool] = Field(default=None, description="Indicates whether the conversation has a clip associated with it.")
+    incognito: Optional[StrictBool] = Field(default=None, description="When true, the conversation is stored only in Redis with a TTL, is not retrievable via list/get, and will be purged automatically.")
     current_time: Optional[StrictStr] = Field(default=None, description="The current time to be used in the prompt, if applicable.")
     tool_selections: Optional[List[ToolSelection]] = Field(default=None, description="The selections of the MCP tools to be used.")
+    expanded_question: Optional[StrictStr] = Field(default=None, description="The expanded version of the original question, if applicable.")
     validate_sparql: Optional[StrictBool] = None
-    __properties: ClassVar[List[str]] = ["question", "username", "dataset", "organizationId", "image_urls", "parent_message_id", "conversation_id", "is_regenerate", "origin", "text_fragmentation", "interface_origin", "system_prompt", "prompt", "additional_fields", "bypass_guardrail", "talk_to_web", "original_question", "oauth_token", "auto_filters", "filters", "exclude_filters", "index_hashes", "apply_map_reduce", "conversation_has_clip", "current_time", "tool_selections", "validate_sparql"]
+    __properties: ClassVar[List[str]] = ["question", "username", "dataset", "organizationId", "image_urls", "parent_message_id", "conversation_id", "is_regenerate", "origin", "text_fragmentation", "interface_origin", "system_prompt", "prompt", "additional_fields", "bypass_guardrail", "talk_to_web", "original_question", "oauth_token", "auto_filters", "filters", "exclude_filters", "index_hashes", "apply_map_reduce", "conversation_has_clip", "incognito", "current_time", "tool_selections", "expanded_question", "validate_sparql"]
 
     @field_validator('interface_origin')
     def interface_origin_validate_enum(cls, value):
@@ -173,8 +175,10 @@ class Text2SparqlPayload(BaseModel):
             "index_hashes": obj.get("index_hashes"),
             "apply_map_reduce": obj.get("apply_map_reduce"),
             "conversation_has_clip": obj.get("conversation_has_clip"),
+            "incognito": obj.get("incognito"),
             "current_time": obj.get("current_time"),
             "tool_selections": [ToolSelection.from_dict(_item) for _item in obj["tool_selections"]] if obj.get("tool_selections") is not None else None,
+            "expanded_question": obj.get("expanded_question"),
             "validate_sparql": obj.get("validate_sparql")
         })
         return _obj

qanswer_sdk/models/thinking_block.py

@@ -0,0 +1,91 @@
+# coding: utf-8
+
+"""
+    QAnswer: Api Documentation
+
+    APIs provided by QAnswer
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, StrictStr
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ThinkingBlock(BaseModel):
+    """
+    ThinkingBlock
+    """ # noqa: E501
+    type: StrictStr
+    thinking: StrictStr
+    signature: StrictStr
+    __properties: ClassVar[List[str]] = ["type", "thinking", "signature"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of ThinkingBlock from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ThinkingBlock from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "type": obj.get("type"),
+            "thinking": obj.get("thinking"),
+            "signature": obj.get("signature")
+        })
+        return _obj
+
+

qanswer_sdk/models/{subject_model.py → update_cells_payload.py}

@@ -17,19 +17,19 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, StrictStr
+from pydantic import BaseModel, ConfigDict, StrictInt, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
-class SubjectModel(BaseModel):
+class UpdateCellsPayload(BaseModel):
     """
-    SubjectModel
+    UpdateCellsPayload
     """ # noqa: E501
-    label: Optional[StrictStr] = None
+    id: Optional[StrictInt] = None
     value: Optional[StrictStr] = None
-    filter_by: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["label", "value", "filter_by"]
+    prompt: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["id", "value", "prompt"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -49,7 +49,7 @@ class SubjectModel(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of SubjectModel from a JSON string"""
+        """Create an instance of UpdateCellsPayload from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -74,7 +74,7 @@ class SubjectModel(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of SubjectModel from a dict"""
+        """Create an instance of UpdateCellsPayload from a dict"""
         if obj is None:
             return None
 
@@ -82,9 +82,9 @@ class SubjectModel(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "label": obj.get("label"),
+            "id": obj.get("id"),
             "value": obj.get("value"),
-            "filter_by": obj.get("filter_by")
+            "prompt": obj.get("prompt")
         })
         return _obj
 

qanswer_sdk/models/{relation_extraction_configuration.py → update_file_es_meta_data.py}

@@ -17,20 +17,20 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, StrictStr
 from typing import Any, ClassVar, Dict, List
-from qanswer_sdk.models.relation_model import RelationModel
-from qanswer_sdk.models.subject_model import SubjectModel
+from qanswer_sdk.models.es_meta_data_value import ESMetaDataValue
 from typing import Optional, Set
 from typing_extensions import Self
 
-class RelationExtractionConfiguration(BaseModel):
+class UpdateFileESMetaData(BaseModel):
     """
-    RelationExtractionConfiguration
+    UpdateFileESMetaData
     """ # noqa: E501
-    subject_entities: List[SubjectModel]
-    relations: List[RelationModel]
-    __properties: ClassVar[List[str]] = ["subject_entities", "relations"]
+    additional_fields: Dict[str, ESMetaDataValue]
+    metadata: Dict[str, ESMetaDataValue]
+    file_id: StrictStr
+    __properties: ClassVar[List[str]] = ["additional_fields", "metadata", "file_id"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -50,7 +50,7 @@ class RelationExtractionConfiguration(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of RelationExtractionConfiguration from a JSON string"""
+        """Create an instance of UpdateFileESMetaData from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -71,25 +71,25 @@ class RelationExtractionConfiguration(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of each item in subject_entities (list)
-        _items = []
-        if self.subject_entities:
-            for _item_subject_entities in self.subject_entities:
-                if _item_subject_entities:
-                    _items.append(_item_subject_entities.to_dict())
-            _dict['subject_entities'] = _items
-        # override the default output from pydantic by calling `to_dict()` of each item in relations (list)
-        _items = []
-        if self.relations:
-            for _item_relations in self.relations:
-                if _item_relations:
-                    _items.append(_item_relations.to_dict())
-            _dict['relations'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each value in additional_fields (dict)
+        _field_dict = {}
+        if self.additional_fields:
+            for _key_additional_fields in self.additional_fields:
+                if self.additional_fields[_key_additional_fields]:
+                    _field_dict[_key_additional_fields] = self.additional_fields[_key_additional_fields].to_dict()
+            _dict['additional_fields'] = _field_dict
+        # override the default output from pydantic by calling `to_dict()` of each value in metadata (dict)
+        _field_dict = {}
+        if self.metadata:
+            for _key_metadata in self.metadata:
+                if self.metadata[_key_metadata]:
+                    _field_dict[_key_metadata] = self.metadata[_key_metadata].to_dict()
+            _dict['metadata'] = _field_dict
         return _dict
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of RelationExtractionConfiguration from a dict"""
+        """Create an instance of UpdateFileESMetaData from a dict"""
         if obj is None:
             return None
 
@@ -97,8 +97,19 @@ class RelationExtractionConfiguration(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "subject_entities": [SubjectModel.from_dict(_item) for _item in obj["subject_entities"]] if obj.get("subject_entities") is not None else None,
-            "relations": [RelationModel.from_dict(_item) for _item in obj["relations"]] if obj.get("relations") is not None else None
+            "additional_fields": dict(
+                (_k, ESMetaDataValue.from_dict(_v))
+                for _k, _v in obj["additional_fields"].items()
+            )
+            if obj.get("additional_fields") is not None
+            else None,
+            "metadata": dict(
+                (_k, ESMetaDataValue.from_dict(_v))
+                for _k, _v in obj["metadata"].items()
+            )
+            if obj.get("metadata") is not None
+            else None,
+            "file_id": obj.get("file_id")
         })
         return _obj
 

qanswer_sdk/models/{update_metadata_model.py → update_metadata_model_input.py}

@@ -19,17 +19,17 @@ import json
 
 from pydantic import BaseModel, ConfigDict, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
-from qanswer_sdk.models.file_es_meta_data import FileESMetaData
+from qanswer_sdk.models.update_file_es_meta_data import UpdateFileESMetaData
 from typing import Optional, Set
 from typing_extensions import Self
 
-class UpdateMetadataModel(BaseModel):
+class UpdateMetadataModelInput(BaseModel):
     """
-    UpdateMetadataModel
+    UpdateMetadataModelInput
     """ # noqa: E501
     username: StrictStr
     dataset: StrictStr
-    data: FileESMetaData
+    data: UpdateFileESMetaData
     conversation_id: Optional[StrictStr] = None
     __properties: ClassVar[List[str]] = ["username", "dataset", "data", "conversation_id"]
 
@@ -51,7 +51,7 @@ class UpdateMetadataModel(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of UpdateMetadataModel from a JSON string"""
+        """Create an instance of UpdateMetadataModelInput from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -79,7 +79,7 @@ class UpdateMetadataModel(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of UpdateMetadataModel from a dict"""
+        """Create an instance of UpdateMetadataModelInput from a dict"""
         if obj is None:
             return None
 
@@ -89,7 +89,7 @@ class UpdateMetadataModel(BaseModel):
         _obj = cls.model_validate({
             "username": obj.get("username"),
             "dataset": obj.get("dataset"),
-            "data": FileESMetaData.from_dict(obj["data"]) if obj.get("data") is not None else None,
+            "data": UpdateFileESMetaData.from_dict(obj["data"]) if obj.get("data") is not None else None,
             "conversation_id": obj.get("conversation_id")
         })
         return _obj

qanswer_sdk/models/{relation_extraction_task_settings_update_payload.py → update_metadata_model_output.py}

@@ -18,19 +18,20 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, StrictStr
-from typing import Any, ClassVar, Dict, List
-from qanswer_sdk.models.relation_extraction_task_update import RelationExtractionTaskUpdate
+from typing import Any, ClassVar, Dict, List, Optional
+from qanswer_sdk.models.update_file_es_meta_data import UpdateFileESMetaData
 from typing import Optional, Set
 from typing_extensions import Self
 
-class RelationExtractionTaskSettingsUpdatePayload(BaseModel):
+class UpdateMetadataModelOutput(BaseModel):
     """
-    RelationExtractionTaskSettingsUpdatePayload
+    UpdateMetadataModelOutput
     """ # noqa: E501
     username: StrictStr
     dataset: StrictStr
-    configs: RelationExtractionTaskUpdate
-    __properties: ClassVar[List[str]] = ["username", "dataset", "configs"]
+    data: UpdateFileESMetaData
+    conversation_id: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["username", "dataset", "data", "conversation_id"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -50,7 +51,7 @@ class RelationExtractionTaskSettingsUpdatePayload(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of RelationExtractionTaskSettingsUpdatePayload from a JSON string"""
+        """Create an instance of UpdateMetadataModelOutput from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -71,14 +72,14 @@ class RelationExtractionTaskSettingsUpdatePayload(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of configs
-        if self.configs:
-            _dict['configs'] = self.configs.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of data
+        if self.data:
+            _dict['data'] = self.data.to_dict()
         return _dict
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of RelationExtractionTaskSettingsUpdatePayload from a dict"""
+        """Create an instance of UpdateMetadataModelOutput from a dict"""
         if obj is None:
             return None
 
@@ -88,7 +89,8 @@ class RelationExtractionTaskSettingsUpdatePayload(BaseModel):
         _obj = cls.model_validate({
             "username": obj.get("username"),
             "dataset": obj.get("dataset"),
-            "configs": RelationExtractionTaskUpdate.from_dict(obj["configs"]) if obj.get("configs") is not None else None
+            "data": UpdateFileESMetaData.from_dict(obj["data"]) if obj.get("data") is not None else None,
+            "conversation_id": obj.get("conversation_id")
         })
         return _obj
 

qanswer_sdk/models/update_qn_a_pair_request.py

@@ -34,7 +34,8 @@ class UpdateQnAPairRequest(BaseModel):
     answer: StrictStr
     source: Optional[StrictStr] = None
     source_type: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["username", "dataset", "connector_id", "file_id", "question", "answer", "source", "source_type"]
+    additional_fields: Optional[Dict[str, Any]] = None
+    __properties: ClassVar[List[str]] = ["username", "dataset", "connector_id", "file_id", "question", "answer", "source", "source_type", "additional_fields"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -94,7 +95,8 @@ class UpdateQnAPairRequest(BaseModel):
             "question": obj.get("question"),
             "answer": obj.get("answer"),
             "source": obj.get("source"),
-            "source_type": obj.get("source_type")
+            "source_type": obj.get("source_type"),
+            "additional_fields": obj.get("additional_fields")
         })
         return _obj
 

qanswer_sdk/models/{relation_extraction_template_payload.py → update_template_cells_payload.py}

@@ -17,21 +17,21 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, StrictStr
-from typing import Any, ClassVar, Dict, List, Optional
-from qanswer_sdk.models.relation_extraction_configuration import RelationExtractionConfiguration
+from pydantic import BaseModel, ConfigDict, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List
+from qanswer_sdk.models.update_cells_payload import UpdateCellsPayload
 from typing import Optional, Set
 from typing_extensions import Self
 
-class RelationExtractionTemplatePayload(BaseModel):
+class UpdateTemplateCellsPayload(BaseModel):
     """
-    RelationExtractionTemplatePayload
+    UpdateTemplateCellsPayload
     """ # noqa: E501
     username: StrictStr
     dataset: StrictStr
-    template_name: Optional[StrictStr] = None
-    template: RelationExtractionConfiguration
-    __properties: ClassVar[List[str]] = ["username", "dataset", "template_name", "template"]
+    template_id: StrictInt
+    cells: List[UpdateCellsPayload]
+    __properties: ClassVar[List[str]] = ["username", "dataset", "template_id", "cells"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -51,7 +51,7 @@ class RelationExtractionTemplatePayload(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of RelationExtractionTemplatePayload from a JSON string"""
+        """Create an instance of UpdateTemplateCellsPayload from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -72,14 +72,18 @@ class RelationExtractionTemplatePayload(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of template
-        if self.template:
-            _dict['template'] = self.template.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in cells (list)
+        _items = []
+        if self.cells:
+            for _item_cells in self.cells:
+                if _item_cells:
+                    _items.append(_item_cells.to_dict())
+            _dict['cells'] = _items
         return _dict
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of RelationExtractionTemplatePayload from a dict"""
+        """Create an instance of UpdateTemplateCellsPayload from a dict"""
         if obj is None:
             return None
 
@@ -89,8 +93,8 @@ class RelationExtractionTemplatePayload(BaseModel):
         _obj = cls.model_validate({
             "username": obj.get("username"),
             "dataset": obj.get("dataset"),
-            "template_name": obj.get("template_name"),
-            "template": RelationExtractionConfiguration.from_dict(obj["template"]) if obj.get("template") is not None else None
+            "template_id": obj.get("template_id"),
+            "cells": [UpdateCellsPayload.from_dict(_item) for _item in obj["cells"]] if obj.get("cells") is not None else None
         })
         return _obj