python-fastllm 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

fastllm/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.1"
+__version__ = "0.0.3"

fastllm/chat.py

@@ -3,8 +3,7 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/07_chat.ipynb.
 
 # %% auto #0
-__all__ = ['haik45', 'sonn45', 'sonn', 'sonn46', 'opus46', 'opus', 'gpt54', 'gpt54m', 'codex54', 'codex55', 'tool_dtls_tag',
-           're_tools', 'token_dtls_tag', 're_token', 'effort', 'remove_cache_ckpts', 'contents', 'stop_reason',
+__all__ = ['tool_dtls_tag', 're_tools', 'token_dtls_tag', 're_token', 'effort', 'remove_cache_ckpts', 'contents', 'stop_reason',
            'mk_msg', 'split_tools', 'fmt2hist', 'mk_msgs', 'cite_footnote', 'postproc', 'lite_mk_func', 'ToolResponse',
            'structured', 'StopResponse', 'FullResponse', 'search_count', 'UsageStats', 'AsyncChat', 'add_warning',
            'astream_with_complete', 'mk_tr_details', 'mk_srv_tc_details', 'StreamFormatter', 'AsyncStreamFormatter',
@@ -21,19 +20,8 @@ from fastcore import imghdr
 from fastcore.xml import Safe
 from dataclasses import dataclass
 
+from .types import *
 from .acomplete import *
-from .acomplete import Msg, Part, PartType, ToolCall, Completion, mk_tool_res_msg, get_model_info
-
-# %% ../nbs/07_chat.ipynb #c4b8f12b
-haik45 = "claude-haiku-4-5"
-sonn45 = "claude-sonnet-4-5"
-sonn = sonn46 = "claude-sonnet-4-6"
-opus46 = "claude-opus-4-6"
-opus = "claude-opus-4-7"
-gpt54 = "gpt-5.4"
-gpt54m = "gpt-5.4-mini"
-codex54 = "gpt-5.4"
-codex55 = "gpt-5.5"
 
 # %% ../nbs/07_chat.ipynb #90f55ad4
 def _bytes2content(data):

fastllm/types.py

@@ -3,7 +3,8 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/00_types.ipynb.
 
 # %% auto #0
-__all__ = ['PartType', 'FinishReason', 'api_registry', 'model_prices_url', 'codex_pricing', 'Part', 'Msg', 'ToolCall',
+__all__ = ['PartType', 'FinishReason', 'api_registry', 'model_prices_url', 'haik45', 'sonn45', 'sonn', 'sonn46', 'opus46', 'opus',
+           'gpt54', 'gpt54m', 'codex54', 'codex55', 'codex53spark', 'codex_pricing', 'Part', 'Msg', 'ToolCall',
            'display_list', 'Usage', 'Completion', 'APIRegistry', 'mk_completion', 'mk_tool_res_msg', 'fn_schema',
            'sys_text', 'part_txt', 'data_url', 'url_mime', 'payload_kwargs', 'get_api_key', 'model_prices_meta',
            'infer_api_name', 'get_model_meta', 'get_model_info']
@@ -151,7 +152,6 @@ class APIRegistry:
 api_registry = APIRegistry()
 
 # %% ../nbs/00_types.ipynb #d58a5f96
-#COMMON
 def mk_completion(resp, model, api_name, vendor_name):
     "Normalize an api response into Completion."
     api = api_registry.apis[api_name]
@@ -167,7 +167,6 @@ def mk_completion(resp, model, api_name, vendor_name):
         raw=resp)
 
 # %% ../nbs/00_types.ipynb #d5322db5
-#COMMON
 def mk_tool_res_msg(tool_calls:list[ToolCall], results:list[str|list]):
     'A util to prepare parallel tool call with str or media list results'
     parts = []
@@ -177,7 +176,6 @@ def mk_tool_res_msg(tool_calls:list[ToolCall], results:list[str|list]):
     return Msg(role="tool", content=parts)
 
 # %% ../nbs/00_types.ipynb #8a8e468b
-#COMMON
 def fn_schema(t):
     "Extract (name, description, parameters) from any tool format."
     if not isinstance(t, dict): return None
@@ -188,7 +186,6 @@ def fn_schema(t):
     return None
 
 # %% ../nbs/00_types.ipynb #d1d48d91
-#COMMON
 def sys_text(system):
     "Extract text from system (str or Part)."
     if system is None: return None
@@ -197,7 +194,6 @@ def sys_text(system):
 def part_txt(p): return p.text if isinstance(p,Part) else p
 
 # %% ../nbs/00_types.ipynb #dc2b75a0
-#COMMON
 _ext_mime = {
     '.jpg':'image/jpeg', '.jpeg':'image/jpeg', '.png':'image/png', '.gif':'image/gif', '.webp':'image/webp',
     '.pdf':'application/pdf',
@@ -250,7 +246,19 @@ def get_model_meta(model, vendor_name=None, tfm=noop):
     elif vendor_name:                                           key = f"{vendor_name}/{model}"
     return dict2obj(tfm(mp.get(key), model, vendor_name))
 
-# %% ../nbs/00_types.ipynb #fbfdeb0a
+# %% ../nbs/00_types.ipynb #60607e23
+haik45 = "claude-haiku-4-5"
+sonn45 = "claude-sonnet-4-5"
+sonn = sonn46 = "claude-sonnet-4-6"
+opus46 = "claude-opus-4-6"
+opus = "claude-opus-4-7"
+gpt54 = "gpt-5.4"
+gpt54m = "gpt-5.4-mini"
+codex54 = "gpt-5.4"
+codex55 = "gpt-5.5"
+codex53spark = "gpt-5.3-codex-spark"
+
+# %% ../nbs/00_types.ipynb #d6d5b98c
 codex_pricing = {
     "input_cost_per_token": 0.10 / 1_000_000,
     "cache_creation_input_token_cost": 0.10 / 1_000_000,
@@ -258,23 +266,31 @@ codex_pricing = {
     "output_cost_per_token": 0.50 / 1_000_000,
 }
 
+_codex_overrides = {
+    codex53spark: dict(
+        supports_vision=False, supports_image_input=False, supports_web_search=True,
+        max_tokens=128000, max_input_tokens=128000, max_output_tokens=128000)
+}
+
+# %% ../nbs/00_types.ipynb #fbfdeb0a
 def get_model_info(mn, vendor_name=None):
     info = get_model_meta(mn, vendor_name)
     # anthropic web search 
-    if 'search_context_cost_per_query' in info:
-        info['supports_web_search'] = True
-    # add reasoning to kimi
+    if 'search_context_cost_per_query' in info: info['supports_web_search'] = True
+    # kimi
     if 'kimi' in mn: 
         if 'k2p6' in mn: info = get_model_meta(mn.replace('k2p6', 'k2p5'), vendor_name)
         info['supports_reasoning'] = True
         info['supports_vision'] = True
         if vendor_name == 'moonshot': info['supports_assistant_prefill'] = True
-    # add web search to gpt
+    # gpt web search
     if mn in ("gpt-5.4", "gpt-5.4-mini"):
         info['supports_web_search'] = True
         info.pop('mode', None)
-    # codex pricing
-    if vendor_name == 'codex': info = merge(info, codex_pricing)
+    # codex updates
+    if vendor_name == 'codex': 
+        info = merge(info, codex_pricing)
+        info |= _codex_overrides.get(mn, {})
     # deepseek v4
     if vendor_name == 'deepseek' and mn in ("deepseek-v4-flash", "deepseek-v4-pro"):
         info = dict(get_model_meta("deepseek/deepseek-v3.2"))

{python_fastllm-0.0.1.dist-info → python_fastllm-0.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-fastllm
-Version: 0.0.1
+Version: 0.0.3
 Author-email: Kerem Turgutlu <keremturgutlu@gmail.com>
 License: Apache-2.0
 Project-URL: Repository, https://github.com/AnswerDotAI/fastllm

{python_fastllm-0.0.1.dist-info → python_fastllm-0.0.3.dist-info}/RECORD

@@ -1,21 +1,21 @@
-fastllm/__init__.py,sha256=sXLh7g3KC4QCFxcZGBTpG2scR7hmmBsMjq6LqRptkRg,22
+fastllm/__init__.py,sha256=4GZKi13lDTD25YBkGakhZyEQZWTER_OWQMNPoH_UM2c,22
 fastllm/_modidx.py,sha256=cOOFPFjpzDYzGBO-4V2gHyxfP9PJKGhxRBzmZ3k4yqs,28453
 fastllm/acomplete.py,sha256=Wl_MdtNe2CuTrmVGPDj5f5pOEnA-ToZqRpPKOHN0J2Q,6764
 fastllm/anthropic.py,sha256=fG20kOv3d3wGKQe8rD5pFWgZHKe-vT-9QJ3nPXh2twY,14615
-fastllm/chat.py,sha256=cibUSgm_8FtFLU8ilCsD8f_2JqvA-KrwJOvHWiDiMrw,29390
+fastllm/chat.py,sha256=RV85NJ9vbmLQ0nU4jsBnjTYNvIYBr1FoPqUtK-HDroc,28941
 fastllm/gemini.py,sha256=E1EYMfV8IMpC_-WzlDrkhz_CJQmzmxvaVUucNgPOqSA,14947
 fastllm/openai_chat.py,sha256=wZ0HI0m9ipy9XVhqmYBXf-BmkVAOipUVwqu9NGB_rJU,10941
 fastllm/openai_responses.py,sha256=Nk5bfTCF2-a17nwvIsf-u39j539v9KIduVfScECItKk,13052
 fastllm/streaming.py,sha256=Ey0ufSYgJREvagMuHuTKBsxMxyS0S_StGuef8taY5PY,7235
-fastllm/types.py,sha256=ZL57Rhgfs_N-tj99B19-BUR8--MyWeyc2vnPjTcjpso,11473
+fastllm/types.py,sha256=5jdE1L4Rj2FclCIx_XiuiLn_aY3wcEp2pPcC9TbZaag,12129
 fastllm/specs/anthropic.json,sha256=VCgTjM2_HoDpCkeu3q_TCOEZLMHriJZLAG3LnDBAgGM,541035
 fastllm/specs/anthropic.yml,sha256=3S3NAKdXB1Nwp-Sn9Gmh4tBnwhGGhMO3DXkGqPXPUYs,724122
 fastllm/specs/gemini.json,sha256=zJGOdvZ2BvCiTENZt0-BDEvNBMl8h6EBmEskle_WBto,309331
 fastllm/specs/openai.with-code-samples.json,sha256=Kto19AW1u8MfxVDJ4cFVBIdZQOIyy8NWylswo57eABU,1995929
 fastllm/specs/openai.with-code-samples.yml,sha256=DlcWGdaeP4k7smVjt6UbyehJ-2XGU3rn3nCIBMDRfYU,2553630
 fastllm/specs/spec_manifest.json,sha256=9tVFwojXFnNqsAxQzCRTP1lgSIM0fXixnrXdv4Cmb0c,653
-python_fastllm-0.0.1.dist-info/METADATA,sha256=OpHIaa2cuqffRF4UcH32RfW2mr93nJ7_x0jBevLjwcU,19546
-python_fastllm-0.0.1.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-python_fastllm-0.0.1.dist-info/entry_points.txt,sha256=dq0chsiRjJYStCOXweFW9L6LpyMTjWu2AabKCbTSbuI,36
-python_fastllm-0.0.1.dist-info/top_level.txt,sha256=F8qodL7nEGUHGmzzqfhNKCTIr1i0D6cvudOnm3z7o0Y,8
-python_fastllm-0.0.1.dist-info/RECORD,,
+python_fastllm-0.0.3.dist-info/METADATA,sha256=LbVdgthLdWt6iBGtLIM88jV_3chcYtTLxLpm4QzP-MI,19546
+python_fastllm-0.0.3.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+python_fastllm-0.0.3.dist-info/entry_points.txt,sha256=dq0chsiRjJYStCOXweFW9L6LpyMTjWu2AabKCbTSbuI,36
+python_fastllm-0.0.3.dist-info/top_level.txt,sha256=F8qodL7nEGUHGmzzqfhNKCTIr1i0D6cvudOnm3z7o0Y,8
+python_fastllm-0.0.3.dist-info/RECORD,,