tool_call_models 0.1.2__tar.gz

tool_call_models-0.1.2/PKG-INFO

@@ -0,0 +1,17 @@
+Metadata-Version: 2.3
+Name: tool_call_models
+Version: 0.1.2
+Summary: A simple package of models used between two services in order to achieve SDUI widgets using LLMs.
+License: MIT
+Author: Aslon Khamidov
+Author-email: hamidovaslon1@gmail.com
+Requires-Python: >=3.11
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: pydantic
+Description-Content-Type: text/markdown
+
+

tool_call_models-0.1.2/pyproject.toml

@@ -0,0 +1,17 @@
+[project]
+name = "tool_call_models"
+version = "0.1.2"
+description = "A simple package of models used between two services in order to achieve SDUI widgets using LLMs."
+authors = [{ name="Aslon Khamidov", email="hamidovaslon1@gmail.com" }]
+readme = "README.md"
+license = "MIT"
+requires-python = ">=3.11"
+
+
+
+dependencies = ["pydantic"]
+
+
+[build-system]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
+build-backend = "poetry.core.masonry.api"

tool_call_models-0.1.2/tool_call_models/__init__.py

@@ -0,0 +1,28 @@
+from .base import BaseToolCallModel
+from .home_balance import HomeBalance, HomeBalanceDetails
+from .paynet import (
+    GetSupplierByCategoryOptions,
+    GetSupplierByCategoryPayload,
+    GetSupplierByCategoryResponse,
+)
+from .smartbazar import (
+    Brand,
+    MainCategory,
+    Merchant,
+    Offer,
+    ProductItem,
+)
+
+__all__ = [
+    "BaseToolCallModel",
+    "HomeBalance",
+    "HomeBalanceDetails",
+    "GetSupplierByCategoryOptions",
+    "GetSupplierByCategoryPayload",
+    "GetSupplierByCategoryResponse",
+    "Brand",
+    "MainCategory",
+    "Merchant",
+    "Offer",
+    "ProductItem",
+]

tool_call_models-0.1.2/tool_call_models/base.py

@@ -0,0 +1,11 @@
+from abc import ABC, abstractmethod
+from typing import Any, Dict, Optional
+
+
+class BaseToolCallModel(ABC):
+    """Base class for all models in the application."""
+
+    @abstractmethod
+    def filter_for_llm(self) -> Dict[str, Any]:
+        """Filter model for LLM."""
+        pass

tool_call_models-0.1.2/tool_call_models/home_balance.py

@@ -0,0 +1,24 @@
+from pydantic import BaseModel, Field, model_validator
+from typing import Dict
+
+
+class HomeBalanceDetails(BaseModel):
+    balance: int
+    # details: Dict[str, int]
+
+    def __init__(self, **data):
+        if "balance" in data:
+            data["balance"] = data["balance"] // 100
+        super().__init__(**data)
+
+
+class HomeBalance(BaseModel):
+    homeName: str
+    services: Dict[str, HomeBalanceDetails] = Field(default_factory=dict)
+
+    @model_validator(mode="before")
+    def extract_services(cls, values):
+        known_keys = {"homeName"}
+        services = {k: v for k, v in values.items() if k not in known_keys}
+        values["services"] = services
+        return values

tool_call_models-0.1.2/tool_call_models/paynet.py

@@ -0,0 +1,20 @@
+from pydantic import BaseModel, Field
+from typing import List
+
+
+class GetSupplierByCategoryOptions(BaseModel):
+    idx: int = Field(alias="id")
+    categoryId: int = Field(alias="categoryId")
+    name: str
+
+
+class GetSupplierByCategoryPayload(BaseModel):
+    idx: int = Field(alias="id", default=None)
+    value: List[GetSupplierByCategoryOptions] = None
+    categoryId: int = Field(alias="categoryId")
+    name: str
+
+
+class GetSupplierByCategoryResponse(BaseModel):
+    description: str
+    payload: List[GetSupplierByCategoryPayload]

tool_call_models-0.1.2/tool_call_models/smartbazar.py

@@ -0,0 +1,37 @@
+from typing import Optional
+from pydantic import BaseModel, Field
+
+
+class Brand(BaseModel):
+    # id: Optional[int] = None
+    name: Optional[str] = None
+
+
+class MainCategory(BaseModel):
+    id: Optional[int] = None
+    name: Optional[str] = None
+
+
+class Merchant(BaseModel):
+    # id: Optional[int] = None
+    name: Optional[str] = None
+
+
+class Offer(BaseModel):
+    original_price: Optional[int] = None
+    price: Optional[int] = None
+    three_month_price: Optional[int] = Field(default=None, alias="3_month_price")
+    six_month_price: Optional[int] = Field(default=None, alias="6_month_price")
+    nine_month_price: Optional[int] = Field(default=None, alias="9_month_price")
+    twelve_month_price: Optional[int] = Field(default=None, alias="12_month_price")
+    eighteen_month_price: Optional[int] = Field(default=None, alias="18_month_price")
+    merchant: Optional[Merchant] = None
+
+
+class ProductItem(BaseModel):
+    # id: Optional[int] = None
+    name_ru: Optional[str] = None
+    brand: Optional[Brand] = None
+    # main_categories: Optional[list[MainCategory]] = None
+
+    offers: Optional[list[Offer]] = None