qanswer-sdk 0.1.0.dev0__py3-none-any.whl

qanswer_sdk/models/connector_type_list_dto.py

@@ -0,0 +1,95 @@
+# coding: utf-8
+
+"""
+    free-text-client
+
+    QAnswer 2.0 API
+
+    The version of the OpenAPI document: 2.0.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
+from typing import Any, ClassVar, Dict, List, Optional
+from qanswer_sdk.models.connector_type_dto import ConnectorTypeDto
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ConnectorTypeListDto(BaseModel):
+    """
+    ConnectorTypeListDto
+    """ # noqa: E501
+    connectors: Optional[List[ConnectorTypeDto]] = None
+    __properties: ClassVar[List[str]] = ["connectors"]
+
+    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 ConnectorTypeListDto 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 connectors (list)
+        _items = []
+        if self.connectors:
+            for _item_connectors in self.connectors:
+                if _item_connectors:
+                    _items.append(_item_connectors.to_dict())
+            _dict['connectors'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ConnectorTypeListDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "connectors": [ConnectorTypeDto.from_dict(_item) for _item in obj["connectors"]] if obj.get("connectors") is not None else None
+        })
+        return _obj
+
+

qanswer_sdk/models/content_item.py

@@ -0,0 +1,106 @@
+# coding: utf-8
+
+"""
+    free-text-client
+
+    QAnswer 2.0 API
+
+    The version of the OpenAPI document: 2.0.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, Optional
+from qanswer_sdk.models.audio_format import AudioFormat
+from qanswer_sdk.models.content_item_type import ContentItemType
+from qanswer_sdk.models.image_url_detail import ImageUrlDetail
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ContentItem(BaseModel):
+    """
+    ContentItem
+    """ # noqa: E501
+    content_item_type: ContentItemType
+    text: Optional[StrictStr] = None
+    url: Optional[StrictStr] = None
+    image_detail: Optional[ImageUrlDetail] = None
+    audio_data: Optional[StrictStr] = None
+    audio_format: Optional[AudioFormat] = None
+    file_id: Optional[StrictStr] = None
+    file_data: Optional[StrictStr] = None
+    file_name: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["content_item_type", "text", "url", "image_detail", "audio_data", "audio_format", "file_id", "file_data", "file_name"]
+
+    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 ContentItem 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 ContentItem from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "content_item_type": obj.get("content_item_type"),
+            "text": obj.get("text"),
+            "url": obj.get("url"),
+            "image_detail": obj.get("image_detail"),
+            "audio_data": obj.get("audio_data"),
+            "audio_format": obj.get("audio_format"),
+            "file_id": obj.get("file_id"),
+            "file_data": obj.get("file_data"),
+            "file_name": obj.get("file_name")
+        })
+        return _obj
+
+

qanswer_sdk/models/content_item_type.py

@@ -0,0 +1,39 @@
+# coding: utf-8
+
+"""
+    free-text-client
+
+    QAnswer 2.0 API
+
+    The version of the OpenAPI document: 2.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+from enum import Enum
+from typing_extensions import Self
+
+
+class ContentItemType(str, Enum):
+    """
+    ContentItemType
+    """
+
+    """
+    allowed enum values
+    """
+    TEXT = 'text'
+    IMAGE_URL = 'image_url'
+    FILE = 'file'
+    INPUT_AUDIO = 'input_audio'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of ContentItemType from a JSON string"""
+        return cls(json.loads(json_str))
+
+

qanswer_sdk/models/conversation_aggregations_request.py

@@ -0,0 +1,107 @@
+# coding: utf-8
+
+"""
+    free-text-client
+
+    QAnswer 2.0 API
+
+    The version of the OpenAPI document: 2.0.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, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from qanswer_sdk.models.search_metadata_filter import SearchMetadataFilter
+from qanswer_sdk.models.task_name import TaskName
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ConversationAggregationsRequest(BaseModel):
+    """
+    ConversationAggregationsRequest
+    """ # noqa: E501
+    username: StrictStr = Field(description="Username who owns the conversation.")
+    dataset: StrictStr = Field(description="Dataset of the conversation.")
+    task_name: TaskName = Field(description="Task whose selected_aggregations gates the response (chat or search).")
+    filters: Optional[List[List[SearchMetadataFilter]]] = Field(default=None, description="Currently selected facet filters. When provided, the aggregations are recomputed live against these filters (faceted search) instead of returning the last stored aggregations.")
+    __properties: ClassVar[List[str]] = ["username", "dataset", "task_name", "filters"]
+
+    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 ConversationAggregationsRequest 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 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
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ConversationAggregationsRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "username": obj.get("username"),
+            "dataset": obj.get("dataset"),
+            "task_name": obj.get("task_name"),
+            "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/conversation_message.py

@@ -0,0 +1,230 @@
+# coding: utf-8
+
+"""
+    free-text-client
+
+    QAnswer 2.0 API
+
+    The version of the OpenAPI document: 2.0.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, field_validator
+from typing import Any, ClassVar, Dict, List, Optional
+from qanswer_sdk.models.aggregations_group import AggregationsGroup
+from qanswer_sdk.models.capability_type import CapabilityType
+from qanswer_sdk.models.content_item import ContentItem
+from qanswer_sdk.models.conversation_message_error_key import ConversationMessageErrorKey
+from qanswer_sdk.models.data_filling_mode import DataFillingMode
+from qanswer_sdk.models.source import Source
+from qanswer_sdk.models.suggested_replies import SuggestedReplies
+from qanswer_sdk.models.task_category import TaskCategory
+from qanswer_sdk.models.thinking_block import ThinkingBlock
+from qanswer_sdk.models.tool_selection import ToolSelection
+from qanswer_sdk.models.warn_visibility import WarnVisibility
+from qanswer_sdk.models.websearch_scraping_site import WebsearchScrapingSite
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ConversationMessage(BaseModel):
+    """
+    ConversationMessage
+    """ # noqa: E501
+    role: StrictStr
+    contents: List[ContentItem]
+    message_id: StrictStr
+    parent_message_id: Optional[StrictStr] = None
+    timestamp: Optional[StrictStr] = '2026-07-18T19:09:09Z'
+    feedback: Optional[StrictInt] = -1
+    origin: Optional[StrictStr] = None
+    sources: Optional[List[Source]] = Field(default=None, description="All sources referenced in the response")
+    all_sources: Optional[List[Source]] = Field(default=None, description="All retrieved sources passed to the llm")
+    task_category: Optional[TaskCategory] = None
+    interface_origin: Optional[StrictStr] = None
+    comment: Optional[StrictStr] = None
+    llm_choice: Optional[StrictStr] = None
+    raw_content: Optional[StrictStr] = None
+    error_key: Optional[ConversationMessageErrorKey] = None
+    prompt_used: Optional[StrictStr] = None
+    mode: Optional[DataFillingMode] = None
+    is_input_data_cropped: Optional[StrictBool] = None
+    aggs: Optional[List[AggregationsGroup]] = None
+    expanded_question: Optional[StrictStr] = None
+    prompt_type: Optional[StrictStr] = None
+    tagged_assistant_name: Optional[StrictStr] = None
+    tagged_assistant_id: Optional[StrictInt] = Field(default=None, description="ID of the tagged assistant")
+    synonyms_used: Optional[List[StrictStr]] = None
+    tool_calls: Optional[List[Dict[str, Any]]] = None
+    tool_call_id: Optional[StrictStr] = None
+    tool_selections: Optional[List[ToolSelection]] = None
+    client_tools: Optional[List[Dict[str, Any]]] = None
+    server_uids: Optional[List[StrictStr]] = None
+    available_tool_names: Optional[List[StrictStr]] = None
+    tool_rerank_query: Optional[StrictStr] = None
+    thinking_blocks: Optional[List[ThinkingBlock]] = None
+    websearch_sites: Optional[List[WebsearchScrapingSite]] = None
+    merge_with_message_id: Optional[StrictStr] = None
+    capability_type: Optional[CapabilityType] = None
+    warn_visibility: Optional[WarnVisibility] = None
+    suggested_replies: Optional[SuggestedReplies] = 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", "raw_content", "error_key", "prompt_used", "mode", "is_input_data_cropped", "aggs", "expanded_question", "prompt_type", "tagged_assistant_name", "tagged_assistant_id", "synonyms_used", "tool_calls", "tool_call_id", "tool_selections", "client_tools", "server_uids", "available_tool_names", "tool_rerank_query", "thinking_blocks", "websearch_sites", "merge_with_message_id", "capability_type", "warn_visibility", "suggested_replies"]
+
+    @field_validator('role')
+    def role_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['assistant', 'user', 'tool']):
+            raise ValueError("must be one of enum values ('assistant', 'user', 'tool')")
+        return value
+
+    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 ConversationMessage 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 contents (list)
+        _items = []
+        if self.contents:
+            for _item_contents in self.contents:
+                if _item_contents:
+                    _items.append(_item_contents.to_dict())
+            _dict['contents'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in sources (list)
+        _items = []
+        if self.sources:
+            for _item_sources in self.sources:
+                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 all_sources (list)
+        _items = []
+        if self.all_sources:
+            for _item_all_sources in self.all_sources:
+                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 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 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
+        # override the default output from pydantic by calling `to_dict()` of suggested_replies
+        if self.suggested_replies:
+            _dict['suggested_replies'] = self.suggested_replies.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ConversationMessage from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "role": obj.get("role"),
+            "contents": [ContentItem.from_dict(_item) for _item in obj["contents"]] if obj.get("contents") is not None else None,
+            "message_id": obj.get("message_id"),
+            "parent_message_id": obj.get("parent_message_id"),
+            "timestamp": obj.get("timestamp") if obj.get("timestamp") is not None else '2026-07-18T19:09:09Z',
+            "feedback": obj.get("feedback") if obj.get("feedback") is not None else -1,
+            "origin": obj.get("origin"),
+            "sources": [Source.from_dict(_item) for _item in obj["sources"]] if obj.get("sources") is not None else None,
+            "all_sources": [Source.from_dict(_item) for _item in obj["all_sources"]] if obj.get("all_sources") is not None else None,
+            "task_category": obj.get("task_category"),
+            "interface_origin": obj.get("interface_origin"),
+            "comment": obj.get("comment"),
+            "llm_choice": obj.get("llm_choice"),
+            "raw_content": obj.get("raw_content"),
+            "error_key": obj.get("error_key"),
+            "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"),
+            "tagged_assistant_name": obj.get("tagged_assistant_name"),
+            "tagged_assistant_id": obj.get("tagged_assistant_id"),
+            "synonyms_used": obj.get("synonyms_used"),
+            "tool_calls": obj.get("tool_calls"),
+            "tool_call_id": obj.get("tool_call_id"),
+            "tool_selections": [ToolSelection.from_dict(_item) for _item in obj["tool_selections"]] if obj.get("tool_selections") is not None else None,
+            "client_tools": obj.get("client_tools"),
+            "server_uids": obj.get("server_uids"),
+            "available_tool_names": obj.get("available_tool_names"),
+            "tool_rerank_query": obj.get("tool_rerank_query"),
+            "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,
+            "merge_with_message_id": obj.get("merge_with_message_id"),
+            "capability_type": obj.get("capability_type"),
+            "warn_visibility": obj.get("warn_visibility"),
+            "suggested_replies": SuggestedReplies.from_dict(obj["suggested_replies"]) if obj.get("suggested_replies") is not None else None
+        })
+        return _obj
+
+

qanswer_sdk/models/conversation_message_error_key.py

@@ -0,0 +1,52 @@
+# coding: utf-8
+
+"""
+    free-text-client
+
+    QAnswer 2.0 API
+
+    The version of the OpenAPI document: 2.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+from enum import Enum
+from typing_extensions import Self
+
+
+class ConversationMessageErrorKey(str, Enum):
+    """
+    ConversationMessageErrorKey
+    """
+
+    """
+    allowed enum values
+    """
+    AUTHENTICATION_ERROR = 'authentication_error'
+    PERMISSION_DENIED = 'permission_denied'
+    CONTEXT_WINDOW_EXCEEDED = 'context_window_exceeded'
+    CONTENT_POLICY_VIOLATION = 'content_policy_violation'
+    REJECTED_REQUEST = 'rejected_request'
+    BAD_REQUEST = 'bad_request'
+    NOT_FOUND = 'not_found'
+    UNPROCESSABLE_ENTITY = 'unprocessable_entity'
+    RATE_LIMIT = 'rate_limit'
+    TIMEOUT = 'timeout'
+    INTERNAL_SERVER_ERROR = 'internal_server_error'
+    BAD_GATEWAY = 'bad_gateway'
+    SERVICE_UNAVAILABLE = 'service_unavailable'
+    API_CONNECTION_ERROR = 'api_connection_error'
+    API_ERROR = 'api_error'
+    ECONSILIUM_TOKEN_ERROR = 'econsilium_token_error'
+    UNKNOWN = 'unknown'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of ConversationMessageErrorKey from a JSON string"""
+        return cls(json.loads(json_str))
+
+