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

qanswer_sdk/models/file_es_meta_data.py

@@ -19,6 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, StrictStr
 from typing import Any, ClassVar, Dict, List
+from qanswer_sdk.models.es_meta_data import ESMetaData
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -26,9 +27,10 @@ class FileESMetaData(BaseModel):
     """
     FileESMetaData
     """ # noqa: E501
-    additional_fields: Dict[str, Dict[str, Any]]
+    additional_fields: Dict[str, ESMetaData]
     file_id: StrictStr
-    __properties: ClassVar[List[str]] = ["additional_fields", "file_id"]
+    metadata: Dict[str, ESMetaData]
+    __properties: ClassVar[List[str]] = ["additional_fields", "file_id", "metadata"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -69,6 +71,20 @@ class FileESMetaData(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # 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
@@ -81,8 +97,19 @@ class FileESMetaData(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "additional_fields": obj.get("additional_fields"),
-            "file_id": obj.get("file_id")
+            "additional_fields": dict(
+                (_k, ESMetaData.from_dict(_v))
+                for _k, _v in obj["additional_fields"].items()
+            )
+            if obj.get("additional_fields") is not None
+            else None,
+            "file_id": obj.get("file_id"),
+            "metadata": dict(
+                (_k, ESMetaData.from_dict(_v))
+                for _k, _v in obj["metadata"].items()
+            )
+            if obj.get("metadata") is not None
+            else None
         })
         return _obj
 

qanswer_sdk/models/g_drive_download_request.py

@@ -32,8 +32,9 @@ class GDriveDownloadRequest(BaseModel):
     account_id: Optional[StrictStr] = None
     connector_id: StrictInt
     quota: Optional[StrictInt] = Field(default=None, description="Quota of the user")
+    additional_fields: Optional[Dict[str, Any]] = None
     ids: GDriveDownloadByIds
-    __properties: ClassVar[List[str]] = ["username", "dataset", "account_id", "connector_id", "quota", "ids"]
+    __properties: ClassVar[List[str]] = ["username", "dataset", "account_id", "connector_id", "quota", "additional_fields", "ids"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -94,6 +95,7 @@ class GDriveDownloadRequest(BaseModel):
             "account_id": obj.get("account_id"),
             "connector_id": obj.get("connector_id"),
             "quota": obj.get("quota"),
+            "additional_fields": obj.get("additional_fields"),
             "ids": GDriveDownloadByIds.from_dict(obj["ids"]) if obj.get("ids") is not None else None
         })
         return _obj

qanswer_sdk/models/generate_ai_cells_payload.py

@@ -18,7 +18,7 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, StrictInt, StrictStr
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -30,7 +30,8 @@ class GenerateAICellsPayload(BaseModel):
     dataset: StrictStr
     template_id: StrictInt
     cells: List[StrictInt]
-    __properties: ClassVar[List[str]] = ["username", "dataset", "template_id", "cells"]
+    additional_fields: Optional[Dict[str, Any]] = None
+    __properties: ClassVar[List[str]] = ["username", "dataset", "template_id", "cells", "additional_fields"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -86,7 +87,8 @@ class GenerateAICellsPayload(BaseModel):
             "username": obj.get("username"),
             "dataset": obj.get("dataset"),
             "template_id": obj.get("template_id"),
-            "cells": obj.get("cells")
+            "cells": obj.get("cells"),
+            "additional_fields": obj.get("additional_fields")
         })
         return _obj
 

qanswer_sdk/models/highlight_audio_payload_model.py

@@ -32,7 +32,8 @@ class HighlightAudioPayloadModel(BaseModel):
     file_id: StrictStr
     references: Optional[ReferenceModel] = None
     task_name: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["username", "dataset", "file_id", "references", "task_name"]
+    additional_fields: Optional[Dict[str, Any]] = None
+    __properties: ClassVar[List[str]] = ["username", "dataset", "file_id", "references", "task_name", "additional_fields"]
 
     @field_validator('task_name')
     def task_name_validate_enum(cls, value):
@@ -40,8 +41,8 @@ class HighlightAudioPayloadModel(BaseModel):
         if value is None:
             return value
 
-        if value not in set(['search', 'chat', 'data_extraction', 'ai_excel', 'report_copilot', 'text2sparql', 'sparql2text', 'chatbot']):
-            raise ValueError("must be one of enum values ('search', 'chat', 'data_extraction', 'ai_excel', 'report_copilot', 'text2sparql', 'sparql2text', 'chatbot')")
+        if value not in set(['search', 'chat', 'ai_excel', 'report_copilot', 'text2sparql', 'sparql2text', 'chatbot']):
+            raise ValueError("must be one of enum values ('search', 'chat', 'ai_excel', 'report_copilot', 'text2sparql', 'sparql2text', 'chatbot')")
         return value
 
     model_config = ConfigDict(
@@ -102,7 +103,8 @@ class HighlightAudioPayloadModel(BaseModel):
             "dataset": obj.get("dataset"),
             "file_id": obj.get("file_id"),
             "references": ReferenceModel.from_dict(obj["references"]) if obj.get("references") is not None else None,
-            "task_name": obj.get("task_name")
+            "task_name": obj.get("task_name"),
+            "additional_fields": obj.get("additional_fields")
         })
         return _obj
 

qanswer_sdk/models/imap_add_payload.py

@@ -18,7 +18,7 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, StrictInt, StrictStr
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
 from qanswer_sdk.models.imap_file_metadata import ImapFileMetadata
 from typing import Optional, Set
 from typing_extensions import Self
@@ -31,8 +31,9 @@ class ImapAddPayload(BaseModel):
     dataset: StrictStr
     connector_id: StrictInt
     quota: StrictInt
+    additional_fields: Optional[Dict[str, Any]] = None
     files: List[ImapFileMetadata]
-    __properties: ClassVar[List[str]] = ["username", "dataset", "connector_id", "quota", "files"]
+    __properties: ClassVar[List[str]] = ["username", "dataset", "connector_id", "quota", "additional_fields", "files"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -96,6 +97,7 @@ class ImapAddPayload(BaseModel):
             "dataset": obj.get("dataset"),
             "connector_id": obj.get("connector_id"),
             "quota": obj.get("quota"),
+            "additional_fields": obj.get("additional_fields"),
             "files": [ImapFileMetadata.from_dict(_item) for _item in obj["files"]] if obj.get("files") is not None else None
         })
         return _obj

qanswer_sdk/models/list_socket_events.py

@@ -33,8 +33,8 @@ class ListSocketEvents(BaseModel):
     def connectors_validate_enum(cls, value):
         """Validates the enum"""
         for i in value:
-            if i not in set(['join_dataset', 'join_conversation', 'stop_conversation', 'chat', 'transcribe', 'connector_indexing', 'data_extraction', 'ai_cell_generation', 'report_copilot_continuation', 'report_copilot_generation_from_slots', 'report_copilot_generation_from_template', 'report_copilot_refinement']):
-                raise ValueError("each list item must be one of ('join_dataset', 'join_conversation', 'stop_conversation', 'chat', 'transcribe', 'connector_indexing', 'data_extraction', 'ai_cell_generation', 'report_copilot_continuation', 'report_copilot_generation_from_slots', 'report_copilot_generation_from_template', 'report_copilot_refinement')")
+            if i not in set(['join_dataset', 'join_conversation', 'stop_conversation', 'chat', 'transcribe', 'connector_indexing', 'websearch_scraping', 'data_extraction', 'ai_cell_generation', 'report_copilot_continuation', 'report_copilot_generation_from_slots', 'report_copilot_generation_from_template', 'report_copilot_refinement']):
+                raise ValueError("each list item must be one of ('join_dataset', 'join_conversation', 'stop_conversation', 'chat', 'transcribe', 'connector_indexing', 'websearch_scraping', 'data_extraction', 'ai_cell_generation', 'report_copilot_continuation', 'report_copilot_generation_from_slots', 'report_copilot_generation_from_template', 'report_copilot_refinement')")
         return value
 
     model_config = ConfigDict(

qanswer_sdk/models/llm_consumption.py

@@ -45,8 +45,8 @@ class LlmConsumption(BaseModel):
         if value is None:
             return value
 
-        if value not in set(['query-expansion', 'response-audit', 'title-generation', 'chat', 'sparql2text', 'guardrail', 'jailbreak', 'human-take-over', 'auto-es-filters-generation', 'question-suggestion', 'language-detection', 'advanced-classification', 'advanced-classification-v2', 'highlight', 'tool-choice', 'pdf-vision-conversion', 'map-reduce-map', 'map-reduce-reduce', 'shorten-instruction', 'auto-bot-description', 'openai-chat-completion', 'mcp-assistant-router', 'mcp-multi-turn-execution', 'mcp-progress-message-generation', 'mcp-progress-tool-bullet-point-generation', 'unknown']):
-            raise ValueError("must be one of enum values ('query-expansion', 'response-audit', 'title-generation', 'chat', 'sparql2text', 'guardrail', 'jailbreak', 'human-take-over', 'auto-es-filters-generation', 'question-suggestion', 'language-detection', 'advanced-classification', 'advanced-classification-v2', 'highlight', 'tool-choice', 'pdf-vision-conversion', 'map-reduce-map', 'map-reduce-reduce', 'shorten-instruction', 'auto-bot-description', 'openai-chat-completion', 'mcp-assistant-router', 'mcp-multi-turn-execution', 'mcp-progress-message-generation', 'mcp-progress-tool-bullet-point-generation', 'unknown')")
+        if value not in set(['query-expansion', 'response-audit', 'title-generation', 'chat', 'sparql2text', 'guardrail', 'jailbreak', 'human-take-over', 'auto-es-filters-generation', 'question-suggestion', 'language-detection', 'advanced-classification', 'advanced-classification-v2', 'highlight', 'tool-choice', 'pdf-vision-conversion', 'map-reduce-map', 'map-reduce-reduce', 'shorten-instruction', 'websearch-selection', 'auto-bot-description', 'openai-chat-completion', 'mcp-assistant-router', 'mcp-multi-turn-execution', 'mcp-progress-message-generation', 'mcp-progress-tool-bullet-point-generation', 'unknown']):
+            raise ValueError("must be one of enum values ('query-expansion', 'response-audit', 'title-generation', 'chat', 'sparql2text', 'guardrail', 'jailbreak', 'human-take-over', 'auto-es-filters-generation', 'question-suggestion', 'language-detection', 'advanced-classification', 'advanced-classification-v2', 'highlight', 'tool-choice', 'pdf-vision-conversion', 'map-reduce-map', 'map-reduce-reduce', 'shorten-instruction', 'websearch-selection', 'auto-bot-description', 'openai-chat-completion', 'mcp-assistant-router', 'mcp-multi-turn-execution', 'mcp-progress-message-generation', 'mcp-progress-tool-bullet-point-generation', 'unknown')")
         return value
 
     model_config = ConfigDict(

qanswer_sdk/models/llm_create_dto.py

@@ -44,10 +44,16 @@ class LlmCreateDto(BaseModel):
     guardrail: Optional[GuardrailEndpoint] = None
     retry_params: Optional[RetryParams] = None
     extra_body: Optional[ExtraBody] = None
+    vertex_location: Optional[StrictStr] = None
+    vertex_credentials: Optional[StrictStr] = None
     tags: Optional[List[StrictStr]] = None
     tool_call_support: Optional[StrictBool] = None
     temperature_support: Optional[StrictBool] = None
     stream_options_support: Optional[StrictBool] = None
+    content_items_support: Optional[StrictBool] = None
+    reasoning_effort: Optional[StrictStr] = None
+    websearch_support: Optional[StrictBool] = None
+    skip_ssl_verification: Optional[StrictBool] = None
     display_name: StrictStr
     engine: StrictStr
     api_key: StrictStr
@@ -57,7 +63,7 @@ class LlmCreateDto(BaseModel):
     name: StrictStr
     cost_input_per_million_token: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="costInputPerMillionToken")
     cost_output_per_million_token: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="costOutputPerMillionToken")
-    __properties: ClassVar[List[str]] = ["modality", "provider", "api_version", "aws_access_key_id", "aws_secret_access_key", "aws_region_name", "aws_inference_profile", "description", "onpremise", "sensitivity_label", "guardrail", "retry_params", "extra_body", "tags", "tool_call_support", "temperature_support", "stream_options_support", "display_name", "engine", "api_key", "endpoint_url", "rag_params", "generation_params", "name", "costInputPerMillionToken", "costOutputPerMillionToken"]
+    __properties: ClassVar[List[str]] = ["modality", "provider", "api_version", "aws_access_key_id", "aws_secret_access_key", "aws_region_name", "aws_inference_profile", "description", "onpremise", "sensitivity_label", "guardrail", "retry_params", "extra_body", "vertex_location", "vertex_credentials", "tags", "tool_call_support", "temperature_support", "stream_options_support", "content_items_support", "reasoning_effort", "websearch_support", "skip_ssl_verification", "display_name", "engine", "api_key", "endpoint_url", "rag_params", "generation_params", "name", "costInputPerMillionToken", "costOutputPerMillionToken"]
 
     @field_validator('modality')
     def modality_validate_enum(cls, value):
@@ -75,8 +81,18 @@ class LlmCreateDto(BaseModel):
         if value is None:
             return value
 
-        if value not in set(['openai', 'azure', 'bedrock', 'mistral', 'anthropic', 'openrouter', 'ollama']):
-            raise ValueError("must be one of enum values ('openai', 'azure', 'bedrock', 'mistral', 'anthropic', 'openrouter', 'ollama')")
+        if value not in set(['openai', 'azure', 'bedrock', 'mistral', 'anthropic', 'openrouter', 'gemini', 'ollama', 'vertex_ai']):
+            raise ValueError("must be one of enum values ('openai', 'azure', 'bedrock', 'mistral', 'anthropic', 'openrouter', 'gemini', 'ollama', 'vertex_ai')")
+        return value
+
+    @field_validator('reasoning_effort')
+    def reasoning_effort_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['none', 'low', 'medium', 'high']):
+            raise ValueError("must be one of enum values ('none', 'low', 'medium', 'high')")
         return value
 
     model_config = ConfigDict(
@@ -158,10 +174,16 @@ class LlmCreateDto(BaseModel):
             "guardrail": GuardrailEndpoint.from_dict(obj["guardrail"]) if obj.get("guardrail") is not None else None,
             "retry_params": RetryParams.from_dict(obj["retry_params"]) if obj.get("retry_params") is not None else None,
             "extra_body": ExtraBody.from_dict(obj["extra_body"]) if obj.get("extra_body") is not None else None,
+            "vertex_location": obj.get("vertex_location"),
+            "vertex_credentials": obj.get("vertex_credentials"),
             "tags": obj.get("tags"),
             "tool_call_support": obj.get("tool_call_support"),
             "temperature_support": obj.get("temperature_support"),
             "stream_options_support": obj.get("stream_options_support"),
+            "content_items_support": obj.get("content_items_support"),
+            "reasoning_effort": obj.get("reasoning_effort"),
+            "websearch_support": obj.get("websearch_support"),
+            "skip_ssl_verification": obj.get("skip_ssl_verification"),
             "display_name": obj.get("display_name"),
             "engine": obj.get("engine"),
             "api_key": obj.get("api_key"),

qanswer_sdk/models/llm_detailed_dto.py

@@ -45,10 +45,16 @@ class LlmDetailedDto(BaseModel):
     guardrail: Optional[GuardrailEndpoint] = None
     retry_params: Optional[RetryParams] = None
     extra_body: Optional[ExtraBody] = None
+    vertex_location: Optional[StrictStr] = None
+    vertex_credentials: Optional[StrictStr] = None
     tags: Optional[List[StrictStr]] = None
     tool_call_support: Optional[StrictBool] = None
     temperature_support: Optional[StrictBool] = None
     stream_options_support: Optional[StrictBool] = None
+    content_items_support: Optional[StrictBool] = None
+    reasoning_effort: Optional[StrictStr] = None
+    websearch_support: Optional[StrictBool] = None
+    skip_ssl_verification: Optional[StrictBool] = None
     display_name: StrictStr
     engine: StrictStr
     api_key: StrictStr
@@ -61,7 +67,9 @@ class LlmDetailedDto(BaseModel):
     cost_input_per_million_token: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="costInputPerMillionToken")
     cost_output_per_million_token: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="costOutputPerMillionToken")
     active: Optional[StrictBool] = None
-    __properties: ClassVar[List[str]] = ["modality", "provider", "api_version", "aws_access_key_id", "aws_secret_access_key", "aws_region_name", "aws_inference_profile", "description", "onpremise", "sensitivity_label", "guardrail", "retry_params", "extra_body", "tags", "tool_call_support", "temperature_support", "stream_options_support", "display_name", "engine", "api_key", "endpoint_url", "rag_params", "generation_params", "name", "id", "context_ranges", "costInputPerMillionToken", "costOutputPerMillionToken", "active"]
+    is_system_default: Optional[StrictBool] = Field(default=None, alias="isSystemDefault")
+    is_organization_default: Optional[StrictBool] = Field(default=None, alias="isOrganizationDefault")
+    __properties: ClassVar[List[str]] = ["modality", "provider", "api_version", "aws_access_key_id", "aws_secret_access_key", "aws_region_name", "aws_inference_profile", "description", "onpremise", "sensitivity_label", "guardrail", "retry_params", "extra_body", "vertex_location", "vertex_credentials", "tags", "tool_call_support", "temperature_support", "stream_options_support", "content_items_support", "reasoning_effort", "websearch_support", "skip_ssl_verification", "display_name", "engine", "api_key", "endpoint_url", "rag_params", "generation_params", "name", "id", "context_ranges", "costInputPerMillionToken", "costOutputPerMillionToken", "active", "isSystemDefault", "isOrganizationDefault"]
 
     @field_validator('modality')
     def modality_validate_enum(cls, value):
@@ -79,8 +87,18 @@ class LlmDetailedDto(BaseModel):
         if value is None:
             return value
 
-        if value not in set(['openai', 'azure', 'bedrock', 'mistral', 'anthropic', 'openrouter', 'ollama']):
-            raise ValueError("must be one of enum values ('openai', 'azure', 'bedrock', 'mistral', 'anthropic', 'openrouter', 'ollama')")
+        if value not in set(['openai', 'azure', 'bedrock', 'mistral', 'anthropic', 'openrouter', 'gemini', 'ollama', 'vertex_ai']):
+            raise ValueError("must be one of enum values ('openai', 'azure', 'bedrock', 'mistral', 'anthropic', 'openrouter', 'gemini', 'ollama', 'vertex_ai')")
+        return value
+
+    @field_validator('reasoning_effort')
+    def reasoning_effort_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['none', 'low', 'medium', 'high']):
+            raise ValueError("must be one of enum values ('none', 'low', 'medium', 'high')")
         return value
 
     model_config = ConfigDict(
@@ -165,10 +183,16 @@ class LlmDetailedDto(BaseModel):
             "guardrail": GuardrailEndpoint.from_dict(obj["guardrail"]) if obj.get("guardrail") is not None else None,
             "retry_params": RetryParams.from_dict(obj["retry_params"]) if obj.get("retry_params") is not None else None,
             "extra_body": ExtraBody.from_dict(obj["extra_body"]) if obj.get("extra_body") is not None else None,
+            "vertex_location": obj.get("vertex_location"),
+            "vertex_credentials": obj.get("vertex_credentials"),
             "tags": obj.get("tags"),
             "tool_call_support": obj.get("tool_call_support"),
             "temperature_support": obj.get("temperature_support"),
             "stream_options_support": obj.get("stream_options_support"),
+            "content_items_support": obj.get("content_items_support"),
+            "reasoning_effort": obj.get("reasoning_effort"),
+            "websearch_support": obj.get("websearch_support"),
+            "skip_ssl_verification": obj.get("skip_ssl_verification"),
             "display_name": obj.get("display_name"),
             "engine": obj.get("engine"),
             "api_key": obj.get("api_key"),
@@ -180,7 +204,9 @@ class LlmDetailedDto(BaseModel):
             "context_ranges": LLMContextRanges.from_dict(obj["context_ranges"]) if obj.get("context_ranges") is not None else None,
             "costInputPerMillionToken": obj.get("costInputPerMillionToken"),
             "costOutputPerMillionToken": obj.get("costOutputPerMillionToken"),
-            "active": obj.get("active")
+            "active": obj.get("active"),
+            "isSystemDefault": obj.get("isSystemDefault"),
+            "isOrganizationDefault": obj.get("isOrganizationDefault")
         })
         return _obj
 

qanswer_sdk/models/llm_dto.py

@@ -32,7 +32,10 @@ class LlmDto(BaseModel):
     sensitivity_label: Optional[StrictStr] = Field(default=None, alias="sensitivityLabel")
     onpremise: Optional[StrictBool] = None
     modality: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["id", "name", "description", "sensitivityLabel", "onpremise", "modality"]
+    tool_call_support: Optional[StrictBool] = Field(default=None, alias="toolCallSupport")
+    reasoning_effort: Optional[StrictStr] = Field(default=None, alias="reasoningEffort")
+    web_search_support: Optional[StrictBool] = Field(default=None, alias="webSearchSupport")
+    __properties: ClassVar[List[str]] = ["id", "name", "description", "sensitivityLabel", "onpremise", "modality", "toolCallSupport", "reasoningEffort", "webSearchSupport"]
 
     @field_validator('modality')
     def modality_validate_enum(cls, value):
@@ -44,6 +47,16 @@ class LlmDto(BaseModel):
             raise ValueError("must be one of enum values ('text', 'text-image')")
         return value
 
+    @field_validator('reasoning_effort')
+    def reasoning_effort_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['none', 'low', 'medium', 'high']):
+            raise ValueError("must be one of enum values ('none', 'low', 'medium', 'high')")
+        return value
+
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,
@@ -100,7 +113,10 @@ class LlmDto(BaseModel):
             "description": obj.get("description"),
             "sensitivityLabel": obj.get("sensitivityLabel"),
             "onpremise": obj.get("onpremise"),
-            "modality": obj.get("modality")
+            "modality": obj.get("modality"),
+            "toolCallSupport": obj.get("toolCallSupport"),
+            "reasoningEffort": obj.get("reasoningEffort"),
+            "webSearchSupport": obj.get("webSearchSupport")
         })
         return _obj
 

qanswer_sdk/models/llm_endpoint.py

@@ -45,10 +45,16 @@ class LLMEndpoint(BaseModel):
     guardrail: Optional[GuardrailEndpoint] = None
     retry_params: Optional[RetryParams] = None
     extra_body: Optional[ExtraBody] = None
+    vertex_location: Optional[StrictStr] = None
+    vertex_credentials: Optional[StrictStr] = None
     tags: Optional[List[StrictStr]] = None
     tool_call_support: Optional[StrictBool] = None
     temperature_support: Optional[StrictBool] = None
     stream_options_support: Optional[StrictBool] = None
+    content_items_support: Optional[StrictBool] = None
+    reasoning_effort: Optional[StrictStr] = None
+    websearch_support: Optional[StrictBool] = None
+    skip_ssl_verification: Optional[StrictBool] = None
     display_name: StrictStr
     engine: StrictStr
     api_key: StrictStr
@@ -58,7 +64,7 @@ class LLMEndpoint(BaseModel):
     name: StrictStr
     id: Optional[StrictInt] = None
     context_ranges: Optional[LLMContextRanges] = None
-    __properties: ClassVar[List[str]] = ["modality", "provider", "api_version", "aws_access_key_id", "aws_secret_access_key", "aws_region_name", "aws_inference_profile", "description", "onpremise", "sensitivity_label", "guardrail", "retry_params", "extra_body", "tags", "tool_call_support", "temperature_support", "stream_options_support", "display_name", "engine", "api_key", "endpoint_url", "rag_params", "generation_params", "name", "id", "context_ranges"]
+    __properties: ClassVar[List[str]] = ["modality", "provider", "api_version", "aws_access_key_id", "aws_secret_access_key", "aws_region_name", "aws_inference_profile", "description", "onpremise", "sensitivity_label", "guardrail", "retry_params", "extra_body", "vertex_location", "vertex_credentials", "tags", "tool_call_support", "temperature_support", "stream_options_support", "content_items_support", "reasoning_effort", "websearch_support", "skip_ssl_verification", "display_name", "engine", "api_key", "endpoint_url", "rag_params", "generation_params", "name", "id", "context_ranges"]
 
     @field_validator('modality')
     def modality_validate_enum(cls, value):
@@ -76,8 +82,18 @@ class LLMEndpoint(BaseModel):
         if value is None:
             return value
 
-        if value not in set(['openai', 'azure', 'bedrock', 'mistral', 'anthropic', 'openrouter', 'ollama']):
-            raise ValueError("must be one of enum values ('openai', 'azure', 'bedrock', 'mistral', 'anthropic', 'openrouter', 'ollama')")
+        if value not in set(['openai', 'azure', 'bedrock', 'mistral', 'anthropic', 'openrouter', 'gemini', 'ollama', 'vertex_ai']):
+            raise ValueError("must be one of enum values ('openai', 'azure', 'bedrock', 'mistral', 'anthropic', 'openrouter', 'gemini', 'ollama', 'vertex_ai')")
+        return value
+
+    @field_validator('reasoning_effort')
+    def reasoning_effort_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['none', 'low', 'medium', 'high']):
+            raise ValueError("must be one of enum values ('none', 'low', 'medium', 'high')")
         return value
 
     model_config = ConfigDict(
@@ -162,10 +178,16 @@ class LLMEndpoint(BaseModel):
             "guardrail": GuardrailEndpoint.from_dict(obj["guardrail"]) if obj.get("guardrail") is not None else None,
             "retry_params": RetryParams.from_dict(obj["retry_params"]) if obj.get("retry_params") is not None else None,
             "extra_body": ExtraBody.from_dict(obj["extra_body"]) if obj.get("extra_body") is not None else None,
+            "vertex_location": obj.get("vertex_location"),
+            "vertex_credentials": obj.get("vertex_credentials"),
             "tags": obj.get("tags"),
             "tool_call_support": obj.get("tool_call_support"),
             "temperature_support": obj.get("temperature_support"),
             "stream_options_support": obj.get("stream_options_support"),
+            "content_items_support": obj.get("content_items_support"),
+            "reasoning_effort": obj.get("reasoning_effort"),
+            "websearch_support": obj.get("websearch_support"),
+            "skip_ssl_verification": obj.get("skip_ssl_verification"),
             "display_name": obj.get("display_name"),
             "engine": obj.get("engine"),
             "api_key": obj.get("api_key"),

qanswer_sdk/models/llm_endpoint_import_failure.py

@@ -0,0 +1,89 @@
+# 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 LLMEndpointImportFailure(BaseModel):
+    """
+    LLMEndpointImportFailure
+    """ # noqa: E501
+    name: StrictStr
+    reason: StrictStr
+    __properties: ClassVar[List[str]] = ["name", "reason"]
+
+    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 LLMEndpointImportFailure 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 LLMEndpointImportFailure from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "name": obj.get("name"),
+            "reason": obj.get("reason")
+        })
+        return _obj
+
+

qanswer_sdk/models/{relation_extraction_data_output.py → llm_endpoint_import_response.py}

@@ -17,18 +17,20 @@ 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.relation_extraction_row_output import RelationExtractionRowOutput
+from pydantic import BaseModel, ConfigDict, StrictInt
+from typing import Any, ClassVar, Dict, List
+from qanswer_sdk.models.llm_endpoint_import_failure import LLMEndpointImportFailure
 from typing import Optional, Set
 from typing_extensions import Self
 
-class RelationExtractionDataOutput(BaseModel):
+class LLMEndpointImportResponse(BaseModel):
     """
-    RelationExtractionDataOutput
+    LLMEndpointImportResponse
     """ # noqa: E501
-    rows: Optional[List[RelationExtractionRowOutput]] = None
-    __properties: ClassVar[List[str]] = ["rows"]
+    success: StrictInt
+    failed: StrictInt
+    fail_list: List[LLMEndpointImportFailure]
+    __properties: ClassVar[List[str]] = ["success", "failed", "fail_list"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -48,7 +50,7 @@ class RelationExtractionDataOutput(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of RelationExtractionDataOutput from a JSON string"""
+        """Create an instance of LLMEndpointImportResponse from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -69,18 +71,18 @@ class RelationExtractionDataOutput(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of each item in rows (list)
+        # override the default output from pydantic by calling `to_dict()` of each item in fail_list (list)
         _items = []
-        if self.rows:
-            for _item_rows in self.rows:
-                if _item_rows:
-                    _items.append(_item_rows.to_dict())
-            _dict['rows'] = _items
+        if self.fail_list:
+            for _item_fail_list in self.fail_list:
+                if _item_fail_list:
+                    _items.append(_item_fail_list.to_dict())
+            _dict['fail_list'] = _items
         return _dict
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of RelationExtractionDataOutput from a dict"""
+        """Create an instance of LLMEndpointImportResponse from a dict"""
         if obj is None:
             return None
 
@@ -88,7 +90,9 @@ class RelationExtractionDataOutput(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "rows": [RelationExtractionRowOutput.from_dict(_item) for _item in obj["rows"]] if obj.get("rows") is not None else None
+            "success": obj.get("success"),
+            "failed": obj.get("failed"),
+            "fail_list": [LLMEndpointImportFailure.from_dict(_item) for _item in obj["fail_list"]] if obj.get("fail_list") is not None else None
         })
         return _obj