-
Notifications
You must be signed in to change notification settings - Fork 7.2k
Feature/chromadb embedding functions #6267 #6648
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Feature/chromadb embedding functions #6267 #6648
Conversation
…on guide- Fix version format from 0.4.0-dev-1 to 0.4.0-dev.1 for all packages- Remove reference to non-existent Microsoft.AutoGen.Extensions package- Add correct extension packages: Aspire, MEAI, and SemanticKernel- Fix typo: RuntimeGatewway -> RuntimeGateway- Improve documentation structure with clear section headersFixes microsoft#6244
Fix issue microsoft#6277 where TextMessage was used but not imported in three code cells of the custom agents documentation, causing NameError when users run the examples. Changes: - Add TextMessage to imports in ArithmeticAgent section - Add TextMessage to imports in GeminiAssistantAgent section - Add TextMessage to imports in Declarative GeminiAssistantAgent section The CountDownAgent section already had the correct import. Fixes microsoft#6277
…osoft#6267) - Add embedding function configuration classes (Default, SentenceTransformer, OpenAI, Custom) - Extend ChromaDBVectorMemoryConfig with embedding_function_config field - Update collection initialization to use custom embedding functions - Add comprehensive tests and demo examples - Maintain backward compatibility with existing code Resolves microsoft#6267
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## main #6648 +/- ##
==========================================
+ Coverage 79.65% 79.67% +0.01%
==========================================
Files 229 231 +2
Lines 17126 17172 +46
==========================================
+ Hits 13642 13682 +40
- Misses 3484 3490 +6
Flags with carried forward coverage won't be shown. Click here to find out more. ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
- Add explicit type annotation for embedding function config variable - Add return statement in test helper function for better type checking - Ensure all type checkers can properly infer types in edge cases
…m:tejas-dharani/autogen into feature/chromadb-embedding-functions-6267
Thanks again for the PR.
|
Thanks for the updates!
|
Thanks for the work on this @tejas-dharani , much appreciated. Thanks again! |
@victordibia Thanks for the feedback and for taking the time to review the PR! You're absolutely right about keeping commits focused - I'll make sure to separate unrelated changes into their own PRs going forward. |
Why are these changes needed?
This PR adds support for configurable embedding functions in ChromaDBVectorMemory, addressing the need for users to customize how embeddings are generated for vector similarity search. Currently, ChromaDB memory is limited to default embedding functions, which restricts flexibility for different use cases that may require specific embedding models or custom embedding logic.
The implementation allows users to:
Related issue number
Closes #6267
Checks