@mytechtoday/augment-extensions 0.2.0 → 0.5.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/README.md +614 -39
- package/augment-extensions/coding-standards/bash/README.md +196 -0
- package/augment-extensions/coding-standards/bash/module.json +163 -0
- package/augment-extensions/coding-standards/bash/rules/naming-conventions.md +336 -0
- package/augment-extensions/coding-standards/bash/rules/universal-standards.md +289 -0
- package/augment-extensions/coding-standards/css/README.md +40 -0
- package/augment-extensions/coding-standards/css/examples/css-examples.css +550 -0
- package/augment-extensions/coding-standards/css/module.json +44 -0
- package/augment-extensions/coding-standards/css/rules/css-modern-features.md +448 -0
- package/augment-extensions/coding-standards/css/rules/css-standards.md +492 -0
- package/augment-extensions/coding-standards/html/README.md +40 -0
- package/augment-extensions/coding-standards/html/examples/html-examples.html +267 -0
- package/augment-extensions/coding-standards/html/examples/responsive-layout.html +505 -0
- package/augment-extensions/coding-standards/html/module.json +44 -0
- package/augment-extensions/coding-standards/html/rules/html-standards.md +349 -0
- package/augment-extensions/coding-standards/html-css-js/README.md +194 -0
- package/augment-extensions/coding-standards/html-css-js/examples/async-examples.js +487 -0
- package/augment-extensions/coding-standards/html-css-js/examples/css-examples.css +550 -0
- package/augment-extensions/coding-standards/html-css-js/examples/dom-examples.js +667 -0
- package/augment-extensions/coding-standards/html-css-js/examples/html-examples.html +267 -0
- package/augment-extensions/coding-standards/html-css-js/examples/javascript-examples.js +612 -0
- package/augment-extensions/coding-standards/html-css-js/examples/responsive-layout.html +505 -0
- package/augment-extensions/coding-standards/html-css-js/module.json +48 -0
- package/augment-extensions/coding-standards/html-css-js/rules/async-patterns.md +515 -0
- package/augment-extensions/coding-standards/html-css-js/rules/css-modern-features.md +448 -0
- package/augment-extensions/coding-standards/html-css-js/rules/css-standards.md +492 -0
- package/augment-extensions/coding-standards/html-css-js/rules/dom-manipulation.md +439 -0
- package/augment-extensions/coding-standards/html-css-js/rules/html-standards.md +349 -0
- package/augment-extensions/coding-standards/html-css-js/rules/javascript-standards.md +486 -0
- package/augment-extensions/coding-standards/html-css-js/rules/performance.md +463 -0
- package/augment-extensions/coding-standards/html-css-js/rules/tooling.md +543 -0
- package/augment-extensions/coding-standards/js/README.md +46 -0
- package/augment-extensions/coding-standards/js/examples/async-examples.js +487 -0
- package/augment-extensions/coding-standards/js/examples/dom-examples.js +667 -0
- package/augment-extensions/coding-standards/js/examples/javascript-examples.js +612 -0
- package/augment-extensions/coding-standards/js/module.json +49 -0
- package/augment-extensions/coding-standards/js/rules/async-patterns.md +515 -0
- package/augment-extensions/coding-standards/js/rules/dom-manipulation.md +439 -0
- package/augment-extensions/coding-standards/js/rules/javascript-standards.md +486 -0
- package/augment-extensions/coding-standards/js/rules/performance.md +463 -0
- package/augment-extensions/coding-standards/js/rules/tooling.md +543 -0
- package/augment-extensions/coding-standards/php/README.md +248 -0
- package/augment-extensions/coding-standards/php/examples/api-endpoint-example.php +204 -0
- package/augment-extensions/coding-standards/php/examples/cli-command-example.php +206 -0
- package/augment-extensions/coding-standards/php/examples/legacy-refactoring-example.php +234 -0
- package/augment-extensions/coding-standards/php/examples/web-application-example.php +211 -0
- package/augment-extensions/coding-standards/php/examples/woocommerce-extension-example.php +215 -0
- package/augment-extensions/coding-standards/php/examples/wordpress-plugin-example.php +189 -0
- package/augment-extensions/coding-standards/php/module.json +166 -0
- package/augment-extensions/coding-standards/php/rules/api-development.md +480 -0
- package/augment-extensions/coding-standards/php/rules/category-configuration.md +332 -0
- package/augment-extensions/coding-standards/php/rules/cli-tools.md +472 -0
- package/augment-extensions/coding-standards/php/rules/cms-integration.md +561 -0
- package/augment-extensions/coding-standards/php/rules/code-quality.md +402 -0
- package/augment-extensions/coding-standards/php/rules/documentation.md +425 -0
- package/augment-extensions/coding-standards/php/rules/ecommerce.md +627 -0
- package/augment-extensions/coding-standards/php/rules/error-handling.md +336 -0
- package/augment-extensions/coding-standards/php/rules/legacy-migration.md +677 -0
- package/augment-extensions/coding-standards/php/rules/naming-conventions.md +279 -0
- package/augment-extensions/coding-standards/php/rules/performance.md +392 -0
- package/augment-extensions/coding-standards/php/rules/psr-standards.md +186 -0
- package/augment-extensions/coding-standards/php/rules/security.md +358 -0
- package/augment-extensions/coding-standards/php/rules/testing.md +403 -0
- package/augment-extensions/coding-standards/php/rules/type-declarations.md +331 -0
- package/augment-extensions/coding-standards/php/rules/web-applications.md +426 -0
- package/augment-extensions/coding-standards/powershell/README.md +154 -0
- package/augment-extensions/coding-standards/powershell/examples/admin-example.ps1 +272 -0
- package/augment-extensions/coding-standards/powershell/examples/automation-example.ps1 +173 -0
- package/augment-extensions/coding-standards/powershell/examples/cloud-example.ps1 +243 -0
- package/augment-extensions/coding-standards/powershell/examples/cross-platform-example.ps1 +297 -0
- package/augment-extensions/coding-standards/powershell/examples/dsc-example.ps1 +224 -0
- package/augment-extensions/coding-standards/powershell/examples/legacy-migration-example.ps1 +340 -0
- package/augment-extensions/coding-standards/powershell/examples/module-example.psm1 +255 -0
- package/augment-extensions/coding-standards/powershell/module.json +165 -0
- package/augment-extensions/coding-standards/powershell/rules/administrative-tools.md +439 -0
- package/augment-extensions/coding-standards/powershell/rules/automation-scripts.md +240 -0
- package/augment-extensions/coding-standards/powershell/rules/cloud-orchestration.md +384 -0
- package/augment-extensions/coding-standards/powershell/rules/configuration-schema.md +383 -0
- package/augment-extensions/coding-standards/powershell/rules/cross-platform-scripts.md +482 -0
- package/augment-extensions/coding-standards/powershell/rules/dsc-configurations.md +296 -0
- package/augment-extensions/coding-standards/powershell/rules/error-handling.md +314 -0
- package/augment-extensions/coding-standards/powershell/rules/legacy-migrations.md +466 -0
- package/augment-extensions/coding-standards/powershell/rules/modules-functions.md +244 -0
- package/augment-extensions/coding-standards/powershell/rules/naming-conventions.md +266 -0
- package/augment-extensions/coding-standards/powershell/rules/performance-optimization.md +209 -0
- package/augment-extensions/coding-standards/powershell/rules/security-practices.md +314 -0
- package/augment-extensions/coding-standards/powershell/rules/testing-guidelines.md +268 -0
- package/augment-extensions/coding-standards/powershell/rules/universal-standards.md +197 -0
- package/augment-extensions/coding-standards/python/README.md +12 -8
- package/augment-extensions/coding-standards/python/examples/best-practices.py +373 -0
- package/augment-extensions/coding-standards/python/module.json +8 -4
- package/augment-extensions/coding-standards/python/rules/async-patterns.md +884 -0
- package/augment-extensions/coding-standards/python/rules/documentation.md +831 -0
- package/augment-extensions/coding-standards/python/rules/error-handling.md +855 -68
- package/augment-extensions/coding-standards/python/rules/testing.md +409 -0
- package/augment-extensions/coding-standards/python/rules/tooling.md +446 -0
- package/augment-extensions/coding-standards/python/rules/type-hints.md +115 -50
- package/augment-extensions/collections/html-css-js/README.md +82 -0
- package/augment-extensions/collections/html-css-js/collection.json +41 -0
- package/augment-extensions/domain-rules/database/README.md +161 -0
- package/augment-extensions/domain-rules/database/examples/flat-database-example.md +793 -0
- package/augment-extensions/domain-rules/database/examples/hybrid-database-example.md +1132 -0
- package/augment-extensions/domain-rules/database/examples/nosql-document-example.md +868 -0
- package/augment-extensions/domain-rules/database/examples/nosql-graph-example.md +805 -0
- package/augment-extensions/domain-rules/database/examples/relational-schema-example.md +621 -0
- package/augment-extensions/domain-rules/database/examples/vector-database-example.md +965 -0
- package/augment-extensions/domain-rules/database/module.json +28 -0
- package/augment-extensions/domain-rules/database/rules/flat-databases.md +624 -0
- package/augment-extensions/domain-rules/database/rules/nosql-databases.md +588 -0
- package/augment-extensions/domain-rules/database/rules/nosql-document-stores.md +856 -0
- package/augment-extensions/domain-rules/database/rules/nosql-graph-databases.md +778 -0
- package/augment-extensions/domain-rules/database/rules/nosql-key-value-stores.md +963 -0
- package/augment-extensions/domain-rules/database/rules/performance-optimization.md +1076 -0
- package/augment-extensions/domain-rules/database/rules/relational-databases.md +697 -0
- package/augment-extensions/domain-rules/database/rules/relational-indexing.md +671 -0
- package/augment-extensions/domain-rules/database/rules/relational-query-optimization.md +607 -0
- package/augment-extensions/domain-rules/database/rules/relational-schema-design.md +907 -0
- package/augment-extensions/domain-rules/database/rules/relational-transactions.md +783 -0
- package/augment-extensions/domain-rules/database/rules/security-standards.md +980 -0
- package/augment-extensions/domain-rules/database/rules/universal-best-practices.md +485 -0
- package/augment-extensions/domain-rules/database/rules/vector-databases.md +521 -0
- package/augment-extensions/domain-rules/database/rules/vector-embeddings.md +858 -0
- package/augment-extensions/domain-rules/database/rules/vector-indexing.md +934 -0
- package/augment-extensions/domain-rules/design/color/themes/catppuccin-latte/README.md +23 -0
- package/augment-extensions/domain-rules/design/color/themes/catppuccin-latte/module.json +26 -0
- package/augment-extensions/domain-rules/design/color/themes/catppuccin-mocha/README.md +23 -0
- package/augment-extensions/domain-rules/design/color/themes/catppuccin-mocha/module.json +26 -0
- package/augment-extensions/domain-rules/design/color/themes/dracula/README.md +23 -0
- package/augment-extensions/domain-rules/design/color/themes/dracula/module.json +26 -0
- package/augment-extensions/domain-rules/design/color/themes/gruvbox-dark/README.md +23 -0
- package/augment-extensions/domain-rules/design/color/themes/gruvbox-dark/module.json +26 -0
- package/augment-extensions/domain-rules/design/color/themes/gruvbox-light/README.md +23 -0
- package/augment-extensions/domain-rules/design/color/themes/gruvbox-light/module.json +26 -0
- package/augment-extensions/domain-rules/design/color/themes/high-contrast/README.md +27 -0
- package/augment-extensions/domain-rules/design/color/themes/high-contrast/module.json +26 -0
- package/augment-extensions/domain-rules/design/color/themes/monokai/README.md +23 -0
- package/augment-extensions/domain-rules/design/color/themes/monokai/module.json +26 -0
- package/augment-extensions/domain-rules/design/color/themes/nord/README.md +23 -0
- package/augment-extensions/domain-rules/design/color/themes/nord/module.json +26 -0
- package/augment-extensions/domain-rules/design/color/themes/one-dark/README.md +23 -0
- package/augment-extensions/domain-rules/design/color/themes/one-dark/module.json +26 -0
- package/augment-extensions/domain-rules/design/color/themes/one-light/README.md +23 -0
- package/augment-extensions/domain-rules/design/color/themes/one-light/module.json +26 -0
- package/augment-extensions/domain-rules/design/color/themes/solarized-dark/README.md +23 -0
- package/augment-extensions/domain-rules/design/color/themes/solarized-dark/module.json +26 -0
- package/augment-extensions/domain-rules/design/color/themes/solarized-light/README.md +23 -0
- package/augment-extensions/domain-rules/design/color/themes/solarized-light/module.json +26 -0
- package/augment-extensions/domain-rules/design/color/themes/tokyo-night/README.md +23 -0
- package/augment-extensions/domain-rules/design/color/themes/tokyo-night/module.json +26 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/README.md +136 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/SCHEMA-VALIDATION-REPORT.md +216 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/examples/brand-kit-example.yaml +292 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/examples/campaign-brief-example.yaml +389 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/examples/content-calendar-example.yaml +643 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/examples/email-newsletter-example.md +376 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/examples/landing-page-example.md +934 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/examples/ppc-ad-copy-example.md +301 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/examples/seo-blog-post-example.md +347 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/examples/social-media-campaign-example.md +606 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/module.json +50 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/rules/affiliate-influencer-marketing.md +593 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/rules/asset-management.md +418 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/rules/brand-consistency.md +210 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/rules/content-marketing.md +337 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/rules/conversion-optimization.md +455 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/rules/direct-sales.md +499 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/rules/email-marketing.md +439 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/rules/legal-compliance.md +227 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/rules/ppc-advertising.md +569 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/rules/seo-optimization.md +470 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/rules/social-media-marketing.md +414 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/rules/universal-marketing.md +177 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/schemas/asset-inventory.schema.json +247 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/schemas/brand-kit.schema.json +326 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/schemas/campaign-brief.schema.json +342 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/schemas/color-palette.schema.json +223 -0
- package/augment-extensions/domain-rules/marketing-standards/seo-sales-marketing/schemas/content-template.schema.json +383 -0
- package/augment-extensions/domain-rules/mcp/README.md +150 -0
- package/augment-extensions/domain-rules/mcp/examples/compressed-example.md +522 -0
- package/augment-extensions/domain-rules/mcp/examples/graph-augmented-example.md +520 -0
- package/augment-extensions/domain-rules/mcp/examples/hybrid-example.md +570 -0
- package/augment-extensions/domain-rules/mcp/examples/state-based-example.md +427 -0
- package/augment-extensions/domain-rules/mcp/examples/token-based-example.md +435 -0
- package/augment-extensions/domain-rules/mcp/examples/vector-based-example.md +502 -0
- package/augment-extensions/domain-rules/mcp/module.json +49 -0
- package/augment-extensions/domain-rules/mcp/rules/compressed-mcp.md +595 -0
- package/augment-extensions/domain-rules/mcp/rules/configuration.md +345 -0
- package/augment-extensions/domain-rules/mcp/rules/graph-augmented-mcp.md +687 -0
- package/augment-extensions/domain-rules/mcp/rules/hybrid-mcp.md +636 -0
- package/augment-extensions/domain-rules/mcp/rules/state-based-mcp.md +484 -0
- package/augment-extensions/domain-rules/mcp/rules/testing-validation.md +360 -0
- package/augment-extensions/domain-rules/mcp/rules/token-based-mcp.md +393 -0
- package/augment-extensions/domain-rules/mcp/rules/universal-rules.md +194 -0
- package/augment-extensions/domain-rules/mcp/rules/vector-based-mcp.md +625 -0
- package/augment-extensions/workflows/beads/module.json +4 -3
- package/augment-extensions/workflows/beads-integration/IMPLEMENTATION-STATUS.md +145 -0
- package/augment-extensions/workflows/beads-integration/README.md +143 -0
- package/augment-extensions/workflows/beads-integration/config/defaults.json +32 -0
- package/augment-extensions/workflows/beads-integration/config/schema.json +140 -0
- package/augment-extensions/workflows/beads-integration/examples/basic-task-generation.md +293 -0
- package/augment-extensions/workflows/beads-integration/module.json +75 -0
- package/augment-extensions/workflows/beads-integration/rules/core-rules.md +219 -0
- package/augment-extensions/workflows/beads-integration/rules/effectiveness-standards.md +256 -0
- package/augment-extensions/workflows/beads-integration/rules/task-generation.md +607 -0
- package/augment-extensions/workflows/database/README.md +195 -0
- package/augment-extensions/workflows/database/ai-prompt-testing.md +295 -0
- package/augment-extensions/workflows/database/examples/migration-example.md +498 -0
- package/augment-extensions/workflows/database/examples/optimization-example.md +496 -0
- package/augment-extensions/workflows/database/examples/schema-design-example.md +444 -0
- package/augment-extensions/workflows/database/module.json +42 -0
- package/augment-extensions/workflows/database/rules/data-migration.md +249 -0
- package/augment-extensions/workflows/database/rules/documentation-standards.md +339 -0
- package/augment-extensions/workflows/database/rules/migration-workflow.md +352 -0
- package/augment-extensions/workflows/database/rules/optimization-workflow.md +435 -0
- package/augment-extensions/workflows/database/rules/schema-design-workflow.md +535 -0
- package/augment-extensions/workflows/database/rules/testing-patterns.md +305 -0
- package/augment-extensions/workflows/database/rules/workflow.md +458 -0
- package/augment-extensions/workflows/openspec/module.json +4 -3
- package/augment-extensions/writing-standards/screenplay/README.md +300 -0
- package/augment-extensions/writing-standards/screenplay/_templates/README.md +121 -0
- package/augment-extensions/writing-standards/screenplay/_templates/genre-template.md +153 -0
- package/augment-extensions/writing-standards/screenplay/_templates/style-template.md +243 -0
- package/augment-extensions/writing-standards/screenplay/_templates/theme-template.md +213 -0
- package/augment-extensions/writing-standards/screenplay/examples/aaa-hollywood-scene.fountain +164 -0
- package/augment-extensions/writing-standards/screenplay/examples/beat-sheet-example.yaml +95 -0
- package/augment-extensions/writing-standards/screenplay/examples/character-profile-example.yaml +116 -0
- package/augment-extensions/writing-standards/screenplay/examples/commercial-30sec.fountain +151 -0
- package/augment-extensions/writing-standards/screenplay/examples/independent-monologue.fountain +67 -0
- package/augment-extensions/writing-standards/screenplay/examples/news-segment.fountain +142 -0
- package/augment-extensions/writing-standards/screenplay/examples/plot-outline-example.yaml +184 -0
- package/augment-extensions/writing-standards/screenplay/examples/tv-episode-teaser.fountain +204 -0
- package/augment-extensions/writing-standards/screenplay/genres/README.md +181 -0
- package/augment-extensions/writing-standards/screenplay/genres/examples/.gitkeep +2 -0
- package/augment-extensions/writing-standards/screenplay/genres/module.json +70 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/.gitkeep +2 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/action.md +399 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/adventure.md +407 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/animation.md +293 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/biographical.md +293 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/comedy.md +401 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/documentary.md +293 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/drama.md +409 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/fantasy.md +293 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/historical.md +293 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/horror.md +268 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/musical.md +294 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/mystery.md +293 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/noir.md +294 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/romance.md +293 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/sci-fi.md +289 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/superhero.md +293 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/thriller.md +294 -0
- package/augment-extensions/writing-standards/screenplay/genres/rules/western.md +293 -0
- package/augment-extensions/writing-standards/screenplay/module.json +124 -0
- package/augment-extensions/writing-standards/screenplay/rules/aaa-hollywood-films.md +339 -0
- package/augment-extensions/writing-standards/screenplay/rules/ai-integration-testing.md +329 -0
- package/augment-extensions/writing-standards/screenplay/rules/character-development.md +169 -0
- package/augment-extensions/writing-standards/screenplay/rules/commercials.md +437 -0
- package/augment-extensions/writing-standards/screenplay/rules/dialogue-writing.md +263 -0
- package/augment-extensions/writing-standards/screenplay/rules/diversity-inclusion.md +261 -0
- package/augment-extensions/writing-standards/screenplay/rules/examples-guide.md +315 -0
- package/augment-extensions/writing-standards/screenplay/rules/formatting-validation.md +413 -0
- package/augment-extensions/writing-standards/screenplay/rules/fountain-format.md +372 -0
- package/augment-extensions/writing-standards/screenplay/rules/independent-films.md +374 -0
- package/augment-extensions/writing-standards/screenplay/rules/live-tv-productions.md +443 -0
- package/augment-extensions/writing-standards/screenplay/rules/narrative-structures.md +207 -0
- package/augment-extensions/writing-standards/screenplay/rules/news-broadcasts.md +444 -0
- package/augment-extensions/writing-standards/screenplay/rules/pacing-timing.md +331 -0
- package/augment-extensions/writing-standards/screenplay/rules/quality-review-checklist.md +334 -0
- package/augment-extensions/writing-standards/screenplay/rules/quick-reference.md +299 -0
- package/augment-extensions/writing-standards/screenplay/rules/screen-continuity.md +263 -0
- package/augment-extensions/writing-standards/screenplay/rules/streaming-content.md +412 -0
- package/augment-extensions/writing-standards/screenplay/rules/trope-management.md +370 -0
- package/augment-extensions/writing-standards/screenplay/rules/tv-series.md +374 -0
- package/augment-extensions/writing-standards/screenplay/rules/universal-formatting.md +339 -0
- package/augment-extensions/writing-standards/screenplay/rules/vscode-integration.md +277 -0
- package/augment-extensions/writing-standards/screenplay/rules/web-content.md +393 -0
- package/augment-extensions/writing-standards/screenplay/schemas/beat-sheet.json +332 -0
- package/augment-extensions/writing-standards/screenplay/schemas/character-profile.json +247 -0
- package/augment-extensions/writing-standards/screenplay/schemas/feature-selection.json +200 -0
- package/augment-extensions/writing-standards/screenplay/schemas/plot-outline.json +233 -0
- package/augment-extensions/writing-standards/screenplay/schemas/screenplay-config.json +245 -0
- package/augment-extensions/writing-standards/screenplay/schemas/trope-inventory.json +221 -0
- package/augment-extensions/writing-standards/screenplay/styles/README.md +159 -0
- package/augment-extensions/writing-standards/screenplay/styles/examples/.gitkeep +2 -0
- package/augment-extensions/writing-standards/screenplay/styles/examples/style-applications.md +1449 -0
- package/augment-extensions/writing-standards/screenplay/styles/module.json +64 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/.gitkeep +2 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/dialogue-centric.md +520 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/ensemble.md +499 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/epic.md +497 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/experimental.md +492 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/flashback.md +509 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/linear.md +490 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/minimalist.md +499 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/non-linear.md +501 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/poetic.md +499 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/realistic.md +498 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/satirical.md +499 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/surreal.md +508 -0
- package/augment-extensions/writing-standards/screenplay/styles/rules/voice-over.md +500 -0
- package/augment-extensions/writing-standards/screenplay/themes/README.md +158 -0
- package/augment-extensions/writing-standards/screenplay/themes/examples/.gitkeep +2 -0
- package/augment-extensions/writing-standards/screenplay/themes/examples/common-mistakes-and-fixes.md +643 -0
- package/augment-extensions/writing-standards/screenplay/themes/examples/complete-scene-example.md +311 -0
- package/augment-extensions/writing-standards/screenplay/themes/examples/individual-theme-examples.md +562 -0
- package/augment-extensions/writing-standards/screenplay/themes/examples/multi-theme-weaving.md +538 -0
- package/augment-extensions/writing-standards/screenplay/themes/examples/theme-application-guide.md +432 -0
- package/augment-extensions/writing-standards/screenplay/themes/examples/theme-integration-across-acts.md +637 -0
- package/augment-extensions/writing-standards/screenplay/themes/module.json +66 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/.gitkeep +2 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/ambition.md +458 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/betrayal.md +490 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/environment.md +458 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/fate.md +459 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/friendship.md +491 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/growth.md +491 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/identity.md +490 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/isolation.md +464 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/justice.md +461 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/love.md +489 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/power.md +494 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/redemption.md +483 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/revenge.md +489 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/survival.md +496 -0
- package/augment-extensions/writing-standards/screenplay/themes/rules/technology.md +463 -0
- package/cli/MODULES.md +302 -0
- package/cli/dist/cli.js +168 -10
- package/cli/dist/cli.js.map +1 -1
- package/cli/dist/commands/catalog.d.ts +13 -0
- package/cli/dist/commands/catalog.d.ts.map +1 -0
- package/cli/dist/commands/catalog.js +104 -0
- package/cli/dist/commands/catalog.js.map +1 -0
- package/cli/dist/commands/gui.d.ts +6 -0
- package/cli/dist/commands/gui.d.ts.map +1 -0
- package/cli/dist/commands/gui.js +211 -0
- package/cli/dist/commands/gui.js.map +1 -0
- package/cli/dist/commands/init.d.ts.map +1 -1
- package/cli/dist/commands/init.js +12 -0
- package/cli/dist/commands/init.js.map +1 -1
- package/cli/dist/commands/install-rules.d.ts +14 -0
- package/cli/dist/commands/install-rules.d.ts.map +1 -0
- package/cli/dist/commands/install-rules.js +127 -0
- package/cli/dist/commands/install-rules.js.map +1 -0
- package/cli/dist/commands/link.d.ts.map +1 -1
- package/cli/dist/commands/link.js +9 -11
- package/cli/dist/commands/link.js.map +1 -1
- package/cli/dist/commands/list.d.ts.map +1 -1
- package/cli/dist/commands/list.js +11 -28
- package/cli/dist/commands/list.js.map +1 -1
- package/cli/dist/commands/mcp.d.ts +48 -0
- package/cli/dist/commands/mcp.d.ts.map +1 -0
- package/cli/dist/commands/mcp.js +229 -0
- package/cli/dist/commands/mcp.js.map +1 -0
- package/cli/dist/commands/self-remove.d.ts +7 -0
- package/cli/dist/commands/self-remove.d.ts.map +1 -0
- package/cli/dist/commands/self-remove.js +179 -0
- package/cli/dist/commands/self-remove.js.map +1 -0
- package/cli/dist/commands/show.d.ts +19 -0
- package/cli/dist/commands/show.d.ts.map +1 -1
- package/cli/dist/commands/show.js +478 -63
- package/cli/dist/commands/show.js.map +1 -1
- package/cli/dist/commands/skill.d.ts +67 -0
- package/cli/dist/commands/skill.d.ts.map +1 -0
- package/cli/dist/commands/skill.js +513 -0
- package/cli/dist/commands/skill.js.map +1 -0
- package/cli/dist/commands/unlink.d.ts +6 -0
- package/cli/dist/commands/unlink.d.ts.map +1 -0
- package/cli/dist/commands/unlink.js +115 -0
- package/cli/dist/commands/unlink.js.map +1 -0
- package/cli/dist/commands/validate.d.ts +6 -0
- package/cli/dist/commands/validate.d.ts.map +1 -0
- package/cli/dist/commands/validate.js +159 -0
- package/cli/dist/commands/validate.js.map +1 -0
- package/cli/dist/types/gui.d.ts +62 -0
- package/cli/dist/types/gui.d.ts.map +1 -0
- package/cli/dist/types/gui.js +30 -0
- package/cli/dist/types/gui.js.map +1 -0
- package/cli/dist/utils/catalog-sync.d.ts +22 -0
- package/cli/dist/utils/catalog-sync.d.ts.map +1 -0
- package/cli/dist/utils/catalog-sync.js +157 -0
- package/cli/dist/utils/catalog-sync.js.map +1 -0
- package/cli/dist/utils/character-count.d.ts +56 -0
- package/cli/dist/utils/character-count.d.ts.map +1 -0
- package/cli/dist/utils/character-count.js +190 -0
- package/cli/dist/utils/character-count.js.map +1 -0
- package/cli/dist/utils/documentation-validator.d.ts +18 -0
- package/cli/dist/utils/documentation-validator.d.ts.map +1 -0
- package/cli/dist/utils/documentation-validator.js +233 -0
- package/cli/dist/utils/documentation-validator.js.map +1 -0
- package/cli/dist/utils/gui-helpers.d.ts +23 -0
- package/cli/dist/utils/gui-helpers.d.ts.map +1 -0
- package/cli/dist/utils/gui-helpers.js +159 -0
- package/cli/dist/utils/gui-helpers.js.map +1 -0
- package/cli/dist/utils/install-rules.d.ts +32 -0
- package/cli/dist/utils/install-rules.d.ts.map +1 -0
- package/cli/dist/utils/install-rules.js +375 -0
- package/cli/dist/utils/install-rules.js.map +1 -0
- package/cli/dist/utils/mcp-integration.d.ts +70 -0
- package/cli/dist/utils/mcp-integration.d.ts.map +1 -0
- package/cli/dist/utils/mcp-integration.js +292 -0
- package/cli/dist/utils/mcp-integration.js.map +1 -0
- package/cli/dist/utils/module-system.d.ts +232 -0
- package/cli/dist/utils/module-system.d.ts.map +1 -0
- package/cli/dist/utils/module-system.js +900 -0
- package/cli/dist/utils/module-system.js.map +1 -0
- package/cli/dist/utils/modules-catalog.d.ts +33 -0
- package/cli/dist/utils/modules-catalog.d.ts.map +1 -0
- package/cli/dist/utils/modules-catalog.js +163 -0
- package/cli/dist/utils/modules-catalog.js.map +1 -0
- package/cli/dist/utils/rule-install-hooks.d.ts +19 -0
- package/cli/dist/utils/rule-install-hooks.d.ts.map +1 -0
- package/cli/dist/utils/rule-install-hooks.js +224 -0
- package/cli/dist/utils/rule-install-hooks.js.map +1 -0
- package/cli/dist/utils/skill-system.d.ts +95 -0
- package/cli/dist/utils/skill-system.d.ts.map +1 -0
- package/cli/dist/utils/skill-system.js +313 -0
- package/cli/dist/utils/skill-system.js.map +1 -0
- package/modules.md +559 -105
- package/package.json +17 -6
|
@@ -0,0 +1,965 @@
|
|
|
1
|
+
# Vector Database Example: Semantic Search Application
|
|
2
|
+
|
|
3
|
+
## Overview
|
|
4
|
+
|
|
5
|
+
This example demonstrates a complete semantic search application using vector databases. It covers:
|
|
6
|
+
- Document ingestion and preprocessing
|
|
7
|
+
- Embedding generation with OpenAI
|
|
8
|
+
- Vector storage in Pinecone and Weaviate
|
|
9
|
+
- Similarity search
|
|
10
|
+
- Hybrid search (vector + keyword)
|
|
11
|
+
- Metadata filtering
|
|
12
|
+
- Sample queries with explanations
|
|
13
|
+
|
|
14
|
+
**Use Case**: Knowledge base search for technical documentation
|
|
15
|
+
|
|
16
|
+
**Tech Stack:**
|
|
17
|
+
- **Vector Database**: Pinecone (managed) or Weaviate (self-hosted)
|
|
18
|
+
- **Embedding Model**: OpenAI `text-embedding-3-small`
|
|
19
|
+
- **Language**: Python
|
|
20
|
+
- **Framework**: LangChain (optional, for RAG)
|
|
21
|
+
|
|
22
|
+
---
|
|
23
|
+
|
|
24
|
+
## Architecture
|
|
25
|
+
|
|
26
|
+
```
|
|
27
|
+
Documents (PDF, Markdown, HTML)
|
|
28
|
+
↓
|
|
29
|
+
Document Loader & Chunker
|
|
30
|
+
↓
|
|
31
|
+
Embedding Generator (OpenAI)
|
|
32
|
+
↓
|
|
33
|
+
Vector Database (Pinecone/Weaviate)
|
|
34
|
+
↓
|
|
35
|
+
Search API (Similarity + Hybrid)
|
|
36
|
+
↓
|
|
37
|
+
Results (Ranked by relevance)
|
|
38
|
+
```
|
|
39
|
+
|
|
40
|
+
---
|
|
41
|
+
|
|
42
|
+
## Setup
|
|
43
|
+
|
|
44
|
+
### Install Dependencies
|
|
45
|
+
|
|
46
|
+
```bash
|
|
47
|
+
pip install openai pinecone-client weaviate-client tiktoken langchain
|
|
48
|
+
```
|
|
49
|
+
|
|
50
|
+
### Environment Variables
|
|
51
|
+
|
|
52
|
+
```bash
|
|
53
|
+
# .env file
|
|
54
|
+
OPENAI_API_KEY=your-openai-api-key
|
|
55
|
+
PINECONE_API_KEY=your-pinecone-api-key
|
|
56
|
+
PINECONE_ENVIRONMENT=us-west1-gcp
|
|
57
|
+
WEAVIATE_URL=http://localhost:8080
|
|
58
|
+
```
|
|
59
|
+
|
|
60
|
+
---
|
|
61
|
+
|
|
62
|
+
## Step 1: Document Ingestion
|
|
63
|
+
|
|
64
|
+
### Load Documents
|
|
65
|
+
|
|
66
|
+
```python
|
|
67
|
+
import os
|
|
68
|
+
from pathlib import Path
|
|
69
|
+
from typing import List, Dict
|
|
70
|
+
|
|
71
|
+
class DocumentLoader:
|
|
72
|
+
"""Load documents from various sources"""
|
|
73
|
+
|
|
74
|
+
def load_markdown_files(self, directory: str) -> List[Dict]:
|
|
75
|
+
"""Load all markdown files from directory"""
|
|
76
|
+
documents = []
|
|
77
|
+
|
|
78
|
+
for file_path in Path(directory).rglob("*.md"):
|
|
79
|
+
with open(file_path, 'r', encoding='utf-8') as f:
|
|
80
|
+
content = f.read()
|
|
81
|
+
|
|
82
|
+
documents.append({
|
|
83
|
+
"id": str(file_path),
|
|
84
|
+
"text": content,
|
|
85
|
+
"metadata": {
|
|
86
|
+
"source": str(file_path),
|
|
87
|
+
"filename": file_path.name,
|
|
88
|
+
"type": "markdown"
|
|
89
|
+
}
|
|
90
|
+
})
|
|
91
|
+
|
|
92
|
+
return documents
|
|
93
|
+
|
|
94
|
+
def load_pdf_files(self, directory: str) -> List[Dict]:
|
|
95
|
+
"""Load PDF files (requires PyPDF2)"""
|
|
96
|
+
import PyPDF2
|
|
97
|
+
documents = []
|
|
98
|
+
|
|
99
|
+
for file_path in Path(directory).rglob("*.pdf"):
|
|
100
|
+
with open(file_path, 'rb') as f:
|
|
101
|
+
pdf_reader = PyPDF2.PdfReader(f)
|
|
102
|
+
text = ""
|
|
103
|
+
|
|
104
|
+
for page in pdf_reader.pages:
|
|
105
|
+
text += page.extract_text()
|
|
106
|
+
|
|
107
|
+
documents.append({
|
|
108
|
+
"id": str(file_path),
|
|
109
|
+
"text": text,
|
|
110
|
+
"metadata": {
|
|
111
|
+
"source": str(file_path),
|
|
112
|
+
"filename": file_path.name,
|
|
113
|
+
"type": "pdf",
|
|
114
|
+
"pages": len(pdf_reader.pages)
|
|
115
|
+
}
|
|
116
|
+
})
|
|
117
|
+
|
|
118
|
+
return documents
|
|
119
|
+
|
|
120
|
+
# Usage
|
|
121
|
+
loader = DocumentLoader()
|
|
122
|
+
documents = loader.load_markdown_files("./docs")
|
|
123
|
+
print(f"Loaded {len(documents)} documents")
|
|
124
|
+
```
|
|
125
|
+
|
|
126
|
+
### Chunk Documents
|
|
127
|
+
|
|
128
|
+
```python
|
|
129
|
+
import tiktoken
|
|
130
|
+
|
|
131
|
+
class DocumentChunker:
|
|
132
|
+
"""Chunk documents into smaller pieces"""
|
|
133
|
+
|
|
134
|
+
def __init__(self, chunk_size: int = 512, overlap: int = 50):
|
|
135
|
+
self.chunk_size = chunk_size
|
|
136
|
+
self.overlap = overlap
|
|
137
|
+
self.tokenizer = tiktoken.get_encoding("cl100k_base")
|
|
138
|
+
|
|
139
|
+
def chunk_by_tokens(self, text: str) -> List[str]:
|
|
140
|
+
"""Split text into chunks by token count"""
|
|
141
|
+
tokens = self.tokenizer.encode(text)
|
|
142
|
+
chunks = []
|
|
143
|
+
|
|
144
|
+
for i in range(0, len(tokens), self.chunk_size - self.overlap):
|
|
145
|
+
chunk_tokens = tokens[i:i + self.chunk_size]
|
|
146
|
+
chunk_text = self.tokenizer.decode(chunk_tokens)
|
|
147
|
+
chunks.append(chunk_text)
|
|
148
|
+
|
|
149
|
+
return chunks
|
|
150
|
+
|
|
151
|
+
def chunk_documents(self, documents: List[Dict]) -> List[Dict]:
|
|
152
|
+
"""Chunk all documents"""
|
|
153
|
+
chunked_docs = []
|
|
154
|
+
|
|
155
|
+
for doc in documents:
|
|
156
|
+
chunks = self.chunk_by_tokens(doc["text"])
|
|
157
|
+
|
|
158
|
+
for i, chunk in enumerate(chunks):
|
|
159
|
+
chunked_docs.append({
|
|
160
|
+
"id": f"{doc['id']}_chunk_{i}",
|
|
161
|
+
"text": chunk,
|
|
162
|
+
"metadata": {
|
|
163
|
+
**doc["metadata"],
|
|
164
|
+
"chunk_index": i,
|
|
165
|
+
"total_chunks": len(chunks),
|
|
166
|
+
"parent_id": doc["id"]
|
|
167
|
+
}
|
|
168
|
+
})
|
|
169
|
+
|
|
170
|
+
return chunked_docs
|
|
171
|
+
|
|
172
|
+
# Usage
|
|
173
|
+
chunker = DocumentChunker(chunk_size=512, overlap=50)
|
|
174
|
+
chunked_documents = chunker.chunk_documents(documents)
|
|
175
|
+
print(f"Created {len(chunked_documents)} chunks from {len(documents)} documents")
|
|
176
|
+
```
|
|
177
|
+
|
|
178
|
+
---
|
|
179
|
+
|
|
180
|
+
## Step 2: Embedding Generation
|
|
181
|
+
|
|
182
|
+
### Generate Embeddings with OpenAI
|
|
183
|
+
|
|
184
|
+
```python
|
|
185
|
+
from openai import OpenAI
|
|
186
|
+
from typing import List
|
|
187
|
+
import time
|
|
188
|
+
|
|
189
|
+
class EmbeddingGenerator:
|
|
190
|
+
"""Generate embeddings using OpenAI"""
|
|
191
|
+
|
|
192
|
+
def __init__(self, api_key: str, model: str = "text-embedding-3-small"):
|
|
193
|
+
self.client = OpenAI(api_key=api_key)
|
|
194
|
+
self.model = model
|
|
195
|
+
|
|
196
|
+
def generate_embedding(self, text: str) -> List[float]:
|
|
197
|
+
"""Generate embedding for a single text"""
|
|
198
|
+
response = self.client.embeddings.create(
|
|
199
|
+
model=self.model,
|
|
200
|
+
input=text
|
|
201
|
+
)
|
|
202
|
+
return response.data[0].embedding
|
|
203
|
+
|
|
204
|
+
def generate_embeddings_batch(self, texts: List[str], batch_size: int = 100) -> List[List[float]]:
|
|
205
|
+
"""Generate embeddings in batches"""
|
|
206
|
+
embeddings = []
|
|
207
|
+
|
|
208
|
+
for i in range(0, len(texts), batch_size):
|
|
209
|
+
batch = texts[i:i + batch_size]
|
|
210
|
+
|
|
211
|
+
# Rate limiting: wait if needed
|
|
212
|
+
time.sleep(0.1)
|
|
213
|
+
|
|
214
|
+
response = self.client.embeddings.create(
|
|
215
|
+
model=self.model,
|
|
216
|
+
input=batch
|
|
217
|
+
)
|
|
218
|
+
|
|
219
|
+
batch_embeddings = [item.embedding for item in response.data]
|
|
220
|
+
embeddings.extend(batch_embeddings)
|
|
221
|
+
|
|
222
|
+
print(f"Generated embeddings for {len(embeddings)}/{len(texts)} texts")
|
|
223
|
+
|
|
224
|
+
return embeddings
|
|
225
|
+
|
|
226
|
+
def embed_documents(self, documents: List[Dict]) -> List[Dict]:
|
|
227
|
+
"""Add embeddings to documents"""
|
|
228
|
+
texts = [doc["text"] for doc in documents]
|
|
229
|
+
embeddings = self.generate_embeddings_batch(texts)
|
|
230
|
+
|
|
231
|
+
for doc, embedding in zip(documents, embeddings):
|
|
232
|
+
doc["embedding"] = embedding
|
|
233
|
+
|
|
234
|
+
return documents
|
|
235
|
+
|
|
236
|
+
# Usage
|
|
237
|
+
import os
|
|
238
|
+
from dotenv import load_dotenv
|
|
239
|
+
|
|
240
|
+
load_dotenv()
|
|
241
|
+
|
|
242
|
+
generator = EmbeddingGenerator(api_key=os.getenv("OPENAI_API_KEY"))
|
|
243
|
+
embedded_documents = generator.embed_documents(chunked_documents)
|
|
244
|
+
print(f"Generated embeddings for {len(embedded_documents)} chunks")
|
|
245
|
+
```
|
|
246
|
+
|
|
247
|
+
---
|
|
248
|
+
|
|
249
|
+
## Step 3: Vector Storage
|
|
250
|
+
|
|
251
|
+
### Option 1: Pinecone (Managed)
|
|
252
|
+
|
|
253
|
+
```python
|
|
254
|
+
import pinecone
|
|
255
|
+
from typing import List, Dict
|
|
256
|
+
|
|
257
|
+
class PineconeVectorStore:
|
|
258
|
+
"""Store and search vectors in Pinecone"""
|
|
259
|
+
|
|
260
|
+
def __init__(self, api_key: str, environment: str, index_name: str):
|
|
261
|
+
pinecone.init(api_key=api_key, environment=environment)
|
|
262
|
+
self.index_name = index_name
|
|
263
|
+
self.index = None
|
|
264
|
+
|
|
265
|
+
def create_index(self, dimension: int = 1536):
|
|
266
|
+
"""Create Pinecone index"""
|
|
267
|
+
if self.index_name not in pinecone.list_indexes():
|
|
268
|
+
pinecone.create_index(
|
|
269
|
+
name=self.index_name,
|
|
270
|
+
dimension=dimension,
|
|
271
|
+
metric="cosine",
|
|
272
|
+
metadata_config={"indexed": ["source", "type", "filename"]}
|
|
273
|
+
)
|
|
274
|
+
|
|
275
|
+
self.index = pinecone.Index(self.index_name)
|
|
276
|
+
print(f"Created/connected to index: {self.index_name}")
|
|
277
|
+
|
|
278
|
+
def upsert_documents(self, documents: List[Dict], batch_size: int = 100):
|
|
279
|
+
"""Upsert documents to Pinecone"""
|
|
280
|
+
for i in range(0, len(documents), batch_size):
|
|
281
|
+
batch = documents[i:i + batch_size]
|
|
282
|
+
|
|
283
|
+
# Prepare vectors for upsert
|
|
284
|
+
vectors = [
|
|
285
|
+
(
|
|
286
|
+
doc["id"],
|
|
287
|
+
doc["embedding"],
|
|
288
|
+
{
|
|
289
|
+
"text": doc["text"],
|
|
290
|
+
**doc["metadata"]
|
|
291
|
+
}
|
|
292
|
+
)
|
|
293
|
+
for doc in batch
|
|
294
|
+
]
|
|
295
|
+
|
|
296
|
+
self.index.upsert(vectors=vectors)
|
|
297
|
+
print(f"Upserted {min(i + batch_size, len(documents))}/{len(documents)} documents")
|
|
298
|
+
|
|
299
|
+
def search(self, query_embedding: List[float], top_k: int = 10, filter: Dict = None):
|
|
300
|
+
"""Search for similar vectors"""
|
|
301
|
+
results = self.index.query(
|
|
302
|
+
vector=query_embedding,
|
|
303
|
+
top_k=top_k,
|
|
304
|
+
filter=filter,
|
|
305
|
+
include_metadata=True
|
|
306
|
+
)
|
|
307
|
+
|
|
308
|
+
return [
|
|
309
|
+
{
|
|
310
|
+
"id": match.id,
|
|
311
|
+
"score": match.score,
|
|
312
|
+
"text": match.metadata.get("text", ""),
|
|
313
|
+
"metadata": match.metadata
|
|
314
|
+
}
|
|
315
|
+
for match in results.matches
|
|
316
|
+
]
|
|
317
|
+
|
|
318
|
+
# Usage
|
|
319
|
+
pinecone_store = PineconeVectorStore(
|
|
320
|
+
api_key=os.getenv("PINECONE_API_KEY"),
|
|
321
|
+
environment=os.getenv("PINECONE_ENVIRONMENT"),
|
|
322
|
+
index_name="knowledge-base"
|
|
323
|
+
)
|
|
324
|
+
|
|
325
|
+
pinecone_store.create_index(dimension=1536)
|
|
326
|
+
pinecone_store.upsert_documents(embedded_documents)
|
|
327
|
+
```
|
|
328
|
+
|
|
329
|
+
### Option 2: Weaviate (Self-Hosted)
|
|
330
|
+
|
|
331
|
+
```python
|
|
332
|
+
import weaviate
|
|
333
|
+
from typing import List, Dict
|
|
334
|
+
|
|
335
|
+
class WeaviateVectorStore:
|
|
336
|
+
"""Store and search vectors in Weaviate"""
|
|
337
|
+
|
|
338
|
+
def __init__(self, url: str):
|
|
339
|
+
self.client = weaviate.Client(url)
|
|
340
|
+
self.class_name = "Document"
|
|
341
|
+
|
|
342
|
+
def create_schema(self):
|
|
343
|
+
"""Create Weaviate schema"""
|
|
344
|
+
schema = {
|
|
345
|
+
"class": self.class_name,
|
|
346
|
+
"description": "Technical documentation chunks",
|
|
347
|
+
"vectorizer": "none", # We provide our own vectors
|
|
348
|
+
"properties": [
|
|
349
|
+
{
|
|
350
|
+
"name": "text",
|
|
351
|
+
"dataType": ["text"],
|
|
352
|
+
"description": "Document text content"
|
|
353
|
+
},
|
|
354
|
+
{
|
|
355
|
+
"name": "source",
|
|
356
|
+
"dataType": ["string"],
|
|
357
|
+
"description": "Source file path"
|
|
358
|
+
},
|
|
359
|
+
{
|
|
360
|
+
"name": "filename",
|
|
361
|
+
"dataType": ["string"],
|
|
362
|
+
"description": "File name"
|
|
363
|
+
},
|
|
364
|
+
{
|
|
365
|
+
"name": "type",
|
|
366
|
+
"dataType": ["string"],
|
|
367
|
+
"description": "Document type (markdown, pdf, etc.)"
|
|
368
|
+
},
|
|
369
|
+
{
|
|
370
|
+
"name": "chunk_index",
|
|
371
|
+
"dataType": ["int"],
|
|
372
|
+
"description": "Chunk index"
|
|
373
|
+
},
|
|
374
|
+
{
|
|
375
|
+
"name": "parent_id",
|
|
376
|
+
"dataType": ["string"],
|
|
377
|
+
"description": "Parent document ID"
|
|
378
|
+
}
|
|
379
|
+
]
|
|
380
|
+
}
|
|
381
|
+
|
|
382
|
+
# Delete class if exists
|
|
383
|
+
if self.client.schema.exists(self.class_name):
|
|
384
|
+
self.client.schema.delete_class(self.class_name)
|
|
385
|
+
|
|
386
|
+
self.client.schema.create_class(schema)
|
|
387
|
+
print(f"Created schema for class: {self.class_name}")
|
|
388
|
+
|
|
389
|
+
def upsert_documents(self, documents: List[Dict], batch_size: int = 100):
|
|
390
|
+
"""Upsert documents to Weaviate"""
|
|
391
|
+
with self.client.batch as batch:
|
|
392
|
+
batch.batch_size = batch_size
|
|
393
|
+
|
|
394
|
+
for i, doc in enumerate(documents):
|
|
395
|
+
properties = {
|
|
396
|
+
"text": doc["text"],
|
|
397
|
+
"source": doc["metadata"].get("source", ""),
|
|
398
|
+
"filename": doc["metadata"].get("filename", ""),
|
|
399
|
+
"type": doc["metadata"].get("type", ""),
|
|
400
|
+
"chunk_index": doc["metadata"].get("chunk_index", 0),
|
|
401
|
+
"parent_id": doc["metadata"].get("parent_id", "")
|
|
402
|
+
}
|
|
403
|
+
|
|
404
|
+
batch.add_data_object(
|
|
405
|
+
data_object=properties,
|
|
406
|
+
class_name=self.class_name,
|
|
407
|
+
vector=doc["embedding"],
|
|
408
|
+
uuid=doc["id"]
|
|
409
|
+
)
|
|
410
|
+
|
|
411
|
+
if (i + 1) % 100 == 0:
|
|
412
|
+
print(f"Upserted {i + 1}/{len(documents)} documents")
|
|
413
|
+
|
|
414
|
+
def search(self, query_embedding: List[float], top_k: int = 10, where_filter: Dict = None):
|
|
415
|
+
"""Search for similar vectors"""
|
|
416
|
+
query = self.client.query.get(
|
|
417
|
+
self.class_name,
|
|
418
|
+
["text", "source", "filename", "type", "chunk_index"]
|
|
419
|
+
).with_near_vector({
|
|
420
|
+
"vector": query_embedding
|
|
421
|
+
}).with_limit(top_k)
|
|
422
|
+
|
|
423
|
+
if where_filter:
|
|
424
|
+
query = query.with_where(where_filter)
|
|
425
|
+
|
|
426
|
+
results = query.do()
|
|
427
|
+
|
|
428
|
+
return [
|
|
429
|
+
{
|
|
430
|
+
"text": item["text"],
|
|
431
|
+
"metadata": {
|
|
432
|
+
"source": item.get("source", ""),
|
|
433
|
+
"filename": item.get("filename", ""),
|
|
434
|
+
"type": item.get("type", ""),
|
|
435
|
+
"chunk_index": item.get("chunk_index", 0)
|
|
436
|
+
}
|
|
437
|
+
}
|
|
438
|
+
for item in results["data"]["Get"][self.class_name]
|
|
439
|
+
]
|
|
440
|
+
|
|
441
|
+
# Usage
|
|
442
|
+
weaviate_store = WeaviateVectorStore(url=os.getenv("WEAVIATE_URL"))
|
|
443
|
+
weaviate_store.create_schema()
|
|
444
|
+
weaviate_store.upsert_documents(embedded_documents)
|
|
445
|
+
```
|
|
446
|
+
|
|
447
|
+
---
|
|
448
|
+
|
|
449
|
+
## Step 4: Similarity Search
|
|
450
|
+
|
|
451
|
+
### Basic Similarity Search
|
|
452
|
+
|
|
453
|
+
```python
|
|
454
|
+
class SemanticSearchEngine:
|
|
455
|
+
"""Semantic search engine using vector database"""
|
|
456
|
+
|
|
457
|
+
def __init__(self, vector_store, embedding_generator):
|
|
458
|
+
self.vector_store = vector_store
|
|
459
|
+
self.embedding_generator = embedding_generator
|
|
460
|
+
|
|
461
|
+
def search(self, query: str, top_k: int = 10) -> List[Dict]:
|
|
462
|
+
"""Search for documents similar to query"""
|
|
463
|
+
# Generate query embedding
|
|
464
|
+
query_embedding = self.embedding_generator.generate_embedding(query)
|
|
465
|
+
|
|
466
|
+
# Search vector database
|
|
467
|
+
results = self.vector_store.search(query_embedding, top_k=top_k)
|
|
468
|
+
|
|
469
|
+
return results
|
|
470
|
+
|
|
471
|
+
def format_results(self, results: List[Dict]) -> str:
|
|
472
|
+
"""Format search results for display"""
|
|
473
|
+
output = []
|
|
474
|
+
|
|
475
|
+
for i, result in enumerate(results, 1):
|
|
476
|
+
output.append(f"\n--- Result {i} (Score: {result.get('score', 'N/A'):.4f}) ---")
|
|
477
|
+
output.append(f"Source: {result['metadata'].get('source', 'Unknown')}")
|
|
478
|
+
output.append(f"Text: {result['text'][:200]}...")
|
|
479
|
+
|
|
480
|
+
return "\n".join(output)
|
|
481
|
+
|
|
482
|
+
# Usage
|
|
483
|
+
search_engine = SemanticSearchEngine(pinecone_store, generator)
|
|
484
|
+
|
|
485
|
+
# Example query
|
|
486
|
+
query = "How do I configure database indexing?"
|
|
487
|
+
results = search_engine.search(query, top_k=5)
|
|
488
|
+
print(search_engine.format_results(results))
|
|
489
|
+
```
|
|
490
|
+
|
|
491
|
+
**Example Output:**
|
|
492
|
+
```
|
|
493
|
+
--- Result 1 (Score: 0.8923) ---
|
|
494
|
+
Source: docs/database/indexing.md
|
|
495
|
+
Text: Database indexing is crucial for query performance. To configure indexes,
|
|
496
|
+
you should first identify frequently queried columns. Create indexes using the
|
|
497
|
+
CREATE INDEX statement...
|
|
498
|
+
|
|
499
|
+
--- Result 2 (Score: 0.8654) ---
|
|
500
|
+
Source: docs/database/performance.md
|
|
501
|
+
Text: Index configuration affects query speed significantly. Best practices include
|
|
502
|
+
creating composite indexes for multi-column queries and monitoring index usage...
|
|
503
|
+
```
|
|
504
|
+
|
|
505
|
+
---
|
|
506
|
+
|
|
507
|
+
## Step 5: Hybrid Search (Vector + Keyword)
|
|
508
|
+
|
|
509
|
+
### Hybrid Search with Metadata Filtering
|
|
510
|
+
|
|
511
|
+
```python
|
|
512
|
+
class HybridSearchEngine:
|
|
513
|
+
"""Hybrid search combining vector similarity and metadata filtering"""
|
|
514
|
+
|
|
515
|
+
def __init__(self, vector_store, embedding_generator):
|
|
516
|
+
self.vector_store = vector_store
|
|
517
|
+
self.embedding_generator = embedding_generator
|
|
518
|
+
|
|
519
|
+
def search_with_filter(
|
|
520
|
+
self,
|
|
521
|
+
query: str,
|
|
522
|
+
top_k: int = 10,
|
|
523
|
+
document_type: str = None,
|
|
524
|
+
source_pattern: str = None
|
|
525
|
+
) -> List[Dict]:
|
|
526
|
+
"""Search with metadata filtering"""
|
|
527
|
+
# Generate query embedding
|
|
528
|
+
query_embedding = self.embedding_generator.generate_embedding(query)
|
|
529
|
+
|
|
530
|
+
# Build filter
|
|
531
|
+
filter_dict = {}
|
|
532
|
+
if document_type:
|
|
533
|
+
filter_dict["type"] = {"$eq": document_type}
|
|
534
|
+
if source_pattern:
|
|
535
|
+
filter_dict["source"] = {"$regex": source_pattern}
|
|
536
|
+
|
|
537
|
+
# Search with filter
|
|
538
|
+
results = self.vector_store.search(
|
|
539
|
+
query_embedding,
|
|
540
|
+
top_k=top_k,
|
|
541
|
+
filter=filter_dict if filter_dict else None
|
|
542
|
+
)
|
|
543
|
+
|
|
544
|
+
return results
|
|
545
|
+
|
|
546
|
+
def search_by_category(self, query: str, category: str, top_k: int = 10) -> List[Dict]:
|
|
547
|
+
"""Search within a specific category"""
|
|
548
|
+
return self.search_with_filter(
|
|
549
|
+
query=query,
|
|
550
|
+
top_k=top_k,
|
|
551
|
+
source_pattern=f".*{category}.*"
|
|
552
|
+
)
|
|
553
|
+
|
|
554
|
+
# Usage
|
|
555
|
+
hybrid_engine = HybridSearchEngine(pinecone_store, generator)
|
|
556
|
+
|
|
557
|
+
# Search only in database documentation
|
|
558
|
+
results = hybrid_engine.search_by_category(
|
|
559
|
+
query="How to optimize queries?",
|
|
560
|
+
category="database",
|
|
561
|
+
top_k=5
|
|
562
|
+
)
|
|
563
|
+
|
|
564
|
+
# Search only markdown files
|
|
565
|
+
results = hybrid_engine.search_with_filter(
|
|
566
|
+
query="API authentication",
|
|
567
|
+
document_type="markdown",
|
|
568
|
+
top_k=5
|
|
569
|
+
)
|
|
570
|
+
```
|
|
571
|
+
|
|
572
|
+
### Hybrid Search with Keyword Boosting
|
|
573
|
+
|
|
574
|
+
```python
|
|
575
|
+
from typing import List, Dict, Set
|
|
576
|
+
|
|
577
|
+
class KeywordBoostingSearch:
|
|
578
|
+
"""Hybrid search with keyword boosting"""
|
|
579
|
+
|
|
580
|
+
def __init__(self, vector_store, embedding_generator):
|
|
581
|
+
self.vector_store = vector_store
|
|
582
|
+
self.embedding_generator = embedding_generator
|
|
583
|
+
|
|
584
|
+
def extract_keywords(self, query: str) -> Set[str]:
|
|
585
|
+
"""Extract important keywords from query"""
|
|
586
|
+
# Simple keyword extraction (can use NLP libraries for better results)
|
|
587
|
+
stopwords = {"the", "a", "an", "in", "on", "at", "to", "for", "of", "and", "or"}
|
|
588
|
+
words = query.lower().split()
|
|
589
|
+
keywords = {word for word in words if word not in stopwords and len(word) > 3}
|
|
590
|
+
return keywords
|
|
591
|
+
|
|
592
|
+
def keyword_match_score(self, text: str, keywords: Set[str]) -> float:
|
|
593
|
+
"""Calculate keyword match score"""
|
|
594
|
+
text_lower = text.lower()
|
|
595
|
+
matches = sum(1 for keyword in keywords if keyword in text_lower)
|
|
596
|
+
return matches / len(keywords) if keywords else 0
|
|
597
|
+
|
|
598
|
+
def hybrid_search(
|
|
599
|
+
self,
|
|
600
|
+
query: str,
|
|
601
|
+
top_k: int = 10,
|
|
602
|
+
vector_weight: float = 0.7,
|
|
603
|
+
keyword_weight: float = 0.3
|
|
604
|
+
) -> List[Dict]:
|
|
605
|
+
"""Hybrid search with weighted scoring"""
|
|
606
|
+
# Extract keywords
|
|
607
|
+
keywords = self.extract_keywords(query)
|
|
608
|
+
|
|
609
|
+
# Vector search (get more results for re-ranking)
|
|
610
|
+
query_embedding = self.embedding_generator.generate_embedding(query)
|
|
611
|
+
vector_results = self.vector_store.search(query_embedding, top_k=top_k * 2)
|
|
612
|
+
|
|
613
|
+
# Re-rank with keyword boosting
|
|
614
|
+
for result in vector_results:
|
|
615
|
+
vector_score = result.get("score", 0)
|
|
616
|
+
keyword_score = self.keyword_match_score(result["text"], keywords)
|
|
617
|
+
|
|
618
|
+
# Combined score
|
|
619
|
+
result["combined_score"] = (
|
|
620
|
+
vector_weight * vector_score +
|
|
621
|
+
keyword_weight * keyword_score
|
|
622
|
+
)
|
|
623
|
+
|
|
624
|
+
# Sort by combined score
|
|
625
|
+
vector_results.sort(key=lambda x: x["combined_score"], reverse=True)
|
|
626
|
+
|
|
627
|
+
return vector_results[:top_k]
|
|
628
|
+
|
|
629
|
+
# Usage
|
|
630
|
+
keyword_search = KeywordBoostingSearch(pinecone_store, generator)
|
|
631
|
+
|
|
632
|
+
results = keyword_search.hybrid_search(
|
|
633
|
+
query="database indexing performance optimization",
|
|
634
|
+
top_k=5,
|
|
635
|
+
vector_weight=0.7,
|
|
636
|
+
keyword_weight=0.3
|
|
637
|
+
)
|
|
638
|
+
|
|
639
|
+
for result in results:
|
|
640
|
+
print(f"Combined Score: {result['combined_score']:.4f}")
|
|
641
|
+
print(f"Text: {result['text'][:100]}...\n")
|
|
642
|
+
```
|
|
643
|
+
|
|
644
|
+
---
|
|
645
|
+
|
|
646
|
+
## Step 6: Sample Queries with Explanations
|
|
647
|
+
|
|
648
|
+
### Query 1: Basic Semantic Search
|
|
649
|
+
|
|
650
|
+
```python
|
|
651
|
+
# Query: "How do I set up authentication?"
|
|
652
|
+
query = "How do I set up authentication?"
|
|
653
|
+
results = search_engine.search(query, top_k=3)
|
|
654
|
+
|
|
655
|
+
# Explanation:
|
|
656
|
+
# - Converts query to embedding vector
|
|
657
|
+
# - Finds documents with similar embeddings (semantic similarity)
|
|
658
|
+
# - Returns top 3 most similar documents
|
|
659
|
+
# - Will find documents about "authentication setup", "auth configuration", etc.
|
|
660
|
+
# even if they don't use exact words "set up authentication"
|
|
661
|
+
```
|
|
662
|
+
|
|
663
|
+
**Why it works:**
|
|
664
|
+
- Semantic understanding: "set up" ≈ "configure" ≈ "initialize"
|
|
665
|
+
- Finds conceptually similar content, not just keyword matches
|
|
666
|
+
|
|
667
|
+
### Query 2: Filtered Search
|
|
668
|
+
|
|
669
|
+
```python
|
|
670
|
+
# Query: "API rate limiting" (only in API documentation)
|
|
671
|
+
query = "API rate limiting"
|
|
672
|
+
results = hybrid_engine.search_with_filter(
|
|
673
|
+
query=query,
|
|
674
|
+
source_pattern=".*api.*",
|
|
675
|
+
top_k=5
|
|
676
|
+
)
|
|
677
|
+
|
|
678
|
+
# Explanation:
|
|
679
|
+
# - Semantic search for "API rate limiting"
|
|
680
|
+
# - Filters results to only include documents from API docs
|
|
681
|
+
# - Combines vector similarity with metadata filtering
|
|
682
|
+
# - More precise results than pure semantic search
|
|
683
|
+
```
|
|
684
|
+
|
|
685
|
+
**Why it works:**
|
|
686
|
+
- Narrows search scope to relevant documentation section
|
|
687
|
+
- Reduces noise from unrelated documents
|
|
688
|
+
- Faster search (fewer vectors to compare)
|
|
689
|
+
|
|
690
|
+
### Query 3: Hybrid Search with Keyword Boosting
|
|
691
|
+
|
|
692
|
+
```python
|
|
693
|
+
# Query: "PostgreSQL index optimization"
|
|
694
|
+
query = "PostgreSQL index optimization"
|
|
695
|
+
results = keyword_search.hybrid_search(
|
|
696
|
+
query=query,
|
|
697
|
+
top_k=5,
|
|
698
|
+
vector_weight=0.6,
|
|
699
|
+
keyword_weight=0.4
|
|
700
|
+
)
|
|
701
|
+
|
|
702
|
+
# Explanation:
|
|
703
|
+
# - Semantic search finds conceptually similar documents
|
|
704
|
+
# - Keyword matching boosts documents containing "PostgreSQL", "index", "optimization"
|
|
705
|
+
# - Weighted combination: 60% semantic similarity, 40% keyword match
|
|
706
|
+
# - Balances semantic understanding with exact term matching
|
|
707
|
+
```
|
|
708
|
+
|
|
709
|
+
**Why it works:**
|
|
710
|
+
- Semantic search finds related concepts (e.g., "database tuning")
|
|
711
|
+
- Keyword boosting prioritizes documents with specific terms (e.g., "PostgreSQL")
|
|
712
|
+
- Best of both worlds: semantic understanding + term precision
|
|
713
|
+
|
|
714
|
+
### Query 4: Multi-Filter Search
|
|
715
|
+
|
|
716
|
+
```python
|
|
717
|
+
# Query: "error handling" (only in Python docs, markdown files)
|
|
718
|
+
query = "error handling"
|
|
719
|
+
results = hybrid_engine.search_with_filter(
|
|
720
|
+
query=query,
|
|
721
|
+
document_type="markdown",
|
|
722
|
+
source_pattern=".*python.*",
|
|
723
|
+
top_k=5
|
|
724
|
+
)
|
|
725
|
+
|
|
726
|
+
# Explanation:
|
|
727
|
+
# - Semantic search for "error handling"
|
|
728
|
+
# - Filter 1: Only markdown files
|
|
729
|
+
# - Filter 2: Only Python documentation
|
|
730
|
+
# - Highly targeted results
|
|
731
|
+
```
|
|
732
|
+
|
|
733
|
+
**Why it works:**
|
|
734
|
+
- Multiple filters narrow search scope significantly
|
|
735
|
+
- Reduces false positives from other languages/formats
|
|
736
|
+
- Faster and more accurate results
|
|
737
|
+
|
|
738
|
+
### Query 5: RAG (Retrieval-Augmented Generation)
|
|
739
|
+
|
|
740
|
+
```python
|
|
741
|
+
def rag_query(question: str, search_engine, llm_client):
|
|
742
|
+
"""Answer question using RAG"""
|
|
743
|
+
# 1. Retrieve relevant context
|
|
744
|
+
results = search_engine.search(question, top_k=5)
|
|
745
|
+
context = "\n\n".join([r["text"] for r in results])
|
|
746
|
+
|
|
747
|
+
# 2. Augment prompt with context
|
|
748
|
+
prompt = f"""
|
|
749
|
+
Context from documentation:
|
|
750
|
+
{context}
|
|
751
|
+
|
|
752
|
+
Question: {question}
|
|
753
|
+
|
|
754
|
+
Answer the question based on the context above. If the context doesn't
|
|
755
|
+
contain enough information, say so.
|
|
756
|
+
"""
|
|
757
|
+
|
|
758
|
+
# 3. Generate answer with LLM
|
|
759
|
+
response = llm_client.chat.completions.create(
|
|
760
|
+
model="gpt-4",
|
|
761
|
+
messages=[{"role": "user", "content": prompt}]
|
|
762
|
+
)
|
|
763
|
+
|
|
764
|
+
answer = response.choices[0].message.content
|
|
765
|
+
|
|
766
|
+
return {
|
|
767
|
+
"answer": answer,
|
|
768
|
+
"sources": [r["metadata"]["source"] for r in results]
|
|
769
|
+
}
|
|
770
|
+
|
|
771
|
+
# Usage
|
|
772
|
+
question = "What are the best practices for database indexing?"
|
|
773
|
+
result = rag_query(question, search_engine, OpenAI(api_key=os.getenv("OPENAI_API_KEY")))
|
|
774
|
+
|
|
775
|
+
print(f"Answer: {result['answer']}")
|
|
776
|
+
print(f"\nSources: {', '.join(result['sources'])}")
|
|
777
|
+
|
|
778
|
+
# Explanation:
|
|
779
|
+
# - Retrieves top 5 most relevant documents
|
|
780
|
+
# - Provides context to LLM
|
|
781
|
+
# - LLM generates answer based on retrieved context
|
|
782
|
+
# - Returns answer with source citations
|
|
783
|
+
```
|
|
784
|
+
|
|
785
|
+
**Why it works:**
|
|
786
|
+
- Combines retrieval (vector search) with generation (LLM)
|
|
787
|
+
- Grounds LLM responses in actual documentation
|
|
788
|
+
- Provides source attribution for verification
|
|
789
|
+
- Reduces hallucinations (LLM making up information)
|
|
790
|
+
|
|
791
|
+
---
|
|
792
|
+
|
|
793
|
+
## Complete Example: End-to-End Workflow
|
|
794
|
+
|
|
795
|
+
```python
|
|
796
|
+
import os
|
|
797
|
+
from dotenv import load_dotenv
|
|
798
|
+
|
|
799
|
+
# Load environment variables
|
|
800
|
+
load_dotenv()
|
|
801
|
+
|
|
802
|
+
# 1. Load documents
|
|
803
|
+
print("Step 1: Loading documents...")
|
|
804
|
+
loader = DocumentLoader()
|
|
805
|
+
documents = loader.load_markdown_files("./docs")
|
|
806
|
+
print(f"Loaded {len(documents)} documents")
|
|
807
|
+
|
|
808
|
+
# 2. Chunk documents
|
|
809
|
+
print("\nStep 2: Chunking documents...")
|
|
810
|
+
chunker = DocumentChunker(chunk_size=512, overlap=50)
|
|
811
|
+
chunked_documents = chunker.chunk_documents(documents)
|
|
812
|
+
print(f"Created {len(chunked_documents)} chunks")
|
|
813
|
+
|
|
814
|
+
# 3. Generate embeddings
|
|
815
|
+
print("\nStep 3: Generating embeddings...")
|
|
816
|
+
generator = EmbeddingGenerator(api_key=os.getenv("OPENAI_API_KEY"))
|
|
817
|
+
embedded_documents = generator.embed_documents(chunked_documents)
|
|
818
|
+
print(f"Generated embeddings for {len(embedded_documents)} chunks")
|
|
819
|
+
|
|
820
|
+
# 4. Store in vector database
|
|
821
|
+
print("\nStep 4: Storing in Pinecone...")
|
|
822
|
+
pinecone_store = PineconeVectorStore(
|
|
823
|
+
api_key=os.getenv("PINECONE_API_KEY"),
|
|
824
|
+
environment=os.getenv("PINECONE_ENVIRONMENT"),
|
|
825
|
+
index_name="knowledge-base"
|
|
826
|
+
)
|
|
827
|
+
pinecone_store.create_index(dimension=1536)
|
|
828
|
+
pinecone_store.upsert_documents(embedded_documents)
|
|
829
|
+
print("Documents stored successfully")
|
|
830
|
+
|
|
831
|
+
# 5. Create search engines
|
|
832
|
+
print("\nStep 5: Creating search engines...")
|
|
833
|
+
search_engine = SemanticSearchEngine(pinecone_store, generator)
|
|
834
|
+
hybrid_engine = HybridSearchEngine(pinecone_store, generator)
|
|
835
|
+
keyword_search = KeywordBoostingSearch(pinecone_store, generator)
|
|
836
|
+
|
|
837
|
+
# 6. Run sample queries
|
|
838
|
+
print("\n" + "="*80)
|
|
839
|
+
print("SAMPLE QUERIES")
|
|
840
|
+
print("="*80)
|
|
841
|
+
|
|
842
|
+
# Query 1: Basic semantic search
|
|
843
|
+
print("\n[Query 1] Basic Semantic Search")
|
|
844
|
+
print("Query: 'How to configure database indexes?'")
|
|
845
|
+
results = search_engine.search("How to configure database indexes?", top_k=3)
|
|
846
|
+
print(search_engine.format_results(results))
|
|
847
|
+
|
|
848
|
+
# Query 2: Filtered search
|
|
849
|
+
print("\n[Query 2] Filtered Search")
|
|
850
|
+
print("Query: 'API authentication' (only markdown files)")
|
|
851
|
+
results = hybrid_engine.search_with_filter(
|
|
852
|
+
query="API authentication",
|
|
853
|
+
document_type="markdown",
|
|
854
|
+
top_k=3
|
|
855
|
+
)
|
|
856
|
+
print(search_engine.format_results(results))
|
|
857
|
+
|
|
858
|
+
# Query 3: Hybrid search with keyword boosting
|
|
859
|
+
print("\n[Query 3] Hybrid Search with Keyword Boosting")
|
|
860
|
+
print("Query: 'PostgreSQL performance optimization'")
|
|
861
|
+
results = keyword_search.hybrid_search(
|
|
862
|
+
query="PostgreSQL performance optimization",
|
|
863
|
+
top_k=3,
|
|
864
|
+
vector_weight=0.7,
|
|
865
|
+
keyword_weight=0.3
|
|
866
|
+
)
|
|
867
|
+
for i, result in enumerate(results, 1):
|
|
868
|
+
print(f"\n--- Result {i} (Combined Score: {result['combined_score']:.4f}) ---")
|
|
869
|
+
print(f"Source: {result['metadata'].get('source', 'Unknown')}")
|
|
870
|
+
print(f"Text: {result['text'][:200]}...")
|
|
871
|
+
|
|
872
|
+
print("\n" + "="*80)
|
|
873
|
+
print("Search system ready!")
|
|
874
|
+
print("="*80)
|
|
875
|
+
```
|
|
876
|
+
|
|
877
|
+
---
|
|
878
|
+
|
|
879
|
+
## Performance Metrics
|
|
880
|
+
|
|
881
|
+
### Measuring Search Quality
|
|
882
|
+
|
|
883
|
+
```python
|
|
884
|
+
def evaluate_search_quality(search_engine, test_queries, ground_truth):
|
|
885
|
+
"""Evaluate search quality using test queries"""
|
|
886
|
+
metrics = {
|
|
887
|
+
"precision_at_5": [],
|
|
888
|
+
"recall_at_5": [],
|
|
889
|
+
"mrr": [] # Mean Reciprocal Rank
|
|
890
|
+
}
|
|
891
|
+
|
|
892
|
+
for query, relevant_docs in zip(test_queries, ground_truth):
|
|
893
|
+
results = search_engine.search(query, top_k=5)
|
|
894
|
+
result_ids = [r["id"] for r in results]
|
|
895
|
+
|
|
896
|
+
# Precision@5
|
|
897
|
+
relevant_found = len(set(result_ids) & set(relevant_docs))
|
|
898
|
+
precision = relevant_found / 5
|
|
899
|
+
metrics["precision_at_5"].append(precision)
|
|
900
|
+
|
|
901
|
+
# Recall@5
|
|
902
|
+
recall = relevant_found / len(relevant_docs) if relevant_docs else 0
|
|
903
|
+
metrics["recall_at_5"].append(recall)
|
|
904
|
+
|
|
905
|
+
# MRR
|
|
906
|
+
for i, result_id in enumerate(result_ids, 1):
|
|
907
|
+
if result_id in relevant_docs:
|
|
908
|
+
metrics["mrr"].append(1 / i)
|
|
909
|
+
break
|
|
910
|
+
else:
|
|
911
|
+
metrics["mrr"].append(0)
|
|
912
|
+
|
|
913
|
+
return {
|
|
914
|
+
"precision_at_5": sum(metrics["precision_at_5"]) / len(metrics["precision_at_5"]),
|
|
915
|
+
"recall_at_5": sum(metrics["recall_at_5"]) / len(metrics["recall_at_5"]),
|
|
916
|
+
"mrr": sum(metrics["mrr"]) / len(metrics["mrr"])
|
|
917
|
+
}
|
|
918
|
+
|
|
919
|
+
# Example
|
|
920
|
+
test_queries = [
|
|
921
|
+
"How to configure database indexes?",
|
|
922
|
+
"API authentication best practices",
|
|
923
|
+
"PostgreSQL performance tuning"
|
|
924
|
+
]
|
|
925
|
+
|
|
926
|
+
ground_truth = [
|
|
927
|
+
["docs/database/indexing.md_chunk_0", "docs/database/indexing.md_chunk_1"],
|
|
928
|
+
["docs/api/auth.md_chunk_0", "docs/api/security.md_chunk_2"],
|
|
929
|
+
["docs/database/postgres.md_chunk_5", "docs/database/performance.md_chunk_3"]
|
|
930
|
+
]
|
|
931
|
+
|
|
932
|
+
metrics = evaluate_search_quality(search_engine, test_queries, ground_truth)
|
|
933
|
+
print(f"Precision@5: {metrics['precision_at_5']:.2%}")
|
|
934
|
+
print(f"Recall@5: {metrics['recall_at_5']:.2%}")
|
|
935
|
+
print(f"MRR: {metrics['mrr']:.4f}")
|
|
936
|
+
```
|
|
937
|
+
|
|
938
|
+
---
|
|
939
|
+
|
|
940
|
+
## Summary
|
|
941
|
+
|
|
942
|
+
**What We Built:**
|
|
943
|
+
1. ✅ Document ingestion pipeline (markdown, PDF)
|
|
944
|
+
2. ✅ Intelligent chunking with overlap
|
|
945
|
+
3. ✅ Embedding generation with OpenAI
|
|
946
|
+
4. ✅ Vector storage in Pinecone/Weaviate
|
|
947
|
+
5. ✅ Semantic similarity search
|
|
948
|
+
6. ✅ Hybrid search (vector + metadata filtering)
|
|
949
|
+
7. ✅ Keyword boosting for precision
|
|
950
|
+
8. ✅ RAG implementation for Q&A
|
|
951
|
+
9. ✅ Performance evaluation metrics
|
|
952
|
+
|
|
953
|
+
**Key Takeaways:**
|
|
954
|
+
- Vector databases enable semantic search (meaning-based, not keyword-based)
|
|
955
|
+
- Chunking strategy affects search quality (512 tokens with 50 token overlap works well)
|
|
956
|
+
- Hybrid search (vector + metadata) provides best results
|
|
957
|
+
- RAG combines retrieval with LLM generation for accurate answers
|
|
958
|
+
- Always measure search quality with precision, recall, and MRR metrics
|
|
959
|
+
|
|
960
|
+
**Next Steps:**
|
|
961
|
+
- See `../rules/vector-databases.md` for vector database fundamentals
|
|
962
|
+
- See `../rules/vector-embeddings.md` for embedding strategies
|
|
963
|
+
- See `../rules/vector-indexing.md` for index optimization
|
|
964
|
+
- Experiment with different chunking strategies and embedding models
|
|
965
|
+
- Tune hybrid search weights for your use case
|