Generating contracts with a raw LLM without templates is like writing code without a linter: it works until hallucinations hit production. Contract number 7, date 30 February, signature from the wrong party—real story from a project where we had to redo such a solution.
Our AI document generation and contract automation services use hybrid document generation combining Jinja2 document templates with LLM for flexibility. We design AI document templates using python-docx and Jinja2. We also implement python-docx AI for document processing and provide legal AI documents validation. For complex scenarios, we employ RAG document management and fine-tuning for documents to improve accuracy. Our MLOps documents pipeline and LLM validation ensure quality.
Here's how we do it.
Understanding the Hybrid Approach
Choosing the Right Approach
| Approach |
Speed |
Field Accuracy |
Text Flexibility |
| Templating (Jinja2 + python-docx) |
<1 sec |
100% |
Low |
| LLM generation with structure |
5-15 sec |
90-95% (hallucination risk) |
High |
| Hybrid (template + LLM) |
1-3 sec |
99% (deterministic fields) + LLM for blocks |
Medium |
The hybrid approach is 3x faster than full LLM generation while maintaining field accuracy—it's what we recommend for most tasks.
Why Hybrid Architecture Is More Effective
We fix the document structure (headers, fields, signatures) via a template with variable substitution. Then we generate 'live' text blocks (subject matter, obligations) using an LLM. This avoids hallucinations in critical fields while keeping flexibility for variable content. Time savings amount to up to 60% on preparing standard documents compared to manual entry. According to Gartner study, hybrid systems reduce document errors by 70%.
Technical Implementation
Our implementation leverages AI document generation, contract automation, and LLM document generation techniques.
Implementing Hybrid Generation in Python
We use a combination of python-docx + Jinja2 for templates and LangChain to call the LLM. Example code below.
from docx import Document
from docx.shared import Pt
import jinja2
def generate_contract(template_path: str, data: ContractData) -> bytes:
doc = Document(template_path)
for paragraph in doc.paragraphs:
for key, value in data.dict().items():
if f"{{{{{key}}}}}" in paragraph.text:
for run in paragraph.runs:
run.text = run.text.replace(f"{{{{{key}}}}}", str(value))
subject_section = find_section(doc, "Subject of the Contract")
generated_subject = llm.generate(
f"Write the 'Subject of the Contract' section for a {data.contract_type}:\n{data.subject_description}"
)
replace_section_content(subject_section, generated_subject)
from io import BytesIO
buffer = BytesIO()
doc.save(buffer)
return buffer.getvalue()
Advanced Techniques: RAG and Versioning
For documents requiring up-to-date data (e.g., legal references), we add RAG (Retrieval-Augmented Generation). We vectorize regulations in ChromaDB, and during generation we retrieve relevant chunks and feed them into the LLM context. This reduces the likelihood of hallucinations on legal norms.
Document templates are code like everything else. Without versioning, you can't roll back after a lawyer's mistake. We use Git for templates + semantic versioning (semver). Each generated document includes a template version ID.
Case Studies and Results
Supply Contract with Hybrid Generation
In one project, we needed to automate supply contracts for a retailer with 500 counterparties. Manual preparation took lawyers up to 2 hours per contract. We implemented a hybrid scheme: 70% of fields (requisites, dates, amounts) were filled from CRM via Jinja2, while sections 'Delivery Procedure' and 'Liability of Parties' were generated via GPT-4 with RAG augmentation from an internal precedent database. As a result, preparation time dropped to 5 minutes per contract, and the return-for-revision rate fell from 30% to 2%. The retailer saved approximately $200,000 annually in legal fees.
Accuracy Comparison by Document Type
| Document Type |
Template Accuracy |
Hybrid Accuracy |
| NDA |
100% |
100% |
| Supply contract |
70% (only fields) |
95% |
| HR order |
100% |
100% |
Services and Timeline
What We Provide and Implementation Steps
- Template design and implementation (python-docx/Jinja2).
- LLM integration (GPT-4, Claude) via LangChain.
- RAG module for legal norms (optional).
- Template versioning in Git.
- Testing on 100+ scenarios.
- Containerization and deployment (Docker, Kubernetes).
- Training materials and support.
Deliverables
- Documentation: detailed specification, user guide, API reference.
- Access: template repository with version control, LLM endpoints.
- Training: 2-3 sessions for your team (documentation and hands-on).
- Support: 3 months of post-deployment assistance (email, chat).
Implementation Checklist: 5 Steps
- Audit: Analyze 5-10 typical documents, identify variables and template blocks.
- Design: Create template structure, choose LLM and vector database (if RAG needed).
- Implementation: Code in Python, test on real data, A/B compare with manual entry.
- Testing: Validate on 100+ data variants, lawyer review.
- Deploy: Containerize (Docker), deploy in your infrastructure (on-prem or cloud).
Timeline and Pricing
- Simple template (one document type) — from 3 to 5 days.
- Comprehensive solution (3+ document types, CRM integration) — from 2 to 4 weeks.
- Pricing is determined individually after audit, with typical costs ranging from $5,000 for a simple template to $20,000 for a comprehensive solution.
With over 10 years of experience and 50+ successful projects, we guarantee quality—each project undergoes mandatory legal validation. Order AI document generation implementation and reduce your documentation preparation costs.
Common Pitfalls
Typical Mistakes in AI Document Generation
- Feeding the entire text to LLM without a template — high risk of hallucinations in fields.
- Ignoring RAG for legal norms — contract may reference outdated laws.
- Lack of template versioning — impossible to roll back erroneous changes.
- No post-processing — dates, amounts, and signatures must be automatically checked.
Our engineers help avoid these mistakes at the design stage. Reach out for a consultation.
For fine-tuning for documents, we offer specialized models. Our MLOps documents pipeline ensures version control. LLM validation is integrated in the final review step.
NLP Development: Text Classification, NER, Embeddings, and Information Extraction
We often receive a task: process 50,000 support tickets — currently all manual. Dataset — 3,000 labeled examples, 12 categories, imbalance: one category occupies 40% of the sample, three at 1-2% each. Baseline accuracy — 78%. Sounds decent until you look at recall for rare classes: 0.31, 0.44, 0.28. These classes — complaints and churn threats — are most important to the business.
This is a typical NLP development project. The problem is not the algorithm but that accuracy is the wrong metric. Our experience across 30+ projects shows: we start by analyzing business metrics and only then choose the model.
Why accuracy is not the right metric for rare classes?
Accuracy ignores imbalance. If the "churn" class appears in 2% of cases, the model can predict "all good" and get 98% accuracy — but the business loses clients. Solution: F1 macro (averaged over all classes) or weighted F1. For NER — strict entity F1 (exact matches only). We guarantee: after choosing the correct metric, model quality becomes measurable and predictable.
Text Classification: From BERT to Distillation
BERT-like models are the standard for classification. ruBERT-base or ruBERT-large from DeepPavlov for Russian. multilingual-e5-large — for multiple languages in one pipeline. XLM-RoBERTa-large — a strong multilingual backbone.
Fine-tuning for classification: add a classification head on top of the [CLS] token, train for 3-5 epochs with lr=2e-5, weight decay=0.01. For imbalance — weighted CrossEntropyLoss or focal loss with gamma=2.0. Contact us — we will show a code snippet.
Imbalance case study. Dataset — 3,000 examples, imbalance 1:20. Solution: class_weight via sklearn + CrossEntropyLoss. Additionally — augmentation of rare classes via backtranslation (ru→en→ru through MarianMT). Recall for rare classes rose from 0.31 to 0.67 with a slight drop in accuracy (76%→74%). Full NLP development end-to-end took 3 weeks.
Distillation for production. BERT-large gives F1 0.89, but inference on CPU — 180ms. Distillation into DistilBERT or ruBERT-tiny2 reduces latency to 25ms with F1 0.84. Export to ONNX Runtime provides an additional 1.5-2x speedup. DistilBERT achieves 7x lower latency than BERT-large with only a 5% drop in macro F1 – a typical production trade-off.
| Model |
F1 macro |
Latency (CPU) |
Size |
| BERT-large |
0.89 |
180 ms |
1.3 GB |
| DistilBERT |
0.84 |
25 ms |
250 MB |
| ruBERT-tiny2 |
0.81 |
12 ms |
120 MB |
| DistilBERT + ONNX |
0.84 |
14 ms |
150 MB |
How to choose between BERT and LLM for your task?
For most classification and extraction tasks, BERT-sized models offer the best trade-off between cost and performance. Shift to LLMs only when the task demands generation, complex reasoning, or zero-shot generalization.
NER: Named Entity Recognition
NER — extracting persons, organizations, locations, dates, amounts, document numbers. For general categories (PER, ORG, LOC), pre-trained models work well. For specialized ones (medical terms, legal concepts) — fine-tuning is needed.
Data annotation. The main cost of an NER project. For a quality model — 500-2,000 labeled sentences per entity type. Tools: Label Studio (open source) or Prodigy (by spaCy creators). IOB2 format — standard.
Architecture. Token classification on top of BERT: each token gets a label (B-PER, I-PER, O). spaCy 3.x with transformer pipeline — a convenient production choice.
Nested entities. Standard IOB models cannot handle nested entities (organization inside an address). For such tasks — span-based NER: SpanBERT or SpERT. More complex but correct.
Post-processing is mandatory. The model predicts tokens — normalized entities are needed. Date — dateparser. Amounts — regex + validation. Names — deduplication via rapidfuzz. Included in our standard delivery.
Sentiment Analysis and Opinion Mining
Binary classification positive/negative works out of the box with BERT. Complexity — aspect-based sentiment analysis (ABSA): "the restaurant has good food but terrible service." For ABSA: aspect extraction (NER) + sentiment per aspect. Joint models BERT-for-ABSA — quality on Russian data is lower due to dataset scarcity. RuSentiment, SentiRuEval — main resources.
For production with simple positive/negative/neutral: distil models are enough. Three classes, balanced dataset, 2,000+ examples — F1 macro 0.82-0.87 in 1-2 days.
Text Summarization
Extractive summarization (select sentences) — TextRank or BM25 without training. Fast, no hallucinations. Good for long documents.
Abstractive (generates new text) — seq2seq: mT5, mBART, FRED-T5, ruT5-large. For production via LLM API (GPT-4, Claude) — often the best cost/quality/speed trade-off.
Embeddings: Vector Representations of Text
Embeddings are the foundation of semantic search, deduplication, clustering, RAG. Quality critically affects downstream tasks.
Models. E5-large-v2, BGE-M3, multilingual-e5-large — strong multilingual embedders. sentence-transformers/paraphrase-multilingual-mpnet-base-v2 — fast option. For Russian: ru-en-RoSBERTa (Skoltech) performs well on semantic textual similarity.
Embedding quality evaluation uses the MTEB benchmark as standard. But top results on MTEB don't guarantee success on a domain dataset — we build domain-specific eval.
Fine-tuning embeddings. If standard models don't give the required Recall@k — contrastive learning on domain pairs with MultipleNegativesRankingLoss. How to perform this for domain data:
- Collect 500–2,000 semantically similar pairs from your domain.
- Apply MultipleNegativesRankingLoss with a batch size of 32–64.
- Train for 1–3 epochs using AdamW (lr=2e-5).
- Evaluate Recall@k on a held-out domain test set.
This approach yields a 5–15% improvement in Recall@k in practice.
Dimensionality and storage. E5-large: 1024 dim, float32 — 4KB per vector. For 10M documents — 40GB. Quantization int8 reduces to 10GB. FAISS IVF_PQ — more compact but with losses. Included in our deployment recommendations.
Information Extraction
Structured extraction is a frequent task. Examples: key contract terms, technical characteristics, dates and amounts from invoices.
- Regex + rule-based. For INN, OGRN, amounts, dates — more reliable than neural networks. No data required.
- NER + post-processing. For variable formats.
- LLM with structured output. GPT‑4 / Claude with JSON schema — for complex documents. Cost: minimal per document. For 10k+ documents/day — we calculate the economics.
We guarantee a hybrid: regex/NER for typical fields + LLM for edge cases. Our guarantee is backed by years of production experience and more than 30 projects.
Work Stages
| Stage |
Duration |
What's included |
| Data and metric analysis |
3-5 days |
Class distribution, text lengths, baseline |
| Baseline (TF‑IDF + LogReg) |
1 day |
Quick estimate of gap with deep models |
| Training and validation |
1-2 weeks |
k‑fold, early stopping, error analysis |
| Deployment (ONNX + FastAPI) |
1-2 weeks |
REST API, batching, monitoring |
| Documentation and training |
2-3 days |
Model card, API docs, team training |
Prototype on existing data — 1-3 weeks. Production system with CI/CD — 1.5–2.5 months. Cost is calculated individually — get a consultation for a project estimate.
What's Included
- Model and pipeline architecture documentation
- Access to the model via REST API (FastAPI + ONNX)
- Client team training (2-hour webinar + Q&A)
- Accuracy guarantee on the agreed test set
- Months of post-delivery support (bug fixes, adaptation to new data)
Our Experience
Years of NLP projects from classification to RAG systems. The team includes ML engineers experienced with Hugging Face, spaCy, LangChain, MLOps. We use vLLM, Kubeflow, Weights & Biases — a production stack, not toys. Contact us to evaluate your NLP project within two days — request a free consultation on your text processing pipeline.