corio 2.2.3__tar.gz → 2.2.5__tar.gz

{corio-2.2.3 → corio-2.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: corio
-Version: 2.2.3
+Version: 2.2.5
 Summary: Collection of high-level tools to simplify everyday development tasks, with a focus on AI/ML
 Author-email: Frontmatter AI <innovative.fowler@mask.pro.fmtr.dev>
 License-Expression: Apache-2.0
@@ -188,6 +188,8 @@ Requires-Dist: httpx; extra == "dns"
 Requires-Dist: httpx_retries; extra == "dns"
 Requires-Dist: logfire; extra == "dns"
 Requires-Dist: logfire[httpx]; extra == "dns"
+Requires-Dist: diskcache; extra == "dns"
+Requires-Dist: cachetools; extra == "dns"
 Provides-Extra: patterns
 Requires-Dist: regex; extra == "patterns"
 Provides-Extra: http
@@ -214,6 +216,9 @@ Requires-Dist: av; extra == "av"
 Provides-Extra: env
 Provides-Extra: env-io
 Requires-Dist: dotenv; extra == "env-io"
+Provides-Extra: toml
+Provides-Extra: toml-write
+Requires-Dist: tomlkit; extra == "toml-write"
 Provides-Extra: ha
 Requires-Dist: dotenv; extra == "ha"
 Provides-Extra: ha-api

{corio-2.2.3 → corio-2.2.5}/corio/ai/agentic.py

@@ -177,43 +177,3 @@ def default_prompt_none_specified(text):
 
 
 StringDefaultNoneSpecified = Annotated[Optional[str], PlainValidator(default_prompt_none_specified)]
-
-
-if __name__ == '__main__':
-    import asyncio
-    from corio import dm
-
-
-    class TestOutput(dm.Base):
-        text: str
-
-
-    class TestDeps(dm.Base):
-        lang: str
-        subject: str
-
-
-    class TaskTest(Task):
-        PROVIDER = Task.PROVIDER_FMTR
-        MODEL_ID = Task.MODEL_ID_FMTR
-        TypeOutput = TestOutput
-        SYSTEM_PROMPT_STATIC = 'Tell the user jokes.'
-
-        @property
-        def tool_output(self) -> OutputSpec[TestOutput]:
-            return NativeOutput(self.TypeOutput)
-
-        def add_system_prompt(self, ctx: RunContext[TestDeps]) -> str:
-            return f'The jokes must be in the {ctx.deps.lang} language.'
-
-        def get_prompt(self, deps: Optional[TestDeps]) -> str:
-            return f'Tell me one about {deps.subject}.'
-
-    task = TaskTest()
-    deps = TestDeps(lang='English', subject='eggs')
-    result1 = task.sync_runner(task.run(deps=deps))
-    result1
-
-    deps = TestDeps(lang='German', subject='sausages')
-    result2 = task.sync_runner(task.run(deps=deps))
-    result2

{corio-2.2.3 → corio-2.2.5}/corio/ai/infer.py

@@ -410,8 +410,3 @@ def tst_tool():
         objs = ToolsCall.from_json(text).apply(BulkInferenceManagerTools.TOOLS)
         obj = objs[0]
         print(obj)
-
-
-if __name__ == '__main__':
-    texts = tst()
-    texts

{corio-2.2.3 → corio-2.2.5}/corio/api.py

@@ -178,8 +178,3 @@ class Base:
         """
         import asyncio
         return asyncio.run(cls.launch_async(*args, **kwargs))
-
-
-
-if __name__ == '__main__':
-    Base.launch()

{corio-2.2.3 → corio-2.2.5}/corio/caching.py

@@ -178,34 +178,3 @@ class TLRU(cachetools.TLRUCache):
 
         """
         return self.dump()
-
-
-
-if __name__ == '__main__':
-    sec10 = timedelta(seconds=10)
-    c = TLRU(ttu_static=sec10, maxsize=2, desc='Test Data')
-    c['test'] = 'val'
-    c['test2'] = 'val2'
-    c['test3'] = 'val3'
-    c
-
-
-
-
-
-
-    path_tmp_cache = Path.cwd().parent.parent / 'data' / 'cache'
-    tc = Disk(path_tmp_cache)
-
-    tc.setdefault('c', Disk).setdefault('c1', Disk)['subkey'] = 0000.1
-    # tc['c']=Disk.Create
-    tc['c']['test'] = False
-    tc['val'] = 123
-    tc.setdefault('b', Disk)
-    tc.setdefault('a', Disk)
-    tc['a']['value2'] = 456
-    tc['a']['value4'] = dict(mykey='myvalue')
-
-    tc['b']['value3'] = [789, True]
-    tc.dump()
-    {}.items()

{corio-2.2.3 → corio-2.2.5}/corio/debug.py

@@ -78,13 +78,3 @@ def debug_shell():
     data = ShellDebug.from_path(path_str)
     trace(is_debug=True)
     data
-
-
-if __name__ == "__main__":
-    import sys
-
-    sys.argv = [
-        'test.py',
-        './fmtr-debug/34e8d492-2f15-419a-8fcb-fe4fa0fa02bb',
-    ]
-    debug_shell()

{corio-2.2.3 → corio-2.2.5}/corio/entrypoint.py

@@ -3,6 +3,7 @@ from pydantic_settings import CliSubCommand
 from corio import dm
 from corio import sec
 from corio import sets
+from corio.path import Path
 
 
 class DocServe(dm.Base):
@@ -42,6 +43,17 @@ class EpTest(dm.Base):
         print("Ran test entrypoint.")
 
 
+class Test(dm.Base):
+    name: str = Path.cwd().name
+
+    def run(self):
+        from corio.infra.project import Project
+
+        project = Project(self.name)
+        is_passed = project.releaser.tester.run()
+        return int(not is_passed)
+
+
 class ShellDebug(dm.Base):
     def run(self):
         from corio import debug
@@ -81,6 +93,7 @@ class Cli(sets.Base, cli_parse_args=True):
     secrets: CliSubCommand[sec.Cli]
     docs: CliSubCommand[Docs]
     pyproject: CliSubCommand[Pyproject]
+    test: CliSubCommand[Test]
     ep_test: CliSubCommand[EpTest]
     shell_debug: CliSubCommand[ShellDebug]
     cache_hfh: CliSubCommand[CacheHfh]

{corio-2.2.3 → corio-2.2.5}/corio/https.py

@@ -44,9 +44,3 @@ class Client(httpx.Client):
 
 
 client = Client()
-
-if __name__ == '__main__':
-    resp = client.get('https://postman-echo.com/delay/5')
-    resp.raise_for_status()
-    print(resp.json())
-    resp

{corio-2.2.3 → corio-2.2.5}/corio/infra/api.py

@@ -47,7 +47,3 @@ class Api(api.Base):
         project = Project(name, pinned=pinned)
 
         project.releaser.run(increment=increment, build=build, release=release, cache=cache)
-
-
-if __name__ == '__main__':
-    Api.launch()

{corio-2.2.3 → corio-2.2.5}/corio/infra/incrementor_pyproject.py

@@ -19,6 +19,9 @@ class IncrementorPyproject(Incrementor):
     ENTRYPOINT_FUNCTION_SEP = "_"
     ENTRYPOINT_FUNC_NAME = "main"
     DEPENDENCIES_SECTION_PATH = ("tool", "corio", "dependencies")
+    TEST_FILENAME_PREFIX = "test_"
+    TEST_FILENAME_SUFFIX = ".py"
+    TOX_REQUIRES = ["tox>=4.22", "tox-uv>=1"]
 
     @cached_property
     def path(self) -> Path:
@@ -274,4 +277,90 @@ class IncrementorPyproject(Incrementor):
         elif "script-files" in setuptools:
             del setuptools["script-files"]
 
+        dependencies = project.get("optional-dependencies", {})
+        envs = self._tox_envs(dependencies=dependencies)
+        tox = ensure_table(data, ("tool", "tox"))
+        tox["requires"] = list(self.TOX_REQUIRES)
+        tox["env_list"] = list(envs.keys())
+        tox["env"] = envs
+
         return data
+
+    @cached_property
+    def _tests_modules(self) -> list[str]:
+        if not self.paths.tests.exists():
+            return []
+
+        modules = []
+        for path in sorted(self.paths.tests.glob(f"{self.TEST_FILENAME_PREFIX}*{self.TEST_FILENAME_SUFFIX}")):
+            module = path.stem.removeprefix(self.TEST_FILENAME_PREFIX)
+            if module:
+                modules.append(module)
+        return modules
+
+    def _tox_get_extras_module(self, module: str, dependencies: dict[str, list[str]]) -> list[str]:
+        extras = ["test"]
+        extras_available = set(dependencies.keys())
+        module_canonical = canonicalize_name(module)
+
+        extras_exact = [extra for extra in extras_available if canonicalize_name(extra) == module_canonical]
+        if extras_exact:
+            extras = sorted(extras_exact) + extras
+
+        extras_children = sorted(
+            extra
+            for extra in extras_available
+            if canonicalize_name(extra).startswith(f"{module_canonical}-")
+        )
+        return extras_children + extras
+
+    def _tox_get_deps_extras(self, extras: list[str], dependencies: dict[str, list[str]]) -> list[str]:
+        deps = []
+        for extra in extras:
+            deps += dependencies.get(extra, [])
+        deps = dedupe(deps)
+        return deps
+
+    def _tox_get_env(
+        self,
+        *,
+        name: str,
+        path_tests: Path,
+        extras: list[str],
+        dependencies: dict[str, list[str]],
+    ) -> dict:
+        if path_tests.is_relative_to(self.paths.repo):
+            path_tests = path_tests.relative_to(self.paths.repo)
+        deps = self._tox_get_deps_extras(extras=extras, dependencies=dependencies)
+        env = {
+            "description": f"Run {name} tests.",
+            "deps": deps,
+            "commands": [["python", "-m", "pytest", "-q", str(path_tests)]],
+        }
+        return env
+
+    def _tox_envs(self, dependencies: dict[str, list[str]]) -> dict[str, dict]:
+        if not self.paths.metadata.test_envs:
+            if not self._tests_modules:
+                return {}
+            return {
+                self.paths.name_ns: self._tox_get_env(
+                    name=self.paths.name_ns,
+                    path_tests=self.paths.tests,
+                    extras=["test"],
+                    dependencies=dependencies,
+                )
+            }
+
+        envs = {}
+        for module in self._tests_modules:
+            extras = self._tox_get_extras_module(module=module, dependencies=dependencies)
+            path_test = self.paths.tests / f"{self.TEST_FILENAME_PREFIX}{module}{self.TEST_FILENAME_SUFFIX}"
+            name = module
+            envs[name] = self._tox_get_env(
+                name=name,
+                path_tests=path_test,
+                extras=extras,
+                dependencies=dependencies,
+            )
+        return envs

{corio-2.2.3 → corio-2.2.5}/corio/infra/releaser.py

@@ -1,7 +1,6 @@
-import shutil
 import subprocess
 from functools import cached_property
-
+import shutil
 import build
 import pygit2 as vcs
 import twine.settings
@@ -571,26 +570,6 @@ class ReleaseDocumentation(Release):
 class Tester(Inherit[Releaser]):
     TEST_FILENAME_PREFIX = "test_"
     TEST_FILENAME_SUFFIX = ".py"
-    TOX_REQUIRES = ["tox>=4.22", "tox-uv>=1"]
-
-    @cached_property
-    def path_config_dir(self) -> Path:
-        return Path.temp() / f"{self.name}-tox"
-
-    @cached_property
-    def path_config(self) -> Path:
-        return self.path_config_dir / "tox.toml"
-
-    @cached_property
-    def dependencies(self) -> dict[str, list[str]]:
-        data = self.paths.pyproject_repo.read_toml()
-        table = data.get("tool", {}).get("corio", {}).get("dependencies", {})
-        dependencies = {
-            str(key): [str(value) for value in values]
-            for key, values in table.items()
-            if isinstance(values, list)
-        }
-        return dependencies
 
     @cached_property
     def modules(self) -> list[str]:
@@ -604,54 +583,10 @@ class Tester(Inherit[Releaser]):
                 modules.append(module)
         return modules
 
-
-    def get_env(self, name: str, path_tests: Path, extras: list[str]) -> dict:
-        if path_tests.is_relative_to(self.paths.repo):
-            path_tests = path_tests.relative_to(self.paths.repo)
-        env = {
-            "description": f"Run {name} tests.",
-            "extras": extras,
-            "commands": [["python", "-m", "pytest", "-q", str(path_tests)]],
-        }
-        return env
-
     @cached_property
-    def env(self) -> dict:
-        return self.get_env(name=self.paths.name_ns, path_tests=self.paths.tests, extras=["test"])
-
-    @cached_property
-    def envs(self) -> dict[str, dict]:
-        if not self.paths.metadata.test_envs:
-            if not self.modules:
-                return {}
-            return {self.paths.name_ns: self.env}
-
-        envs = {}
-        extras_available = set(self.dependencies.keys())
-
-        for module in self.modules:
-            extras = ["test"]
-            if module in extras_available:
-                extras.insert(0, module)
-
-            path_test = self.paths.tests / f"{self.TEST_FILENAME_PREFIX}{module}{self.TEST_FILENAME_SUFFIX}"
-            name = f"{self.paths.name_ns}.{module}"
-            envs[name] = self.get_env(name=name, path_tests=path_test, extras=extras)
-
-        return envs
-
-    @cached_property
-    def data(self) -> dict:
-        data = {
-            "requires": self.TOX_REQUIRES,
-            "env_list": list(self.envs.keys()),
-            "env": self.envs,
-        }
-        return data
-
-    def write_config(self):
-        self.path_config_dir.mkdir(parents=True, exist_ok=True)
-        self.path_config.write_toml(self.data)
+    def env_list(self) -> list[str]:
+        data = self.paths.pyproject_repo.read_toml()
+        return list(data.get("tool", {}).get("tox", {}).get("env_list", []))
 
     def run_subprocess(self) -> int:
         command = [
@@ -660,7 +595,7 @@ class Tester(Inherit[Releaser]):
             "tox-uv",
             "tox",
             "-c",
-            str(self.path_config),
+            str(self.paths.pyproject_repo),
             "--root",
             str(self.paths.repo),
             "--workdir",
@@ -686,17 +621,15 @@ class Tester(Inherit[Releaser]):
 
     @logger.instrument('Running test suite for "{self.paths.name_ns}"...')
     def run(self) -> bool:
-        if not self.envs:
+        if not self.modules:
             logger.warning(f'No tests found under "{self.paths.tests}". Skipping.')
             return True
 
-        logger.info(f'Generating temporary tox config: "{self.path_config}"')
-        self.write_config()
-        try:
-            code = self.run_subprocess()
-        finally:
-            self.path_config.unlink(missing_ok=True)
-            shutil.rmtree(self.path_config_dir, ignore_errors=True)
+        if not self.env_list:
+            logger.warning(f'No tox envs found in "{self.paths.pyproject_repo}". Skipping.')
+            return True
+
+        code = self.run_subprocess()
 
         if code == 0:
             logger.info("All test environments passed.")

{corio-2.2.3 → corio-2.2.5}/corio/interface/interface.py

@@ -183,6 +183,3 @@ class Test(Base[Context]):
     def __init__(self):
         controls = [ft.Text(self.TITLE)]
         super().__init__(controls=controls)
-
-if __name__ == "__main__":
-    Test.launch()

{corio-2.2.3 → corio-2.2.5}/corio/logs.py

@@ -148,9 +148,3 @@ def get_native_level_from_otel(otel_name: str) -> int:
 
 
 logger = get_logger(name=Constants.LIBRARY_NAME)
-
-if __name__ == '__main__':
-    logger.info('Hello World')
-    logger.warning('test warning')
-    logger.debug('Hello World')
-    logger

{corio-2.2.3 → corio-2.2.5}/corio/openai.py

@@ -33,8 +33,3 @@ def get_text(prompt, model='gpt-4o'):
     )
     text = completion.choices[0].message.content
     return text
-
-
-if __name__ == '__main__':
-    text = get_text('hi')
-    text

{corio-2.2.3 → corio-2.2.5}/corio/path/path.py

@@ -909,7 +909,3 @@ class PathsSearchData:
             org = None
             name = next(iter(parts))
         return org, name
-
-if __name__ == "__main__":
-    paths = PackagePaths()
-    paths

{corio-2.2.3 → corio-2.2.5}/corio/patterns.py

@@ -272,7 +272,3 @@ class Transformer:
         logger.debug(f'Finished transforming: {history_str}')
 
         return previous
-
-
-if __name__ == '__main__':
-    ...

{corio-2.2.3 → corio-2.2.5}/corio/pdf.py

@@ -216,16 +216,3 @@ class Document(pm.Document):
             documents.append(document)
 
         return documents
-
-if __name__ == '__main__':
-    from corio.path import Path
-
-    PATH_DATA = Path.data()
-    # PATH_PDF=PATH_DATA/'chib.pdf'
-    PATH_PDF = PATH_DATA / 'kvm.pdf'
-    assert PATH_PDF.exists()
-
-    doc = Document(PATH_PDF)
-    data = doc.data
-    md = doc.to_markdown()
-    md