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

qanswer_sdk/models/chat_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 ChatTaskSettings(BaseModel):
     ChatTaskSettings
     """ # 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,16 +35,17 @@ class ChatTaskSettings(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
     chat_initial_message: Optional[StrictStr] = None
     human_takeover: Optional[StrictBool] = 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", "chat_initial_message", "human_takeover"]
+    __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", "chat_initial_message", "human_takeover"]
 
     @field_validator('bot_answer_length')
     def bot_answer_length_validate_enum(cls, value):
@@ -96,9 +96,9 @@ class ChatTaskSettings(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
@@ -112,7 +112,6 @@ class ChatTaskSettings(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"),
@@ -120,12 +119,13 @@ class ChatTaskSettings(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"),
             "chat_initial_message": obj.get("chat_initial_message"),
             "human_takeover": obj.get("human_takeover")

qanswer_sdk/models/chat_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 ChatTaskUpdate(BaseModel):
     ChatTaskUpdate
     """ # 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,15 +36,17 @@ class ChatTaskUpdate(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
-    slots_values: Optional[Dict[str, StrictStr]] = None
     selected_aggregations: Optional[List[StrictStr]] = None
     chat_initial_message: Optional[StrictStr] = None
     human_takeover: Optional[StrictBool] = 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", "slots_values", "selected_aggregations", "chat_initial_message", "human_takeover"]
+    __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", "chat_initial_message", "human_takeover"]
 
     @field_validator('bot_answer_length')
     def bot_answer_length_validate_enum(cls, value):
@@ -94,6 +97,9 @@ class ChatTaskUpdate(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
@@ -115,11 +121,13 @@ class ChatTaskUpdate(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"),
-            "slots_values": obj.get("slots_values"),
             "selected_aggregations": obj.get("selected_aggregations"),
             "chat_initial_message": obj.get("chat_initial_message"),
             "human_takeover": obj.get("human_takeover")

qanswer_sdk/models/chatbot_chat_payload.py

@@ -19,6 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from qanswer_sdk.models.search_metadata_filter import SearchMetadataFilter
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -41,7 +42,9 @@ class ChatbotChatPayload(BaseModel):
     oauth_token: Optional[StrictStr] = Field(default=None, description="The oauth token")
     agentic_chatbot_enabled: Optional[StrictBool] = Field(default=None, description="Whether to enable agentic chatbot")
     current_time: Optional[StrictStr] = Field(default=None, description="The current time in ISO 8601 format, e.g., 2023-10-05T14:48:00Z")
-    __properties: ClassVar[List[str]] = ["question", "username", "conversation_id", "llm_choice", "image_urls", "parent_message_id", "is_regenerate", "system_message", "addon_system_message", "additional_fields", "bypass_guardrail", "origin", "oauth_token", "agentic_chatbot_enabled", "current_time"]
+    auto_filters: Optional[StrictBool] = Field(default=None, description="Determines whether the auto filters should be used for the response.")
+    filters: Optional[List[List[SearchMetadataFilter]]] = None
+    __properties: ClassVar[List[str]] = ["question", "username", "conversation_id", "llm_choice", "image_urls", "parent_message_id", "is_regenerate", "system_message", "addon_system_message", "additional_fields", "bypass_guardrail", "origin", "oauth_token", "agentic_chatbot_enabled", "current_time", "auto_filters", "filters"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -82,6 +85,15 @@ class ChatbotChatPayload(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of each item in filters (list of list)
+        _items = []
+        if self.filters:
+            for _item_filters in self.filters:
+                if _item_filters:
+                    _items.append(
+                         [_inner_item.to_dict() for _inner_item in _item_filters if _inner_item is not None]
+                    )
+            _dict['filters'] = _items
         return _dict
 
     @classmethod
@@ -108,7 +120,12 @@ class ChatbotChatPayload(BaseModel):
             "origin": obj.get("origin"),
             "oauth_token": obj.get("oauth_token"),
             "agentic_chatbot_enabled": obj.get("agentic_chatbot_enabled"),
-            "current_time": obj.get("current_time")
+            "current_time": obj.get("current_time"),
+            "auto_filters": obj.get("auto_filters"),
+            "filters": [
+                    [SearchMetadataFilter.from_dict(_inner_item) for _inner_item in _item]
+                    for _item in obj["filters"]
+                ] if obj.get("filters") is not None else None
         })
         return _obj
 

qanswer_sdk/models/chatbot_conversation_model.py

@@ -58,6 +58,16 @@ class ChatbotConversationModel(BaseModel):
             raise ValueError("must be one of enum values ('search', 'chat', 'widget', 'microsoft_teams', 'discord', 'slack', 'chatbot_playground', 'undefined')")
         return value
 
+    @field_validator('shared_entity')
+    def shared_entity_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['USER', 'TEAM', 'ORGANIZATION']):
+            raise ValueError("must be one of enum values ('USER', 'TEAM', 'ORGANIZATION')")
+        return value
+
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

qanswer_sdk/models/chatbot_response.py

@@ -19,6 +19,8 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
+from qanswer_sdk.models.aggregations_group import AggregationsGroup
+from qanswer_sdk.models.search_metadata_filter import SearchMetadataFilter
 from qanswer_sdk.models.source import Source
 from typing import Optional, Set
 from typing_extensions import Self
@@ -35,7 +37,9 @@ class ChatbotResponse(BaseModel):
     sources: Optional[List[Source]] = Field(default=None, description="The sources used for generation")
     is_input_data_cropped: Optional[StrictBool] = Field(default=None, description="Indicates whether the input data was cropped during prompting.")
     mode: Optional[StrictStr] = Field(default=None, description="The mode of the response generation.")
-    __properties: ClassVar[List[str]] = ["conversation_id", "message_id", "ai_response", "finish_reason", "is_regenerate", "sources", "is_input_data_cropped", "mode"]
+    aggs: Optional[List[AggregationsGroup]] = Field(default=None, description="Aggregations response containing additional metadata or statistics related to the response.")
+    filters: Optional[List[SearchMetadataFilter]] = Field(default=None, description="Filters response containing additional metadata or statistics related to the response.")
+    __properties: ClassVar[List[str]] = ["conversation_id", "message_id", "ai_response", "finish_reason", "is_regenerate", "sources", "is_input_data_cropped", "mode", "aggs", "filters"]
 
     @field_validator('mode')
     def mode_validate_enum(cls, value):
@@ -93,6 +97,20 @@ class ChatbotResponse(BaseModel):
                 if _item_sources:
                     _items.append(_item_sources.to_dict())
             _dict['sources'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in aggs (list)
+        _items = []
+        if self.aggs:
+            for _item_aggs in self.aggs:
+                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 filters (list)
+        _items = []
+        if self.filters:
+            for _item_filters in self.filters:
+                if _item_filters:
+                    _items.append(_item_filters.to_dict())
+            _dict['filters'] = _items
         return _dict
 
     @classmethod
@@ -112,7 +130,9 @@ class ChatbotResponse(BaseModel):
             "is_regenerate": obj.get("is_regenerate"),
             "sources": [Source.from_dict(_item) for _item in obj["sources"]] if obj.get("sources") is not None else None,
             "is_input_data_cropped": obj.get("is_input_data_cropped"),
-            "mode": obj.get("mode")
+            "mode": obj.get("mode"),
+            "aggs": [AggregationsGroup.from_dict(_item) for _item in obj["aggs"]] if obj.get("aggs") is not None else None,
+            "filters": [SearchMetadataFilter.from_dict(_item) for _item in obj["filters"]] if obj.get("filters") is not None else None
         })
         return _obj
 

qanswer_sdk/models/{user_chatbot_setting_payload.py → chatbot_setting_dto.py}

@@ -19,17 +19,18 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from qanswer_sdk.models.llm_dto import LlmDto
 from typing import Optional, Set
 from typing_extensions import Self
 
-class UserChatbotSettingPayload(BaseModel):
+class ChatbotSettingDto(BaseModel):
     """
-    UserChatbotSettingPayload
+    ChatbotSettingDto
     """ # noqa: E501
     system_message: Optional[StrictStr] = Field(default=None, alias="systemMessage")
-    llm_choice: Optional[StrictStr] = Field(default=None, alias="llmChoice")
     agentic_chatbot_enabled: Optional[StrictBool] = Field(default=None, alias="agenticChatbotEnabled")
-    __properties: ClassVar[List[str]] = ["systemMessage", "llmChoice", "agenticChatbotEnabled"]
+    llm: Optional[LlmDto] = None
+    __properties: ClassVar[List[str]] = ["systemMessage", "agenticChatbotEnabled", "llm"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -49,7 +50,7 @@ class UserChatbotSettingPayload(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of UserChatbotSettingPayload from a JSON string"""
+        """Create an instance of ChatbotSettingDto from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -70,16 +71,14 @@ class UserChatbotSettingPayload(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of llm
+        if self.llm:
+            _dict['llm'] = self.llm.to_dict()
         # set to None if system_message (nullable) is None
         # and model_fields_set contains the field
         if self.system_message is None and "system_message" in self.model_fields_set:
             _dict['systemMessage'] = None
 
-        # set to None if llm_choice (nullable) is None
-        # and model_fields_set contains the field
-        if self.llm_choice is None and "llm_choice" in self.model_fields_set:
-            _dict['llmChoice'] = None
-
         # set to None if agentic_chatbot_enabled (nullable) is None
         # and model_fields_set contains the field
         if self.agentic_chatbot_enabled is None and "agentic_chatbot_enabled" in self.model_fields_set:
@@ -89,7 +88,7 @@ class UserChatbotSettingPayload(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of UserChatbotSettingPayload from a dict"""
+        """Create an instance of ChatbotSettingDto from a dict"""
         if obj is None:
             return None
 
@@ -98,8 +97,8 @@ class UserChatbotSettingPayload(BaseModel):
 
         _obj = cls.model_validate({
             "systemMessage": obj.get("systemMessage"),
-            "llmChoice": obj.get("llmChoice"),
-            "agenticChatbotEnabled": obj.get("agenticChatbotEnabled")
+            "agenticChatbotEnabled": obj.get("agenticChatbotEnabled"),
+            "llm": LlmDto.from_dict(obj["llm"]) if obj.get("llm") is not None else None
         })
         return _obj
 

qanswer_sdk/models/chatbot_setting_request.py

@@ -0,0 +1,96 @@
+# 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, Field, StrictBool, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ChatbotSettingRequest(BaseModel):
+    """
+    ChatbotSettingRequest
+    """ # noqa: E501
+    system_message: Optional[StrictStr] = Field(default=None, alias="systemMessage")
+    llm_id: Optional[StrictInt] = Field(default=None, alias="llmId")
+    agentic_chatbot_enabled: Optional[StrictBool] = Field(default=None, alias="agenticChatbotEnabled")
+    __properties: ClassVar[List[str]] = ["systemMessage", "llmId", "agenticChatbotEnabled"]
+
+    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 ChatbotSettingRequest 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,
+        )
+        # set to None if llm_id (nullable) is None
+        # and model_fields_set contains the field
+        if self.llm_id is None and "llm_id" in self.model_fields_set:
+            _dict['llmId'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ChatbotSettingRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "systemMessage": obj.get("systemMessage"),
+            "llmId": obj.get("llmId"),
+            "agenticChatbotEnabled": obj.get("agenticChatbotEnabled")
+        })
+        return _obj
+
+

qanswer_sdk/models/{question_completion.py → clip_connector_file_structure.py}

@@ -22,13 +22,13 @@ from typing import Any, ClassVar, Dict, List
 from typing import Optional, Set
 from typing_extensions import Self
 
-class QuestionCompletion(BaseModel):
+class ClipConnectorFileStructure(BaseModel):
     """
-    QuestionCompletion
+    ClipConnectorFileStructure
     """ # noqa: E501
-    question: StrictStr
-    completion: StrictStr
-    __properties: ClassVar[List[str]] = ["question", "completion"]
+    file_id: StrictStr
+    display_name: StrictStr
+    __properties: ClassVar[List[str]] = ["file_id", "display_name"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -48,7 +48,7 @@ class QuestionCompletion(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of QuestionCompletion from a JSON string"""
+        """Create an instance of ClipConnectorFileStructure from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -73,7 +73,7 @@ class QuestionCompletion(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of QuestionCompletion from a dict"""
+        """Create an instance of ClipConnectorFileStructure from a dict"""
         if obj is None:
             return None
 
@@ -81,8 +81,8 @@ class QuestionCompletion(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "question": obj.get("question"),
-            "completion": obj.get("completion")
+            "file_id": obj.get("file_id"),
+            "display_name": obj.get("display_name")
         })
         return _obj
 

qanswer_sdk/models/clip_connector_structure.py

@@ -0,0 +1,97 @@
+# 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, StrictInt
+from typing import Any, ClassVar, Dict, List
+from qanswer_sdk.models.clip_connector_file_structure import ClipConnectorFileStructure
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ClipConnectorStructure(BaseModel):
+    """
+    ClipConnectorStructure
+    """ # noqa: E501
+    connector_id: StrictInt
+    files_required: List[ClipConnectorFileStructure]
+    __properties: ClassVar[List[str]] = ["connector_id", "files_required"]
+
+    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 ClipConnectorStructure 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,
+        )
+        # override the default output from pydantic by calling `to_dict()` of each item in files_required (list)
+        _items = []
+        if self.files_required:
+            for _item_files_required in self.files_required:
+                if _item_files_required:
+                    _items.append(_item_files_required.to_dict())
+            _dict['files_required'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ClipConnectorStructure from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "connector_id": obj.get("connector_id"),
+            "files_required": [ClipConnectorFileStructure.from_dict(_item) for _item in obj["files_required"]] if obj.get("files_required") is not None else None
+        })
+        return _obj
+
+

qanswer_sdk/models/connector_model.py

@@ -41,8 +41,8 @@ class ConnectorModel(BaseModel):
     @field_validator('connector_type')
     def connector_type_validate_enum(cls, value):
         """Validates the enum"""
-        if value not in set(['document', 'website', 'qna', 'pinecone', 'rdf', 'websearch', 'econsilium', 'econsilium_search_index', 'publication_office', 'onedrive', 'sharepoint', 'gdrive', 'onenote']):
-            raise ValueError("must be one of enum values ('document', 'website', 'qna', 'pinecone', 'rdf', 'websearch', 'econsilium', 'econsilium_search_index', 'publication_office', 'onedrive', 'sharepoint', 'gdrive', 'onenote')")
+        if value not in set(['document', 'website', 'qna', 'pinecone', 'rdf', 'websearch', 'econsilium', 'econsilium_search_index', 'publication_office', 'onedrive', 'sharepoint', 'gdrive', 'onenote', 'imap']):
+            raise ValueError("must be one of enum values ('document', 'website', 'qna', 'pinecone', 'rdf', 'websearch', 'econsilium', 'econsilium_search_index', 'publication_office', 'onedrive', 'sharepoint', 'gdrive', 'onenote', 'imap')")
         return value
 
     model_config = ConfigDict(

qanswer_sdk/models/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
@@ -43,13 +42,13 @@ class ConversationMessage(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
-    __properties: ClassVar[List[str]] = ["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"]
+    prompt_type: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["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"]
 
     @field_validator('role')
     def role_validate_enum(cls, value):
@@ -135,13 +134,6 @@ class ConversationMessage(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:
@@ -174,12 +166,12 @@ class ConversationMessage(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")
+            "expanded_question": obj.get("expanded_question"),
+            "prompt_type": obj.get("prompt_type")
         })
         return _obj