amsdal_ml 0.1.3__tar.gz → 0.1.4__tar.gz

amsdal_ml-0.1.4/CLAUDE.md

@@ -0,0 +1,171 @@
+# CLAUDE.md
+
+This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
+
+## Project Overview
+
+amsdal-ml is a machine learning plugin for the AMSDAL Framework that provides embeddings, vector search, and AI-driven features. It supports both synchronous and asynchronous modes, with primary focus on async operations using OpenAI models.
+
+## Development Commands
+
+### Environment Setup
+```bash
+# Install dependencies using hatch/uv
+pip install --upgrade uv hatch==1.14.2
+hatch env create
+hatch run sync
+```
+
+### Testing
+```bash
+# Run all tests with coverage
+hatch run cov
+
+# Run specific test file
+hatch run test tests/test_openai_model.py
+
+# Run tests with pytest directly (after env setup)
+pytest tests/
+pytest tests/agents_tests/  # Run agent-specific tests
+```
+
+### Code Quality
+```bash
+# Run all checks (style + typing)
+hatch run all
+
+# Style checks only
+hatch run style
+
+# Format code (fix style issues)
+hatch run fmt
+
+# Type checking
+hatch run typing
+```
+
+### Dependency Management
+```bash
+# Sync dependencies
+hatch run sync
+
+# Update lock file
+hatch run lock
+
+# Upgrade all dependencies
+hatch run lock-upgrade
+```
+
+### AMSDAL CLI Commands
+```bash
+# Generate new model
+amsdal generate model ModelName --format py
+
+# Generate property for model
+amsdal generate property --model ModelName property_name
+
+# Generate transaction
+amsdal generate transaction TransactionName
+
+# Generate hook
+amsdal generate hook --model ModelName on_create
+```
+
+## Architecture
+
+### Core Components
+
+**ML Models** (`amsdal_ml/ml_models/`)
+- Abstract base class `MLModel` defines the interface for all ML models
+- Supports both sync/async invoke and streaming methods
+- Primary implementation uses OpenAI API
+- All models must implement `setup()`, `teardown()`, `invoke()`, `ainvoke()`, `stream()`, and `astream()`
+- Custom error hierarchy: `ModelError`, `ModelConnectionError`, `ModelRateLimitError`, `ModelAPIError`
+
+**ML Ingesting** (`amsdal_ml/ml_ingesting/`)
+- `MLIngesting` abstract base handles text generation and embedding creation from data
+- Creates `EmbeddingData` records that link embeddings to source objects
+- Supports chunk-based processing with configurable depth and token limits
+- Both sync/async methods for text generation and embedding
+
+**ML Retrievers** (`amsdal_ml/ml_retrievers/`)
+- `MLRetriever` provides semantic search via similarity_search/asimilarity_search
+- Returns `RetrievalChunk` objects with object metadata, chunk text, distance, and tags
+- Supports filtering by include/exclude tags
+- Configurable k parameter for number of results
+
+**Agents** (`amsdal_ml/agents/`)
+- Abstract `Agent` base class for Q&A and task-oriented agents
+- Async-first design (sync methods raise NotImplementedError)
+- Returns `AgentOutput` with answer, used_tools, and citations
+- Supports streaming responses via `astream()`
+- File attachments supported through `FileAttachment` interface
+
+**MCP Integration**
+- **Server** (`amsdal_ml/mcp_server/`): Exposes retriever search as MCP tool via stdio
+- **Client** (`amsdal_ml/mcp_client/`): Supports both stdio and HTTP transports for calling MCP tools
+- Server accepts base64-encoded AMSDAL config for initialization
+
+**File I/O** (`amsdal_ml/fileio/`)
+- `BaseFileLoader` abstract class for uploading files to ML providers
+- `FileAttachment` represents processed attachments (types: PLAIN_TEXT, FILE_ID)
+- `FileItem` helper for creating attachments from paths, bytes, or strings
+
+### Data Models
+
+**EmbeddingModel** (`amsdal_ml/models/embedding_model.py`)
+- Core model storing embeddings in database
+- Links to source object via `data_object_class` and `data_object_id`
+- Stores 1536-dimensional vectors (OpenAI text-embedding-3-small default)
+- Includes chunk_index, raw_text, tags, and ml_metadata fields
+
+### Configuration
+
+**MLConfig** (`amsdal_ml/ml_config.py`)
+- Loaded from `.env` file using pydantic-settings
+- Key settings:
+  - `ml_model_class`: Path to ML model implementation
+  - `ml_retriever_class`: Path to retriever implementation
+  - `ml_ingesting_class`: Path to ingesting implementation
+  - `llm_model_name`: Default 'gpt-4o'
+  - `embed_model_name`: Default 'text-embedding-3-small'
+  - `embed_max_depth`, `embed_max_chunks`, `embed_max_tokens_per_chunk`: Chunking parameters
+  - `retriever_default_k`: Number of results for similarity search
+  - `openai_api_key`, `claude_api_key`: API credentials
+  - `embedding_targets`: List of models to embed
+
+**Database Config** (`config.yml`)
+- Defines AMSDAL connections (sqlite_history, sqlite_state, lock)
+- Resources config maps repository and lakehouse to connections
+- Set `async_mode: true` for async operations
+
+## Code Style
+
+- Python 3.11+ required
+- Uses Ruff for linting and formatting with 120-char line length
+- Single quotes enforced (`quote-style = "single"`)
+- Import ordering: force-single-line with order-by-type
+- Type checking via mypy with strict settings (disallow_any_generics, check_untyped_defs)
+- Excludes migrations directory from linting
+
+## Testing
+
+- Uses pytest with pytest-asyncio for async tests
+- Test fixtures in `tests/conftest.py` provide mocked OpenAI clients
+- `OPENAI_API_KEY` set to dummy value in tests via fixture
+- Coverage tracking with coverage.py
+
+## CI/CD
+
+The project uses self-hosted runners with two jobs:
+1. **license-check**: Validates third-party licenses using `license_check.py`
+2. **test-lint**: Runs on Python 3.11 and 3.12, executes `hatch run all` (style+typing) and `hatch run cov`
+
+## Key Patterns
+
+1. **Async-First**: Most components prioritize async methods; sync methods often raise NotImplementedError
+2. **Abstract Base Classes**: Heavy use of ABCs to define interfaces for models, retrievers, ingesters, and agents
+3. **Configuration via Pydantic**: Settings loaded from environment with type validation
+4. **AMSDAL Integration**: Uses AMSDAL's model system, manager, and connection framework
+5. **Chunking Strategy**: Text split into chunks with metadata preservation for better embedding quality
+6. **Tag-Based Filtering**: Embeddings tagged for fine-grained retrieval control

amsdal_ml-0.1.4/PKG-INFO

@@ -0,0 +1,235 @@
+Metadata-Version: 2.4
+Name: amsdal_ml
+Version: 0.1.4
+Summary: amsdal_ml plugin for AMSDAL Framework
+Requires-Python: >=3.11
+Requires-Dist: aiohttp==3.12.15
+Requires-Dist: amsdal-cli>=0.5.7
+Requires-Dist: amsdal-data>=0.5.9
+Requires-Dist: amsdal-models>=0.5.9
+Requires-Dist: amsdal-utils>=0.5.4
+Requires-Dist: amsdal>=0.5.6
+Requires-Dist: mcp>=0.1
+Requires-Dist: openai==1.100.2
+Requires-Dist: pydantic-settings==2.10.1
+Requires-Dist: pydantic==2.11.7
+Description-Content-Type: text/markdown
+
+# AMSDAL ML
+
+[![CI](https://github.com/amsdal/amsdal_ml/actions/workflows/ci.yml/badge.svg)](https://github.com/amsdal/amsdal_ml/actions/workflows/ci.yml)
+[![Python 3.11+](https://img.shields.io/badge/python-3.11+-blue.svg)](https://www.python.org/downloads/)
+
+Machine learning plugin for the AMSDAL Framework, providing embeddings, vector search, semantic retrieval, and AI agents with support for OpenAI models.
+
+## Features
+
+- **Vector Embeddings**: Generate and store embeddings for any AMSDAL model with automatic chunking
+- **Semantic Search**: Query your data using natural language with tag-based filtering
+- **AI Agents**: Build Q&A systems with streaming support and citation tracking
+- **Async-First**: Optimized for high-performance async operations
+- **MCP Integration**: Expose and consume tools via Model Context Protocol (stdio/HTTP)
+- **File Attachments**: Process and embed documents with built-in loaders
+- **Extensible**: Abstract base classes for custom models, retrievers, and ingesters
+
+## Installation
+
+```bash
+pip install amsdal-ml
+```
+
+### Requirements
+
+- Python 3.11 or higher
+- AMSDAL Framework 0.5.6+
+- OpenAI API key (for default implementations)
+
+## Quick Start
+
+### 1. Configuration
+
+Create a `.env` file in your project root:
+
+```env
+OPENAI_API_KEY=sk-your-api-key-here
+async_mode=true
+ml_model_class=amsdal_ml.ml_models.openai_model.OpenAIModel
+ml_retriever_class=amsdal_ml.ml_retrievers.openai_retriever.OpenAIRetriever
+ml_ingesting_class=amsdal_ml.ml_ingesting.openai_ingesting.OpenAIIngesting
+```
+
+Create a `config.yml` for AMSDAL connections:
+
+```yaml
+application_name: my-ml-app
+async_mode: true
+connections:
+  - name: sqlite_state
+    backend: sqlite-state-async
+    credentials:
+      - db_path: ./warehouse/state.sqlite3
+      - check_same_thread: false
+  - name: lock
+    backend: amsdal_data.lock.implementations.thread_lock.ThreadLock
+resources_config:
+  repository:
+    default: sqlite_state
+  lock: lock
+```
+
+### 2. Generate Embeddings
+
+```python
+from amsdal_ml.ml_ingesting.openai_ingesting import OpenAIIngesting
+from amsdal_ml.ml_config import ml_config
+
+# Initialize ingesting
+ingester = OpenAIIngesting(
+    model=MyModel,
+    embedding_field='embedding',
+)
+
+# Generate embeddings for an instance
+instance = MyModel(content='Your text here')
+embeddings = await ingester.agenerate_embeddings(instance)
+await ingester.asave(embeddings, instance)
+```
+
+### 3. Semantic Search
+
+```python
+from amsdal_ml.ml_retrievers.openai_retriever import OpenAIRetriever
+
+retriever = OpenAIRetriever()
+
+# Search for relevant content
+results = await retriever.asimilarity_search(
+    query='What is machine learning?',
+    k=5,
+    include_tags=['documentation']
+)
+
+for chunk in results:
+    print(f'{chunk.object_class}:{chunk.object_id} - {chunk.raw_text}')
+```
+
+### 4. Build an AI Agent
+
+```python
+from amsdal_ml.agents.default_qa_agent import DefaultQAAgent
+
+agent = DefaultQAAgent()
+
+# Ask questions
+output = await agent.arun('Explain vector embeddings')
+print(output.answer)
+print(f'Used tools: {output.used_tools}')
+
+# Stream responses
+async for chunk in agent.astream('What is semantic search?'):
+    print(chunk, end='', flush=True)
+```
+
+## Architecture
+
+### Core Components
+
+- **`MLModel`**: Abstract interface for LLM inference (invoke, stream, with attachments)
+- **`MLIngesting`**: Generate text and embeddings from data objects with chunking
+- **`MLRetriever`**: Semantic similarity search with tag-based filtering
+- **`Agent`**: Q&A and task-oriented agents with streaming and citations
+- **`EmbeddingModel`**: Database model storing 1536-dimensional vectors linked to source objects
+- **MCP Server/Client**: Expose retrievers as tools or consume external MCP services
+
+### Configuration
+
+All settings are managed via `MLConfig` in `.env`:
+
+```env
+# Model Configuration
+llm_model_name=gpt-4o
+llm_temperature=0.0
+embed_model_name=text-embedding-3-small
+
+# Chunking Parameters
+embed_max_depth=2
+embed_max_chunks=10
+embed_max_tokens_per_chunk=800
+
+# Retrieval Settings
+retriever_default_k=8
+```
+
+## Development
+
+### Setup
+
+```bash
+# Install dependencies
+pip install --upgrade uv hatch==1.14.2
+hatch env create
+hatch run sync
+```
+
+### Testing
+
+```bash
+# Run all tests with coverage
+hatch run cov
+
+# Run specific tests
+hatch run test tests/test_openai_model.py
+
+# Watch mode
+pytest tests/ -v
+```
+
+### Code Quality
+
+```bash
+# Run all checks (style + typing)
+hatch run all
+
+# Format code
+hatch run fmt
+
+# Type checking
+hatch run typing
+```
+
+### AMSDAL CLI
+
+```bash
+# Generate a new model
+amsdal generate model MyModel --format py
+
+# Generate property
+amsdal generate property --model MyModel embedding_field
+
+# Generate transaction
+amsdal generate transaction ProcessEmbeddings
+
+# Generate hook
+amsdal generate hook --model MyModel on_create
+```
+
+## MCP Server
+
+Run the retriever as an MCP server for integration with Claude Desktop or other MCP clients:
+
+```bash
+python -m amsdal_ml.mcp_server.server_retriever_stdio \
+  --amsdal-config "$(echo '{"async_mode": true, ...}' | base64)"
+```
+
+The server exposes a `search` tool for semantic search in your knowledge base.
+
+## License
+
+See `amsdal_ml/Third-Party Materials - AMSDAL Dependencies - License Notices.md` for dependency licenses.
+
+## Links
+
+- [AMSDAL Framework](https://github.com/amsdal/amsdal)
+- [Documentation](https://docs.amsdal.com)
+- [Issue Tracker](https://github.com/amsdal/amsdal_ml/issues)

amsdal_ml-0.1.4/README.md

@@ -0,0 +1,218 @@
+# AMSDAL ML
+
+[![CI](https://github.com/amsdal/amsdal_ml/actions/workflows/ci.yml/badge.svg)](https://github.com/amsdal/amsdal_ml/actions/workflows/ci.yml)
+[![Python 3.11+](https://img.shields.io/badge/python-3.11+-blue.svg)](https://www.python.org/downloads/)
+
+Machine learning plugin for the AMSDAL Framework, providing embeddings, vector search, semantic retrieval, and AI agents with support for OpenAI models.
+
+## Features
+
+- **Vector Embeddings**: Generate and store embeddings for any AMSDAL model with automatic chunking
+- **Semantic Search**: Query your data using natural language with tag-based filtering
+- **AI Agents**: Build Q&A systems with streaming support and citation tracking
+- **Async-First**: Optimized for high-performance async operations
+- **MCP Integration**: Expose and consume tools via Model Context Protocol (stdio/HTTP)
+- **File Attachments**: Process and embed documents with built-in loaders
+- **Extensible**: Abstract base classes for custom models, retrievers, and ingesters
+
+## Installation
+
+```bash
+pip install amsdal-ml
+```
+
+### Requirements
+
+- Python 3.11 or higher
+- AMSDAL Framework 0.5.6+
+- OpenAI API key (for default implementations)
+
+## Quick Start
+
+### 1. Configuration
+
+Create a `.env` file in your project root:
+
+```env
+OPENAI_API_KEY=sk-your-api-key-here
+async_mode=true
+ml_model_class=amsdal_ml.ml_models.openai_model.OpenAIModel
+ml_retriever_class=amsdal_ml.ml_retrievers.openai_retriever.OpenAIRetriever
+ml_ingesting_class=amsdal_ml.ml_ingesting.openai_ingesting.OpenAIIngesting
+```
+
+Create a `config.yml` for AMSDAL connections:
+
+```yaml
+application_name: my-ml-app
+async_mode: true
+connections:
+  - name: sqlite_state
+    backend: sqlite-state-async
+    credentials:
+      - db_path: ./warehouse/state.sqlite3
+      - check_same_thread: false
+  - name: lock
+    backend: amsdal_data.lock.implementations.thread_lock.ThreadLock
+resources_config:
+  repository:
+    default: sqlite_state
+  lock: lock
+```
+
+### 2. Generate Embeddings
+
+```python
+from amsdal_ml.ml_ingesting.openai_ingesting import OpenAIIngesting
+from amsdal_ml.ml_config import ml_config
+
+# Initialize ingesting
+ingester = OpenAIIngesting(
+    model=MyModel,
+    embedding_field='embedding',
+)
+
+# Generate embeddings for an instance
+instance = MyModel(content='Your text here')
+embeddings = await ingester.agenerate_embeddings(instance)
+await ingester.asave(embeddings, instance)
+```
+
+### 3. Semantic Search
+
+```python
+from amsdal_ml.ml_retrievers.openai_retriever import OpenAIRetriever
+
+retriever = OpenAIRetriever()
+
+# Search for relevant content
+results = await retriever.asimilarity_search(
+    query='What is machine learning?',
+    k=5,
+    include_tags=['documentation']
+)
+
+for chunk in results:
+    print(f'{chunk.object_class}:{chunk.object_id} - {chunk.raw_text}')
+```
+
+### 4. Build an AI Agent
+
+```python
+from amsdal_ml.agents.default_qa_agent import DefaultQAAgent
+
+agent = DefaultQAAgent()
+
+# Ask questions
+output = await agent.arun('Explain vector embeddings')
+print(output.answer)
+print(f'Used tools: {output.used_tools}')
+
+# Stream responses
+async for chunk in agent.astream('What is semantic search?'):
+    print(chunk, end='', flush=True)
+```
+
+## Architecture
+
+### Core Components
+
+- **`MLModel`**: Abstract interface for LLM inference (invoke, stream, with attachments)
+- **`MLIngesting`**: Generate text and embeddings from data objects with chunking
+- **`MLRetriever`**: Semantic similarity search with tag-based filtering
+- **`Agent`**: Q&A and task-oriented agents with streaming and citations
+- **`EmbeddingModel`**: Database model storing 1536-dimensional vectors linked to source objects
+- **MCP Server/Client**: Expose retrievers as tools or consume external MCP services
+
+### Configuration
+
+All settings are managed via `MLConfig` in `.env`:
+
+```env
+# Model Configuration
+llm_model_name=gpt-4o
+llm_temperature=0.0
+embed_model_name=text-embedding-3-small
+
+# Chunking Parameters
+embed_max_depth=2
+embed_max_chunks=10
+embed_max_tokens_per_chunk=800
+
+# Retrieval Settings
+retriever_default_k=8
+```
+
+## Development
+
+### Setup
+
+```bash
+# Install dependencies
+pip install --upgrade uv hatch==1.14.2
+hatch env create
+hatch run sync
+```
+
+### Testing
+
+```bash
+# Run all tests with coverage
+hatch run cov
+
+# Run specific tests
+hatch run test tests/test_openai_model.py
+
+# Watch mode
+pytest tests/ -v
+```
+
+### Code Quality
+
+```bash
+# Run all checks (style + typing)
+hatch run all
+
+# Format code
+hatch run fmt
+
+# Type checking
+hatch run typing
+```
+
+### AMSDAL CLI
+
+```bash
+# Generate a new model
+amsdal generate model MyModel --format py
+
+# Generate property
+amsdal generate property --model MyModel embedding_field
+
+# Generate transaction
+amsdal generate transaction ProcessEmbeddings
+
+# Generate hook
+amsdal generate hook --model MyModel on_create
+```
+
+## MCP Server
+
+Run the retriever as an MCP server for integration with Claude Desktop or other MCP clients:
+
+```bash
+python -m amsdal_ml.mcp_server.server_retriever_stdio \
+  --amsdal-config "$(echo '{"async_mode": true, ...}' | base64)"
+```
+
+The server exposes a `search` tool for semantic search in your knowledge base.
+
+## License
+
+See `amsdal_ml/Third-Party Materials - AMSDAL Dependencies - License Notices.md` for dependency licenses.
+
+## Links
+
+- [AMSDAL Framework](https://github.com/amsdal/amsdal)
+- [Documentation](https://docs.amsdal.com)
+- [Issue Tracker](https://github.com/amsdal/amsdal_ml/issues)

amsdal_ml-0.1.4/RELEASE.md

@@ -0,0 +1,180 @@
+# Release Guide
+
+Follow these steps each time you release a new version of `amsdal-ml`.
+
+---
+
+## Step 0: Check Working Directory
+
+Before starting, ensure you have a clean working directory:
+
+```bash
+git status
+```
+
+**If you have uncommitted changes:**
+1. Review the changes: `git diff`
+2. Decide what to do:
+   - **Commit them now** if they should be part of this release:
+     ```bash
+     git add .
+     git commit -m "Your commit message"
+     git push origin main
+     ```
+   - **Stash them** if they're unrelated to this release:
+     ```bash
+     git stash
+     # After release, restore with: git stash pop
+     ```
+   - **Discard them** if they're not needed:
+     ```bash
+     git restore .  # Be careful - this cannot be undone!
+     ```
+
+---
+
+## Step 1: Update Version Number
+
+Edit `amsdal_ml/__about__.py`:
+
+```python
+__version__ = '0.1.4'  # Change to your new version
+```
+
+**Version format**: `MAJOR.MINOR.PATCH`
+- PATCH: Bug fixes (e.g., `0.1.3` → `0.1.4`)
+- MINOR: New features (e.g., `0.1.4` → `0.2.0`)
+- MAJOR: Breaking changes (e.g., `0.2.0` → `1.0.0`)
+
+---
+
+## Step 2: Update Changelogs
+
+### 2a. Update `latest-changelogs.md`
+
+Replace entire content with your new release notes:
+
+```markdown
+## [v0.1.4](https://pypi.org/project/amsdal_ml/0.1.4/) - 2025-10-15
+
+### Description of changes
+
+- First change
+- Second change
+- Third change
+```
+
+### 2b. Update `change-logs.md`
+
+Prepend the same content to the top of the file (keep existing entries below).
+
+---
+
+## Step 3: Run Quality Checks
+
+```bash
+hatch run all
+hatch run cov
+```
+
+All checks must pass before continuing.
+
+---
+
+## Step 4: Create Release Branch
+
+```bash
+git checkout -b release/v0.1.4
+```
+
+---
+
+## Step 5: Commit and Push
+
+```bash
+git add amsdal_ml/__about__.py latest-changelogs.md change-logs.md
+git commit -m "Release v0.1.4"
+git push origin release/v0.1.4
+```
+
+---
+
+## Step 6: Create Pull Request
+
+1. Go to: https://github.com/amsdal/amsdal_ml/pulls
+2. Click "New pull request"
+3. Set base: `main` ← compare: `release/v0.1.4`
+4. Title: `Release v0.1.4`
+5. Add description with changelog content
+6. Create and merge the PR
+
+---
+
+## Step 7: Checkout Main and Pull
+
+```bash
+git checkout main
+git pull origin main
+```
+
+---
+
+## Step 8: Create and Push Tag
+
+```bash
+git tag -a v0.1.4 -m "Release v0.1.4"
+git push origin v0.1.4
+```
+
+**Important**: Tag must start with `v` (e.g., `v0.1.4`)
+
+---
+
+## Step 9: Monitor CI/CD
+
+Go to: https://github.com/amsdal/amsdal_ml/actions
+
+Wait for all jobs to complete:
+1. ✅ License check
+2. ✅ Build
+3. ✅ Publish to PyPI
+4. ✅ Create GitHub release
+
+---
+
+## Step 10: Verify Release
+
+### Check PyPI
+https://pypi.org/project/amsdal-ml/
+
+### Check GitHub Releases
+https://github.com/amsdal/amsdal_ml/releases
+
+### Test Installation
+```bash
+pip install --upgrade amsdal-ml
+python -c "import amsdal_ml; print(amsdal_ml.__version__)"
+```
+
+---
+
+## Done! ✅
+
+---
+
+## Troubleshooting
+
+**If CI fails:**
+1. Check logs at https://github.com/amsdal/amsdal_ml/actions
+2. Fix the issue
+3. Delete and recreate the tag:
+   ```bash
+   git tag -d v0.1.4
+   git push origin :refs/tags/v0.1.4
+   git tag -a v0.1.4 -m "Release v0.1.4"
+   git push origin v0.1.4
+   ```
+
+**If you need to rollback:**
+- Release a new patch version with the fix (recommended)
+- Or yank the release from PyPI using `twine`

amsdal_ml-0.1.4/amsdal_ml/__about__.py

@@ -0,0 +1 @@
+__version__ = '0.1.4'

{amsdal_ml-0.1.3 → amsdal_ml-0.1.4}/amsdal_ml/agents/retriever_tool.py

@@ -27,7 +27,16 @@ class RetrieverArgs(BaseModel):
     exclude_tags: Optional[list[str]] = None
 
 
-_retriever = OpenAIRetriever()
+class _RetrieverSingleton:
+    """Singleton holder for lazy retriever initialization."""
+    _instance: Optional[OpenAIRetriever] = None
+
+    @classmethod
+    def get(cls) -> OpenAIRetriever:
+        """Lazy initialization of retriever to ensure env vars are loaded."""
+        if cls._instance is None:
+            cls._instance = OpenAIRetriever()
+        return cls._instance
 
 
 async def retriever_search(
@@ -39,7 +48,8 @@ async def retriever_search(
     logging.info(
         f"retriever_search called with query={query}, k={k}, include_tags={include_tags}, exclude_tags={exclude_tags}"
     )
-    chunks = await _retriever.asimilarity_search(
+    retriever = _RetrieverSingleton.get()
+    chunks = await retriever.asimilarity_search(
         query=query,
         k=k,
         include_tags=include_tags,

{amsdal_ml-0.1.3 → amsdal_ml-0.1.4}/amsdal_ml/mcp_client/stdio_client.py

@@ -94,7 +94,7 @@ class StdioClient(ToolClient):
 
         if not self._persist:
             async with AsyncExitStack() as stack:
-                params = StdioServerParameters(command=self._command, args=self._args)
+                params = StdioServerParameters(command=self._command, args=self._args, env=os.environ.copy())
                 rx, tx = await stack.enter_async_context(stdio_client(params))
                 s = await stack.enter_async_context(ClientSession(rx, tx))
                 await s.initialize()
@@ -123,7 +123,7 @@ class StdioClient(ToolClient):
     async def call(self, tool_name: str, args: dict[str, Any], *, timeout: float | None = None) -> Any:
         if not self._persist:
             async with AsyncExitStack() as stack:
-                params = StdioServerParameters(command=self._command, args=self._args)
+                params = StdioServerParameters(command=self._command, args=self._args, env=os.environ.copy())
                 rx, tx = await stack.enter_async_context(stdio_client(params))
                 s = await stack.enter_async_context(ClientSession(rx, tx))
                 await s.initialize()

{amsdal_ml-0.1.3 → amsdal_ml-0.1.4}/change-logs.md

@@ -1,3 +1,13 @@
+## [v0.1.4](https://pypi.org/project/amsdal_ml/0.1.4/) - 2025-10-15
+
+### Fixed retriever initialization in K8s environments
+
+- Fixed lazy initialization of OpenAIRetriever to ensure env vars are loaded
+- Added missing env parameter to stdio_client for non-persistent sessions
+- Environment variables now properly passed to MCP stdio subprocesses
+- Updated README.md to be production-ready
+- Added RELEASE.md with step-by-step release guide
+
 ## [v0.1.3](https://pypi.org/project/amsdal_ml/0.1.3/) - 2025-10-13
 
 ### Pass env vars into stdio server

amsdal_ml-0.1.4/latest-changelogs.md

@@ -0,0 +1,9 @@
+## [v0.1.4](https://pypi.org/project/amsdal_ml/0.1.4/) - 2025-10-15
+
+### Fixed retriever initialization in K8s environments
+
+- Fixed lazy initialization of OpenAIRetriever to ensure env vars are loaded
+- Added missing env parameter to stdio_client for non-persistent sessions
+- Environment variables now properly passed to MCP stdio subprocesses
+- Updated README.md to be production-ready
+- Added RELEASE.md with step-by-step release guide

amsdal_ml-0.1.3/PKG-INFO

@@ -1,69 +0,0 @@
-Metadata-Version: 2.4
-Name: amsdal_ml
-Version: 0.1.3
-Summary: amsdal_ml plugin for AMSDAL Framework
-Requires-Python: >=3.11
-Requires-Dist: aiohttp==3.12.15
-Requires-Dist: amsdal-cli>=0.5.7
-Requires-Dist: amsdal-data>=0.5.9
-Requires-Dist: amsdal-models>=0.5.9
-Requires-Dist: amsdal-utils>=0.5.4
-Requires-Dist: amsdal>=0.5.6
-Requires-Dist: mcp>=0.1
-Requires-Dist: openai==1.100.2
-Requires-Dist: pydantic-settings==2.10.1
-Requires-Dist: pydantic==2.11.7
-Description-Content-Type: text/markdown
-
-# amsdal-ml
-
-This plugin extends the AMSDAL Framework with machine learning utilities, 
-including custom models for embeddings, properties for ML metadata, and 
-hooks for working with vector search and AI-driven features.
-
-## Plugin Structure
-
-- `src/models/` - Contains model definitions in Python format
-- `src/transactions/` - Contains transaction definitions
-- `pyproject.toml` - Plugin configuration file
-- `config.yml` - Configuration for connections
-
-## Installing this Plugin
-
-To use this plugin in an AMSDAL application:
-
-1. Copy the plugin directory to your AMSDAL application
-2. Import the models and transactions as needed
-3. Register the plugin in your application configuration
-
-## Development
-
-This plugin uses sync mode.
-
-### Adding Models
-
-```bash
-amsdal generate model ModelName --format py
-```
-
-### Adding Properties
-
-```bash
-amsdal generate property --model ModelName property_name
-```
-
-### Adding Transactions
-
-```bash
-amsdal generate transaction TransactionName
-```
-
-### Adding Hooks
-
-```bash
-amsdal generate hook --model ModelName on_create
-```
-
-## Testing
-
-Test your plugin by integrating it with an AMSDAL application and running the application's test suite.

amsdal_ml-0.1.3/README.md

@@ -1,52 +0,0 @@
-# amsdal-ml
-
-This plugin extends the AMSDAL Framework with machine learning utilities, 
-including custom models for embeddings, properties for ML metadata, and 
-hooks for working with vector search and AI-driven features.
-
-## Plugin Structure
-
-- `src/models/` - Contains model definitions in Python format
-- `src/transactions/` - Contains transaction definitions
-- `pyproject.toml` - Plugin configuration file
-- `config.yml` - Configuration for connections
-
-## Installing this Plugin
-
-To use this plugin in an AMSDAL application:
-
-1. Copy the plugin directory to your AMSDAL application
-2. Import the models and transactions as needed
-3. Register the plugin in your application configuration
-
-## Development
-
-This plugin uses sync mode.
-
-### Adding Models
-
-```bash
-amsdal generate model ModelName --format py
-```
-
-### Adding Properties
-
-```bash
-amsdal generate property --model ModelName property_name
-```
-
-### Adding Transactions
-
-```bash
-amsdal generate transaction TransactionName
-```
-
-### Adding Hooks
-
-```bash
-amsdal generate hook --model ModelName on_create
-```
-
-## Testing
-
-Test your plugin by integrating it with an AMSDAL application and running the application's test suite.

amsdal_ml-0.1.3/amsdal_ml/__about__.py

@@ -1 +0,0 @@
-__version__ = '0.1.3'

amsdal_ml-0.1.3/latest-changelogs.md

@@ -1,6 +0,0 @@
-## [v0.1.3](https://pypi.org/project/amsdal_ml/0.1.3/) - 2025-10-13
-
-### Pass env vars into stdio server
-
-- Pass env vars into stdio server
-- cleanup of app.py