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

qanswer_sdk/models/dataset_detail_kg.py

@@ -20,14 +20,16 @@ import json
 from datetime import datetime
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
+from qanswer_sdk.models.ai_assistant_restricted_domain import AiAssistantRestrictedDomain
 from qanswer_sdk.models.dataset_config_request import DatasetConfigRequest
+from qanswer_sdk.models.embedder_consumption import EmbedderConsumption
 from qanswer_sdk.models.entity_description import EntityDescription
 from qanswer_sdk.models.llm_consumption import LlmConsumption
 from qanswer_sdk.models.logo import Logo
 from qanswer_sdk.models.shared_organization_access import SharedOrganizationAccess
 from qanswer_sdk.models.shared_team_access import SharedTeamAccess
 from qanswer_sdk.models.user import User
-from qanswer_sdk.models.user_dataset_access_control_origin_header import UserDatasetAccessControlOriginHeader
+from qanswer_sdk.models.user_dataset_mcp_tool_selection import UserDatasetMcpToolSelection
 from qanswer_sdk.models.user_dataset_shared import UserDatasetShared
 from typing import Optional, Set
 from typing_extensions import Self
@@ -53,12 +55,14 @@ class DatasetDetailKg(BaseModel):
     shared_users: Optional[List[UserDatasetShared]] = Field(default=None, alias="sharedUsers")
     shared_team_accesses: Optional[List[SharedTeamAccess]] = Field(default=None, alias="sharedTeamAccesses")
     shared_organization_accesses: Optional[List[SharedOrganizationAccess]] = Field(default=None, alias="sharedOrganizationAccesses")
-    access_control_origin_headers: Optional[List[UserDatasetAccessControlOriginHeader]] = Field(default=None, alias="accessControlOriginHeaders")
+    mcp_tool_selections: Optional[List[UserDatasetMcpToolSelection]] = Field(default=None, alias="mcpToolSelections")
+    ai_assistant_restricted_domains: Optional[List[AiAssistantRestrictedDomain]] = Field(default=None, alias="aiAssistantRestrictedDomains")
     entity_description: Optional[EntityDescription] = Field(default=None, alias="entityDescription")
     logo: Optional[Logo] = None
     llm_consumptions: Optional[List[LlmConsumption]] = Field(default=None, alias="llmConsumptions")
+    embedder_consumptions: Optional[List[EmbedderConsumption]] = Field(default=None, alias="embedderConsumptions")
     config: Optional[DatasetConfigRequest] = None
-    __properties: ClassVar[List[str]] = ["id", "dataset", "name", "description", "user", "uploaded", "indexed", "access", "accessType", "size", "created", "indexType", "languages", "assistantTemplatePublished", "sharedUsers", "sharedTeamAccesses", "sharedOrganizationAccesses", "accessControlOriginHeaders", "entityDescription", "logo", "llmConsumptions", "config"]
+    __properties: ClassVar[List[str]] = ["id", "dataset", "name", "description", "user", "uploaded", "indexed", "access", "accessType", "size", "created", "indexType", "languages", "assistantTemplatePublished", "sharedUsers", "sharedTeamAccesses", "sharedOrganizationAccesses", "mcpToolSelections", "aiAssistantRestrictedDomains", "entityDescription", "logo", "llmConsumptions", "embedderConsumptions", "config"]
 
     @field_validator('access_type')
     def access_type_validate_enum(cls, value):
@@ -133,13 +137,20 @@ class DatasetDetailKg(BaseModel):
                 if _item_shared_organization_accesses:
                     _items.append(_item_shared_organization_accesses.to_dict())
             _dict['sharedOrganizationAccesses'] = _items
-        # override the default output from pydantic by calling `to_dict()` of each item in access_control_origin_headers (list)
+        # override the default output from pydantic by calling `to_dict()` of each item in mcp_tool_selections (list)
         _items = []
-        if self.access_control_origin_headers:
-            for _item_access_control_origin_headers in self.access_control_origin_headers:
-                if _item_access_control_origin_headers:
-                    _items.append(_item_access_control_origin_headers.to_dict())
-            _dict['accessControlOriginHeaders'] = _items
+        if self.mcp_tool_selections:
+            for _item_mcp_tool_selections in self.mcp_tool_selections:
+                if _item_mcp_tool_selections:
+                    _items.append(_item_mcp_tool_selections.to_dict())
+            _dict['mcpToolSelections'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in ai_assistant_restricted_domains (list)
+        _items = []
+        if self.ai_assistant_restricted_domains:
+            for _item_ai_assistant_restricted_domains in self.ai_assistant_restricted_domains:
+                if _item_ai_assistant_restricted_domains:
+                    _items.append(_item_ai_assistant_restricted_domains.to_dict())
+            _dict['aiAssistantRestrictedDomains'] = _items
         # override the default output from pydantic by calling `to_dict()` of entity_description
         if self.entity_description:
             _dict['entityDescription'] = self.entity_description.to_dict()
@@ -153,6 +164,13 @@ class DatasetDetailKg(BaseModel):
                 if _item_llm_consumptions:
                     _items.append(_item_llm_consumptions.to_dict())
             _dict['llmConsumptions'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in embedder_consumptions (list)
+        _items = []
+        if self.embedder_consumptions:
+            for _item_embedder_consumptions in self.embedder_consumptions:
+                if _item_embedder_consumptions:
+                    _items.append(_item_embedder_consumptions.to_dict())
+            _dict['embedderConsumptions'] = _items
         # override the default output from pydantic by calling `to_dict()` of config
         if self.config:
             _dict['config'] = self.config.to_dict()
@@ -185,10 +203,12 @@ class DatasetDetailKg(BaseModel):
             "sharedUsers": [UserDatasetShared.from_dict(_item) for _item in obj["sharedUsers"]] if obj.get("sharedUsers") is not None else None,
             "sharedTeamAccesses": [SharedTeamAccess.from_dict(_item) for _item in obj["sharedTeamAccesses"]] if obj.get("sharedTeamAccesses") is not None else None,
             "sharedOrganizationAccesses": [SharedOrganizationAccess.from_dict(_item) for _item in obj["sharedOrganizationAccesses"]] if obj.get("sharedOrganizationAccesses") is not None else None,
-            "accessControlOriginHeaders": [UserDatasetAccessControlOriginHeader.from_dict(_item) for _item in obj["accessControlOriginHeaders"]] if obj.get("accessControlOriginHeaders") is not None else None,
+            "mcpToolSelections": [UserDatasetMcpToolSelection.from_dict(_item) for _item in obj["mcpToolSelections"]] if obj.get("mcpToolSelections") is not None else None,
+            "aiAssistantRestrictedDomains": [AiAssistantRestrictedDomain.from_dict(_item) for _item in obj["aiAssistantRestrictedDomains"]] if obj.get("aiAssistantRestrictedDomains") is not None else None,
             "entityDescription": EntityDescription.from_dict(obj["entityDescription"]) if obj.get("entityDescription") is not None else None,
             "logo": Logo.from_dict(obj["logo"]) if obj.get("logo") is not None else None,
             "llmConsumptions": [LlmConsumption.from_dict(_item) for _item in obj["llmConsumptions"]] if obj.get("llmConsumptions") is not None else None,
+            "embedderConsumptions": [EmbedderConsumption.from_dict(_item) for _item in obj["embedderConsumptions"]] if obj.get("embedderConsumptions") is not None else None,
             "config": DatasetConfigRequest.from_dict(obj["config"]) if obj.get("config") is not None else None
         })
         return _obj

qanswer_sdk/models/dataset_schema.py

@@ -64,15 +64,17 @@ class DatasetSchema(BaseModel):
     creation_date: Optional[StrictStr] = None
     use_training: Optional[StrictBool] = None
     embedding_model: Optional[StrictStr] = None
+    reindexing_failed: Optional[StrictBool] = None
     data_storage_preference: Optional[StrictStr] = None
     logs_visibility_policy: Optional[StrictStr] = None
     dataset_type: Optional[StrictStr] = None
     categories: Optional[List[StrictStr]] = None
     is_template: Optional[StrictBool] = None
+    incognito: Optional[StrictBool] = None
     widget_configs: WidgetConfigs
     n_tokens: Optional[StrictInt] = None
     embedding_model_id: Optional[StrictInt] = None
-    __properties: ClassVar[List[str]] = ["id", "dataset", "username", "index_hash", "language", "is_updating", "use_reader", "use_ranker", "use_fallback", "top_k_retriever", "top_k_reader", "top_k_ranker", "return_no_ans", "no_ans_boost", "max_seq_len", "doc_stride", "max_ans_len", "weight_title", "weight_text", "split_by", "split_len", "split_overlap", "bot_name", "bot_description", "bot_mode", "bot_knowledge", "bot_answer_length", "bot_seed", "bot_temperature", "prompt", "number_of_references", "stream_speed", "initial_message", "track_geo", "creation_date", "use_training", "embedding_model", "data_storage_preference", "logs_visibility_policy", "dataset_type", "categories", "is_template", "widget_configs", "n_tokens", "embedding_model_id"]
+    __properties: ClassVar[List[str]] = ["id", "dataset", "username", "index_hash", "language", "is_updating", "use_reader", "use_ranker", "use_fallback", "top_k_retriever", "top_k_reader", "top_k_ranker", "return_no_ans", "no_ans_boost", "max_seq_len", "doc_stride", "max_ans_len", "weight_title", "weight_text", "split_by", "split_len", "split_overlap", "bot_name", "bot_description", "bot_mode", "bot_knowledge", "bot_answer_length", "bot_seed", "bot_temperature", "prompt", "number_of_references", "stream_speed", "initial_message", "track_geo", "creation_date", "use_training", "embedding_model", "reindexing_failed", "data_storage_preference", "logs_visibility_policy", "dataset_type", "categories", "is_template", "incognito", "widget_configs", "n_tokens", "embedding_model_id"]
 
     @field_validator('data_storage_preference')
     def data_storage_preference_validate_enum(cls, value):
@@ -195,11 +197,13 @@ class DatasetSchema(BaseModel):
             "creation_date": obj.get("creation_date"),
             "use_training": obj.get("use_training"),
             "embedding_model": obj.get("embedding_model"),
+            "reindexing_failed": obj.get("reindexing_failed"),
             "data_storage_preference": obj.get("data_storage_preference"),
             "logs_visibility_policy": obj.get("logs_visibility_policy"),
             "dataset_type": obj.get("dataset_type"),
             "categories": obj.get("categories"),
             "is_template": obj.get("is_template"),
+            "incognito": obj.get("incognito"),
             "widget_configs": WidgetConfigs.from_dict(obj["widget_configs"]) if obj.get("widget_configs") is not None else None,
             "n_tokens": obj.get("n_tokens"),
             "embedding_model_id": obj.get("embedding_model_id")

qanswer_sdk/models/dataset_update_object.py

@@ -70,7 +70,10 @@ class DatasetUpdateObject(BaseModel):
     split_by: Optional[StrictStr] = Field(default=None, description="Split method")
     split_len: Optional[StrictInt] = Field(default=None, description="Split length")
     split_overlap: Optional[StrictInt] = Field(default=None, description="Split overlap")
-    __properties: ClassVar[List[str]] = ["dataset", "username", "is_updating", "use_reader", "use_ranker", "use_fallback", "top_k_retriever", "top_k_reader", "return_no_ans", "no_ans_boost", "max_seq_len", "doc_stride", "max_ans_len", "weight_title", "weight_text", "bot_name", "bot_description", "bot_mode", "bot_knowledge", "bot_answer_length", "bot_seed", "bot_temperature", "prompt", "number_of_references", "data_storage_preference", "stream_speed", "categories", "initial_message", "widget_init_msg", "widget_theme", "widget_open_default", "widget_font", "widget_colorPrimary", "widget_colorSecondary", "widget_align", "widget_offsetX", "widget_offsetY", "widget_show_powered_by", "widget_show_export_conversation", "track_geo", "use_training", "split_by", "split_len", "split_overlap"]
+    embedding_model_id: Optional[StrictInt] = Field(default=None, description="Embedding model id in case of changing the embedder, otherwise None")
+    reindexing_failed: Optional[StrictBool] = None
+    additional_fields: Optional[Dict[str, Any]] = Field(default=None, description="Additional fields for future extensions")
+    __properties: ClassVar[List[str]] = ["dataset", "username", "is_updating", "use_reader", "use_ranker", "use_fallback", "top_k_retriever", "top_k_reader", "return_no_ans", "no_ans_boost", "max_seq_len", "doc_stride", "max_ans_len", "weight_title", "weight_text", "bot_name", "bot_description", "bot_mode", "bot_knowledge", "bot_answer_length", "bot_seed", "bot_temperature", "prompt", "number_of_references", "data_storage_preference", "stream_speed", "categories", "initial_message", "widget_init_msg", "widget_theme", "widget_open_default", "widget_font", "widget_colorPrimary", "widget_colorSecondary", "widget_align", "widget_offsetX", "widget_offsetY", "widget_show_powered_by", "widget_show_export_conversation", "track_geo", "use_training", "split_by", "split_len", "split_overlap", "embedding_model_id", "reindexing_failed", "additional_fields"]
 
     @field_validator('data_storage_preference')
     def data_storage_preference_validate_enum(cls, value):
@@ -176,7 +179,10 @@ class DatasetUpdateObject(BaseModel):
             "use_training": obj.get("use_training"),
             "split_by": obj.get("split_by"),
             "split_len": obj.get("split_len"),
-            "split_overlap": obj.get("split_overlap")
+            "split_overlap": obj.get("split_overlap"),
+            "embedding_model_id": obj.get("embedding_model_id"),
+            "reindexing_failed": obj.get("reindexing_failed"),
+            "additional_fields": obj.get("additional_fields")
         })
         return _obj
 

qanswer_sdk/models/document_content_update_model.py

@@ -32,7 +32,8 @@ class DocumentContentUpdateModel(BaseModel):
     connector_id: StrictInt
     content: StrictStr
     conversation_id: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["username", "dataset", "filename", "connector_id", "content", "conversation_id"]
+    additional_fields: Optional[Dict[str, Any]] = None
+    __properties: ClassVar[List[str]] = ["username", "dataset", "filename", "connector_id", "content", "conversation_id", "additional_fields"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -90,7 +91,8 @@ class DocumentContentUpdateModel(BaseModel):
             "filename": obj.get("filename"),
             "connector_id": obj.get("connector_id"),
             "content": obj.get("content"),
-            "conversation_id": obj.get("conversation_id")
+            "conversation_id": obj.get("conversation_id"),
+            "additional_fields": obj.get("additional_fields")
         })
         return _obj
 

qanswer_sdk/models/econsilium_search_index_add_payload.py

@@ -31,9 +31,10 @@ class EconsiliumSearchIndexAddPayload(BaseModel):
     dataset: StrictStr
     connector_id: StrictInt
     quota: StrictInt
+    additional_fields: Optional[Dict[str, Any]] = None
     oauth_token: Optional[StrictStr] = None
     files: List[EconsiliumFileMetadata]
-    __properties: ClassVar[List[str]] = ["username", "dataset", "connector_id", "quota", "oauth_token", "files"]
+    __properties: ClassVar[List[str]] = ["username", "dataset", "connector_id", "quota", "additional_fields", "oauth_token", "files"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -97,6 +98,7 @@ class EconsiliumSearchIndexAddPayload(BaseModel):
             "dataset": obj.get("dataset"),
             "connector_id": obj.get("connector_id"),
             "quota": obj.get("quota"),
+            "additional_fields": obj.get("additional_fields"),
             "oauth_token": obj.get("oauth_token"),
             "files": [EconsiliumFileMetadata.from_dict(_item) for _item in obj["files"]] if obj.get("files") is not None else None
         })

qanswer_sdk/models/embedder_create_dto.py

@@ -37,7 +37,10 @@ class EmbedderCreateDto(BaseModel):
     aws_secret_access_key: Optional[StrictStr] = None
     aws_region_name: Optional[StrictStr] = None
     aws_inference_profile: Optional[StrictStr] = None
+    vertex_location: Optional[StrictStr] = None
+    vertex_credentials: Optional[StrictStr] = None
     batch_size: Optional[StrictInt] = None
+    skip_ssl_verification: Optional[StrictBool] = None
     display_name: StrictStr
     engine: StrictStr
     api_key: Optional[StrictStr] = None
@@ -45,7 +48,7 @@ class EmbedderCreateDto(BaseModel):
     embedding_size: StrictInt
     context_window: StrictInt
     cost_input_per_million_token: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="costInputPerMillionToken")
-    __properties: ClassVar[List[str]] = ["modality", "provider", "endpoint_url", "description", "onpremise", "sensitivity_label", "api_version", "aws_access_key_id", "aws_secret_access_key", "aws_region_name", "aws_inference_profile", "batch_size", "display_name", "engine", "api_key", "name", "embedding_size", "context_window", "costInputPerMillionToken"]
+    __properties: ClassVar[List[str]] = ["modality", "provider", "endpoint_url", "description", "onpremise", "sensitivity_label", "api_version", "aws_access_key_id", "aws_secret_access_key", "aws_region_name", "aws_inference_profile", "vertex_location", "vertex_credentials", "batch_size", "skip_ssl_verification", "display_name", "engine", "api_key", "name", "embedding_size", "context_window", "costInputPerMillionToken"]
 
     @field_validator('modality')
     def modality_validate_enum(cls, value):
@@ -63,8 +66,8 @@ class EmbedderCreateDto(BaseModel):
         if value is None:
             return value
 
-        if value not in set(['openai', 'azure', 'bedrock', 'ollama']):
-            raise ValueError("must be one of enum values ('openai', 'azure', 'bedrock', '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
 
     model_config = ConfigDict(
@@ -129,7 +132,10 @@ class EmbedderCreateDto(BaseModel):
             "aws_secret_access_key": obj.get("aws_secret_access_key"),
             "aws_region_name": obj.get("aws_region_name"),
             "aws_inference_profile": obj.get("aws_inference_profile"),
+            "vertex_location": obj.get("vertex_location"),
+            "vertex_credentials": obj.get("vertex_credentials"),
             "batch_size": obj.get("batch_size"),
+            "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/embedder_detailed_dto.py

@@ -37,7 +37,10 @@ class EmbedderDetailedDto(BaseModel):
     aws_secret_access_key: Optional[StrictStr] = None
     aws_region_name: Optional[StrictStr] = None
     aws_inference_profile: Optional[StrictStr] = None
+    vertex_location: Optional[StrictStr] = None
+    vertex_credentials: Optional[StrictStr] = None
     batch_size: Optional[StrictInt] = None
+    skip_ssl_verification: Optional[StrictBool] = None
     display_name: StrictStr
     engine: StrictStr
     api_key: Optional[StrictStr] = None
@@ -47,7 +50,7 @@ class EmbedderDetailedDto(BaseModel):
     id: Optional[StrictInt] = None
     cost_input_per_million_token: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="costInputPerMillionToken")
     active: Optional[StrictBool] = None
-    __properties: ClassVar[List[str]] = ["modality", "provider", "endpoint_url", "description", "onpremise", "sensitivity_label", "api_version", "aws_access_key_id", "aws_secret_access_key", "aws_region_name", "aws_inference_profile", "batch_size", "display_name", "engine", "api_key", "name", "embedding_size", "context_window", "id", "costInputPerMillionToken", "active"]
+    __properties: ClassVar[List[str]] = ["modality", "provider", "endpoint_url", "description", "onpremise", "sensitivity_label", "api_version", "aws_access_key_id", "aws_secret_access_key", "aws_region_name", "aws_inference_profile", "vertex_location", "vertex_credentials", "batch_size", "skip_ssl_verification", "display_name", "engine", "api_key", "name", "embedding_size", "context_window", "id", "costInputPerMillionToken", "active"]
 
     @field_validator('modality')
     def modality_validate_enum(cls, value):
@@ -65,8 +68,8 @@ class EmbedderDetailedDto(BaseModel):
         if value is None:
             return value
 
-        if value not in set(['openai', 'azure', 'bedrock', 'ollama']):
-            raise ValueError("must be one of enum values ('openai', 'azure', 'bedrock', '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
 
     model_config = ConfigDict(
@@ -131,7 +134,10 @@ class EmbedderDetailedDto(BaseModel):
             "aws_secret_access_key": obj.get("aws_secret_access_key"),
             "aws_region_name": obj.get("aws_region_name"),
             "aws_inference_profile": obj.get("aws_inference_profile"),
+            "vertex_location": obj.get("vertex_location"),
+            "vertex_credentials": obj.get("vertex_credentials"),
             "batch_size": obj.get("batch_size"),
+            "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/embedder_update_dto.py

@@ -37,12 +37,15 @@ class EmbedderUpdateDto(BaseModel):
     aws_secret_access_key: Optional[StrictStr] = None
     aws_region_name: Optional[StrictStr] = None
     aws_inference_profile: Optional[StrictStr] = None
+    vertex_location: Optional[StrictStr] = None
+    vertex_credentials: Optional[StrictStr] = None
     batch_size: Optional[StrictInt] = None
+    skip_ssl_verification: Optional[StrictBool] = None
     display_name: Optional[StrictStr] = None
     engine: Optional[StrictStr] = None
     api_key: Optional[StrictStr] = None
     cost_input_per_million_token: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="costInputPerMillionToken")
-    __properties: ClassVar[List[str]] = ["modality", "provider", "endpoint_url", "description", "onpremise", "sensitivity_label", "api_version", "aws_access_key_id", "aws_secret_access_key", "aws_region_name", "aws_inference_profile", "batch_size", "display_name", "engine", "api_key", "costInputPerMillionToken"]
+    __properties: ClassVar[List[str]] = ["modality", "provider", "endpoint_url", "description", "onpremise", "sensitivity_label", "api_version", "aws_access_key_id", "aws_secret_access_key", "aws_region_name", "aws_inference_profile", "vertex_location", "vertex_credentials", "batch_size", "skip_ssl_verification", "display_name", "engine", "api_key", "costInputPerMillionToken"]
 
     @field_validator('modality')
     def modality_validate_enum(cls, value):
@@ -60,8 +63,8 @@ class EmbedderUpdateDto(BaseModel):
         if value is None:
             return value
 
-        if value not in set(['openai', 'azure', 'bedrock', 'ollama']):
-            raise ValueError("must be one of enum values ('openai', 'azure', 'bedrock', '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
 
     model_config = ConfigDict(
@@ -126,7 +129,10 @@ class EmbedderUpdateDto(BaseModel):
             "aws_secret_access_key": obj.get("aws_secret_access_key"),
             "aws_region_name": obj.get("aws_region_name"),
             "aws_inference_profile": obj.get("aws_inference_profile"),
+            "vertex_location": obj.get("vertex_location"),
+            "vertex_credentials": obj.get("vertex_credentials"),
             "batch_size": obj.get("batch_size"),
+            "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/{relation_model.py → embedding_endpoint_import_failure.py}

@@ -18,18 +18,17 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, StrictStr
-from typing import Any, ClassVar, Dict, List, Optional
+from typing import Any, ClassVar, Dict, List
 from typing import Optional, Set
 from typing_extensions import Self
 
-class RelationModel(BaseModel):
+class EmbeddingEndpointImportFailure(BaseModel):
     """
-    RelationModel
+    EmbeddingEndpointImportFailure
     """ # noqa: E501
     name: StrictStr
-    task_description: Optional[StrictStr] = None
-    prompt: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["name", "task_description", "prompt"]
+    reason: StrictStr
+    __properties: ClassVar[List[str]] = ["name", "reason"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -49,7 +48,7 @@ class RelationModel(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of RelationModel from a JSON string"""
+        """Create an instance of EmbeddingEndpointImportFailure from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -74,7 +73,7 @@ class RelationModel(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of RelationModel from a dict"""
+        """Create an instance of EmbeddingEndpointImportFailure from a dict"""
         if obj is None:
             return None
 
@@ -83,8 +82,7 @@ class RelationModel(BaseModel):
 
         _obj = cls.model_validate({
             "name": obj.get("name"),
-            "task_description": obj.get("task_description"),
-            "prompt": obj.get("prompt")
+            "reason": obj.get("reason")
         })
         return _obj
 

qanswer_sdk/models/embedding_endpoint_import_response.py

@@ -0,0 +1,99 @@
+# 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.embedding_endpoint_import_failure import EmbeddingEndpointImportFailure
+from typing import Optional, Set
+from typing_extensions import Self
+
+class EmbeddingEndpointImportResponse(BaseModel):
+    """
+    EmbeddingEndpointImportResponse
+    """ # noqa: E501
+    success: StrictInt
+    failed: StrictInt
+    fail_list: List[EmbeddingEndpointImportFailure]
+    __properties: ClassVar[List[str]] = ["success", "failed", "fail_list"]
+
+    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 EmbeddingEndpointImportResponse 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 fail_list (list)
+        _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 EmbeddingEndpointImportResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "success": obj.get("success"),
+            "failed": obj.get("failed"),
+            "fail_list": [EmbeddingEndpointImportFailure.from_dict(_item) for _item in obj["fail_list"]] if obj.get("fail_list") is not None else None
+        })
+        return _obj
+
+

qanswer_sdk/models/es_meta_data.py

@@ -0,0 +1,98 @@
+# 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, StrictBool, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ESMetaData(BaseModel):
+    """
+    ESMetaData
+    """ # noqa: E501
+    value: StrictStr
+    type: StrictStr
+    editable: StrictBool
+    __properties: ClassVar[List[str]] = ["value", "type", "editable"]
+
+    @field_validator('type')
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['keyword', 'text', 'long', 'date', 'freetext', 'boolean']):
+            raise ValueError("must be one of enum values ('keyword', 'text', 'long', 'date', 'freetext', 'boolean')")
+        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 ESMetaData 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 ESMetaData from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "value": obj.get("value"),
+            "type": obj.get("type"),
+            "editable": obj.get("editable")
+        })
+        return _obj
+
+

qanswer_sdk/models/{get_file_es_meta_data_response.py → es_meta_data_value.py}

@@ -22,14 +22,12 @@ from typing import Any, ClassVar, Dict, List
 from typing import Optional, Set
 from typing_extensions import Self
 
-class GetFileESMetaDataResponse(BaseModel):
+class ESMetaDataValue(BaseModel):
     """
-    GetFileESMetaDataResponse
+    ESMetaDataValue
     """ # noqa: E501
-    additional_fields: Dict[str, Dict[str, Any]]
-    file_id: StrictStr
-    metadata: Dict[str, Dict[str, Any]]
-    __properties: ClassVar[List[str]] = ["additional_fields", "file_id", "metadata"]
+    value: StrictStr
+    __properties: ClassVar[List[str]] = ["value"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -49,7 +47,7 @@ class GetFileESMetaDataResponse(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of GetFileESMetaDataResponse from a JSON string"""
+        """Create an instance of ESMetaDataValue from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -74,7 +72,7 @@ class GetFileESMetaDataResponse(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of GetFileESMetaDataResponse from a dict"""
+        """Create an instance of ESMetaDataValue from a dict"""
         if obj is None:
             return None
 
@@ -82,9 +80,7 @@ class GetFileESMetaDataResponse(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "additional_fields": obj.get("additional_fields"),
-            "file_id": obj.get("file_id"),
-            "metadata": obj.get("metadata")
+            "value": obj.get("value")
         })
         return _obj
 

qanswer_sdk/models/feedback_update.py

@@ -35,7 +35,8 @@ class FeedbackUpdate(BaseModel):
     modified_response: Optional[StrictStr] = None
     comment: Optional[StrictStr] = None
     approved: Optional[StrictBool] = None
-    __properties: ClassVar[List[str]] = ["username", "dataset", "message_id", "conversation_id", "rate", "modified_question", "modified_response", "comment", "approved"]
+    additional_fields: Optional[Dict[str, Any]] = None
+    __properties: ClassVar[List[str]] = ["username", "dataset", "message_id", "conversation_id", "rate", "modified_question", "modified_response", "comment", "approved", "additional_fields"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -96,7 +97,8 @@ class FeedbackUpdate(BaseModel):
             "modified_question": obj.get("modified_question"),
             "modified_response": obj.get("modified_response"),
             "comment": obj.get("comment"),
-            "approved": obj.get("approved")
+            "approved": obj.get("approved"),
+            "additional_fields": obj.get("additional_fields")
         })
         return _obj