stores 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

stores/indexes/base_index.py

@@ -1,5 +1,4 @@
 import asyncio
-import functools
 import inspect
 import logging
 import re
@@ -18,6 +17,8 @@ from typing import (
     get_type_hints,
 )
 
+from makefun import create_function
+
 from stores.format import ProviderFormat, format_tools
 from stores.parse import llm_parse_json
 from stores.utils import check_duplicates
@@ -209,8 +210,15 @@ def wrap_tool(tool: Callable):
             # Inject default values within wrapper
             bound_args = original_signature.bind(*args, **kwargs)
             bound_args.apply_defaults()
-            _cast_bound_args(bound_args)
             # Inject correct Literals
+            for k, v in bound_args.arguments.items():
+                if (
+                    v is None
+                    and original_signature.parameters[k].default is not Parameter.empty
+                ):
+                    bound_args.arguments[k] = original_signature.parameters[k].default
+
+            _cast_bound_args(bound_args)
             for k, v in bound_args.arguments.items():
                 if k in literal_maps:
                     param = original_signature.parameters[k]
@@ -219,11 +227,17 @@ def wrap_tool(tool: Callable):
                     )
             return tool(*bound_args.args, **bound_args.kwargs)
 
-    functools.update_wrapper(wrapper, tool)
-    wrapper.__signature__ = new_sig
-    wrapper._wrapped = True
+    wrapped = create_function(
+        new_sig,
+        wrapper,
+        qualname=tool.__name__,
+        doc=inspect.getdoc(tool),
+    )
+
+    wrapped.__name__ = tool.__name__
+    wrapped._wrapped = True
 
-    return wrapper
+    return wrapped
 
 
 class BaseIndex:

{stores-0.1.0.dist-info → stores-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stores
-Version: 0.1.0
+Version: 0.1.1
 Summary: Repository of Python functions and tools for LLMs
 License-File: LICENSE
 Requires-Python: >=3.10

{stores-0.1.0.dist-info → stores-0.1.1.dist-info}/RECORD

@@ -4,12 +4,12 @@ stores/format.py,sha256=LduYBVDiUDB1J1HDyu9jHrRG1V97pw6C5g76OirpJJY,7792
 stores/parse.py,sha256=HYPNPzQod2vpu1Cln7yQ8aVkZT1Mw2IN0sZ2A1DIaqE,4967
 stores/utils.py,sha256=GPWT6lCoGobwP3PlEOHyJfKyd0dobamjyErcR7lgm7M,242
 stores/indexes/__init__.py,sha256=s-RNqml8uGREQhxwSdDoxcbcxeD8soB9BcL5dBKsQfI,215
-stores/indexes/base_index.py,sha256=1DP3cb1njE-MHuVXHVlyQlao2APeyWjRA2VT4RwW6BU,10293
+stores/indexes/base_index.py,sha256=K4--_OAFyPNESaisD0JPNRIEffCnjw9nziahDgjdZHU,10708
 stores/indexes/index.py,sha256=ckyisaoD_0Uw3dWXpMYYhNWxMXcfykwJIY8lUdpNntA,2113
 stores/indexes/local_index.py,sha256=NS4AWzeMVZ51YPo0eJ_udbF5aASuksG-88YXLDn41fQ,2880
 stores/indexes/remote_index.py,sha256=-iKO0q9MS1CiyzY1lu90FxHs0pVR6z82_7Lk7jX9wFk,2523
 stores/indexes/venv_utils.py,sha256=odeKBphn6H2Jwz6pyy4pvEZFB48Kd1rroAuMiJjcPLE,11696
-stores-0.1.0.dist-info/METADATA,sha256=EyJIq_xAsOTaBc0h5dh3H_N58qlXS5CvP4BwIakuOK8,3076
-stores-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-stores-0.1.0.dist-info/licenses/LICENSE,sha256=VTidYE7_Dam0Dwyq095EhhDIqi47g03oVpLAHQgKws0,1066
-stores-0.1.0.dist-info/RECORD,,
+stores-0.1.1.dist-info/METADATA,sha256=FwbKEQMpPJNB_wPgKRU_kM2lhmUNocmL2VIwBP7UZOE,3076
+stores-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+stores-0.1.1.dist-info/licenses/LICENSE,sha256=VTidYE7_Dam0Dwyq095EhhDIqi47g03oVpLAHQgKws0,1066
+stores-0.1.1.dist-info/RECORD,,