ai-microcore 5.0.0.dev7__tar.gz → 5.0.0.dev8__tar.gz

{ai_microcore-5.0.0.dev7 → ai_microcore-5.0.0.dev8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-microcore
-Version: 5.0.0.dev7
+Version: 5.0.0.dev8
 Summary: # Minimalistic Foundation for AI Applications
 Keywords: llm,large language models,ai,similarity search,ai search,gpt,openai,framework,adapter,anthropic,google gemini,google vertex ai
 Author-email: Vitalii Stepanenko <mail@vitaliy.in>

{ai_microcore-5.0.0.dev7 → ai_microcore-5.0.0.dev8}/microcore/__init__.py

@@ -79,7 +79,7 @@ def model_names() -> list[str]:
     Return a list of available model names from the default LLM client.
     """
     if env().default_client is None:
-        raise ValueError("No default LLM client supporting models list configured.")
+        raise ValueError("No default LLM client is configured.")
     return env().default_client.model_names()
 
 
@@ -231,4 +231,4 @@ __all__ = [
     # "wrappers",
 ]
 
-__version__ = "5.0.0.dev7"
+__version__ = "5.0.0dev8"

{ai_microcore-5.0.0.dev7 → ai_microcore-5.0.0.dev8}/microcore/_llm_functions.py

@@ -267,9 +267,7 @@ def llm(
             save_cache(cache_name, response)
     [h(response) for h in env().llm_after_handlers]
     if tries > 0:
-        retry_params = dict(**kwargs)
-        retry_params["retries"] = tries - 1
-        setattr(response, "_retry_callback", lambda: llm(prompt, **retry_params))
+        setattr(response, "_retry_callback", lambda: llm(prompt, retries=tries - 1, **kwargs))
     if parse_json:
         parsing_params = parse_json if isinstance(parse_json, dict) else {}
         return response.parse_json(**parsing_params)
@@ -294,6 +292,13 @@ async def allm(
         parse_json (bool|dict):
             If True, parses response as JSON,
             alternatively non-empty dict can be used as parse_json arguments.
+            Default is False (no parsing).
+        file_cache (bool | str):
+            If True or non-empty string, enables file caching of LLM responses.
+            If string, it will be used as cache prefix.
+            When enabled, identical requests with identical parameters
+            will return cached responses instead of making new API calls.
+            Default is False (no caching).
         **kwargs: Parameters supported by the LLM API.
 
             See parameters supported by the OpenAI:
@@ -374,8 +379,13 @@ async def allm(
                 if file_cache:
                     delete_cache(cache_name)
                 return await allm(
-                    prompt, retries=tries - 1, parse_json=parse_json, **kwargs
+                    prompt,
+                    retries=tries - 1,
+                    parse_json=parse_json,
+                    file_cache=file_cache,
+                    **kwargs
                 )
+            raise e
     return response
 
 

{ai_microcore-5.0.0.dev7 → ai_microcore-5.0.0.dev8}/microcore/ai_func/__init__.py

@@ -80,7 +80,7 @@ def func_metadata(func, name=None) -> Dict[str, Any]:
 
     # Add descriptions from parsed docstring to parameters
     for param in parsed_docstring.params:
-        if param.arg_name in metadata.get("args", []):
+        if param.arg_name in metadata.get("args", {}):
             metadata["args"][param.arg_name]["docstr"] = param.description
 
     return metadata
@@ -190,6 +190,7 @@ def extract_tag_tool_params(
     if len(tags) > 1:
         if raise_errors:
             raise ValueError("Response contains multiple tags when only one expected")
+        logging.warning("Response contains multiple tags, but only the first one will be used.")
     tag, attrs, content = tags[0]
     return tag, [content], attrs
 

{ai_microcore-5.0.0.dev7 → ai_microcore-5.0.0.dev8}/microcore/file_storage.py

@@ -1,5 +1,15 @@
 """
-File storage functions
+File storage functionality.
+
+
+Provides a Storage class for file operations within a configured storage directory.
+Supports automatic file numbering, backups, JSON serialization, encoding detection,
+file copying, directory listing, etc.
+
+Usage:
+    from microcore import storage
+    storage.write("data.txt", "content")
+    content = storage.read("data.txt")
 """
 
 import fnmatch
@@ -15,6 +25,7 @@ from ._env import config
 from .utils import file_link, list_files
 
 _missing = object()
+"""Sentinel value to distinguish between None and 'not provided'."""
 
 
 @dataclass
@@ -224,8 +235,6 @@ class Storage:
         if isinstance(content, bytes):
             if encoding is not None:
                 logging.warning("Encoding is ignored when writing bytes content")
-            if append:
-                raise ValueError("Cannot append bytes content")
 
         if rewrite_existing is None:
             rewrite_existing = True
@@ -268,18 +277,12 @@ class Storage:
                 if file_name != fn_incremented:
                     os.rename(self.path / file_name, self.path / fn_incremented)
         (self.path / file_name).parent.mkdir(parents=True, exist_ok=True)
-        if append:
-            with (self.path / file_name).open(
-                mode="a",
-                encoding=encoding if not isinstance(content, bytes) else None,
-            ) as file:
-                file.write(content)
+        if isinstance(content, bytes):
+            with (self.path / file_name).open(mode="ab" if append else "wb") as f:
+                f.write(content)
         else:
-            if isinstance(content, bytes):
-                with (self.path / file_name).open(mode="wb") as file:
-                    file.write(content)
-            else:
-                (self.path / file_name).write_text(content, encoding=encoding)
+            with (self.path / file_name).open(mode="a" if append else "w", encoding=encoding) as f:
+                f.write(content)
         return file_name
 
     def clean(self, path: str | Path):

{ai_microcore-5.0.0.dev7 → ai_microcore-5.0.0.dev8}/microcore/images.py

@@ -87,7 +87,7 @@ class FileImage(ImageInterface):
 
 
 class FileImageList(ImageListInterface):
-    files: list[str] | None
+    files: list[str] | None = None
 
     def images(self) -> list[FileImage]:
         if not self.files:

{ai_microcore-5.0.0.dev7 → ai_microcore-5.0.0.dev8}/microcore/llm/google_genai.py

@@ -163,8 +163,9 @@ class AsyncGoogleClient(BaseAsyncAIClient):
     def __init__(self, client: GoogleClient):
         self.sync_client = client
 
-    async def load_models(self) -> dict:
-        raise NotImplementedError
+    async def load_models(self, **kwargs) -> dict:
+        models = await self.sync_client.genai_client.aio.models.list(**kwargs)
+        return {model.name: model for model in models}
 
     async def generate(
         self,

{ai_microcore-5.0.0.dev7 → ai_microcore-5.0.0.dev8}/microcore/llm/openai.py

@@ -42,8 +42,7 @@ class AsyncOpenAIClient(BaseAsyncAIClient):
         config = self.sync_client.config
         args, options = _prepare_llm_arguments(config, kwargs)
         if is_image_model(args["model"]):
-            # Note: image generation is synchronous
-            return _generate_image(
+            return await _generate_image_async(
                 prompt,
                 args,
                 self.oai_client,
@@ -294,12 +293,8 @@ def _oai_image_response_to_images(response: ImagesResponse) -> list[Image]:
     return images
 
 
-def _generate_image(
-    prompt,
-    args,
-    connection: openai.OpenAI | openai.AsyncOpenAI,
-    options
-) -> ImageGenerationResponse | None:
+def _prepare_image_generation(prompt, args):
+    """Prepare prompt and images for image generation (shared logic)."""
     def convert_input_image(image: ImageInterface):
         if isinstance(image, FileImage):
             return open(image.file, "rb")
@@ -333,6 +328,32 @@ def _generate_image(
     if save and args.get("response_format", "b64_json") != "b64_json":
         raise ValueError("Only 'b64_json' response format is supported.")
 
+    return prompt, images, save
+
+
+def _image_generation_response(
+        response: ImagesResponse,
+        save: bool,
+        options: dict,
+) -> ImageGenerationResponse | None:
+    check_for_errors(response)
+    images = _oai_image_response_to_images(response)
+    response_attrs = response.__dict__.copy()
+    result = make_image_generation_response(images, save, response_attrs)
+    for cb in options["callbacks"]:
+        cb(result)
+    return result
+
+
+def _generate_image(
+    prompt,
+    args,
+    connection: openai.OpenAI,
+    options
+) -> ImageGenerationResponse | None:
+    """Synchronous version of image generation."""
+    prompt, images, save = _prepare_image_generation(prompt, args)
+
     if not images:
         response: ImagesResponse = connection.images.generate(prompt=prompt, **args)
     else:
@@ -341,10 +362,24 @@ def _generate_image(
             prompt=prompt,
             **args
         )
-    check_for_errors(response)
-    images = _oai_image_response_to_images(response)
-    response_attrs = response.__dict__.copy()
-    result = make_image_generation_response(images, save, response_attrs)
-    for cb in options["callbacks"]:
-        cb(result)
-    return result
+    return _image_generation_response(response, save, options)
+
+
+async def _generate_image_async(
+    prompt,
+    args,
+    connection: openai.AsyncOpenAI,
+    options
+) -> ImageGenerationResponse | None:
+    """Asynchronous version of image generation."""
+    prompt, images, save = _prepare_image_generation(prompt, args)
+
+    if not images:
+        response: ImagesResponse = await connection.images.generate(prompt=prompt, **args)
+    else:
+        response: ImagesResponse = await connection.images.edit(
+            image=images,
+            prompt=prompt,
+            **args
+        )
+    return _image_generation_response(response, save, options)