Files
encoach_backend_v4/custom_addons/encoach_vector/services/indexer.py
Yamen Ahmad 6a62a43d61 feat: Generation Page AI workflows + AI/Vector modules + exam session fixes
Generation Page (complete rebuild):
- Full production-parity exam generation wizard with 4 IELTS modules
- Reading: AI passage gen, 5 exercise types (MCQ, Fill, Write, T/F, Match)
- Listening: 4 section types, AI context gen, TTS audio gen (ElevenLabs)
- Writing: Task 1/2, AI instruction gen, word limits, marks
- Speaking: 3 parts, AI script gen, avatar video gen (7 avatars)
- Per-module config: timer, CEFR difficulty, access, approval, rubrics
- Exam submission workflow (draft/published)

Exam Structures:
- New encoach.exam.structure model + CRUD controller
- ExamStructuresPage wired to real API

AI Module (encoach_ai):
- OpenAI service, ElevenLabs TTS, AWS Polly, ELAI avatars
- AI settings model with Odoo config parameters
- 7 generation endpoints (passage, exercises, instructions, scripts, context)

Vector Module (encoach_vector):
- pgvector integration for RAG-based content search
- Embedding service with sentence-transformers

Exam Session Fixes:
- Fixed ExamSession.tsx field mapping (question_type→type, exam_title→title)
- Fixed submit payload to include attempt_id and answers
- Fixed normalizeType to handle null/undefined

Tested: 12/12 API tests passed, browser-verified with real OpenAI calls
Made-with: Cursor
2026-04-11 14:27:03 +04:00

128 lines
3.6 KiB
Python

"""Indexer — batch-indexes existing Odoo records into the vector store."""
import logging
_logger = logging.getLogger(__name__)
MODEL_CONFIG = [
{
'model': 'op.course',
'content_type': 'course',
'text_field': 'name',
'description_field': 'description',
'metadata_fields': [],
},
{
'model': 'encoach.resource',
'content_type': 'resource',
'text_field': 'name',
'description_field': 'content',
'metadata_fields': ['type', 'cefr_level', 'difficulty'],
},
{
'model': 'encoach.question',
'content_type': 'question',
'text_field': 'name',
'description_field': None,
'metadata_fields': ['question_type', 'difficulty', 'skill'],
},
{
'model': 'encoach.course.module',
'content_type': 'module',
'text_field': 'name',
'description_field': 'description',
'metadata_fields': ['skill'],
},
{
'model': 'encoach.ai.generation.log',
'content_type': 'generation_log',
'text_field': 'brief',
'description_field': 'generated_content',
'metadata_fields': ['course_type', 'status'],
},
]
def _get_text(record, config):
"""Extract indexable text from a record."""
parts = []
text_field = config.get('text_field', 'name')
if hasattr(record, text_field):
val = getattr(record, text_field)
if val:
parts.append(str(val))
desc_field = config.get('description_field')
if desc_field and hasattr(record, desc_field):
val = getattr(record, desc_field)
if val:
parts.append(str(val)[:2000])
return ' '.join(parts).strip()
def _get_metadata(record, config):
"""Extract metadata dict from a record."""
meta = {}
for f in config.get('metadata_fields', []):
if hasattr(record, f):
val = getattr(record, f)
if val:
meta[f] = str(val) if not isinstance(val, (int, float, bool)) else val
return meta
def index_model(env, config, batch_size=100):
"""Index all records of a single model."""
model_name = config['model']
Model = env.get(model_name)
if Model is None:
_logger.warning("Model %s not found, skipping", model_name)
return 0
Model = Model.sudo()
from .embedding_service import EmbeddingService
svc = EmbeddingService(env)
total = Model.search_count([])
indexed = 0
offset = 0
while offset < total:
records = Model.search([], limit=batch_size, offset=offset, order='id')
batch_data = []
for rec in records:
text = _get_text(rec, config)
if text:
batch_data.append({
'id': rec.id,
'text': text,
'metadata': _get_metadata(rec, config),
})
if batch_data:
indexed += svc.bulk_index(config['content_type'], batch_data)
offset += batch_size
env.cr.commit()
_logger.info("Indexed %d/%d records for %s", indexed, total, model_name)
return indexed
def index_all(env, batch_size=100):
"""Index all configured models."""
total = 0
for config in MODEL_CONFIG:
try:
total += index_model(env, config, batch_size)
except Exception:
_logger.exception("Failed to index %s", config['model'])
_logger.info("Total records indexed: %d", total)
return total
def cron_reindex(env):
"""Cron entry point for periodic re-indexing."""
_logger.info("Starting scheduled vector re-index")
return index_all(env)