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

qanswer_sdk/models/socket_conversation_message.py

@@ -0,0 +1,280 @@
+# 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.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
+
+class SocketConversationMessage(BaseModel):
+    """
+    SocketConversationMessage
+    """ # noqa: E501
+    username: StrictStr = Field(description="Username of the user")
+    jwt: Optional[StrictStr] = Field(default=None, description="JTW token for authentification to join the conversation.")
+    api_key: Optional[StrictStr] = Field(default=None, description="Api ley for authentification to join the conversation.")
+    dataset: StrictStr = Field(description="Name of the dataset")
+    conversation_id: StrictStr = Field(description="ID of the conversation")
+    user: Optional[StrictStr] = Field(default=None, description="User that is joining the conversation")
+    role: StrictStr
+    contents: List[ContentItem]
+    message_id: StrictStr
+    parent_message_id: Optional[StrictStr] = None
+    timestamp: Optional[StrictStr] = '2026-04-14T13:50:20Z'
+    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[StrictStr] = None
+    interface_origin: Optional[StrictStr] = None
+    comment: Optional[StrictStr] = None
+    llm_choice: Optional[StrictStr] = None
+    error_key: Optional[StrictStr] = 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
+    tagged_assistant_name: Optional[StrictStr] = None
+    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
+    server_uids: Optional[List[StrictStr]] = None
+    thinking_blocks: Optional[List[ThinkingBlock]] = None
+    websearch_sites: Optional[List[WebsearchScrapingSite]] = None
+    merge_with_message_id: Optional[StrictStr] = None
+    capability_type: Optional[StrictStr] = None
+    warn_visibility: Optional[StrictStr] = 'none'
+    status: StrictStr
+    __properties: ClassVar[List[str]] = ["username", "jwt", "api_key", "dataset", "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", "merge_with_message_id", "capability_type", "warn_visibility", "status"]
+
+    @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
+
+    @field_validator('task_category')
+    def task_category_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        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', 'request_initial_message', 'econsilium-request-for-new-token']):
+            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', 'request_initial_message', 'econsilium-request-for-new-token')")
+        return value
+
+    @field_validator('error_key')
+    def error_key_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['authentication_error', 'permission_denied', 'context_window_exceeded', 'content_policy_violation', 'rejected_request', 'bad_request', 'not_found', 'unprocessable_entity', 'rate_limit', 'timeout', 'internal_server_error', 'bad_gateway', 'service_unavailable', 'api_connection_error', 'api_error', 'econsilium_token_error', 'unknown']):
+            raise ValueError("must be one of enum values ('authentication_error', 'permission_denied', 'context_window_exceeded', 'content_policy_violation', 'rejected_request', 'bad_request', 'not_found', 'unprocessable_entity', 'rate_limit', 'timeout', 'internal_server_error', 'bad_gateway', 'service_unavailable', 'api_connection_error', 'api_error', 'econsilium_token_error', 'unknown')")
+        return value
+
+    @field_validator('mode')
+    def mode_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['rag', 'full-context', 'rag-with-map-reduce', 'full_context_with_map_reduce']):
+            raise ValueError("must be one of enum values ('rag', 'full-context', 'rag-with-map-reduce', 'full_context_with_map_reduce')")
+        return value
+
+    @field_validator('capability_type')
+    def capability_type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['web_search', 'retrieve_documents']):
+            raise ValueError("must be one of enum values ('web_search', 'retrieve_documents')")
+        return value
+
+    @field_validator('warn_visibility')
+    def warn_visibility_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['shared', 'shared_anonymized', 'incognito', 'not_shared', 'none']):
+            raise ValueError("must be one of enum values ('shared', 'shared_anonymized', 'incognito', 'not_shared', 'none')")
+        return value
+
+    @field_validator('status')
+    def status_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['COMPLETING', 'COMPLETED', 'THINKING', 'THINKING_COMPLETED', 'SOMETHING_WENT_WRONG']):
+            raise ValueError("must be one of enum values ('COMPLETING', 'COMPLETED', 'THINKING', 'THINKING_COMPLETED', 'SOMETHING_WENT_WRONG')")
+        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 SocketConversationMessage 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
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of SocketConversationMessage 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"),
+            "jwt": obj.get("jwt"),
+            "api_key": obj.get("api_key"),
+            "dataset": obj.get("dataset"),
+            "conversation_id": obj.get("conversation_id"),
+            "user": obj.get("user"),
+            "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-04-14T13:50:20Z',
+            "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"),
+            "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"),
+            "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,
+            "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,
+            "merge_with_message_id": obj.get("merge_with_message_id"),
+            "capability_type": obj.get("capability_type"),
+            "warn_visibility": obj.get("warn_visibility") if obj.get("warn_visibility") is not None else 'none',
+            "status": obj.get("status")
+        })
+        return _obj
+
+

qanswer_sdk/models/socket_events_q_answer.py

@@ -0,0 +1,51 @@
+# 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 SocketEventsQAnswer(str, Enum):
+    """
+    All events that can be send over sockets
+    """
+
+    """
+    allowed enum values
+    """
+    JOIN_USER = 'join_user'
+    JOIN_DATASET = 'join_dataset'
+    JOIN_CONVERSATION = 'join_conversation'
+    STOP_CONVERSATION = 'stop_conversation'
+    CHAT = 'chat'
+    TRANSCRIBE = 'transcribe'
+    CONNECTOR_INDEXING = 'connector_indexing'
+    WEBSEARCH_SCRAPING = 'websearch_scraping'
+    DATA_EXTRACTION = 'data_extraction'
+    AI_CELL_GENERATION = 'ai_cell_generation'
+    REPORT_COPILOT_CONTINUATION = 'report_copilot_continuation'
+    REPORT_COPILOT_GENERATION_FROM_SLOTS = 'report_copilot_generation_from_slots'
+    REPORT_COPILOT_GENERATION_FROM_TEMPLATE = 'report_copilot_generation_from_template'
+    REPORT_COPILOT_REFINEMENT = 'report_copilot_refinement'
+    FEEDBACK_EVALUATION_PROGRESS = 'feedback_evaluation_progress'
+    NOTIFICATION = 'notification'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of SocketEventsQAnswer from a JSON string"""
+        return cls(json.loads(json_str))
+
+

qanswer_sdk/models/socket_feedback_evaluation_progress.py

@@ -0,0 +1,114 @@
+# 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.feedback_model import FeedbackModel
+from typing import Optional, Set
+from typing_extensions import Self
+
+class SocketFeedbackEvaluationProgress(BaseModel):
+    """
+    Progress event emitted after each feedback is evaluated.
+    """ # noqa: E501
+    username: StrictStr = Field(description="Username of the user")
+    jwt: Optional[StrictStr] = Field(default=None, description="JTW token for authentification to join the conversation.")
+    api_key: Optional[StrictStr] = Field(default=None, description="Api ley for authentification to join the conversation.")
+    dataset: Optional[StrictStr] = Field(default=None, description="Dataset of the evaluated feedback when available.")
+    organization_id: Optional[StrictInt] = Field(default=None, description="Organization ID of the evaluated feedback when available.")
+    chatbot_feedbacks: Optional[StrictBool] = Field(default=False, description="Whether the evaluated feedback are chatbot feedbacks when available.")
+    status: StrictStr
+    feedback: FeedbackModel
+    error_message: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["username", "jwt", "api_key", "dataset", "organization_id", "chatbot_feedbacks", "status", "feedback", "error_message"]
+
+    @field_validator('status')
+    def status_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['COMPLETING', 'COMPLETED', 'THINKING', 'THINKING_COMPLETED', 'SOMETHING_WENT_WRONG']):
+            raise ValueError("must be one of enum values ('COMPLETING', 'COMPLETED', 'THINKING', 'THINKING_COMPLETED', 'SOMETHING_WENT_WRONG')")
+        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 SocketFeedbackEvaluationProgress 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 feedback
+        if self.feedback:
+            _dict['feedback'] = self.feedback.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of SocketFeedbackEvaluationProgress 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"),
+            "jwt": obj.get("jwt"),
+            "api_key": obj.get("api_key"),
+            "dataset": obj.get("dataset"),
+            "organization_id": obj.get("organization_id"),
+            "chatbot_feedbacks": obj.get("chatbot_feedbacks") if obj.get("chatbot_feedbacks") is not None else False,
+            "status": obj.get("status"),
+            "feedback": FeedbackModel.from_dict(obj["feedback"]) if obj.get("feedback") is not None else None,
+            "error_message": obj.get("error_message")
+        })
+        return _obj
+
+

qanswer_sdk/models/socket_file_metadata.py

@@ -0,0 +1,207 @@
+# 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 datetime import datetime
+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.source_metadata import SourceMetadata
+from typing import Optional, Set
+from typing_extensions import Self
+
+class SocketFileMetadata(BaseModel):
+    """
+    SocketFileMetadata
+    """ # noqa: E501
+    file_id: StrictStr
+    title: Optional[StrictStr] = None
+    name: Optional[StrictStr] = None
+    file_name: StrictStr
+    display_name: Optional[StrictStr] = None
+    original_file_name: StrictStr
+    file_extension: StrictStr
+    additional_fields: Dict[str, Any]
+    author: Optional[StrictStr] = None
+    subject: Optional[StrictStr] = None
+    creator: Optional[StrictStr] = None
+    last_modified_by: Optional[StrictStr] = None
+    created_at: Optional[datetime] = None
+    modified_at: Optional[datetime] = None
+    category: Optional[StrictStr] = None
+    comments: Optional[StrictStr] = None
+    keywords: Optional[StrictStr] = None
+    source_metadata: Optional[SourceMetadata] = None
+    username: StrictStr = Field(description="Username of the user")
+    dataset: StrictStr = Field(description="Name of the dataset")
+    connector_id: StrictInt
+    connector_name: StrictStr
+    connector_type: StrictStr
+    usage: Optional[StrictInt] = 0
+    status: StrictStr
+    failure_reason: Optional[StrictStr] = None
+    source: StrictStr
+    source_type: Optional[StrictStr] = None
+    weight: Optional[Union[StrictFloat, StrictInt]] = 1.0
+    mime_type: Optional[StrictStr] = None
+    hashed_filename: Optional[StrictStr] = None
+    size: Optional[StrictInt] = None
+    do_ocr: Optional[StrictBool] = False
+    do_llm: Optional[StrictBool] = False
+    tabular_parse_method: Optional[StrictStr] = None
+    account_id: Optional[StrictStr] = None
+    favicon: Optional[StrictStr] = None
+    jwt: Optional[StrictStr] = Field(default=None, description="JTW token for authentification to join the conversation.")
+    api_key: Optional[StrictStr] = Field(default=None, description="Api ley for authentification to join the conversation.")
+    size_in_mb: Optional[Union[StrictFloat, StrictInt]] = 0.0
+    number_of_files: Optional[StrictInt] = 0
+    has_files_pending_upload: Optional[StrictBool] = False
+    n_files_to_index: Optional[StrictInt] = 0
+    all_files_to_index: Optional[StrictInt] = 0
+    is_assistant_indexing: Optional[StrictBool] = False
+    __properties: ClassVar[List[str]] = ["file_id", "title", "name", "file_name", "display_name", "original_file_name", "file_extension", "additional_fields", "author", "subject", "creator", "last_modified_by", "created_at", "modified_at", "category", "comments", "keywords", "source_metadata", "username", "dataset", "connector_id", "connector_name", "connector_type", "usage", "status", "failure_reason", "source", "source_type", "weight", "mime_type", "hashed_filename", "size", "do_ocr", "do_llm", "tabular_parse_method", "account_id", "favicon", "jwt", "api_key", "size_in_mb", "number_of_files", "has_files_pending_upload", "n_files_to_index", "all_files_to_index", "is_assistant_indexing"]
+
+    @field_validator('status')
+    def status_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['Submitted', 'Uploaded', 'Converted', 'Indexed', 'QuotaExceeded', 'Failed', 'MalwareDetected', 'NeedsToBeUploaded']):
+            raise ValueError("must be one of enum values ('Submitted', 'Uploaded', 'Converted', 'Indexed', 'QuotaExceeded', 'Failed', 'MalwareDetected', 'NeedsToBeUploaded')")
+        return value
+
+    @field_validator('failure_reason')
+    def failure_reason_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['format_not_supported', 'file_empty', 'pdf_conversion_failed', 'embedding_failed', 'indexing_failed', 'quota_exceeded', 'illegal_filename', 'malware_detected', 'duplicate_filename', 'unknown_error', 'website_not_reachable', 'remote_file_rejected', 'other']):
+            raise ValueError("must be one of enum values ('format_not_supported', 'file_empty', 'pdf_conversion_failed', 'embedding_failed', 'indexing_failed', 'quota_exceeded', 'illegal_filename', 'malware_detected', 'duplicate_filename', 'unknown_error', 'website_not_reachable', 'remote_file_rejected', 'other')")
+        return value
+
+    @field_validator('source_type')
+    def source_type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['document', 'website', 'qna']):
+            raise ValueError("must be one of enum values ('document', 'website', 'qna')")
+        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 SocketFileMetadata 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 source_metadata
+        if self.source_metadata:
+            _dict['source_metadata'] = self.source_metadata.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of SocketFileMetadata from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "file_id": obj.get("file_id"),
+            "title": obj.get("title"),
+            "name": obj.get("name"),
+            "file_name": obj.get("file_name"),
+            "display_name": obj.get("display_name"),
+            "original_file_name": obj.get("original_file_name"),
+            "file_extension": obj.get("file_extension"),
+            "additional_fields": obj.get("additional_fields"),
+            "author": obj.get("author"),
+            "subject": obj.get("subject"),
+            "creator": obj.get("creator"),
+            "last_modified_by": obj.get("last_modified_by"),
+            "created_at": obj.get("created_at"),
+            "modified_at": obj.get("modified_at"),
+            "category": obj.get("category"),
+            "comments": obj.get("comments"),
+            "keywords": obj.get("keywords"),
+            "source_metadata": SourceMetadata.from_dict(obj["source_metadata"]) if obj.get("source_metadata") is not None else None,
+            "username": obj.get("username"),
+            "dataset": obj.get("dataset"),
+            "connector_id": obj.get("connector_id"),
+            "connector_name": obj.get("connector_name"),
+            "connector_type": obj.get("connector_type"),
+            "usage": obj.get("usage") if obj.get("usage") is not None else 0,
+            "status": obj.get("status"),
+            "failure_reason": obj.get("failure_reason"),
+            "source": obj.get("source"),
+            "source_type": obj.get("source_type"),
+            "weight": obj.get("weight") if obj.get("weight") is not None else 1.0,
+            "mime_type": obj.get("mime_type"),
+            "hashed_filename": obj.get("hashed_filename"),
+            "size": obj.get("size"),
+            "do_ocr": obj.get("do_ocr") if obj.get("do_ocr") is not None else False,
+            "do_llm": obj.get("do_llm") if obj.get("do_llm") is not None else False,
+            "tabular_parse_method": obj.get("tabular_parse_method"),
+            "account_id": obj.get("account_id"),
+            "favicon": obj.get("favicon"),
+            "jwt": obj.get("jwt"),
+            "api_key": obj.get("api_key"),
+            "size_in_mb": obj.get("size_in_mb") if obj.get("size_in_mb") is not None else 0.0,
+            "number_of_files": obj.get("number_of_files") if obj.get("number_of_files") is not None else 0,
+            "has_files_pending_upload": obj.get("has_files_pending_upload") if obj.get("has_files_pending_upload") is not None else False,
+            "n_files_to_index": obj.get("n_files_to_index") if obj.get("n_files_to_index") is not None else 0,
+            "all_files_to_index": obj.get("all_files_to_index") if obj.get("all_files_to_index") is not None else 0,
+            "is_assistant_indexing": obj.get("is_assistant_indexing") if obj.get("is_assistant_indexing") is not None else False
+        })
+        return _obj
+
+