@r1z4x/tezgah 1.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +738 -0
- package/LICENSE +21 -0
- package/MANIFEST +753 -0
- package/README.de.md +169 -0
- package/README.es.md +168 -0
- package/README.fr.md +166 -0
- package/README.md +161 -0
- package/README.tr.md +163 -0
- package/agents/tezgah-reviewer.md +70 -0
- package/bin/codegen +337 -0
- package/bin/consult +797 -0
- package/bin/tezgah-adhd +50 -0
- package/bin/tezgah-agents +45 -0
- package/bin/tezgah-capture +66 -0
- package/bin/tezgah-context +96 -0
- package/bin/tezgah-decisions +213 -0
- package/bin/tezgah-design +953 -0
- package/bin/tezgah-docs +1199 -0
- package/bin/tezgah-doctor +497 -0
- package/bin/tezgah-dsh +37 -0
- package/bin/tezgah-dsh.cmd +70 -0
- package/bin/tezgah-gate +75 -0
- package/bin/tezgah-import-ai-research +439 -0
- package/bin/tezgah-index +32 -0
- package/bin/tezgah-lessons +148 -0
- package/bin/tezgah-mcp +322 -0
- package/bin/tezgah-migrate +324 -0
- package/bin/tezgah-pony +50 -0
- package/bin/tezgah-research +772 -0
- package/bin/tezgah-rollback +93 -0
- package/bin/tezgah-route +124 -0
- package/bin/tezgah-setup +5798 -0
- package/bin/tezgah-status +383 -0
- package/bin/tezgah-task +607 -0
- package/bin/tezgah-taste +423 -0
- package/bin/tezgah-triage +485 -0
- package/bin/tezgah.js +36 -0
- package/commands/adhd.md +21 -0
- package/commands/ponytail.md +21 -0
- package/docs/README.md +112 -0
- package/docs/architecture.md +226 -0
- package/docs/citations-baseline.json +59 -0
- package/docs/clarity-baseline.json +116 -0
- package/docs/contract.md +416 -0
- package/docs/environment.md +141 -0
- package/docs/evidence.md +831 -0
- package/docs/feature-audit.md +139 -0
- package/docs/gate.md +407 -0
- package/docs/glossary.md +201 -0
- package/docs/hosts.md +400 -0
- package/docs/index.json +471 -0
- package/docs/judge.md +193 -0
- package/docs/layers.md +189 -0
- package/docs/models.md +160 -0
- package/docs/operations.md +666 -0
- package/docs/research.md +809 -0
- package/docs/results/paired-outcome-2026-10.jsonl +254 -0
- package/docs/results/paired-outcome-2026-10.md +204 -0
- package/docs/skills.md +236 -0
- package/docs/status-line.md +315 -0
- package/docs/testing.md +287 -0
- package/docs/vision.md +117 -0
- package/hooks/hooks.json +33 -0
- package/hooks/projects-auto-init.py +98 -0
- package/hooks/projects-posttooluse.py +149 -0
- package/hooks/projects-pretooluse.py +40 -0
- package/hooks/projects-stop.py +60 -0
- package/hooks/tezgah_agents.py +817 -0
- package/hooks/tezgah_apps.py +266 -0
- package/hooks/tezgah_attest.py +255 -0
- package/hooks/tezgah_clarity.py +171 -0
- package/hooks/tezgah_components.py +140 -0
- package/hooks/tezgah_context.py +2953 -0
- package/hooks/tezgah_embed.py +765 -0
- package/hooks/tezgah_gate.py +2736 -0
- package/hooks/tezgah_guard.py +177 -0
- package/hooks/tezgah_index.py +124 -0
- package/hooks/tezgah_integrity.py +5141 -0
- package/hooks/tezgah_judge.py +487 -0
- package/hooks/tezgah_lang.py +230 -0
- package/hooks/tezgah_lessons.py +210 -0
- package/hooks/tezgah_models.py +855 -0
- package/hooks/tezgah_paths.py +1132 -0
- package/hooks/tezgah_policy.py +1039 -0
- package/hooks/tezgah_rank.py +77 -0
- package/hooks/tezgah_replay.py +1109 -0
- package/hooks/tezgah_research.py +5176 -0
- package/hooks/tezgah_shapes.py +305 -0
- package/hooks/tezgah_skill_pick.py +227 -0
- package/hooks/tezgah_snapshot.py +490 -0
- package/hooks/tezgah_stopspec.py +320 -0
- package/hooks/tezgah_task.py +458 -0
- package/hooks/tezgah_taste.py +201 -0
- package/hooks/tezgah_untrusted.py +112 -0
- package/hooks/tezgah_update.py +430 -0
- package/hosts/codex/hook.py +268 -0
- package/hosts/codex/hooks.json +26 -0
- package/hosts/cursor/hook.py +461 -0
- package/hosts/cursor/hooks.json +44 -0
- package/hosts/dsh/hooks.json +21 -0
- package/hosts/dsh/statusline/lib/client.js +138 -0
- package/hosts/dsh/statusline/lib/index.js +104 -0
- package/hosts/dsh/statusline/package.json +28 -0
- package/hosts/omp/hook.py +277 -0
- package/hosts/omp/tezgah-hook.ts.in +701 -0
- package/hosts/opencode/plugins/tezgah.js +2742 -0
- package/hosts/opencode/tui/tezgah-tui.tsx +151 -0
- package/output-styles/tezgah.md +142 -0
- package/package.json +32 -0
- package/packaging/build.sh +342 -0
- package/packaging/install.ps1 +166 -0
- package/packaging/install.sh +56 -0
- package/packaging/upgrade.sh +142 -0
- package/skills/ai-research/0-autoresearch-skill/SKILL.md +411 -0
- package/skills/ai-research/0-autoresearch-skill/references/agent-continuity.md +95 -0
- package/skills/ai-research/0-autoresearch-skill/references/progress-reporting.md +166 -0
- package/skills/ai-research/0-autoresearch-skill/references/skill-routing.md +219 -0
- package/skills/ai-research/0-autoresearch-skill/templates/findings.md +44 -0
- package/skills/ai-research/0-autoresearch-skill/templates/research-log.md +41 -0
- package/skills/ai-research/01-model-architecture/litgpt/SKILL.md +469 -0
- package/skills/ai-research/01-model-architecture/litgpt/references/custom-models.md +569 -0
- package/skills/ai-research/01-model-architecture/litgpt/references/distributed-training.md +452 -0
- package/skills/ai-research/01-model-architecture/litgpt/references/supported-models.md +337 -0
- package/skills/ai-research/01-model-architecture/litgpt/references/training-recipes.md +620 -0
- package/skills/ai-research/01-model-architecture/mamba/SKILL.md +260 -0
- package/skills/ai-research/01-model-architecture/mamba/references/architecture-details.md +207 -0
- package/skills/ai-research/01-model-architecture/mamba/references/benchmarks.md +256 -0
- package/skills/ai-research/01-model-architecture/mamba/references/training-guide.md +389 -0
- package/skills/ai-research/01-model-architecture/nanogpt/SKILL.md +290 -0
- package/skills/ai-research/01-model-architecture/nanogpt/references/architecture.md +383 -0
- package/skills/ai-research/01-model-architecture/nanogpt/references/data.md +477 -0
- package/skills/ai-research/01-model-architecture/nanogpt/references/training.md +565 -0
- package/skills/ai-research/01-model-architecture/rwkv/SKILL.md +260 -0
- package/skills/ai-research/01-model-architecture/rwkv/references/architecture-details.md +345 -0
- package/skills/ai-research/01-model-architecture/rwkv/references/rwkv7.md +387 -0
- package/skills/ai-research/01-model-architecture/rwkv/references/state-management.md +370 -0
- package/skills/ai-research/01-model-architecture/torchtitan/SKILL.md +358 -0
- package/skills/ai-research/01-model-architecture/torchtitan/references/checkpoint.md +182 -0
- package/skills/ai-research/01-model-architecture/torchtitan/references/custom-models.md +259 -0
- package/skills/ai-research/01-model-architecture/torchtitan/references/float8.md +134 -0
- package/skills/ai-research/01-model-architecture/torchtitan/references/fsdp.md +127 -0
- package/skills/ai-research/02-tokenization/huggingface-tokenizers/SKILL.md +516 -0
- package/skills/ai-research/02-tokenization/huggingface-tokenizers/references/algorithms.md +654 -0
- package/skills/ai-research/02-tokenization/huggingface-tokenizers/references/integration.md +638 -0
- package/skills/ai-research/02-tokenization/huggingface-tokenizers/references/pipeline.md +724 -0
- package/skills/ai-research/02-tokenization/huggingface-tokenizers/references/training.md +566 -0
- package/skills/ai-research/02-tokenization/sentencepiece/SKILL.md +235 -0
- package/skills/ai-research/02-tokenization/sentencepiece/references/algorithms.md +201 -0
- package/skills/ai-research/02-tokenization/sentencepiece/references/training.md +305 -0
- package/skills/ai-research/03-fine-tuning/axolotl/SKILL.md +158 -0
- package/skills/ai-research/03-fine-tuning/axolotl/references/api.md +5549 -0
- package/skills/ai-research/03-fine-tuning/axolotl/references/dataset-formats.md +1030 -0
- package/skills/ai-research/03-fine-tuning/axolotl/references/index.md +16 -0
- package/skills/ai-research/03-fine-tuning/axolotl/references/other.md +3564 -0
- package/skills/ai-research/03-fine-tuning/llama-factory/SKILL.md +80 -0
- package/skills/ai-research/03-fine-tuning/llama-factory/references/_images.md +24 -0
- package/skills/ai-research/03-fine-tuning/llama-factory/references/advanced.md +1056 -0
- package/skills/ai-research/03-fine-tuning/llama-factory/references/getting_started.md +350 -0
- package/skills/ai-research/03-fine-tuning/llama-factory/references/index.md +20 -0
- package/skills/ai-research/03-fine-tuning/llama-factory/references/other.md +32 -0
- package/skills/ai-research/03-fine-tuning/peft/SKILL.md +431 -0
- package/skills/ai-research/03-fine-tuning/peft/references/advanced-usage.md +515 -0
- package/skills/ai-research/03-fine-tuning/peft/references/troubleshooting.md +481 -0
- package/skills/ai-research/03-fine-tuning/unsloth/SKILL.md +80 -0
- package/skills/ai-research/03-fine-tuning/unsloth/references/index.md +8 -0
- package/skills/ai-research/03-fine-tuning/unsloth/references/llms.md +83 -0
- package/skills/ai-research/04-mechanistic-interpretability/nnsight/SKILL.md +436 -0
- package/skills/ai-research/04-mechanistic-interpretability/nnsight/references/README.md +79 -0
- package/skills/ai-research/04-mechanistic-interpretability/nnsight/references/api.md +345 -0
- package/skills/ai-research/04-mechanistic-interpretability/nnsight/references/tutorials.md +301 -0
- package/skills/ai-research/04-mechanistic-interpretability/pyvene/SKILL.md +473 -0
- package/skills/ai-research/04-mechanistic-interpretability/pyvene/references/README.md +74 -0
- package/skills/ai-research/04-mechanistic-interpretability/pyvene/references/api.md +384 -0
- package/skills/ai-research/04-mechanistic-interpretability/pyvene/references/tutorials.md +377 -0
- package/skills/ai-research/04-mechanistic-interpretability/saelens/SKILL.md +386 -0
- package/skills/ai-research/04-mechanistic-interpretability/saelens/references/README.md +71 -0
- package/skills/ai-research/04-mechanistic-interpretability/saelens/references/api.md +334 -0
- package/skills/ai-research/04-mechanistic-interpretability/saelens/references/tutorials.md +319 -0
- package/skills/ai-research/04-mechanistic-interpretability/transformer-lens/SKILL.md +346 -0
- package/skills/ai-research/04-mechanistic-interpretability/transformer-lens/references/README.md +55 -0
- package/skills/ai-research/04-mechanistic-interpretability/transformer-lens/references/api.md +363 -0
- package/skills/ai-research/04-mechanistic-interpretability/transformer-lens/references/tutorials.md +340 -0
- package/skills/ai-research/05-data-processing/nemo-curator/SKILL.md +383 -0
- package/skills/ai-research/05-data-processing/nemo-curator/references/deduplication.md +88 -0
- package/skills/ai-research/05-data-processing/nemo-curator/references/filtering.md +103 -0
- package/skills/ai-research/05-data-processing/ray-data/SKILL.md +326 -0
- package/skills/ai-research/05-data-processing/ray-data/references/integration.md +83 -0
- package/skills/ai-research/05-data-processing/ray-data/references/transformations.md +84 -0
- package/skills/ai-research/06-post-training/grpo-rl-training/README.md +98 -0
- package/skills/ai-research/06-post-training/grpo-rl-training/SKILL.md +572 -0
- package/skills/ai-research/06-post-training/miles/SKILL.md +315 -0
- package/skills/ai-research/06-post-training/miles/references/api-reference.md +142 -0
- package/skills/ai-research/06-post-training/miles/references/troubleshooting.md +353 -0
- package/skills/ai-research/06-post-training/openrlhf/SKILL.md +249 -0
- package/skills/ai-research/06-post-training/openrlhf/references/algorithm-comparison.md +405 -0
- package/skills/ai-research/06-post-training/openrlhf/references/custom-rewards.md +531 -0
- package/skills/ai-research/06-post-training/openrlhf/references/hybrid-engine.md +288 -0
- package/skills/ai-research/06-post-training/openrlhf/references/multi-node-training.md +455 -0
- package/skills/ai-research/06-post-training/simpo/SKILL.md +219 -0
- package/skills/ai-research/06-post-training/simpo/references/datasets.md +479 -0
- package/skills/ai-research/06-post-training/simpo/references/hyperparameters.md +453 -0
- package/skills/ai-research/06-post-training/simpo/references/loss-functions.md +351 -0
- package/skills/ai-research/06-post-training/slime/SKILL.md +464 -0
- package/skills/ai-research/06-post-training/slime/references/api-reference.md +393 -0
- package/skills/ai-research/06-post-training/slime/references/troubleshooting.md +387 -0
- package/skills/ai-research/06-post-training/torchforge/SKILL.md +433 -0
- package/skills/ai-research/06-post-training/torchforge/references/api-reference.md +328 -0
- package/skills/ai-research/06-post-training/torchforge/references/troubleshooting.md +410 -0
- package/skills/ai-research/06-post-training/trl-fine-tuning/SKILL.md +455 -0
- package/skills/ai-research/06-post-training/trl-fine-tuning/references/dpo-variants.md +228 -0
- package/skills/ai-research/06-post-training/trl-fine-tuning/references/online-rl.md +83 -0
- package/skills/ai-research/06-post-training/trl-fine-tuning/references/reward-modeling.md +123 -0
- package/skills/ai-research/06-post-training/trl-fine-tuning/references/sft-training.md +169 -0
- package/skills/ai-research/06-post-training/verl/SKILL.md +391 -0
- package/skills/ai-research/06-post-training/verl/references/api-reference.md +302 -0
- package/skills/ai-research/06-post-training/verl/references/troubleshooting.md +392 -0
- package/skills/ai-research/07-safety-alignment/constitutional-ai/SKILL.md +290 -0
- package/skills/ai-research/07-safety-alignment/llamaguard/SKILL.md +337 -0
- package/skills/ai-research/07-safety-alignment/nemo-guardrails/SKILL.md +297 -0
- package/skills/ai-research/07-safety-alignment/prompt-guard/SKILL.md +312 -0
- package/skills/ai-research/08-distributed-training/accelerate/SKILL.md +332 -0
- package/skills/ai-research/08-distributed-training/accelerate/references/custom-plugins.md +454 -0
- package/skills/ai-research/08-distributed-training/accelerate/references/megatron-integration.md +490 -0
- package/skills/ai-research/08-distributed-training/accelerate/references/performance.md +526 -0
- package/skills/ai-research/08-distributed-training/deepspeed/SKILL.md +141 -0
- package/skills/ai-research/08-distributed-training/deepspeed/references/08.md +18 -0
- package/skills/ai-research/08-distributed-training/deepspeed/references/09.md +174 -0
- package/skills/ai-research/08-distributed-training/deepspeed/references/2020.md +379 -0
- package/skills/ai-research/08-distributed-training/deepspeed/references/2023.md +280 -0
- package/skills/ai-research/08-distributed-training/deepspeed/references/assets.md +180 -0
- package/skills/ai-research/08-distributed-training/deepspeed/references/index.md +36 -0
- package/skills/ai-research/08-distributed-training/deepspeed/references/mii.md +119 -0
- package/skills/ai-research/08-distributed-training/deepspeed/references/other.md +1192 -0
- package/skills/ai-research/08-distributed-training/megatron-core/SKILL.md +366 -0
- package/skills/ai-research/08-distributed-training/megatron-core/references/benchmarks.md +250 -0
- package/skills/ai-research/08-distributed-training/megatron-core/references/parallelism-guide.md +405 -0
- package/skills/ai-research/08-distributed-training/megatron-core/references/production-examples.md +474 -0
- package/skills/ai-research/08-distributed-training/megatron-core/references/training-recipes.md +548 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/SKILL.md +231 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/references/pytorch_dcp_async_recipe.md +14 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/references/pytorch_dcp_overview.md +19 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/references/pytorch_dcp_recipe.md +18 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/references/pytorch_ddp_notes.md +10 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/references/pytorch_device_mesh_tutorial.md +22 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/references/pytorch_examples_fsdp2.md +15 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/references/pytorch_fsdp1_api.md +9 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/references/pytorch_fsdp2_tutorial.md +45 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/references/pytorch_fully_shard_api.md +49 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/references/pytorch_tp_tutorial.md +22 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/references/ray_train_fsdp2_example.md +13 -0
- package/skills/ai-research/08-distributed-training/pytorch-fsdp2/references/torchtitan_fsdp_notes.md +14 -0
- package/skills/ai-research/08-distributed-training/pytorch-lightning/SKILL.md +346 -0
- package/skills/ai-research/08-distributed-training/pytorch-lightning/references/callbacks.md +437 -0
- package/skills/ai-research/08-distributed-training/pytorch-lightning/references/distributed.md +491 -0
- package/skills/ai-research/08-distributed-training/pytorch-lightning/references/hyperparameter-tuning.md +557 -0
- package/skills/ai-research/08-distributed-training/ray-train/SKILL.md +406 -0
- package/skills/ai-research/08-distributed-training/ray-train/references/multi-node.md +629 -0
- package/skills/ai-research/09-infrastructure/lambda-labs/SKILL.md +545 -0
- package/skills/ai-research/09-infrastructure/lambda-labs/references/advanced-usage.md +612 -0
- package/skills/ai-research/09-infrastructure/lambda-labs/references/troubleshooting.md +531 -0
- package/skills/ai-research/09-infrastructure/modal/SKILL.md +341 -0
- package/skills/ai-research/09-infrastructure/modal/references/advanced-usage.md +504 -0
- package/skills/ai-research/09-infrastructure/modal/references/troubleshooting.md +495 -0
- package/skills/ai-research/09-infrastructure/skypilot/SKILL.md +509 -0
- package/skills/ai-research/09-infrastructure/skypilot/references/advanced-usage.md +492 -0
- package/skills/ai-research/09-infrastructure/skypilot/references/troubleshooting.md +571 -0
- package/skills/ai-research/10-optimization/awq/SKILL.md +310 -0
- package/skills/ai-research/10-optimization/awq/references/advanced-usage.md +325 -0
- package/skills/ai-research/10-optimization/awq/references/troubleshooting.md +345 -0
- package/skills/ai-research/10-optimization/bitsandbytes/SKILL.md +411 -0
- package/skills/ai-research/10-optimization/bitsandbytes/references/memory-optimization.md +522 -0
- package/skills/ai-research/10-optimization/bitsandbytes/references/qlora-training.md +522 -0
- package/skills/ai-research/10-optimization/bitsandbytes/references/quantization-formats.md +448 -0
- package/skills/ai-research/10-optimization/flash-attention/SKILL.md +367 -0
- package/skills/ai-research/10-optimization/flash-attention/references/benchmarks.md +216 -0
- package/skills/ai-research/10-optimization/flash-attention/references/transformers-integration.md +294 -0
- package/skills/ai-research/10-optimization/gguf/SKILL.md +427 -0
- package/skills/ai-research/10-optimization/gguf/references/advanced-usage.md +505 -0
- package/skills/ai-research/10-optimization/gguf/references/troubleshooting.md +443 -0
- package/skills/ai-research/10-optimization/gptq/SKILL.md +450 -0
- package/skills/ai-research/10-optimization/gptq/references/calibration.md +338 -0
- package/skills/ai-research/10-optimization/gptq/references/integration.md +130 -0
- package/skills/ai-research/10-optimization/gptq/references/troubleshooting.md +96 -0
- package/skills/ai-research/10-optimization/hqq/SKILL.md +445 -0
- package/skills/ai-research/10-optimization/hqq/references/advanced-usage.md +529 -0
- package/skills/ai-research/10-optimization/hqq/references/troubleshooting.md +504 -0
- package/skills/ai-research/10-optimization/ml-training-recipes/SKILL.md +319 -0
- package/skills/ai-research/10-optimization/ml-training-recipes/references/architecture.md +320 -0
- package/skills/ai-research/10-optimization/ml-training-recipes/references/biomedical.md +498 -0
- package/skills/ai-research/10-optimization/ml-training-recipes/references/domain-specific.md +626 -0
- package/skills/ai-research/10-optimization/ml-training-recipes/references/experiment-loop.md +129 -0
- package/skills/ai-research/10-optimization/ml-training-recipes/references/optimizers.md +330 -0
- package/skills/ai-research/10-optimization/ml-training-recipes/references/scaling-and-selection.md +428 -0
- package/skills/ai-research/11-evaluation/bigcode-evaluation-harness/SKILL.md +405 -0
- package/skills/ai-research/11-evaluation/bigcode-evaluation-harness/references/benchmarks.md +394 -0
- package/skills/ai-research/11-evaluation/bigcode-evaluation-harness/references/custom-tasks.md +425 -0
- package/skills/ai-research/11-evaluation/bigcode-evaluation-harness/references/issues.md +395 -0
- package/skills/ai-research/11-evaluation/lm-evaluation-harness/SKILL.md +490 -0
- package/skills/ai-research/11-evaluation/lm-evaluation-harness/references/api-evaluation.md +491 -0
- package/skills/ai-research/11-evaluation/lm-evaluation-harness/references/benchmark-guide.md +489 -0
- package/skills/ai-research/11-evaluation/lm-evaluation-harness/references/custom-tasks.md +603 -0
- package/skills/ai-research/11-evaluation/lm-evaluation-harness/references/distributed-eval.md +520 -0
- package/skills/ai-research/11-evaluation/nemo-evaluator/SKILL.md +494 -0
- package/skills/ai-research/11-evaluation/nemo-evaluator/references/adapter-system.md +341 -0
- package/skills/ai-research/11-evaluation/nemo-evaluator/references/configuration.md +448 -0
- package/skills/ai-research/11-evaluation/nemo-evaluator/references/custom-benchmarks.md +316 -0
- package/skills/ai-research/11-evaluation/nemo-evaluator/references/execution-backends.md +362 -0
- package/skills/ai-research/12-inference-serving/llama-cpp/SKILL.md +258 -0
- package/skills/ai-research/12-inference-serving/llama-cpp/references/optimization.md +90 -0
- package/skills/ai-research/12-inference-serving/llama-cpp/references/quantization.md +214 -0
- package/skills/ai-research/12-inference-serving/llama-cpp/references/server.md +126 -0
- package/skills/ai-research/12-inference-serving/sglang/SKILL.md +442 -0
- package/skills/ai-research/12-inference-serving/sglang/references/deployment.md +491 -0
- package/skills/ai-research/12-inference-serving/sglang/references/radix-attention.md +414 -0
- package/skills/ai-research/12-inference-serving/sglang/references/structured-generation.md +542 -0
- package/skills/ai-research/12-inference-serving/tensorrt-llm/SKILL.md +187 -0
- package/skills/ai-research/12-inference-serving/tensorrt-llm/references/multi-gpu.md +299 -0
- package/skills/ai-research/12-inference-serving/tensorrt-llm/references/optimization.md +243 -0
- package/skills/ai-research/12-inference-serving/tensorrt-llm/references/serving.md +471 -0
- package/skills/ai-research/12-inference-serving/vllm/SKILL.md +364 -0
- package/skills/ai-research/12-inference-serving/vllm/references/optimization.md +227 -0
- package/skills/ai-research/12-inference-serving/vllm/references/quantization.md +285 -0
- package/skills/ai-research/12-inference-serving/vllm/references/server-deployment.md +256 -0
- package/skills/ai-research/12-inference-serving/vllm/references/troubleshooting.md +448 -0
- package/skills/ai-research/13-mlops/mlflow/SKILL.md +704 -0
- package/skills/ai-research/13-mlops/mlflow/references/deployment.md +745 -0
- package/skills/ai-research/13-mlops/mlflow/references/model-registry.md +771 -0
- package/skills/ai-research/13-mlops/mlflow/references/tracking.md +681 -0
- package/skills/ai-research/13-mlops/swanlab/SKILL.md +406 -0
- package/skills/ai-research/13-mlops/swanlab/references/integrations.md +305 -0
- package/skills/ai-research/13-mlops/swanlab/references/visualization.md +234 -0
- package/skills/ai-research/13-mlops/tensorboard/SKILL.md +629 -0
- package/skills/ai-research/13-mlops/tensorboard/references/integrations.md +639 -0
- package/skills/ai-research/13-mlops/tensorboard/references/profiling.md +546 -0
- package/skills/ai-research/13-mlops/tensorboard/references/visualization.md +621 -0
- package/skills/ai-research/13-mlops/weights-and-biases/SKILL.md +590 -0
- package/skills/ai-research/13-mlops/weights-and-biases/references/artifacts.md +585 -0
- package/skills/ai-research/13-mlops/weights-and-biases/references/integrations.md +701 -0
- package/skills/ai-research/13-mlops/weights-and-biases/references/sweeps.md +848 -0
- package/skills/ai-research/14-agents/a-evolve/SKILL.md +383 -0
- package/skills/ai-research/14-agents/a-evolve/references/README.md +831 -0
- package/skills/ai-research/14-agents/a-evolve/references/api.md +594 -0
- package/skills/ai-research/14-agents/a-evolve/references/architecture.md +417 -0
- package/skills/ai-research/14-agents/a-evolve/references/design-patterns.md +462 -0
- package/skills/ai-research/14-agents/a-evolve/references/examples.md +352 -0
- package/skills/ai-research/14-agents/a-evolve/references/issues.md +591 -0
- package/skills/ai-research/14-agents/a-evolve/references/releases.md +112 -0
- package/skills/ai-research/14-agents/a-evolve/references/tutorials.md +819 -0
- package/skills/ai-research/14-agents/autogpt/SKILL.md +403 -0
- package/skills/ai-research/14-agents/autogpt/references/advanced-usage.md +536 -0
- package/skills/ai-research/14-agents/autogpt/references/troubleshooting.md +421 -0
- package/skills/ai-research/14-agents/crewai/SKILL.md +498 -0
- package/skills/ai-research/14-agents/crewai/references/flows.md +439 -0
- package/skills/ai-research/14-agents/crewai/references/tools.md +430 -0
- package/skills/ai-research/14-agents/crewai/references/troubleshooting.md +481 -0
- package/skills/ai-research/14-agents/langchain/SKILL.md +480 -0
- package/skills/ai-research/14-agents/langchain/references/agents.md +500 -0
- package/skills/ai-research/14-agents/langchain/references/integration.md +563 -0
- package/skills/ai-research/14-agents/langchain/references/rag.md +601 -0
- package/skills/ai-research/14-agents/llamaindex/SKILL.md +569 -0
- package/skills/ai-research/14-agents/llamaindex/references/agents.md +84 -0
- package/skills/ai-research/14-agents/llamaindex/references/data_connectors.md +109 -0
- package/skills/ai-research/14-agents/llamaindex/references/query_engines.md +407 -0
- package/skills/ai-research/15-rag/chroma/SKILL.md +406 -0
- package/skills/ai-research/15-rag/chroma/references/integration.md +39 -0
- package/skills/ai-research/15-rag/faiss/SKILL.md +221 -0
- package/skills/ai-research/15-rag/faiss/references/index_types.md +281 -0
- package/skills/ai-research/15-rag/pinecone/SKILL.md +358 -0
- package/skills/ai-research/15-rag/pinecone/references/deployment.md +182 -0
- package/skills/ai-research/15-rag/qdrant/SKILL.md +493 -0
- package/skills/ai-research/15-rag/qdrant/references/advanced-usage.md +649 -0
- package/skills/ai-research/15-rag/qdrant/references/troubleshooting.md +632 -0
- package/skills/ai-research/15-rag/sentence-transformers/SKILL.md +255 -0
- package/skills/ai-research/15-rag/sentence-transformers/references/models.md +124 -0
- package/skills/ai-research/16-prompt-engineering/dspy/SKILL.md +590 -0
- package/skills/ai-research/16-prompt-engineering/dspy/references/examples.md +664 -0
- package/skills/ai-research/16-prompt-engineering/dspy/references/modules.md +476 -0
- package/skills/ai-research/16-prompt-engineering/dspy/references/optimizers.md +567 -0
- package/skills/ai-research/16-prompt-engineering/guidance/SKILL.md +572 -0
- package/skills/ai-research/16-prompt-engineering/guidance/references/backends.md +555 -0
- package/skills/ai-research/16-prompt-engineering/guidance/references/constraints.md +675 -0
- package/skills/ai-research/16-prompt-engineering/guidance/references/examples.md +768 -0
- package/skills/ai-research/16-prompt-engineering/instructor/SKILL.md +740 -0
- package/skills/ai-research/16-prompt-engineering/instructor/references/examples.md +108 -0
- package/skills/ai-research/16-prompt-engineering/instructor/references/providers.md +71 -0
- package/skills/ai-research/16-prompt-engineering/instructor/references/validation.md +607 -0
- package/skills/ai-research/16-prompt-engineering/outlines/SKILL.md +652 -0
- package/skills/ai-research/16-prompt-engineering/outlines/references/backends.md +616 -0
- package/skills/ai-research/16-prompt-engineering/outlines/references/examples.md +774 -0
- package/skills/ai-research/16-prompt-engineering/outlines/references/json_generation.md +653 -0
- package/skills/ai-research/17-observability/langsmith/SKILL.md +422 -0
- package/skills/ai-research/17-observability/langsmith/references/advanced-usage.md +549 -0
- package/skills/ai-research/17-observability/langsmith/references/troubleshooting.md +538 -0
- package/skills/ai-research/17-observability/phoenix/SKILL.md +475 -0
- package/skills/ai-research/17-observability/phoenix/references/advanced-usage.md +620 -0
- package/skills/ai-research/17-observability/phoenix/references/troubleshooting.md +539 -0
- package/skills/ai-research/18-multimodal/audiocraft/SKILL.md +564 -0
- package/skills/ai-research/18-multimodal/audiocraft/references/advanced-usage.md +667 -0
- package/skills/ai-research/18-multimodal/audiocraft/references/troubleshooting.md +505 -0
- package/skills/ai-research/18-multimodal/blip-2/SKILL.md +564 -0
- package/skills/ai-research/18-multimodal/blip-2/references/advanced-usage.md +681 -0
- package/skills/ai-research/18-multimodal/blip-2/references/troubleshooting.md +527 -0
- package/skills/ai-research/18-multimodal/clip/SKILL.md +253 -0
- package/skills/ai-research/18-multimodal/clip/references/applications.md +208 -0
- package/skills/ai-research/18-multimodal/cosmos-policy/SKILL.md +402 -0
- package/skills/ai-research/18-multimodal/cosmos-policy/references/libero-commands.md +175 -0
- package/skills/ai-research/18-multimodal/cosmos-policy/references/robocasa-commands.md +147 -0
- package/skills/ai-research/18-multimodal/llava/SKILL.md +304 -0
- package/skills/ai-research/18-multimodal/llava/references/training.md +198 -0
- package/skills/ai-research/18-multimodal/openpi/SKILL.md +480 -0
- package/skills/ai-research/18-multimodal/openpi/references/checkpoints-and-env-map.md +86 -0
- package/skills/ai-research/18-multimodal/openpi/references/config-recipes.md +50 -0
- package/skills/ai-research/18-multimodal/openpi/references/pytorch-gotchas.md +49 -0
- package/skills/ai-research/18-multimodal/openpi/references/remote-client-pattern.md +48 -0
- package/skills/ai-research/18-multimodal/openpi/references/training-debugging.md +46 -0
- package/skills/ai-research/18-multimodal/openvla-oft/SKILL.md +441 -0
- package/skills/ai-research/18-multimodal/openvla-oft/references/aloha-workflow.md +141 -0
- package/skills/ai-research/18-multimodal/openvla-oft/references/config-troubleshooting.md +97 -0
- package/skills/ai-research/18-multimodal/openvla-oft/references/libero-workflow.md +115 -0
- package/skills/ai-research/18-multimodal/openvla-oft/references/paper-and-checkpoints.md +84 -0
- package/skills/ai-research/18-multimodal/segment-anything/SKILL.md +500 -0
- package/skills/ai-research/18-multimodal/segment-anything/references/advanced-usage.md +590 -0
- package/skills/ai-research/18-multimodal/segment-anything/references/troubleshooting.md +485 -0
- package/skills/ai-research/18-multimodal/stable-diffusion/SKILL.md +519 -0
- package/skills/ai-research/18-multimodal/stable-diffusion/references/advanced-usage.md +717 -0
- package/skills/ai-research/18-multimodal/stable-diffusion/references/troubleshooting.md +556 -0
- package/skills/ai-research/18-multimodal/whisper/SKILL.md +317 -0
- package/skills/ai-research/18-multimodal/whisper/references/languages.md +190 -0
- package/skills/ai-research/19-emerging-techniques/knowledge-distillation/SKILL.md +458 -0
- package/skills/ai-research/19-emerging-techniques/knowledge-distillation/references/minillm.md +335 -0
- package/skills/ai-research/19-emerging-techniques/long-context/SKILL.md +536 -0
- package/skills/ai-research/19-emerging-techniques/long-context/references/extension_methods.md +469 -0
- package/skills/ai-research/19-emerging-techniques/long-context/references/fine_tuning.md +612 -0
- package/skills/ai-research/19-emerging-techniques/long-context/references/rope.md +403 -0
- package/skills/ai-research/19-emerging-techniques/model-merging/SKILL.md +497 -0
- package/skills/ai-research/19-emerging-techniques/model-merging/references/coefficient-tuning.md +415 -0
- package/skills/ai-research/19-emerging-techniques/model-merging/references/evaluation.md +463 -0
- package/skills/ai-research/19-emerging-techniques/model-merging/references/examples.md +429 -0
- package/skills/ai-research/19-emerging-techniques/model-merging/references/methods.md +353 -0
- package/skills/ai-research/19-emerging-techniques/model-pruning/SKILL.md +495 -0
- package/skills/ai-research/19-emerging-techniques/model-pruning/references/wanda.md +348 -0
- package/skills/ai-research/19-emerging-techniques/moe-training/SKILL.md +526 -0
- package/skills/ai-research/19-emerging-techniques/moe-training/references/architectures.md +433 -0
- package/skills/ai-research/19-emerging-techniques/moe-training/references/inference.md +349 -0
- package/skills/ai-research/19-emerging-techniques/moe-training/references/training.md +426 -0
- package/skills/ai-research/19-emerging-techniques/speculative-decoding/SKILL.md +467 -0
- package/skills/ai-research/19-emerging-techniques/speculative-decoding/references/lookahead.md +310 -0
- package/skills/ai-research/19-emerging-techniques/speculative-decoding/references/medusa.md +351 -0
- package/skills/ai-research/20-ml-paper-writing/academic-plotting/SKILL.md +479 -0
- package/skills/ai-research/20-ml-paper-writing/academic-plotting/references/data-visualization.md +509 -0
- package/skills/ai-research/20-ml-paper-writing/academic-plotting/references/diagram-generation.md +395 -0
- package/skills/ai-research/20-ml-paper-writing/academic-plotting/references/style-guide.md +305 -0
- package/skills/ai-research/20-ml-paper-writing/ml-paper-writing/SKILL.md +982 -0
- package/skills/ai-research/20-ml-paper-writing/ml-paper-writing/references/checklists.md +348 -0
- package/skills/ai-research/20-ml-paper-writing/ml-paper-writing/references/citation-workflow.md +563 -0
- package/skills/ai-research/20-ml-paper-writing/ml-paper-writing/references/reviewer-guidelines.md +349 -0
- package/skills/ai-research/20-ml-paper-writing/ml-paper-writing/references/sources.md +165 -0
- package/skills/ai-research/20-ml-paper-writing/ml-paper-writing/references/writing-guide.md +477 -0
- package/skills/ai-research/20-ml-paper-writing/presenting-conference-talks/SKILL.md +298 -0
- package/skills/ai-research/20-ml-paper-writing/presenting-conference-talks/references/slide-templates.md +505 -0
- package/skills/ai-research/20-ml-paper-writing/systems-paper-writing/SKILL.md +270 -0
- package/skills/ai-research/20-ml-paper-writing/systems-paper-writing/references/checklist.md +279 -0
- package/skills/ai-research/20-ml-paper-writing/systems-paper-writing/references/reviewer-guidelines.md +116 -0
- package/skills/ai-research/20-ml-paper-writing/systems-paper-writing/references/section-blueprints.md +400 -0
- package/skills/ai-research/20-ml-paper-writing/systems-paper-writing/references/systems-conferences.md +261 -0
- package/skills/ai-research/20-ml-paper-writing/systems-paper-writing/references/writing-patterns.md +231 -0
- package/skills/ai-research/21-research-ideation/brainstorming-research-ideas/SKILL.md +384 -0
- package/skills/ai-research/21-research-ideation/creative-thinking-for-research/SKILL.md +366 -0
- package/skills/ai-research/22-agent-native-research-artifact/compiler/SKILL.md +245 -0
- package/skills/ai-research/22-agent-native-research-artifact/compiler/references/ara-schema.md +439 -0
- package/skills/ai-research/22-agent-native-research-artifact/compiler/references/exploration-tree-spec.md +125 -0
- package/skills/ai-research/22-agent-native-research-artifact/compiler/references/validation-checklist.md +149 -0
- package/skills/ai-research/22-agent-native-research-artifact/research-manager/SKILL.md +324 -0
- package/skills/ai-research/22-agent-native-research-artifact/research-manager/references/event-taxonomy.md +124 -0
- package/skills/ai-research/22-agent-native-research-artifact/research-manager/references/provenance-tags.md +190 -0
- package/skills/ai-research/22-agent-native-research-artifact/research-manager/references/session-protocol.md +162 -0
- package/skills/ai-research/22-agent-native-research-artifact/rigor-reviewer/SKILL.md +322 -0
- package/skills/ai-research/22-agent-native-research-artifact/rigor-reviewer/references/review-dimensions.md +182 -0
- package/skills/ai-research/SKILL.md +101 -0
- package/skills/ai-research/SOURCE +38 -0
- package/skills/ai-research/index/1-frame.md +18 -0
- package/skills/ai-research/index/2-data.md +21 -0
- package/skills/ai-research/index/3-train.md +48 -0
- package/skills/ai-research/index/4-measure.md +25 -0
- package/skills/ai-research/index/5-run.md +23 -0
- package/skills/ai-research/index/6-write.md +35 -0
- package/skills/ai-research/library.json +3456 -0
- package/skills/analyze-app/EVALS.md +82 -0
- package/skills/analyze-app/SKILL.md +188 -0
- package/skills/design-contract/EVALS.md +80 -0
- package/skills/design-contract/SKILL.md +171 -0
- package/skills/design-library/EVALS.md +85 -0
- package/skills/design-library/INDEX.md +59 -0
- package/skills/design-library/SKILL.md +58 -0
- package/skills/design-library/SOURCE +131 -0
- package/skills/design-library/adaptive-interfaces/skills/colour-independence/SKILL.md +90 -0
- package/skills/design-library/adaptive-interfaces/skills/flexible-typography/SKILL.md +83 -0
- package/skills/design-library/adaptive-interfaces/skills/information-density/SKILL.md +87 -0
- package/skills/design-library/adaptive-interfaces/skills/preference-audit/SKILL.md +70 -0
- package/skills/design-library/adaptive-interfaces/skills/responsive-accessibility/SKILL.md +86 -0
- package/skills/design-library/adaptive-interfaces/skills/responsive-review/SKILL.md +76 -0
- package/skills/design-library/adaptive-interfaces/skills/simplified-views/SKILL.md +97 -0
- package/skills/design-library/adaptive-interfaces/skills/specify/SKILL.md +63 -0
- package/skills/design-library/adaptive-interfaces/skills/user-preference-respect/SKILL.md +100 -0
- package/skills/design-library/cognitive-accessibility/skills/adaptive-personalisation/SKILL.md +61 -0
- package/skills/design-library/cognitive-accessibility/skills/assess-load/SKILL.md +48 -0
- package/skills/design-library/cognitive-accessibility/skills/cognitive-load-assessment/SKILL.md +60 -0
- package/skills/design-library/cognitive-accessibility/skills/contextual-help-design/SKILL.md +61 -0
- package/skills/design-library/cognitive-accessibility/skills/error-prevention-recovery/SKILL.md +71 -0
- package/skills/design-library/cognitive-accessibility/skills/focus-attention-design/SKILL.md +67 -0
- package/skills/design-library/cognitive-accessibility/skills/memory-load-reduction/SKILL.md +60 -0
- package/skills/design-library/cognitive-accessibility/skills/plain-language-design/SKILL.md +78 -0
- package/skills/design-library/cognitive-accessibility/skills/review/SKILL.md +56 -0
- package/skills/design-library/cognitive-accessibility/skills/simplify/SKILL.md +39 -0
- package/skills/design-library/cognitive-accessibility/skills/wayfinding-navigation/SKILL.md +64 -0
- package/skills/design-library/design-research/skills/behavioural-analytics/SKILL.md +33 -0
- package/skills/design-library/design-research/skills/qual-quant-triangulation/SKILL.md +33 -0
- package/skills/design-library/inclusive-interaction/skills/audit/SKILL.md +50 -0
- package/skills/design-library/inclusive-interaction/skills/design-flow/SKILL.md +58 -0
- package/skills/design-library/inclusive-interaction/skills/feedback-and-status/SKILL.md +76 -0
- package/skills/design-library/inclusive-interaction/skills/gesture-alternatives/SKILL.md +70 -0
- package/skills/design-library/inclusive-interaction/skills/keyboard-navigation/SKILL.md +99 -0
- package/skills/design-library/inclusive-interaction/skills/keyboard-review/SKILL.md +54 -0
- package/skills/design-library/inclusive-interaction/skills/motion-sensitivity/SKILL.md +79 -0
- package/skills/design-library/inclusive-interaction/skills/multi-modal-input/SKILL.md +70 -0
- package/skills/design-library/inclusive-interaction/skills/touch-target-design/SKILL.md +71 -0
- package/skills/design-library/inclusive-interaction/skills/voice-interaction/SKILL.md +68 -0
- package/skills/design-library/visual-critique/skills/critique-affordance/SKILL.md +45 -0
- package/skills/design-library/visual-critique/skills/critique-brand-consistency/SKILL.md +42 -0
- package/skills/design-library/visual-critique/skills/critique-color/SKILL.md +44 -0
- package/skills/design-library/visual-critique/skills/critique-composition/SKILL.md +46 -0
- package/skills/design-library/visual-critique/skills/critique-information-density/SKILL.md +45 -0
- package/skills/design-library/visual-critique/skills/critique-typography/SKILL.md +45 -0
- package/skills/design-library/visual-critique/skills/critique-visual-hierarchy/SKILL.md +40 -0
- package/skills/feature-audit/EVALS.md +83 -0
- package/skills/feature-audit/SKILL.md +326 -0
- package/skills/harness/EVALS.md +81 -0
- package/skills/harness/SKILL.md +88 -0
- package/skills/i-have-adhd/EVALS.md +80 -0
- package/skills/i-have-adhd/SKILL.md +183 -0
- package/skills/no-ai-slop/EVALS.md +89 -0
- package/skills/no-ai-slop/SKILL.md +209 -0
- package/skills/no-ai-slop/eval.md +52 -0
- package/skills/plan-add/EVALS.md +79 -0
- package/skills/plan-add/SKILL.md +179 -0
- package/skills/plan-add/render_table.py +72 -0
- package/skills/plan-status/EVALS.md +78 -0
- package/skills/plan-status/SKILL.md +52 -0
- package/skills/plan-sync/EVALS.md +80 -0
- package/skills/plan-sync/SKILL.md +69 -0
- package/skills/pm-frameworks/SKILL.md +33 -0
- package/skills/pm-frameworks/SOURCE +39 -0
- package/skills/pm-frameworks/intended-vs-implemented/SKILL.md +42 -0
- package/skills/pm-frameworks/opportunity-solution-tree/SKILL.md +64 -0
- package/skills/ponytail/EVALS.md +80 -0
- package/skills/ponytail/SKILL.md +124 -0
- package/skills/product-analysis/EVALS.md +103 -0
- package/skills/product-analysis/SKILL.md +361 -0
- package/skills/research/EVALS.md +104 -0
- package/skills/research/SKILL.md +570 -0
- package/skills/rl-env/SKILL.md +53 -0
- package/skills/rl-env/SOURCE +41 -0
- package/skills/rl-env/generate-nemo-gym-env/SKILL.md +169 -0
- package/skills/rl-env/generate-nemo-gym-env/references/architecture.md +114 -0
- package/skills/rl-env/generate-openenv-env/SKILL.md +183 -0
- package/skills/rl-env/generate-openenv-env/references/architecture.md +82 -0
- package/skills/rl-env/generate-ors-env/SKILL.md +209 -0
- package/skills/rl-env/generate-ors-env/references/architecture.md +113 -0
- package/skills/rl-env/generate-verifiers-env/SKILL.md +188 -0
- package/skills/rl-env/generate-verifiers-env/references/architecture.md +119 -0
- package/skills/rl-env/rl-env-from-description/SKILL.md +124 -0
- package/skills/rl-env/rl-env-from-description/references/interview.md +75 -0
- package/skills/rl-env/rl-env-from-description/references/nemo_gym.md +41 -0
- package/skills/rl-env/rl-env-from-description/references/openenv.md +35 -0
- package/skills/rl-env/rl-env-from-description/references/ors.md +50 -0
- package/skills/rl-env/rl-env-from-description/references/verifiers.md +39 -0
- package/skills/tezgah-contract/EVALS.md +78 -0
- package/skills/tezgah-contract/SKILL.md +792 -0
- package/statusline.py +129 -0
- package/workflows/graph-impact.js +153 -0
- package/workflows/graph-map.js +158 -0
- package/workflows/graph-review.js +170 -0
|
@@ -0,0 +1,664 @@
|
|
|
1
|
+
<!-- vendored from orchestra-research/AI-research-SKILLs@773a52944ba4747a18bd4ae9ade53fff041adcbc 16-prompt-engineering/dspy/references/examples.md; MIT (c) Orchestra Research -->
|
|
2
|
+
# DSPy Real-World Examples
|
|
3
|
+
|
|
4
|
+
Practical examples of building production systems with DSPy.
|
|
5
|
+
|
|
6
|
+
## Table of Contents
|
|
7
|
+
- RAG Systems
|
|
8
|
+
- Agent Systems
|
|
9
|
+
- Classification
|
|
10
|
+
- Data Processing
|
|
11
|
+
- Multi-Stage Pipelines
|
|
12
|
+
|
|
13
|
+
## RAG Systems
|
|
14
|
+
|
|
15
|
+
### Basic RAG
|
|
16
|
+
|
|
17
|
+
```python
|
|
18
|
+
import dspy
|
|
19
|
+
|
|
20
|
+
class BasicRAG(dspy.Module):
|
|
21
|
+
def __init__(self, num_passages=3):
|
|
22
|
+
super().__init__()
|
|
23
|
+
self.retrieve = dspy.Retrieve(k=num_passages)
|
|
24
|
+
self.generate = dspy.ChainOfThought("context, question -> answer")
|
|
25
|
+
|
|
26
|
+
def forward(self, question):
|
|
27
|
+
passages = self.retrieve(question).passages
|
|
28
|
+
context = "\n\n".join(passages)
|
|
29
|
+
return self.generate(context=context, question=question)
|
|
30
|
+
|
|
31
|
+
# Configure retriever (example with Chroma)
|
|
32
|
+
from dspy.retrieve.chromadb_rm import ChromadbRM
|
|
33
|
+
|
|
34
|
+
retriever = ChromadbRM(
|
|
35
|
+
collection_name="my_docs",
|
|
36
|
+
persist_directory="./chroma_db",
|
|
37
|
+
k=3
|
|
38
|
+
)
|
|
39
|
+
dspy.settings.configure(rm=retriever)
|
|
40
|
+
|
|
41
|
+
# Use RAG
|
|
42
|
+
rag = BasicRAG()
|
|
43
|
+
result = rag(question="What is DSPy?")
|
|
44
|
+
print(result.answer)
|
|
45
|
+
```
|
|
46
|
+
|
|
47
|
+
### Optimized RAG
|
|
48
|
+
|
|
49
|
+
```python
|
|
50
|
+
from dspy.teleprompt import BootstrapFewShot
|
|
51
|
+
|
|
52
|
+
# Training data with question-answer pairs
|
|
53
|
+
trainset = [
|
|
54
|
+
dspy.Example(
|
|
55
|
+
question="What is retrieval augmented generation?",
|
|
56
|
+
answer="RAG combines retrieval of relevant documents with generation..."
|
|
57
|
+
).with_inputs("question"),
|
|
58
|
+
# ... more examples
|
|
59
|
+
]
|
|
60
|
+
|
|
61
|
+
# Define metric
|
|
62
|
+
def answer_correctness(example, pred, trace=None):
|
|
63
|
+
# Check if answer contains key information
|
|
64
|
+
return example.answer.lower() in pred.answer.lower()
|
|
65
|
+
|
|
66
|
+
# Optimize RAG
|
|
67
|
+
optimizer = BootstrapFewShot(metric=answer_correctness)
|
|
68
|
+
optimized_rag = optimizer.compile(rag, trainset=trainset)
|
|
69
|
+
|
|
70
|
+
# Optimized RAG performs better on similar questions
|
|
71
|
+
result = optimized_rag(question="Explain RAG systems")
|
|
72
|
+
```
|
|
73
|
+
|
|
74
|
+
### Multi-Hop RAG
|
|
75
|
+
|
|
76
|
+
```python
|
|
77
|
+
class MultiHopRAG(dspy.Module):
|
|
78
|
+
"""RAG that follows chains of reasoning across documents."""
|
|
79
|
+
|
|
80
|
+
def __init__(self):
|
|
81
|
+
super().__init__()
|
|
82
|
+
self.retrieve = dspy.Retrieve(k=3)
|
|
83
|
+
self.generate_query = dspy.ChainOfThought("question -> search_query")
|
|
84
|
+
self.generate_answer = dspy.ChainOfThought("context, question -> answer")
|
|
85
|
+
|
|
86
|
+
def forward(self, question):
|
|
87
|
+
# First retrieval
|
|
88
|
+
query1 = self.generate_query(question=question).search_query
|
|
89
|
+
passages1 = self.retrieve(query1).passages
|
|
90
|
+
|
|
91
|
+
# Generate follow-up query based on first results
|
|
92
|
+
context1 = "\n".join(passages1)
|
|
93
|
+
query2 = self.generate_query(
|
|
94
|
+
question=f"Based on: {context1}\nFollow-up: {question}"
|
|
95
|
+
).search_query
|
|
96
|
+
|
|
97
|
+
# Second retrieval
|
|
98
|
+
passages2 = self.retrieve(query2).passages
|
|
99
|
+
|
|
100
|
+
# Combine all context
|
|
101
|
+
all_context = "\n\n".join(passages1 + passages2)
|
|
102
|
+
|
|
103
|
+
# Generate final answer
|
|
104
|
+
return self.generate_answer(context=all_context, question=question)
|
|
105
|
+
|
|
106
|
+
# Use multi-hop RAG
|
|
107
|
+
multi_rag = MultiHopRAG()
|
|
108
|
+
result = multi_rag(question="Who wrote the book that inspired Blade Runner?")
|
|
109
|
+
# Hop 1: Find "Blade Runner was based on..."
|
|
110
|
+
# Hop 2: Find author of that book
|
|
111
|
+
```
|
|
112
|
+
|
|
113
|
+
### RAG with Reranking
|
|
114
|
+
|
|
115
|
+
```python
|
|
116
|
+
class RerankedRAG(dspy.Module):
|
|
117
|
+
"""RAG with learned reranking of retrieved passages."""
|
|
118
|
+
|
|
119
|
+
def __init__(self):
|
|
120
|
+
super().__init__()
|
|
121
|
+
self.retrieve = dspy.Retrieve(k=10) # Get more candidates
|
|
122
|
+
self.rerank = dspy.Predict("question, passage -> relevance_score: float")
|
|
123
|
+
self.answer = dspy.ChainOfThought("context, question -> answer")
|
|
124
|
+
|
|
125
|
+
def forward(self, question):
|
|
126
|
+
# Retrieve candidates
|
|
127
|
+
passages = self.retrieve(question).passages
|
|
128
|
+
|
|
129
|
+
# Rerank passages
|
|
130
|
+
scored_passages = []
|
|
131
|
+
for passage in passages:
|
|
132
|
+
score = float(self.rerank(
|
|
133
|
+
question=question,
|
|
134
|
+
passage=passage
|
|
135
|
+
).relevance_score)
|
|
136
|
+
scored_passages.append((score, passage))
|
|
137
|
+
|
|
138
|
+
# Take top 3 after reranking
|
|
139
|
+
top_passages = [p for _, p in sorted(scored_passages, reverse=True)[:3]]
|
|
140
|
+
context = "\n\n".join(top_passages)
|
|
141
|
+
|
|
142
|
+
# Generate answer from reranked context
|
|
143
|
+
return self.answer(context=context, question=question)
|
|
144
|
+
```
|
|
145
|
+
|
|
146
|
+
## Agent Systems
|
|
147
|
+
|
|
148
|
+
### ReAct Agent
|
|
149
|
+
|
|
150
|
+
```python
|
|
151
|
+
from dspy.predict import ReAct
|
|
152
|
+
|
|
153
|
+
# Define tools
|
|
154
|
+
def search_wikipedia(query: str) -> str:
|
|
155
|
+
"""Search Wikipedia for information."""
|
|
156
|
+
import wikipedia
|
|
157
|
+
try:
|
|
158
|
+
return wikipedia.summary(query, sentences=3)
|
|
159
|
+
except:
|
|
160
|
+
return "No results found"
|
|
161
|
+
|
|
162
|
+
def calculate(expression: str) -> str:
|
|
163
|
+
"""Evaluate mathematical expression safely."""
|
|
164
|
+
try:
|
|
165
|
+
# Use safe eval
|
|
166
|
+
result = eval(expression, {"__builtins__": {}}, {})
|
|
167
|
+
return str(result)
|
|
168
|
+
except:
|
|
169
|
+
return "Invalid expression"
|
|
170
|
+
|
|
171
|
+
def search_web(query: str) -> str:
|
|
172
|
+
"""Search the web."""
|
|
173
|
+
# Your web search implementation
|
|
174
|
+
return results
|
|
175
|
+
|
|
176
|
+
# Create agent signature
|
|
177
|
+
class ResearchAgent(dspy.Signature):
|
|
178
|
+
"""Answer questions using available tools."""
|
|
179
|
+
question = dspy.InputField()
|
|
180
|
+
answer = dspy.OutputField()
|
|
181
|
+
|
|
182
|
+
# Create ReAct agent
|
|
183
|
+
agent = ReAct(ResearchAgent, tools=[search_wikipedia, calculate, search_web])
|
|
184
|
+
|
|
185
|
+
# Agent decides which tools to use
|
|
186
|
+
result = agent(question="What is the population of France divided by 10?")
|
|
187
|
+
# Agent:
|
|
188
|
+
# 1. Thinks: "Need population of France"
|
|
189
|
+
# 2. Acts: search_wikipedia("France population")
|
|
190
|
+
# 3. Thinks: "Got 67 million, need to divide"
|
|
191
|
+
# 4. Acts: calculate("67000000 / 10")
|
|
192
|
+
# 5. Returns: "6,700,000"
|
|
193
|
+
```
|
|
194
|
+
|
|
195
|
+
### Multi-Agent System
|
|
196
|
+
|
|
197
|
+
```python
|
|
198
|
+
class MultiAgentSystem(dspy.Module):
|
|
199
|
+
"""System with specialized agents for different tasks."""
|
|
200
|
+
|
|
201
|
+
def __init__(self):
|
|
202
|
+
super().__init__()
|
|
203
|
+
|
|
204
|
+
# Router agent
|
|
205
|
+
self.router = dspy.Predict("question -> agent_type: str")
|
|
206
|
+
|
|
207
|
+
# Specialized agents
|
|
208
|
+
self.research_agent = ReAct(
|
|
209
|
+
ResearchAgent,
|
|
210
|
+
tools=[search_wikipedia, search_web]
|
|
211
|
+
)
|
|
212
|
+
self.math_agent = dspy.ProgramOfThought("problem -> answer")
|
|
213
|
+
self.reasoning_agent = dspy.ChainOfThought("question -> answer")
|
|
214
|
+
|
|
215
|
+
def forward(self, question):
|
|
216
|
+
# Route to appropriate agent
|
|
217
|
+
agent_type = self.router(question=question).agent_type
|
|
218
|
+
|
|
219
|
+
if agent_type == "research":
|
|
220
|
+
return self.research_agent(question=question)
|
|
221
|
+
elif agent_type == "math":
|
|
222
|
+
return self.math_agent(problem=question)
|
|
223
|
+
else:
|
|
224
|
+
return self.reasoning_agent(question=question)
|
|
225
|
+
|
|
226
|
+
# Use multi-agent system
|
|
227
|
+
mas = MultiAgentSystem()
|
|
228
|
+
result = mas(question="What is 15% of the GDP of France?")
|
|
229
|
+
# Routes to research_agent for GDP, then to math_agent for calculation
|
|
230
|
+
```
|
|
231
|
+
|
|
232
|
+
## Classification
|
|
233
|
+
|
|
234
|
+
### Binary Classifier
|
|
235
|
+
|
|
236
|
+
```python
|
|
237
|
+
class SentimentClassifier(dspy.Module):
|
|
238
|
+
def __init__(self):
|
|
239
|
+
super().__init__()
|
|
240
|
+
self.classify = dspy.Predict("text -> sentiment: str")
|
|
241
|
+
|
|
242
|
+
def forward(self, text):
|
|
243
|
+
return self.classify(text=text)
|
|
244
|
+
|
|
245
|
+
# Training data
|
|
246
|
+
trainset = [
|
|
247
|
+
dspy.Example(text="I love this!", sentiment="positive").with_inputs("text"),
|
|
248
|
+
dspy.Example(text="Terrible experience", sentiment="negative").with_inputs("text"),
|
|
249
|
+
# ... more examples
|
|
250
|
+
]
|
|
251
|
+
|
|
252
|
+
# Optimize
|
|
253
|
+
def accuracy(example, pred, trace=None):
|
|
254
|
+
return example.sentiment == pred.sentiment
|
|
255
|
+
|
|
256
|
+
optimizer = BootstrapFewShot(metric=accuracy, max_bootstrapped_demos=5)
|
|
257
|
+
classifier = SentimentClassifier()
|
|
258
|
+
optimized_classifier = optimizer.compile(classifier, trainset=trainset)
|
|
259
|
+
|
|
260
|
+
# Use classifier
|
|
261
|
+
result = optimized_classifier(text="This product is amazing!")
|
|
262
|
+
print(result.sentiment) # "positive"
|
|
263
|
+
```
|
|
264
|
+
|
|
265
|
+
### Multi-Class Classifier
|
|
266
|
+
|
|
267
|
+
```python
|
|
268
|
+
class TopicClassifier(dspy.Module):
|
|
269
|
+
def __init__(self):
|
|
270
|
+
super().__init__()
|
|
271
|
+
self.classify = dspy.ChainOfThought(
|
|
272
|
+
"text -> category: str, confidence: float"
|
|
273
|
+
)
|
|
274
|
+
|
|
275
|
+
def forward(self, text):
|
|
276
|
+
result = self.classify(text=text)
|
|
277
|
+
return dspy.Prediction(
|
|
278
|
+
category=result.category,
|
|
279
|
+
confidence=float(result.confidence)
|
|
280
|
+
)
|
|
281
|
+
|
|
282
|
+
# Define categories in signature
|
|
283
|
+
class TopicSignature(dspy.Signature):
|
|
284
|
+
"""Classify text into one of: technology, sports, politics, entertainment."""
|
|
285
|
+
text = dspy.InputField()
|
|
286
|
+
category = dspy.OutputField(desc="one of: technology, sports, politics, entertainment")
|
|
287
|
+
confidence = dspy.OutputField(desc="0.0 to 1.0")
|
|
288
|
+
|
|
289
|
+
classifier = dspy.ChainOfThought(TopicSignature)
|
|
290
|
+
result = classifier(text="The Lakers won the championship")
|
|
291
|
+
print(result.category) # "sports"
|
|
292
|
+
print(result.confidence) # 0.95
|
|
293
|
+
```
|
|
294
|
+
|
|
295
|
+
### Hierarchical Classifier
|
|
296
|
+
|
|
297
|
+
```python
|
|
298
|
+
class HierarchicalClassifier(dspy.Module):
|
|
299
|
+
"""Two-stage classification: coarse then fine-grained."""
|
|
300
|
+
|
|
301
|
+
def __init__(self):
|
|
302
|
+
super().__init__()
|
|
303
|
+
self.coarse = dspy.Predict("text -> broad_category: str")
|
|
304
|
+
self.fine_tech = dspy.Predict("text -> tech_subcategory: str")
|
|
305
|
+
self.fine_sports = dspy.Predict("text -> sports_subcategory: str")
|
|
306
|
+
|
|
307
|
+
def forward(self, text):
|
|
308
|
+
# Stage 1: Broad category
|
|
309
|
+
broad = self.coarse(text=text).broad_category
|
|
310
|
+
|
|
311
|
+
# Stage 2: Fine-grained based on broad
|
|
312
|
+
if broad == "technology":
|
|
313
|
+
fine = self.fine_tech(text=text).tech_subcategory
|
|
314
|
+
elif broad == "sports":
|
|
315
|
+
fine = self.fine_sports(text=text).sports_subcategory
|
|
316
|
+
else:
|
|
317
|
+
fine = "other"
|
|
318
|
+
|
|
319
|
+
return dspy.Prediction(broad_category=broad, fine_category=fine)
|
|
320
|
+
```
|
|
321
|
+
|
|
322
|
+
## Data Processing
|
|
323
|
+
|
|
324
|
+
### Text Summarization
|
|
325
|
+
|
|
326
|
+
```python
|
|
327
|
+
class AdaptiveSummarizer(dspy.Module):
|
|
328
|
+
"""Summarizes text to target length."""
|
|
329
|
+
|
|
330
|
+
def __init__(self):
|
|
331
|
+
super().__init__()
|
|
332
|
+
self.summarize = dspy.ChainOfThought("text, target_length -> summary")
|
|
333
|
+
|
|
334
|
+
def forward(self, text, target_length="3 sentences"):
|
|
335
|
+
return self.summarize(text=text, target_length=target_length)
|
|
336
|
+
|
|
337
|
+
# Use summarizer
|
|
338
|
+
summarizer = AdaptiveSummarizer()
|
|
339
|
+
long_text = "..." # Long article
|
|
340
|
+
|
|
341
|
+
short_summary = summarizer(long_text, target_length="1 sentence")
|
|
342
|
+
medium_summary = summarizer(long_text, target_length="3 sentences")
|
|
343
|
+
detailed_summary = summarizer(long_text, target_length="1 paragraph")
|
|
344
|
+
```
|
|
345
|
+
|
|
346
|
+
### Information Extraction
|
|
347
|
+
|
|
348
|
+
```python
|
|
349
|
+
from pydantic import BaseModel, Field
|
|
350
|
+
|
|
351
|
+
class PersonInfo(BaseModel):
|
|
352
|
+
name: str = Field(description="Full name")
|
|
353
|
+
age: int = Field(description="Age in years")
|
|
354
|
+
occupation: str = Field(description="Job title")
|
|
355
|
+
location: str = Field(description="City and country")
|
|
356
|
+
|
|
357
|
+
class ExtractPerson(dspy.Signature):
|
|
358
|
+
"""Extract person information from text."""
|
|
359
|
+
text = dspy.InputField()
|
|
360
|
+
person: PersonInfo = dspy.OutputField()
|
|
361
|
+
|
|
362
|
+
extractor = dspy.TypedPredictor(ExtractPerson)
|
|
363
|
+
|
|
364
|
+
text = "Dr. Jane Smith, 42, is a neuroscientist at Stanford University in Palo Alto, California."
|
|
365
|
+
result = extractor(text=text)
|
|
366
|
+
|
|
367
|
+
print(result.person.name) # "Dr. Jane Smith"
|
|
368
|
+
print(result.person.age) # 42
|
|
369
|
+
print(result.person.occupation) # "neuroscientist"
|
|
370
|
+
print(result.person.location) # "Palo Alto, California"
|
|
371
|
+
```
|
|
372
|
+
|
|
373
|
+
### Batch Processing
|
|
374
|
+
|
|
375
|
+
```python
|
|
376
|
+
class BatchProcessor(dspy.Module):
|
|
377
|
+
"""Process large datasets efficiently."""
|
|
378
|
+
|
|
379
|
+
def __init__(self):
|
|
380
|
+
super().__init__()
|
|
381
|
+
self.process = dspy.Predict("text -> processed_text")
|
|
382
|
+
|
|
383
|
+
def forward(self, texts):
|
|
384
|
+
# Batch processing for efficiency
|
|
385
|
+
return self.process.batch([{"text": t} for t in texts])
|
|
386
|
+
|
|
387
|
+
# Process 1000 documents
|
|
388
|
+
processor = BatchProcessor()
|
|
389
|
+
results = processor(texts=large_dataset)
|
|
390
|
+
|
|
391
|
+
# Results are returned in order
|
|
392
|
+
for original, result in zip(large_dataset, results):
|
|
393
|
+
print(f"{original} -> {result.processed_text}")
|
|
394
|
+
```
|
|
395
|
+
|
|
396
|
+
## Multi-Stage Pipelines
|
|
397
|
+
|
|
398
|
+
### Document Processing Pipeline
|
|
399
|
+
|
|
400
|
+
```python
|
|
401
|
+
class DocumentPipeline(dspy.Module):
|
|
402
|
+
"""Multi-stage document processing."""
|
|
403
|
+
|
|
404
|
+
def __init__(self):
|
|
405
|
+
super().__init__()
|
|
406
|
+
self.extract = dspy.Predict("document -> key_points")
|
|
407
|
+
self.classify = dspy.Predict("key_points -> category")
|
|
408
|
+
self.summarize = dspy.ChainOfThought("key_points, category -> summary")
|
|
409
|
+
self.tag = dspy.Predict("summary -> tags")
|
|
410
|
+
|
|
411
|
+
def forward(self, document):
|
|
412
|
+
# Stage 1: Extract key points
|
|
413
|
+
key_points = self.extract(document=document).key_points
|
|
414
|
+
|
|
415
|
+
# Stage 2: Classify
|
|
416
|
+
category = self.classify(key_points=key_points).category
|
|
417
|
+
|
|
418
|
+
# Stage 3: Summarize
|
|
419
|
+
summary = self.summarize(
|
|
420
|
+
key_points=key_points,
|
|
421
|
+
category=category
|
|
422
|
+
).summary
|
|
423
|
+
|
|
424
|
+
# Stage 4: Generate tags
|
|
425
|
+
tags = self.tag(summary=summary).tags
|
|
426
|
+
|
|
427
|
+
return dspy.Prediction(
|
|
428
|
+
key_points=key_points,
|
|
429
|
+
category=category,
|
|
430
|
+
summary=summary,
|
|
431
|
+
tags=tags
|
|
432
|
+
)
|
|
433
|
+
```
|
|
434
|
+
|
|
435
|
+
### Quality Control Pipeline
|
|
436
|
+
|
|
437
|
+
```python
|
|
438
|
+
class QualityControlPipeline(dspy.Module):
|
|
439
|
+
"""Generate output and verify quality."""
|
|
440
|
+
|
|
441
|
+
def __init__(self):
|
|
442
|
+
super().__init__()
|
|
443
|
+
self.generate = dspy.ChainOfThought("prompt -> output")
|
|
444
|
+
self.verify = dspy.Predict("output -> is_valid: bool, issues: str")
|
|
445
|
+
self.improve = dspy.ChainOfThought("output, issues -> improved_output")
|
|
446
|
+
|
|
447
|
+
def forward(self, prompt, max_iterations=3):
|
|
448
|
+
output = self.generate(prompt=prompt).output
|
|
449
|
+
|
|
450
|
+
for _ in range(max_iterations):
|
|
451
|
+
# Verify output
|
|
452
|
+
verification = self.verify(output=output)
|
|
453
|
+
|
|
454
|
+
if verification.is_valid:
|
|
455
|
+
return dspy.Prediction(output=output, iterations=_ + 1)
|
|
456
|
+
|
|
457
|
+
# Improve based on issues
|
|
458
|
+
output = self.improve(
|
|
459
|
+
output=output,
|
|
460
|
+
issues=verification.issues
|
|
461
|
+
).improved_output
|
|
462
|
+
|
|
463
|
+
return dspy.Prediction(output=output, iterations=max_iterations)
|
|
464
|
+
```
|
|
465
|
+
|
|
466
|
+
## Production Tips
|
|
467
|
+
|
|
468
|
+
### 1. Caching for Performance
|
|
469
|
+
|
|
470
|
+
```python
|
|
471
|
+
from functools import lru_cache
|
|
472
|
+
|
|
473
|
+
class CachedRAG(dspy.Module):
|
|
474
|
+
def __init__(self):
|
|
475
|
+
super().__init__()
|
|
476
|
+
self.retrieve = dspy.Retrieve(k=3)
|
|
477
|
+
self.generate = dspy.ChainOfThought("context, question -> answer")
|
|
478
|
+
|
|
479
|
+
@lru_cache(maxsize=1000)
|
|
480
|
+
def forward(self, question):
|
|
481
|
+
passages = self.retrieve(question).passages
|
|
482
|
+
context = "\n".join(passages)
|
|
483
|
+
return self.generate(context=context, question=question).answer
|
|
484
|
+
```
|
|
485
|
+
|
|
486
|
+
### 2. Error Handling
|
|
487
|
+
|
|
488
|
+
```python
|
|
489
|
+
class RobustModule(dspy.Module):
|
|
490
|
+
def __init__(self):
|
|
491
|
+
super().__init__()
|
|
492
|
+
self.process = dspy.ChainOfThought("input -> output")
|
|
493
|
+
|
|
494
|
+
def forward(self, input):
|
|
495
|
+
try:
|
|
496
|
+
result = self.process(input=input)
|
|
497
|
+
return result
|
|
498
|
+
except Exception as e:
|
|
499
|
+
# Log error
|
|
500
|
+
print(f"Error processing {input}: {e}")
|
|
501
|
+
# Return fallback
|
|
502
|
+
return dspy.Prediction(output="Error: could not process input")
|
|
503
|
+
```
|
|
504
|
+
|
|
505
|
+
### 3. Monitoring
|
|
506
|
+
|
|
507
|
+
```python
|
|
508
|
+
class MonitoredModule(dspy.Module):
|
|
509
|
+
def __init__(self):
|
|
510
|
+
super().__init__()
|
|
511
|
+
self.process = dspy.ChainOfThought("input -> output")
|
|
512
|
+
self.call_count = 0
|
|
513
|
+
self.errors = 0
|
|
514
|
+
|
|
515
|
+
def forward(self, input):
|
|
516
|
+
self.call_count += 1
|
|
517
|
+
|
|
518
|
+
try:
|
|
519
|
+
result = self.process(input=input)
|
|
520
|
+
return result
|
|
521
|
+
except Exception as e:
|
|
522
|
+
self.errors += 1
|
|
523
|
+
raise
|
|
524
|
+
|
|
525
|
+
def get_stats(self):
|
|
526
|
+
return {
|
|
527
|
+
"calls": self.call_count,
|
|
528
|
+
"errors": self.errors,
|
|
529
|
+
"error_rate": self.errors / max(self.call_count, 1)
|
|
530
|
+
}
|
|
531
|
+
```
|
|
532
|
+
|
|
533
|
+
### 4. A/B Testing
|
|
534
|
+
|
|
535
|
+
```python
|
|
536
|
+
class ABTestModule(dspy.Module):
|
|
537
|
+
"""Run two variants and compare."""
|
|
538
|
+
|
|
539
|
+
def __init__(self, variant_a, variant_b):
|
|
540
|
+
super().__init__()
|
|
541
|
+
self.variant_a = variant_a
|
|
542
|
+
self.variant_b = variant_b
|
|
543
|
+
self.a_calls = 0
|
|
544
|
+
self.b_calls = 0
|
|
545
|
+
|
|
546
|
+
def forward(self, input, variant="a"):
|
|
547
|
+
if variant == "a":
|
|
548
|
+
self.a_calls += 1
|
|
549
|
+
return self.variant_a(input=input)
|
|
550
|
+
else:
|
|
551
|
+
self.b_calls += 1
|
|
552
|
+
return self.variant_b(input=input)
|
|
553
|
+
|
|
554
|
+
# Compare two optimizers
|
|
555
|
+
baseline = dspy.ChainOfThought("question -> answer")
|
|
556
|
+
optimized = BootstrapFewShot(...).compile(baseline, trainset=trainset)
|
|
557
|
+
|
|
558
|
+
ab_test = ABTestModule(variant_a=baseline, variant_b=optimized)
|
|
559
|
+
|
|
560
|
+
# Route 50% to each
|
|
561
|
+
import random
|
|
562
|
+
variant = "a" if random.random() < 0.5 else "b"
|
|
563
|
+
result = ab_test(input=question, variant=variant)
|
|
564
|
+
```
|
|
565
|
+
|
|
566
|
+
## Complete Example: Customer Support Bot
|
|
567
|
+
|
|
568
|
+
```python
|
|
569
|
+
import dspy
|
|
570
|
+
from dspy.teleprompt import BootstrapFewShot
|
|
571
|
+
|
|
572
|
+
class CustomerSupportBot(dspy.Module):
|
|
573
|
+
"""Complete customer support system."""
|
|
574
|
+
|
|
575
|
+
def __init__(self):
|
|
576
|
+
super().__init__()
|
|
577
|
+
|
|
578
|
+
# Classify intent
|
|
579
|
+
self.classify_intent = dspy.Predict("message -> intent: str")
|
|
580
|
+
|
|
581
|
+
# Specialized handlers
|
|
582
|
+
self.technical_handler = dspy.ChainOfThought("message, history -> response")
|
|
583
|
+
self.billing_handler = dspy.ChainOfThought("message, history -> response")
|
|
584
|
+
self.general_handler = dspy.Predict("message, history -> response")
|
|
585
|
+
|
|
586
|
+
# Retrieve relevant docs
|
|
587
|
+
self.retrieve = dspy.Retrieve(k=3)
|
|
588
|
+
|
|
589
|
+
# Conversation history
|
|
590
|
+
self.history = []
|
|
591
|
+
|
|
592
|
+
def forward(self, message):
|
|
593
|
+
# Classify intent
|
|
594
|
+
intent = self.classify_intent(message=message).intent
|
|
595
|
+
|
|
596
|
+
# Retrieve relevant documentation
|
|
597
|
+
docs = self.retrieve(message).passages
|
|
598
|
+
context = "\n".join(docs)
|
|
599
|
+
|
|
600
|
+
# Add context to history
|
|
601
|
+
history_str = "\n".join(self.history)
|
|
602
|
+
full_message = f"Context: {context}\n\nMessage: {message}"
|
|
603
|
+
|
|
604
|
+
# Route to appropriate handler
|
|
605
|
+
if intent == "technical":
|
|
606
|
+
response = self.technical_handler(
|
|
607
|
+
message=full_message,
|
|
608
|
+
history=history_str
|
|
609
|
+
).response
|
|
610
|
+
elif intent == "billing":
|
|
611
|
+
response = self.billing_handler(
|
|
612
|
+
message=full_message,
|
|
613
|
+
history=history_str
|
|
614
|
+
).response
|
|
615
|
+
else:
|
|
616
|
+
response = self.general_handler(
|
|
617
|
+
message=full_message,
|
|
618
|
+
history=history_str
|
|
619
|
+
).response
|
|
620
|
+
|
|
621
|
+
# Update history
|
|
622
|
+
self.history.append(f"User: {message}")
|
|
623
|
+
self.history.append(f"Bot: {response}")
|
|
624
|
+
|
|
625
|
+
return dspy.Prediction(response=response, intent=intent)
|
|
626
|
+
|
|
627
|
+
# Training data
|
|
628
|
+
trainset = [
|
|
629
|
+
dspy.Example(
|
|
630
|
+
message="My account isn't working",
|
|
631
|
+
intent="technical",
|
|
632
|
+
response="I'd be happy to help. What error are you seeing?"
|
|
633
|
+
).with_inputs("message"),
|
|
634
|
+
# ... more examples
|
|
635
|
+
]
|
|
636
|
+
|
|
637
|
+
# Define metric
|
|
638
|
+
def response_quality(example, pred, trace=None):
|
|
639
|
+
# Check if response is helpful
|
|
640
|
+
if len(pred.response) < 20:
|
|
641
|
+
return 0.0
|
|
642
|
+
if example.intent != pred.intent:
|
|
643
|
+
return 0.3
|
|
644
|
+
return 1.0
|
|
645
|
+
|
|
646
|
+
# Optimize
|
|
647
|
+
optimizer = BootstrapFewShot(metric=response_quality)
|
|
648
|
+
bot = CustomerSupportBot()
|
|
649
|
+
optimized_bot = optimizer.compile(bot, trainset=trainset)
|
|
650
|
+
|
|
651
|
+
# Use in production
|
|
652
|
+
optimized_bot.save("models/support_bot_v1.json")
|
|
653
|
+
|
|
654
|
+
# Later, load and use
|
|
655
|
+
loaded_bot = CustomerSupportBot()
|
|
656
|
+
loaded_bot.load("models/support_bot_v1.json")
|
|
657
|
+
response = loaded_bot(message="I can't log in")
|
|
658
|
+
```
|
|
659
|
+
|
|
660
|
+
## Resources
|
|
661
|
+
|
|
662
|
+
- **Documentation**: https://dspy.ai
|
|
663
|
+
- **Examples Repo**: https://github.com/stanfordnlp/dspy/tree/main/examples
|
|
664
|
+
- **Discord**: https://discord.gg/XCGy2WDCQB
|