python-fastllm 0.0.1__tar.gz → 0.0.3__tar.gz

{python_fastllm-0.0.1 → python_fastllm-0.0.3}/PKG-INFO

@@ -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.3/fastllm/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.0.3"

{python_fastllm-0.0.1 → python_fastllm-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):

{python_fastllm-0.0.1 → python_fastllm-0.0.3}/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 → python_fastllm-0.0.3}/pyproject.toml

@@ -27,7 +27,7 @@ fastllm = "fastllm._modidx:d"
 version = {attr = "fastllm.__version__"}
 
 [tool.setuptools.packages.find]
-include = ["fastllm"]
+include = ["fastllm*"]
 
 [tool.setuptools.package-data]
 fastllm = ["specs/*.yml", "specs/*.json"]
@@ -35,3 +35,5 @@ fastllm = ["specs/*.yml", "specs/*.json"]
 [tool.nbdev]
 allowed_metadata_keys = ['solveit']
 allowed_cell_metadata_keys = ['solveit_ai']
+custom_sidebar = false
+lib_path = "fastllm"

{python_fastllm-0.0.1 → python_fastllm-0.0.3}/python_fastllm.egg-info/PKG-INFO

@@ -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/fastllm/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.0.1"