cocoindex 0.2.9__cp311-abi3-macosx_11_0_arm64.whl → 0.2.10__cp311-abi3-macosx_11_0_arm64.whl

cocoindex/__init__.py

@@ -6,7 +6,12 @@ from . import functions, sources, targets, cli, utils
 
 from . import targets as storages  # Deprecated: Use targets instead
 
-from .auth_registry import AuthEntryReference, add_auth_entry, add_transient_auth_entry
+from .auth_registry import (
+    AuthEntryReference,
+    add_auth_entry,
+    add_transient_auth_entry,
+    ref_auth_entry,
+)
 from .flow import FlowBuilder, DataScope, DataSlice, Flow, transform_flow
 from .flow import flow_def
 from .flow import EvaluateAndDumpOptions, GeneratedField
@@ -14,7 +19,7 @@ from .flow import FlowLiveUpdater, FlowLiveUpdaterOptions, FlowUpdaterStatusUpda
 from .flow import open_flow
 from .flow import add_flow_def, remove_flow  # DEPRECATED
 from .flow import update_all_flows_async, setup_all_flows, drop_all_flows
-from .lib import init, start_server, stop
+from .lib import settings, init, start_server, stop
 from .llm import LlmSpec, LlmApiType
 from .index import VectorSimilarityMetric, VectorIndexDef, IndexOptions
 from .setting import DatabaseConnectionSpec, Settings, ServerSettings
@@ -65,6 +70,7 @@ __all__ = [
     "setup_all_flows",
     "drop_all_flows",
     # Lib
+    "settings",
     "init",
     "start_server",
     "stop",

cocoindex/cli.py

@@ -91,8 +91,6 @@ def _load_user_app(app_target: str) -> None:
 
 
 def _initialize_cocoindex_in_process() -> None:
-    settings = setting.Settings.from_env()
-    lib.init(settings)
     atexit.register(lib.stop)
 
 

cocoindex/flow.py

@@ -694,16 +694,11 @@ class Flow:
     """
 
     _name: str
-    _full_name: str
     _lazy_engine_flow: Callable[[], _engine.Flow] | None
 
-    def __init__(
-        self, name: str, full_name: str, engine_flow_creator: Callable[[], _engine.Flow]
-    ):
+    def __init__(self, name: str, engine_flow_creator: Callable[[], _engine.Flow]):
         validate_flow_name(name)
-        validate_full_flow_name(full_name)
         self._name = name
-        self._full_name = full_name
         engine_flow = None
         lock = Lock()
 
@@ -762,7 +757,7 @@ class Flow:
         """
         Get the full name of the flow.
         """
-        return self._full_name
+        return get_flow_full_name(self._name)
 
     def update(self, /, *, reexport_targets: bool = False) -> _engine.IndexUpdateInfo:
         """
@@ -861,9 +856,10 @@ def _create_lazy_flow(
     The flow will be built the first time when it's really needed.
     """
     flow_name = _flow_name_builder.build_name(name, prefix="_flow_")
-    flow_full_name = get_flow_full_name(flow_name)
 
     def _create_engine_flow() -> _engine.Flow:
+        flow_full_name = get_flow_full_name(flow_name)
+        validate_full_flow_name(flow_full_name)
         flow_builder_state = _FlowBuilderState(flow_full_name)
         root_scope = DataScope(
             flow_builder_state, flow_builder_state.engine_flow_builder.root_scope()
@@ -873,7 +869,7 @@ def _create_lazy_flow(
             execution_context.event_loop
         )
 
-    return Flow(flow_name, flow_full_name, _create_engine_flow)
+    return Flow(flow_name, _create_engine_flow)
 
 
 _flows_lock = Lock()

cocoindex/lib.py

@@ -2,12 +2,57 @@
 Library level functions and states.
 """
 
+import threading
 import warnings
-from typing import Callable, Any
 
 from . import _engine  # type: ignore
 from . import flow, setting
 from .convert import dump_engine_object
+from .validation import validate_app_namespace_name
+from typing import Any, Callable, overload
+
+
+def prepare_settings(settings: setting.Settings) -> Any:
+    """Prepare the settings for the engine."""
+    if settings.app_namespace:
+        validate_app_namespace_name(settings.app_namespace)
+    return dump_engine_object(settings)
+
+
+_engine.set_settings_fn(lambda: prepare_settings(setting.Settings.from_env()))
+
+
+_prev_settings_fn: Callable[[], setting.Settings] | None = None
+_prev_settings_fn_lock: threading.Lock = threading.Lock()
+
+
+@overload
+def settings(fn: Callable[[], setting.Settings]) -> Callable[[], setting.Settings]: ...
+@overload
+def settings(
+    fn: None,
+) -> Callable[[Callable[[], setting.Settings]], Callable[[], setting.Settings]]: ...
+def settings(fn: Callable[[], setting.Settings] | None = None) -> Any:
+    """
+    Decorate a function that returns a settings.Settings object.
+    It registers the function as a settings provider.
+    """
+
+    def _inner(fn: Callable[[], setting.Settings]) -> Callable[[], setting.Settings]:
+        global _prev_settings_fn  # pylint: disable=global-statement
+        with _prev_settings_fn_lock:
+            if _prev_settings_fn is not None:
+                warnings.warn(
+                    f"Setting a new settings function will override the previous one {_prev_settings_fn}."
+                )
+            _prev_settings_fn = fn
+        _engine.set_settings_fn(lambda: prepare_settings(fn()))
+        return fn
+
+    if fn is not None:
+        return _inner(fn)
+    else:
+        return _inner
 
 
 def init(settings: setting.Settings | None = None) -> None:
@@ -16,9 +61,7 @@ def init(settings: setting.Settings | None = None) -> None:
 
     If the settings are not provided, they are loaded from the environment variables.
     """
-    settings = settings or setting.Settings.from_env()
-    _engine.init(dump_engine_object(settings))
-    setting.set_app_namespace(settings.app_namespace)
+    _engine.init(prepare_settings(settings) if settings is not None else None)
 
 
 def start_server(settings: setting.ServerSettings) -> None:

cocoindex/setting.py

@@ -6,16 +6,15 @@ import os
 
 from typing import Callable, Self, Any, overload
 from dataclasses import dataclass
-from .validation import validate_app_namespace_name
-
-_app_namespace: str = ""
+from . import _engine  # type: ignore
 
 
 def get_app_namespace(*, trailing_delimiter: str | None = None) -> str:
     """Get the application namespace. Append the `trailing_delimiter` if not empty."""
-    if _app_namespace == "" or trailing_delimiter is None:
-        return _app_namespace
-    return f"{_app_namespace}{trailing_delimiter}"
+    app_namespace: str = _engine.get_app_namespace()
+    if app_namespace == "" or trailing_delimiter is None:
+        return app_namespace
+    return f"{app_namespace}{trailing_delimiter}"
 
 
 def split_app_namespace(full_name: str, delimiter: str) -> tuple[str, str]:
@@ -26,14 +25,6 @@ def split_app_namespace(full_name: str, delimiter: str) -> tuple[str, str]:
     return (parts[0], parts[1])
 
 
-def set_app_namespace(app_namespace: str) -> None:
-    """Set the application namespace."""
-    if app_namespace:
-        validate_app_namespace_name(app_namespace)
-    global _app_namespace  # pylint: disable=global-statement
-    _app_namespace = app_namespace
-
-
 @dataclass
 class DatabaseConnectionSpec:
     """

{cocoindex-0.2.9.dist-info → cocoindex-0.2.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cocoindex
-Version: 0.2.9
+Version: 0.2.10
 Classifier: Development Status :: 3 - Alpha
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent

{cocoindex-0.2.9.dist-info → cocoindex-0.2.10.dist-info}/RECORD

@@ -1,21 +1,21 @@
-cocoindex-0.2.9.dist-info/METADATA,sha256=pAGAALnR660k-iGmmhH0NMTqmJx2eILu99lkLDXBerc,12998
-cocoindex-0.2.9.dist-info/WHEEL,sha256=cVaoL47Ex1FxzwnkO_WCjy3a1Wl6mtZbBPTvTiNCHdY,103
-cocoindex-0.2.9.dist-info/entry_points.txt,sha256=_NretjYVzBdNTn7dK-zgwr7YfG2afz1u1uSE-5bZXF8,46
-cocoindex-0.2.9.dist-info/licenses/THIRD_PARTY_NOTICES.html,sha256=F5Dm7y2cDyW5bZNG8a0bGXes0pVKHiaMrbcedg7chCk,716358
-cocoindex/__init__.py,sha256=sLpSVO5Cotgn_82lawxvXnaqfa-qj33rytWBAe2MTtU,2201
-cocoindex/_engine.abi3.so,sha256=TJ65BlgZdmaIPDE9wiviwrGTP9wZnn7ceAN3TJeaAbQ,66494288
+cocoindex-0.2.10.dist-info/METADATA,sha256=W1EtSw4PWmq2rT-jS5OA_3x85UIXrJmgtyJLRo7Zhgc,12999
+cocoindex-0.2.10.dist-info/WHEEL,sha256=cVaoL47Ex1FxzwnkO_WCjy3a1Wl6mtZbBPTvTiNCHdY,103
+cocoindex-0.2.10.dist-info/entry_points.txt,sha256=_NretjYVzBdNTn7dK-zgwr7YfG2afz1u1uSE-5bZXF8,46
+cocoindex-0.2.10.dist-info/licenses/THIRD_PARTY_NOTICES.html,sha256=s4PFFZPP5i9n3Gw4VuPYKYYU5pc2bK9W9ipjRkQoygw,717804
+cocoindex/__init__.py,sha256=r0WRAMjHW5gq85fgvxD_EcWW1um4irffpY-ifMrnhKw,2264
+cocoindex/_engine.abi3.so,sha256=3PKDk9hOXtmaXbQiy_7Cl0xxR2bUVE8lB8SDxCGw5eU,67079984
 cocoindex/auth_registry.py,sha256=PE1-kVkcyC1G2C_V7b1kvYzeq73OFQehWKQP7ln7fJ8,1478
-cocoindex/cli.py,sha256=laLLHtEQvEt7Ua4xulsWEI_dH5IpHiaXyTWY8p1VF_c,21665
+cocoindex/cli.py,sha256=69X30bFTFdM7c0_6lgIHR19CeQ7UEkobEQYihy8IdOQ,21599
 cocoindex/convert.py,sha256=Eh9Co37BtW_PK3Oi-pFEiFt8cc_6g7XLcurV-NeH6GU,22090
-cocoindex/flow.py,sha256=wKT_RWeB9_sm764aNrOl2XU-RJIu4EK_sIkjcyZI8jA,37669
+cocoindex/flow.py,sha256=GGk2wtCGfP9cfgdk23SmwR2yBLAuoCDvBGmzmp-XEzM,37591
 cocoindex/functions.py,sha256=09erNt3WbzY9l1KER-akBF2O5-6xEahV2ORBECaL6yk,12260
 cocoindex/index.py,sha256=j93B9jEvvLXHtpzKWL88SY6wCGEoPgpsQhEGHlyYGFg,540
-cocoindex/lib.py,sha256=f--9dAYd84CZosbDZqNW0oGbBLsY3dXiUTR1VrfQ_QY,817
+cocoindex/lib.py,sha256=0XheDF7fiFdqExpdqzU-VKun_Zll6DwZ5JfTm7u42aY,2284
 cocoindex/llm.py,sha256=Pv_cdnRngTLtuLU9AUmS8izIHhcKVnuBNolC33f9BDI,851
 cocoindex/op.py,sha256=3jTEpmzHPjg-euZhXjQqgF0i8V2LHUdkq7bp4AAIHRM,21627
 cocoindex/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cocoindex/runtime.py,sha256=povilB3HH3y1JF-yxKwU-pD8n2WnAqyQxIgvXXHNc60,1080
-cocoindex/setting.py,sha256=rynB2qriPXysEDXp1YV_riqfyiGInR62tcWeDu8MMko,5214
+cocoindex/setting.py,sha256=1Dx8ktjwf-8BiXrbsmfn5Mzudb2SQYqFdRnSNGVKaLk,4960
 cocoindex/setup.py,sha256=7uIHKN4FOCuoidPXcKyGTrkqpkl9luL49-6UcnMxYzw,3068
 cocoindex/sources.py,sha256=FYz7cWYasLGDaYoIEQ1dF2uprgUETHWsTIrIS7n6pQE,3188
 cocoindex/subprocess_exec.py,sha256=r1xO84uek4VP4I6i87JMwsH5xFm3vKW0ABvgn0jskt4,10088
@@ -30,4 +30,4 @@ cocoindex/typing.py,sha256=lEQYIzAGVKQp6RnhyeopY9Q7xEED7yQj3ZMxvTPblV8,14200
 cocoindex/user_app_loader.py,sha256=bc3Af-gYRxJ9GpObtpjegZY855oQBCv5FGkrkWV2yGY,1873
 cocoindex/utils.py,sha256=hUhX-XV6XGCtJSEIpBOuDv6VvqImwPlgBxztBTw7u0U,598
 cocoindex/validation.py,sha256=PZnJoby4sLbsmPv9fOjOQXuefjfZ7gmtsiTGU8SH-tc,3090
-cocoindex-0.2.9.dist-info/RECORD,,
+cocoindex-0.2.10.dist-info/RECORD,,

{cocoindex-0.2.9.dist-info → cocoindex-0.2.10.dist-info}/licenses/THIRD_PARTY_NOTICES.html

@@ -45,7 +45,7 @@
         <h2>Overview of licenses:</h2>
         <ul class="licenses-overview">
             <li><a href="#Apache-2.0">Apache License 2.0</a> (412)</li>
-            <li><a href="#MIT">MIT License</a> (124)</li>
+            <li><a href="#MIT">MIT License</a> (125)</li>
             <li><a href="#Unicode-3.0">Unicode License v3</a> (19)</li>
             <li><a href="#ISC">ISC License</a> (8)</li>
             <li><a href="#BSD-3-Clause">BSD 3-Clause &quot;New&quot; or &quot;Revised&quot; License</a> (6)</li>
@@ -2428,7 +2428,7 @@ Software.
                 <h3 id="Apache-2.0">Apache License 2.0</h3>
                 <h4>Used by:</h4>
                 <ul class="license-used-by">
-                    <li><a href=" https://crates.io/crates/cocoindex ">cocoindex 0.2.9</a></li>
+                    <li><a href=" https://crates.io/crates/cocoindex ">cocoindex 0.2.10</a></li>
                     <li><a href=" https://github.com/awesomized/crc-fast-rust ">crc-fast 1.3.0</a></li>
                     <li><a href=" https://github.com/qdrant/rust-client ">qdrant-client 1.15.0</a></li>
                 </ul>
@@ -11376,6 +11376,20 @@ OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR
 IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 DEALINGS IN THE SOFTWARE.
 </pre>
+            </li>
+            <li class="license">
+                <h3 id="MIT">MIT License</h3>
+                <h4>Used by:</h4>
+                <ul class="license-used-by">
+                    <li><a href=" https://github.com/JoranHonig/tree-sitter-solidity ">tree-sitter-solidity 1.2.13</a></li>
+                </ul>
+                <pre class="license-text">Copyright (c) 2020 Joran Honig
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the &quot;Software&quot;), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED &quot;AS IS&quot;, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.</pre>
             </li>
             <li class="license">
                 <h3 id="MIT">MIT License</h3>