kash-shell 0.3.28__py3-none-any.whl → 0.3.33__py3-none-any.whl

kash/llm_utils/llm_features.py

@@ -1,72 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass
-from typing import Literal, TypeAlias
-
-from prettyfmt import custom_key_sort
-
-from kash.llm_utils.llm_names import LLMName
-from kash.llm_utils.llms import LLM
-
-Speed: TypeAlias = Literal["fast", "medium", "slow"]
-
-ContextSize: TypeAlias = Literal["small", "medium", "large"]
-
-ModelSize: TypeAlias = Literal["small", "medium", "large"]
-
-
-@dataclass(frozen=True)
-class LLMFeatures:
-    speed: Speed | None = None
-    context_size: ContextSize | None = None
-    model_size: ModelSize | None = None
-    structured_output: bool | None = None
-    thinking: bool = False
-
-    def satisfies(self, features: LLMFeatures) -> bool:
-        return all(
-            getattr(self, attr) == getattr(features, attr)
-            for attr in features.__dataclass_fields__
-            if getattr(self, attr) is not None
-        )
-
-
-def pick_llm(desired_features: LLMFeatures) -> LLMName:
-    """
-    Pick the preferred model that satisfies the desired features.
-    """
-    satisfied_models: list[LLMName] = [
-        llm for llm, features in FEATURES.items() if features.satisfies(desired_features)
-    ]
-    satisfied_models.sort(key=custom_key_sort(preferred_llms))
-    if not satisfied_models:
-        raise ValueError(f"No model found for features: {desired_features}")
-    return satisfied_models[0]
-
-
-FEATURES = {
-    LLM.o3_mini: LLMFeatures(
-        speed="fast",
-        context_size="small",
-        model_size="small",
-        structured_output=True,
-        thinking=True,
-    ),
-    # FIXME
-}
-
-preferred_llms: list[LLMName] = [
-    LLM.o4_mini,
-    LLM.o3,
-    LLM.o3_mini,
-    LLM.o1_mini,
-    LLM.o1,
-    LLM.gpt_4o_mini,
-    LLM.gpt_4o,
-    LLM.gpt_4,
-    LLM.claude_4_sonnet,
-    LLM.claude_4_opus,
-    LLM.claude_3_7_sonnet,
-    LLM.claude_3_5_haiku,
-    LLM.gemini_2_5_pro,
-]

kash/web_gen/simple_webpage.py

@@ -1,55 +0,0 @@
-from kash.model.items_model import Item
-from kash.utils.file_utils.file_formats_model import Format
-from kash.web_gen.template_render import render_web_template
-
-
-def simple_webpage_render(
-    item: Item,
-    page_template: str = "simple_webpage.html.jinja",
-    add_title_h1: bool = True,
-    show_theme_toggle: bool = False,
-) -> str:
-    """
-    Generate a simple web page from a single item.
-    If `add_title_h1` is True, the title will be inserted as an h1 heading above the body.
-    """
-    return render_web_template(
-        template_filename=page_template,
-        data={
-            "title": item.pick_title(),
-            "add_title_h1": add_title_h1,
-            "content_html": item.body_as_html(),
-            "thumbnail_url": item.thumbnail_url,
-            "enable_themes": show_theme_toggle,
-            "show_theme_toggle": show_theme_toggle,
-        },
-    )
-
-
-## Tests
-
-
-def test_render():
-    import os
-
-    from kash.model.items_model import ItemType
-
-    # Create a test item
-    item = Item(
-        type=ItemType.doc,
-        format=Format.html,
-        title="A Simple Web Page",
-        body="<p>This is a simple web page with <b>HTML content</b>.</p>",
-    )
-
-    # Generate HTML
-    html = simple_webpage_render(item)
-
-    os.makedirs("tmp", exist_ok=True)
-    with open("tmp/simple_webpage.html", "w") as f:
-        f.write(html)
-    print("Rendered simple webpage to tmp/simple_webpage.html")
-
-    # Basic validation
-    assert item.title and item.title in html
-    assert "<b>HTML content</b>" in html