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

qanswer_sdk/models/imap_search_response.py

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

qanswer_sdk/models/llm.py

@@ -0,0 +1,129 @@
+# coding: utf-8
+
+"""
+    QAnswer: Api Documentation
+
+    APIs provided by QAnswer
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictFloat, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional, Union
+from qanswer_sdk.models.logo import Logo
+from typing import Optional, Set
+from typing_extensions import Self
+
+class Llm(BaseModel):
+    """
+    Llm
+    """ # noqa: E501
+    id: Optional[StrictInt] = None
+    python_id: Optional[StrictInt] = Field(default=None, alias="pythonId")
+    name: Optional[StrictStr] = None
+    display_name: Optional[StrictStr] = Field(default=None, alias="displayName")
+    cost_input_token: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="costInputToken")
+    cost_output_token: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="costOutputToken")
+    sensitivity_level: Optional[StrictStr] = Field(default=None, alias="sensitivityLevel")
+    is_active: Optional[StrictBool] = Field(default=None, alias="isActive")
+    logo: Optional[Logo] = None
+    organization_llms: Optional[List[OrganizationLlm]] = Field(default=None, alias="organizationLlms")
+    user_chatbot_settings: Optional[List[UserChatbotSetting]] = Field(default=None, alias="userChatbotSettings")
+    __properties: ClassVar[List[str]] = ["id", "pythonId", "name", "displayName", "costInputToken", "costOutputToken", "sensitivityLevel", "isActive", "logo", "organizationLlms", "userChatbotSettings"]
+
+    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 Llm 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 logo
+        if self.logo:
+            _dict['logo'] = self.logo.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in organization_llms (list)
+        _items = []
+        if self.organization_llms:
+            for _item_organization_llms in self.organization_llms:
+                if _item_organization_llms:
+                    _items.append(_item_organization_llms.to_dict())
+            _dict['organizationLlms'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in user_chatbot_settings (list)
+        _items = []
+        if self.user_chatbot_settings:
+            for _item_user_chatbot_settings in self.user_chatbot_settings:
+                if _item_user_chatbot_settings:
+                    _items.append(_item_user_chatbot_settings.to_dict())
+            _dict['userChatbotSettings'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of Llm from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "id": obj.get("id"),
+            "pythonId": obj.get("pythonId"),
+            "name": obj.get("name"),
+            "displayName": obj.get("displayName"),
+            "costInputToken": obj.get("costInputToken"),
+            "costOutputToken": obj.get("costOutputToken"),
+            "sensitivityLevel": obj.get("sensitivityLevel"),
+            "isActive": obj.get("isActive"),
+            "logo": Logo.from_dict(obj["logo"]) if obj.get("logo") is not None else None,
+            "organizationLlms": [OrganizationLlm.from_dict(_item) for _item in obj["organizationLlms"]] if obj.get("organizationLlms") is not None else None,
+            "userChatbotSettings": [UserChatbotSetting.from_dict(_item) for _item in obj["userChatbotSettings"]] if obj.get("userChatbotSettings") is not None else None
+        })
+        return _obj
+
+from qanswer_sdk.models.organization_llm import OrganizationLlm
+from qanswer_sdk.models.user_chatbot_setting import UserChatbotSetting
+# TODO: Rewrite to not use raise_errors
+Llm.model_rebuild(raise_errors=False)
+

qanswer_sdk/models/llm_consumption.py

@@ -0,0 +1,118 @@
+# 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 datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, StrictFloat, StrictInt
+from typing import Any, ClassVar, Dict, List, Optional, Union
+from qanswer_sdk.models.llm import Llm
+from qanswer_sdk.models.user import User
+from typing import Optional, Set
+from typing_extensions import Self
+
+class LlmConsumption(BaseModel):
+    """
+    LlmConsumption
+    """ # noqa: E501
+    id: Optional[StrictInt] = None
+    created: Optional[datetime] = None
+    user: Optional[User] = None
+    dataset: Optional[UserDataset] = None
+    llm: Optional[Llm] = None
+    input_tokens: Optional[StrictInt] = Field(default=None, alias="inputTokens")
+    input_cost: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="inputCost")
+    output_tokens: Optional[StrictInt] = Field(default=None, alias="outputTokens")
+    output_cost: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="outputCost")
+    __properties: ClassVar[List[str]] = ["id", "created", "user", "dataset", "llm", "inputTokens", "inputCost", "outputTokens", "outputCost"]
+
+    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 LlmConsumption 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 user
+        if self.user:
+            _dict['user'] = self.user.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of dataset
+        if self.dataset:
+            _dict['dataset'] = self.dataset.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of llm
+        if self.llm:
+            _dict['llm'] = self.llm.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of LlmConsumption from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "id": obj.get("id"),
+            "created": obj.get("created"),
+            "user": User.from_dict(obj["user"]) if obj.get("user") is not None else None,
+            "dataset": UserDataset.from_dict(obj["dataset"]) if obj.get("dataset") is not None else None,
+            "llm": Llm.from_dict(obj["llm"]) if obj.get("llm") is not None else None,
+            "inputTokens": obj.get("inputTokens"),
+            "inputCost": obj.get("inputCost"),
+            "outputTokens": obj.get("outputTokens"),
+            "outputCost": obj.get("outputCost")
+        })
+        return _obj
+
+from qanswer_sdk.models.user_dataset import UserDataset
+# TODO: Rewrite to not use raise_errors
+LlmConsumption.model_rebuild(raise_errors=False)
+

qanswer_sdk/models/{pageable.py → llm_context_ranges.py}

@@ -17,20 +17,20 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from pydantic import BaseModel, ConfigDict, StrictInt
 from typing import Any, ClassVar, Dict, List, Optional
-from typing_extensions import Annotated
 from typing import Optional, Set
 from typing_extensions import Self
 
-class Pageable(BaseModel):
+class LLMContextRanges(BaseModel):
     """
-    Pageable
+    LLMContextRanges
     """ # noqa: E501
-    page: Optional[Annotated[int, Field(strict=True, ge=0)]] = None
-    size: Optional[Annotated[int, Field(strict=True, ge=1)]] = None
-    sort: Optional[List[StrictStr]] = None
-    __properties: ClassVar[List[str]] = ["page", "size", "sort"]
+    min_context_window: Optional[StrictInt] = None
+    max_context_window: Optional[StrictInt] = None
+    min_max_tokens: Optional[StrictInt] = None
+    max_max_tokens: Optional[StrictInt] = None
+    __properties: ClassVar[List[str]] = ["min_context_window", "max_context_window", "min_max_tokens", "max_max_tokens"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -50,7 +50,7 @@ class Pageable(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of Pageable from a JSON string"""
+        """Create an instance of LLMContextRanges from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -75,7 +75,7 @@ class Pageable(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of Pageable from a dict"""
+        """Create an instance of LLMContextRanges from a dict"""
         if obj is None:
             return None
 
@@ -83,9 +83,10 @@ class Pageable(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "page": obj.get("page"),
-            "size": obj.get("size"),
-            "sort": obj.get("sort")
+            "min_context_window": obj.get("min_context_window"),
+            "max_context_window": obj.get("max_context_window"),
+            "min_max_tokens": obj.get("min_max_tokens"),
+            "max_max_tokens": obj.get("max_max_tokens")
         })
         return _obj
 

qanswer_sdk/models/{llm_cost.py → llm_cost_filter.py}

@@ -23,13 +23,15 @@ from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
-class LlmCost(BaseModel):
+class LlmCostFilter(BaseModel):
     """
-    LlmCost
+    LlmCostFilter
     """ # noqa: E501
     user_id: Optional[StrictInt] = Field(default=None, alias="userId")
     ai_assistant_id: Optional[StrictInt] = Field(default=None, alias="aiAssistantId")
-    llm: Optional[StrictStr] = None
+    organization_id: Optional[StrictInt] = Field(default=None, alias="organizationId")
+    team_id: Optional[StrictInt] = Field(default=None, alias="teamId")
+    llm_id: Optional[StrictInt] = Field(default=None, alias="llmId")
     start_date: Optional[datetime] = Field(default=None, alias="startDate")
     end_date: Optional[datetime] = Field(default=None, alias="endDate")
     group_by: Optional[StrictStr] = Field(default=None, alias="groupBy")
@@ -38,7 +40,7 @@ class LlmCost(BaseModel):
     sort_by: Optional[StrictStr] = Field(default=None, alias="sortBy")
     sort_dir: Optional[StrictStr] = Field(default=None, alias="sortDir")
     order: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["userId", "aiAssistantId", "llm", "startDate", "endDate", "groupBy", "page", "size", "sortBy", "sortDir", "order"]
+    __properties: ClassVar[List[str]] = ["userId", "aiAssistantId", "organizationId", "teamId", "llmId", "startDate", "endDate", "groupBy", "page", "size", "sortBy", "sortDir", "order"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -58,7 +60,7 @@ class LlmCost(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of LlmCost from a JSON string"""
+        """Create an instance of LlmCostFilter from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -83,7 +85,7 @@ class LlmCost(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of LlmCost from a dict"""
+        """Create an instance of LlmCostFilter from a dict"""
         if obj is None:
             return None
 
@@ -93,7 +95,9 @@ class LlmCost(BaseModel):
         _obj = cls.model_validate({
             "userId": obj.get("userId"),
             "aiAssistantId": obj.get("aiAssistantId"),
-            "llm": obj.get("llm"),
+            "organizationId": obj.get("organizationId"),
+            "teamId": obj.get("teamId"),
+            "llmId": obj.get("llmId"),
             "startDate": obj.get("startDate"),
             "endDate": obj.get("endDate"),
             "groupBy": obj.get("groupBy"),

qanswer_sdk/models/llm_cost_list.py

@@ -0,0 +1,101 @@
+# coding: utf-8
+
+"""
+    QAnswer: Api Documentation
+
+    APIs provided by QAnswer
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictInt
+from typing import Any, ClassVar, Dict, List, Optional
+from qanswer_sdk.models.cost_summary_dto import CostSummaryDto
+from typing import Optional, Set
+from typing_extensions import Self
+
+class LlmCostList(BaseModel):
+    """
+    LlmCostList
+    """ # noqa: E501
+    cost_summaries: Optional[List[CostSummaryDto]] = Field(default=None, alias="costSummaries")
+    page: Optional[StrictInt] = None
+    page_size: Optional[StrictInt] = Field(default=None, alias="pageSize")
+    total: Optional[StrictInt] = None
+    __properties: ClassVar[List[str]] = ["costSummaries", "page", "pageSize", "total"]
+
+    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 LlmCostList 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 cost_summaries (list)
+        _items = []
+        if self.cost_summaries:
+            for _item_cost_summaries in self.cost_summaries:
+                if _item_cost_summaries:
+                    _items.append(_item_cost_summaries.to_dict())
+            _dict['costSummaries'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of LlmCostList from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "costSummaries": [CostSummaryDto.from_dict(_item) for _item in obj["costSummaries"]] if obj.get("costSummaries") is not None else None,
+            "page": obj.get("page"),
+            "pageSize": obj.get("pageSize"),
+            "total": obj.get("total")
+        })
+        return _obj
+
+