lmcore 0.1.1__py3-none-any.whl

lmcore/__init__.py

@@ -0,0 +1,13 @@
+from .config import load_config
+from .llm import get_chat_client, get_embed_client, register_provider
+from .logging_utils import configure_logging, get_logger, Colors
+
+__all__ = [
+    "load_config",
+    "get_chat_client",
+    "get_embed_client",
+    "register_provider",
+    "configure_logging",
+    "get_logger",
+    "Colors",
+]

lmcore/config.py

@@ -0,0 +1,24 @@
+from dotenv import load_dotenv
+import os
+
+
+def load_config(required: list[str], optional: dict[str, str] = {}) -> dict:
+    load_dotenv()
+
+    config = {}
+    missing = []
+
+    for key in required:
+        val = os.getenv(key)
+        if val is None:
+            missing.append(key)
+        else:
+            config[key] = val
+
+    if missing:
+        raise EnvironmentError(f"Missing required env vars: {missing}")
+
+    for key, default in optional.items():
+        config[key] = os.getenv(key, default)
+
+    return config

lmcore/llm.py

@@ -0,0 +1,102 @@
+from __future__ import annotations
+import yaml
+import os
+from typing import Callable
+
+_registry: dict[str, Callable] = {}
+
+
+def register_provider(name: str, factory_fn: Callable) -> None:
+    _registry[name] = factory_fn
+
+
+def _load_arch(models_yaml_path: str, arch: str) -> dict:
+    if not os.path.exists(models_yaml_path):
+        raise FileNotFoundError(f"models.yaml not found at: {models_yaml_path}")
+
+    with open(models_yaml_path, "r") as f:
+        data = yaml.safe_load(f)
+
+    if arch not in data:
+        available = list(data.keys())
+        raise KeyError(f"Arch '{arch}' not found in {models_yaml_path}. Available: {available}")
+
+    return data[arch]
+
+
+def _resolve_api_key(cfg: dict) -> dict:
+    if "api_key_env" in cfg:
+        key = os.getenv(cfg["api_key_env"])
+        if key is None:
+            raise EnvironmentError(f"Env var '{cfg['api_key_env']}' is not set")
+        cfg = {**cfg, "api_key": key}
+    if "base_url_env" in cfg:
+        url = os.getenv(cfg["base_url_env"])
+        if url is None:
+            raise EnvironmentError(f"Env var '{cfg['base_url_env']}' is not set")
+        cfg = {**cfg, "base_url": url}
+    return cfg
+
+
+def get_chat_client(models_yaml_path: str, arch: str, index: int = 0):
+    arch_config = _load_arch(models_yaml_path, arch)
+
+    if "chat" not in arch_config:
+        raise ValueError(f"Arch '{arch}' has no chat config in {models_yaml_path}")
+
+    clients = arch_config["chat"]
+    if index >= len(clients):
+        raise IndexError(
+            f"Arch '{arch}' has {len(clients)} chat client(s), index {index} is out of range"
+        )
+
+    cfg = _resolve_api_key(clients[index])
+    provider = cfg["provider"]
+
+    if provider not in _registry:
+        raise ValueError(f"Provider '{provider}' is not registered. Use register_provider() to add it.")
+
+    return _registry[provider](cfg)
+
+
+def get_embed_client(models_yaml_path: str, arch: str, index: int = 0):
+    arch_config = _load_arch(models_yaml_path, arch)
+
+    if "embed" not in arch_config:
+        raise ValueError(f"Arch '{arch}' has no embed config in {models_yaml_path}")
+
+    clients = arch_config["embed"]
+    if index >= len(clients):
+        raise IndexError(
+            f"Arch '{arch}' has {len(clients)} embed client(s), index {index} is out of range"
+        )
+
+    cfg = _resolve_api_key(clients[index])
+    provider = cfg["provider"]
+
+    if provider not in _registry:
+        raise ValueError(f"Provider '{provider}' is not registered. Use register_provider() to add it.")
+
+    return _registry[provider](cfg)
+
+
+# --- Built-in provider factories ---
+
+def _build_ollama(cfg: dict):
+    from ollama import Client
+    return Client(host=cfg.get("base_url", "http://localhost:11434"))
+
+
+def _build_openai(cfg: dict):
+    from openai import OpenAI
+    return OpenAI(api_key=cfg["api_key"], base_url=cfg.get("base_url"))
+
+
+def _build_anthropic(cfg: dict):
+    from anthropic import Anthropic
+    return Anthropic(api_key=cfg["api_key"])
+
+
+register_provider("ollama", _build_ollama)
+register_provider("openai", _build_openai)
+register_provider("anthropic", _build_anthropic)

lmcore/logging_utils.py

@@ -0,0 +1,49 @@
+import logging
+
+
+class Colors:
+    HEADER    = '\033[95m'
+    OKBLUE    = '\033[94m'
+    OKCYAN    = '\033[96m'
+    OKGREEN   = '\033[92m'
+    WARNING   = '\033[93m'
+    FAIL      = '\033[91m'
+    ENDC      = '\033[0m'
+    BOLD      = '\033[1m'
+    UNDERLINE = '\033[4m'
+    DIM       = '\033[2m'
+
+
+class _ColorFormatter(logging.Formatter):
+    _LEVEL_COLORS = {
+        logging.DEBUG:    Colors.DIM,
+        logging.INFO:     Colors.OKCYAN,
+        logging.WARNING:  Colors.WARNING,
+        logging.ERROR:    Colors.FAIL,
+        logging.CRITICAL: Colors.FAIL + Colors.BOLD,
+    }
+
+    def format(self, record: logging.LogRecord) -> str:
+        color = self._LEVEL_COLORS.get(record.levelno, "")
+        record.levelname = f"{color}{record.levelname}{Colors.ENDC}"
+        record.name = f"{Colors.OKBLUE}{record.name}{Colors.ENDC}"
+        return super().format(record)
+
+
+def configure_logging(level: str = "INFO") -> None:
+    numeric = getattr(logging, level.upper(), logging.INFO)
+
+    handler = logging.StreamHandler()
+    handler.setFormatter(
+        _ColorFormatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+    )
+
+    root = logging.getLogger()
+    if not root.handlers:
+        root.addHandler(handler)
+
+    root.setLevel(numeric)
+
+
+def get_logger(name: str) -> logging.Logger:
+    return logging.getLogger(name)

lmcore-0.1.1.dist-info/METADATA

@@ -0,0 +1,242 @@
+Metadata-Version: 2.4
+Name: lmcore
+Version: 0.1.1
+Summary: Personal core toolkit for AI projects — config loading, LLM client factory, and logging.
+Author-email: Luigi Medrano <luigimedrano03@gmail.com>
+License: MIT
+Project-URL: Homepage, https://github.com/lm45562/lmcore
+Project-URL: Repository, https://github.com/lm45562/lmcore
+Keywords: ai,llm,openai,anthropic,ollama,config,logging
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: pyyaml>=6.0
+Provides-Extra: ollama
+Requires-Dist: ollama>=0.2.0; extra == "ollama"
+Provides-Extra: openai
+Requires-Dist: openai>=1.0.0; extra == "openai"
+Provides-Extra: anthropic
+Requires-Dist: anthropic>=0.25.0; extra == "anthropic"
+Provides-Extra: all
+Requires-Dist: ollama>=0.2.0; extra == "all"
+Requires-Dist: openai>=1.0.0; extra == "all"
+Requires-Dist: anthropic>=0.25.0; extra == "all"
+
+# lmcore
+
+Personal core toolkit for AI projects. Install once, call with inputs, never edit the package.
+
+Covers three things every project needs:
+- **Config** — load and validate `.env` secrets
+- **LLM** — build AI clients from a `models.yaml` file
+- **Logging** — colored, consistent logging across all modules
+
+---
+
+## Install
+
+```bash
+pip install lmcore
+```
+
+With specific provider dependencies:
+
+```bash
+pip install lmcore[openai]
+pip install lmcore[anthropic]
+pip install lmcore[ollama]
+pip install lmcore[all]       # all three providers
+```
+
+---
+
+## Config
+
+Call `load_config` once at startup. Pass the keys your project needs — the package handles loading from `.env`, validating, and returning a plain dict.
+
+```python
+from lmcore import load_config
+
+cfg = load_config(
+    required=["OPENAI_API_KEY", "CHROMA_API_KEY"],
+    optional={"ACTIVE_ARCH": "arch_1", "LOG_LEVEL": "INFO"}
+)
+```
+
+- `required` — must exist in `.env`, raises `EnvironmentError` if any are missing
+- `optional` — uses the default value if not set in `.env`
+
+---
+
+## LLM
+
+### models.yaml
+
+Lives in your **project root**. Define only what your project uses — `chat` and `embed` are both optional. Each section is a list so you can define multiple clients and reference them by index.
+
+```yaml
+arch_1:
+  chat:
+    - provider: openai
+      model: gpt-4o
+      api_key_env: OPENAI_API_KEY
+    - provider: anthropic
+      model: claude-sonnet-4-6
+      api_key_env: ANTHROPIC_API_KEY
+  embed:
+    - provider: ollama
+      model: nomic-embed-text
+      base_url_env: OLLAMA_URL
+
+arch_2:
+  chat:
+    - provider: anthropic
+      model: claude-opus-4-8
+      api_key_env: ANTHROPIC_API_KEY
+```
+
+**Supported fields per client:**
+
+| Field | Required | Description |
+|---|---|---|
+| `provider` | yes | `ollama`, `openai`, or `anthropic` |
+| `model` | yes | model name string |
+| `api_key_env` | when needed | env var name that holds the API key |
+| `base_url` | no | override default endpoint (e.g. local Ollama) |
+| `base_url_env` | no | env var name that holds the base URL (alternative to `base_url`) |
+
+### Getting clients
+
+```python
+from lmcore import get_chat_client, get_embed_client
+
+# index defaults to 0 — covers most single-client projects
+chat  = get_chat_client("models.yaml", arch="arch_1")
+embed = get_embed_client("models.yaml", arch="arch_1")
+
+# specify index for multiple clients
+openai_client    = get_chat_client("models.yaml", arch="arch_1", index=0)
+anthropic_client = get_chat_client("models.yaml", arch="arch_1", index=1)
+```
+
+### Adding a custom provider
+
+If you need a provider not built into the package, register it in your project:
+
+```python
+from lmcore import register_provider
+
+def _build_groq(cfg):
+    from groq import Groq
+    return Groq(api_key=cfg["api_key"])
+
+register_provider("groq", _build_groq)
+```
+
+Once registered, `groq` works as a `provider:` value in `models.yaml` like any built-in.
+
+---
+
+## Logging
+
+```python
+from lmcore import configure_logging, get_logger
+
+configure_logging()          # call once at app startup
+configure_logging("DEBUG")   # optional level override
+
+logger = get_logger(__name__)
+
+logger.info("Starting up")
+logger.warning("Something looks off")
+logger.error("Connection failed")
+```
+
+Color scheme:
+
+| Color | Level |
+|---|---|
+| Cyan | INFO |
+| Yellow | WARNING |
+| Red | ERROR / CRITICAL |
+| Dim | DEBUG |
+
+---
+
+## Full Example
+
+**Project layout:**
+```
+my-project/
+├── .env
+├── models.yaml
+└── app/
+    └── core/
+        ├── config.py
+        └── llm.py
+```
+
+**`.env`:**
+```
+OPENAI_API_KEY=sk-...
+ACTIVE_ARCH=arch_1
+```
+
+**`models.yaml`:**
+```yaml
+arch_1:
+  chat:
+    - provider: openai
+      model: gpt-4o
+      api_key_env: OPENAI_API_KEY
+    - provider: ollama
+      model: llama3
+      base_url_env: OLLAMA_URL
+```
+
+**`app/core/config.py`:**
+```python
+from lmcore import load_config, configure_logging
+
+configure_logging()
+
+cfg = load_config(
+    required=["OPENAI_API_KEY"],
+    optional={"ACTIVE_ARCH": "arch_1"}
+)
+```
+
+**`app/core/llm.py`:**
+```python
+from lmcore import get_chat_client
+from app.core.config import cfg
+
+chat = get_chat_client("models.yaml", arch=cfg["ACTIVE_ARCH"])
+```
+
+**Anywhere in the project:**
+```python
+from lmcore import get_logger
+from app.core.llm import chat
+
+logger = get_logger(__name__)
+
+response = chat.chat(
+    model="gpt-4o",
+    messages=[{"role": "user", "content": "Summarize this document."}]
+)
+logger.info(f"Done: {response.choices[0].message.content}")
+```
+
+---
+
+## Built-in Providers
+
+| Provider | `provider:` value | Install extra |
+|---|---|---|
+| Ollama | `ollama` | `pip install lmcore[ollama]` |
+| OpenAI | `openai` | `pip install lmcore[openai]` |
+| Anthropic | `anthropic` | `pip install lmcore[anthropic]` |

lmcore-0.1.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+lmcore/__init__.py,sha256=3V0bvc0IzxKobeEOAet0vbDl8KgOaoQn54n-KFd-P6Q,330
+lmcore/config.py,sha256=HxKjWKUQSMeVBSl5gaN13BfZmfMFz72NS6FxwjK6FlU,524
+lmcore/llm.py,sha256=UWlNs2A7MhHA9f9v7RDO3waAQ21Xe3z18051iBGgP7o,3109
+lmcore/logging_utils.py,sha256=8gDxkZupI_JN9nxTRy1XYHfV-tWJ3BrMqVtXexf8ytE,1338
+lmcore-0.1.1.dist-info/METADATA,sha256=FyvwOYjjobVVoreIcuqkSBgt9f4-XjQjuJlq_A5zdMM,5806
+lmcore-0.1.1.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+lmcore-0.1.1.dist-info/top_level.txt,sha256=tM4MX4grJmBML_00ZV6k23k1SLugbZ42qhRWi3TZjTk,7
+lmcore-0.1.1.dist-info/RECORD,,

lmcore-0.1.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

lmcore-0.1.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+lmcore