kash-shell 0.3.24__py3-none-any.whl → 0.3.25__py3-none-any.whl

kash/embeddings/embeddings.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import ast
 from collections.abc import Iterable
 from pathlib import Path
-from typing import TYPE_CHECKING, TypeAlias, cast
+from typing import TYPE_CHECKING, TypeAlias
 
 from pydantic.dataclasses import dataclass
 from strif import abbrev_list
@@ -65,7 +65,6 @@ class Embeddings:
     @classmethod
     def embed(cls, keyvals: list[KeyVal], model=DEFAULT_EMBEDDING_MODEL) -> Embeddings:
         from litellm import embedding
-        from litellm.types.utils import EmbeddingResponse
 
         init_litellm()
 
@@ -82,9 +81,8 @@ class Embeddings:
             keys = [kv[0] for kv in batch]
             texts = [kv[1] for kv in batch]
 
-            response: EmbeddingResponse = cast(
-                EmbeddingResponse, embedding(model=model.litellm_name, input=texts)
-            )
+            response = embedding(model=model.litellm_name, input=texts)
+
             if not response.data:
                 raise ValueError("No embedding response data")
 

kash/embeddings/text_similarity.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, cast
+from typing import TYPE_CHECKING
 
 from funlog import log_calls
 
@@ -24,12 +24,9 @@ def cosine_relatedness(x: ArrayLike, y: ArrayLike) -> float:
 def embed_query(model: EmbeddingModel, query: str) -> EmbeddingResponse:
     import litellm
     from litellm import embedding
-    from litellm.types.utils import EmbeddingResponse
 
     try:
-        response: EmbeddingResponse = cast(
-            EmbeddingResponse, embedding(model=model.litellm_name, input=[query])
-        )
+        response = embedding(model=model.litellm_name, input=[query])
     except litellm.exceptions.APIError as e:
         log.info("API error embedding query: %s", e)
         raise ApiResultError(str(e))

kash/help/help_embeddings.py

@@ -3,6 +3,8 @@ from __future__ import annotations
 from dataclasses import dataclass, field
 from pathlib import Path
 
+from typing_extensions import override
+
 from kash.config.logger import get_logger
 from kash.embeddings.embeddings import Embeddings
 from kash.embeddings.text_similarity import rank_by_relatedness
@@ -19,6 +21,7 @@ class DocKey:
     doc_type: HelpDocType
     index: int
 
+    @override
     def __str__(self) -> str:
         return f"{self.doc_type.value}:{self.index}"
 

kash/llm_utils/llm_features.py

@@ -68,5 +68,5 @@ preferred_llms: list[LLMName] = [
     LLM.claude_4_opus,
     LLM.claude_3_7_sonnet,
     LLM.claude_3_5_haiku,
-    LLM.gemini_2_5_pro_preview_05_06,
+    LLM.gemini_2_5_pro,
 ]

kash/llm_utils/llms.py

@@ -15,6 +15,7 @@ class LLM(LLMName, Enum):
     # https://platform.openai.com/docs/models
     o4_mini = LLMName("o4-mini")
     o3 = LLMName("o3")
+    o3_pro = LLMName("o3-pro")
     o3_mini = LLMName("o3-mini")
     o1 = LLMName("o1")
     o1_mini = LLMName("o1-mini")
@@ -35,13 +36,9 @@ class LLM(LLMName, Enum):
     claude_3_5_haiku = LLMName("claude-3-5-haiku-latest")
 
     # https://ai.google.dev/gemini-api/docs/models
-    gemini_2_5_pro_preview_06_05 = LLMName("gemini/gemini-2.5-pro-preview-06-05")
-    gemini_2_5_pro_preview_05_06 = LLMName("gemini/gemini-2.5-pro-preview-05-06")
-    gemini_2_5_pro_preview_03_25 = LLMName("gemini/gemini-2.5-pro-preview-03-25")
-    gemini_2_5_flash_preview = LLMName("gemini-2.5-flash-preview-05-20")
-    gemini_2_0_flash = LLMName("gemini/gemini-2_0-flash")
-    gemini_2_0_flash_lite = LLMName("gemini/gemini-2.0-flash-lite")
-    gemini_2_0_pro_exp_02_05 = LLMName("gemini/gemini-2.0-pro-exp-02-05")
+    gemini_2_5_pro = LLMName("gemini/gemini-2.5-pro")
+    gemini_2_5_flash = LLMName("gemini/gemini-2.5-flash")
+    gemini_2_5_flash_lite = LLMName("gemini-2.5-flash-lite-preview-06-17")
 
     # https://docs.x.ai/docs/models
     xai_grok_3 = LLMName("xai/grok-3")
@@ -56,6 +53,7 @@ class LLM(LLMName, Enum):
     deepseek_reasoner = LLMName("deepseek/deepseek-reasoner")
 
     # https://console.groq.com/docs/models
+    groq_gemma2_9b_it = LLMName("groq/gemma2-9b-it")
     groq_llama_3_1_8b_instant = LLMName("groq/llama-3.1-8b-instant")
     groq_llama_3_3_70b_versatile = LLMName("groq/llama-3.3-70b-versatile")
     groq_deepseek_r1_distill_llama_70b = LLMName("groq/deepseek-r1-distill-llama-70b")

kash/model/params_model.py

@@ -209,7 +209,7 @@ A list of parameter declarations, possibly with default values.
 DEFAULT_CAREFUL_LLM = LLM.o3
 DEFAULT_STRUCTURED_LLM = LLM.gpt_4o
 DEFAULT_STANDARD_LLM = LLM.claude_4_sonnet
-DEFAULT_FAST_LLM = LLM.o1_mini
+DEFAULT_FAST_LLM = LLM.gpt_4o
 
 
 # Parameters set globally such as in the workspace.

kash/utils/text_handling/doc_normalization.py

@@ -21,6 +21,7 @@ def normalize_formatting(
     format: Format | None,
     support_ansi: bool = True,
     cleanups: bool = True,
+    smartquotes: bool = True,
 ) -> str:
     """
     Normalize formatting. Currently only normalizes Markdown and leaves plaintext
@@ -35,6 +36,7 @@ def normalize_formatting(
             text,
             line_wrapper=line_wrap_by_sentence(len_fn=len_fn, is_markdown=True),
             cleanups=cleanups,
+            smartquotes=smartquotes,
         )
     elif format == Format.plaintext:
         # Consider plaintext a raw format and don't normalize.

kash/web_gen/templates/base_styles.css.jinja

@@ -22,6 +22,7 @@
   {% endblock root_variables %}
 }
 
+/* CSS color definitions. */
 {{ color_defs|safe }}
 
 {% block selection_styles %}
@@ -145,7 +146,7 @@ h2 + h3 {
 }
 
 h3 {
-  font-size: 1.18rem;
+  font-size: 1.15rem;
   margin-top: 1.4rem;
   margin-bottom: 0.7rem;
 }
@@ -662,6 +663,12 @@ sup {
     max-width: none;
   }
   
+  /* Smaller table text on mobile. */
+  table code,
+  table pre {
+    font-size: var(--font-size-mono-small);
+  }
+  
   ul, ol {
     margin-left: 1rem;
   }

{kash_shell-0.3.24.dist-info → kash_shell-0.3.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kash-shell
-Version: 0.3.24
+Version: 0.3.25
 Summary: The knowledge agent shell (core)
 Project-URL: Repository, https://github.com/jlevy/kash-shell
 Author-email: Joshua Levy <joshua@cal.berkeley.edu>
@@ -28,7 +28,7 @@ Requires-Dist: curl-cffi>=0.11.4
 Requires-Dist: deepgram-sdk>=3.10.1
 Requires-Dist: dunamai>=1.23.0
 Requires-Dist: fastapi>=0.115.11
-Requires-Dist: flowmark>=0.4.6
+Requires-Dist: flowmark>=0.4.8
 Requires-Dist: frontmatter-format>=0.2.1
 Requires-Dist: funlog>=0.2.0
 Requires-Dist: httpx[brotli]>=0.28.1

{kash_shell-0.3.24.dist-info → kash_shell-0.3.25.dist-info}/RECORD

@@ -81,8 +81,8 @@ kash/docs_base/recipes/general_system_commands.sh,sha256=rFNPuLj3Md09L5yzWx6ILja
 kash/docs_base/recipes/python_dev_commands.sh,sha256=9vJsQiDZKJ7ShokFnzc2Jsto6n87MNRkbgcc2Ee9Kro,179
 kash/docs_base/recipes/tldr_standard_commands.sh,sha256=7nPES55aT45HF3eDhQRrEUiWRpPdvvp40Sg88uADa80,60491
 kash/embeddings/cosine.py,sha256=QTWPWUHivXjxCM6APSqij_-4mywM2BVVm0xb0hu7FHA,1587
-kash/embeddings/embeddings.py,sha256=l1mDCTi296_hP_jvY8lJ8-WTBq43FWS4jy3kNvKJy44,4448
-kash/embeddings/text_similarity.py,sha256=BOo9Vcs5oi2Zs5La56uTkPMHo65XSd4qz_yr6GTfUA4,1924
+kash/embeddings/embeddings.py,sha256=q0f_yGAN8Hkx32t_k6smVSVWogbJFbTEEKay9uzUt4o,4311
+kash/embeddings/text_similarity.py,sha256=HgoDffF0vuecjKN-l1MxHwvfJBYJORemmyO0WVQkyBI,1798
 kash/exec/__init__.py,sha256=Najls8No143yoj_KAaOQgo8ufC2LWCB_DwwEQ-8nDM0,1277
 kash/exec/action_decorators.py,sha256=kCEqQFN1MIhRbFeIEGux956LzsiwonhyIIrJ8Pq9zPk,16765
 kash/exec/action_exec.py,sha256=tLL3ESIuxqS_gSOD4eGRYLuTzesRMTyKb_Uy3i_28VU,19022
@@ -117,7 +117,7 @@ kash/help/assistant.py,sha256=R0XHNi-h51QoQ7rGouD2chrDPGomYaPQUqJdvpjlCs8,11535
 kash/help/assistant_instructions.py,sha256=jW5XAsmLx8YZMKSDJgWnqo9Vwe7VuiTURQHjKBqr_L8,2549
 kash/help/assistant_output.py,sha256=9sM-OVLc6eMSOkxyovB88dNlsknFpf8Wz89Zp5PuEA8,1668
 kash/help/function_param_info.py,sha256=yGuFLVZoDF1E1YsXqGwJhuY2uJB3R1B1EC77MEMSc-U,1700
-kash/help/help_embeddings.py,sha256=le7yqxGOtzIU_LUkl-Ef-7WQGuWa0wAOx9umZGva8zg,2787
+kash/help/help_embeddings.py,sha256=lcUv_Y402gHoJiH30Z85KpaezwARtN-eM3mvA6TtGK4,2841
 kash/help/help_lookups.py,sha256=0dtuLWEXncqhJCijC98IA9stBDNNcJewt1JYqMLkTx4,2029
 kash/help/help_pages.py,sha256=TaKsE26R-pZTrK4Pa593DK5osdJodFHaVm5pZpjqgaI,3894
 kash/help/help_printing.py,sha256=eZbZdyJC158JiXcEk2zvUmqYbYzbYOpHvxEhC1kIN-Q,6086
@@ -131,10 +131,10 @@ kash/llm_utils/fuzzy_parsing.py,sha256=bbG2Y7i5w6kxAVPAixyluv3MDS2hW_pkhnJpVOLHZ
 kash/llm_utils/init_litellm.py,sha256=5Fn9uW4P7lfEO8Rk1EJJUzDEGNjw-PDvxFgHlKDf-Ok,409
 kash/llm_utils/llm_api_keys.py,sha256=nTB9wSFfHTOXvqshSQCQGCPxUwOW1U7oslngya8nHkw,1168
 kash/llm_utils/llm_completion.py,sha256=SzeWGRrsjuN1TXdPwscYG6whLQkHclITtwTvQK19GE0,5436
-kash/llm_utils/llm_features.py,sha256=tgcz8qQByZ80noybXS2144K9eCw1iOFxPYkCK2kJJOw,1861
+kash/llm_utils/llm_features.py,sha256=ZpQhHiRDtF1BY8HGxTQH0dIeM3rux7a9CvJ0JO2KAlI,1847
 kash/llm_utils/llm_messages.py,sha256=70QwIIvdwo-h4Jfn_6MbEHb3LTUjUmzg_v_dU_Ey__g,1174
 kash/llm_utils/llm_names.py,sha256=VZbdKnoeBx_luB5YQ-Rz37gMt3_FcueJdp40ZaQbpUA,3620
-kash/llm_utils/llms.py,sha256=Dhwb3DIk9xoSGxFJAegIwD7xq32-Gfw83yPniIFNt84,4036
+kash/llm_utils/llms.py,sha256=Sm7SdNVE0pVIOpAggIgGp9s_Waw8vq6hNOCu-GTc6GI,3788
 kash/local_server/__init__.py,sha256=AyNpvCOJlQF6A4DnlYKRMbbfRNzdizEA-ytp-F2SLZU,162
 kash/local_server/local_server.py,sha256=EugjL30VM0pWdZDsiQxU-o6EdEa082qlGd_7RHvI5tk,5863
 kash/local_server/local_server_commands.py,sha256=ZMp1DpYgg-MJ0iqH0DHfhWKDpFqNRG_txkRdODIr9mU,1661
@@ -171,7 +171,7 @@ kash/model/language_list.py,sha256=I3RIbxTseVmPdhExQimimEv18Gmy2ImMbpXe0-_t1Qw,4
 kash/model/llm_actions_model.py,sha256=a29uXVNfS2CiqvM7HPdC6H9A23rSQQihAideuBLMH8g,2110
 kash/model/media_model.py,sha256=ZnlZ-FkswbAIGpUAuNqLce1WDZK-WbnwHn2ipg8x7-0,3511
 kash/model/operations_model.py,sha256=WmU-xeWGsqMLVN369dQEyVGU8T7G_KyLLsj6YFc5sVw,6517
-kash/model/params_model.py,sha256=s2pyM4AcWdis3CAQGEFI97EA-NjG5UxTJgeyuHLXp88,15084
+kash/model/params_model.py,sha256=vSFMjWfksysEhAOQ2Ge2ZEvRqgUmIL7febxcI3-7Smg,15083
 kash/model/paths_model.py,sha256=KDFm7wan7hjObHbnV2rR8-jsyLTVqbKcwFdKeLFRtdM,15889
 kash/model/preconditions_model.py,sha256=-IfsVR0NkQhq_3hUTXzK2bFYAd--3YjSwUiDKHVQQqk,2887
 kash/shell/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -242,7 +242,7 @@ kash/utils/rich_custom/multitask_status.py,sha256=ViNx35akEaV-fihef57_b3l8FnLytN
 kash/utils/rich_custom/rich_char_transform.py,sha256=3M89tViKM0y31VHsDoHi5eHFWlv5ME7F4p35IdDxnrw,2616
 kash/utils/rich_custom/rich_indent.py,sha256=nz72yNpUuYjOsaPNVmxM81oEQm-GKEfQkNsuWmv16G0,2286
 kash/utils/rich_custom/rich_markdown_fork.py,sha256=M_JRaSAyHrSg-wuLv9C9P7SkehSim3lwkqQPuMIFkVw,26551
-kash/utils/text_handling/doc_normalization.py,sha256=C211eSof8PUDVCqQtShuC4AMJpTZeBK8GHlGATp3c9E,2976
+kash/utils/text_handling/doc_normalization.py,sha256=UtVfaSHE5k_n7BzTbOkc5FUJtXiTbaw9QUl03AnCvLs,3043
 kash/utils/text_handling/escape_html_tags.py,sha256=8pC3JgoKRtdnbnOu8DiWrlvNR6GAqjwhGbQgl3jiFG4,6441
 kash/utils/text_handling/markdown_render.py,sha256=LHPdJc__2ejBx7iwkp_P9wIePNmiVSgwu4-uhamVjms,3791
 kash/utils/text_handling/markdown_utils.py,sha256=tAPMZulEcJOmWQHnZVKRqstThmsLB0mm7Fey5DHMFvc,31985
@@ -262,7 +262,7 @@ kash/web_gen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kash/web_gen/simple_webpage.py,sha256=ks_0ljxCeS2-gAAEaUc1JEnzY3JY0nzqGFiyyqyRuZs,1537
 kash/web_gen/tabbed_webpage.py,sha256=e0GGG1bQ4CQegpJgOIT2KpyM6cmwN_BQ9eJSsi4BjgY,4945
 kash/web_gen/template_render.py,sha256=aypo6UanouftV4RpxgNm6JdquelI52fV0IlihdA3yjE,1908
-kash/web_gen/templates/base_styles.css.jinja,sha256=kLFEntQ7SHkbG9BEJTLtsiJCP5ldflILMXHAZGKjaF0,14942
+kash/web_gen/templates/base_styles.css.jinja,sha256=COMDe2UVJ8Uk2Jnce697EtRq8SC5-TeXyhQjrMksRxE,15088
 kash/web_gen/templates/base_webpage.html.jinja,sha256=gwxXMSC_eY-stu8uT5AVyQJ-Ppy2GyWwq9vEdTEbVi0,12887
 kash/web_gen/templates/content_styles.css.jinja,sha256=qwMKnjDRdjXxNfUgSrZEAkliHSLZ9OMsWGulonp-1Zs,3764
 kash/web_gen/templates/explain_view.html.jinja,sha256=DNw5Iw5SrhIUFRGB4qNvfcKXsBHVbEJVURGdhvyC75Q,949
@@ -294,8 +294,8 @@ kash/xonsh_custom/xonsh_modern_tools.py,sha256=mj_b34LZXfE8MJe9EpDmp5JZ0tDM1biYN
 kash/xonsh_custom/xonsh_ranking_completer.py,sha256=ZRGiAfoEgqgnlq2-ReUVEaX5oOgW1DQ9WxIv2OJLuTo,5620
 kash/xontrib/fnm.py,sha256=V2tsOdmIDgbFbZSfMLpsvDIwwJJqiYnOkOySD1cXNXw,3700
 kash/xontrib/kash_extension.py,sha256=FLIMlgR3C_6A1fwKE-Ul0nmmpJSszVPbAriinUyQ8Zg,1896
-kash_shell-0.3.24.dist-info/METADATA,sha256=zgoOSgrSaBs5rmbe9RaoUHai2XJTLc3YFLTdM73MwWU,32726
-kash_shell-0.3.24.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-kash_shell-0.3.24.dist-info/entry_points.txt,sha256=SQraWDAo8SqYpthLXThei0mf_hGGyhYBUO-Er_0HcwI,85
-kash_shell-0.3.24.dist-info/licenses/LICENSE,sha256=rCh2PsfYeiU6FK_0wb58kHGm_Fj5c43fdcHEexiVzIo,34562
-kash_shell-0.3.24.dist-info/RECORD,,
+kash_shell-0.3.25.dist-info/METADATA,sha256=vxaqN6N2fP79pTjdzslq61uL-DXCs2nrJMuPt014n4U,32726
+kash_shell-0.3.25.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kash_shell-0.3.25.dist-info/entry_points.txt,sha256=SQraWDAo8SqYpthLXThei0mf_hGGyhYBUO-Er_0HcwI,85
+kash_shell-0.3.25.dist-info/licenses/LICENSE,sha256=rCh2PsfYeiU6FK_0wb58kHGm_Fj5c43fdcHEexiVzIo,34562
+kash_shell-0.3.25.dist-info/RECORD,,