Skip to content

feat: add AG2 multi-agent RAG integration with Milvus#1585

Open
faridun-ag2 wants to merge 1 commit intomilvus-io:masterfrom
faridun-ag2:feat/ag2-multiagent-rag-integration
Open

feat: add AG2 multi-agent RAG integration with Milvus#1585
faridun-ag2 wants to merge 1 commit intomilvus-io:masterfrom
faridun-ag2:feat/ag2-multiagent-rag-integration

Conversation

@faridun-ag2
Copy link
Copy Markdown

Summary

Add an integration notebook for AG2 (formerly AutoGen), a multi-agent conversation framework.

  • integration/ag2/ag2_multiagent_rag_with_milvus.ipynb — Complete notebook

How it works

  1. Documents are embedded using OpenAI embeddings and stored in Milvus (using Milvus Lite)
  2. AG2 Research Agent uses a registered search_documents tool to query Milvus
  3. AG2 Analyst Agent synthesizes retrieved information into grounded answers
  4. Agents collaborate via AG2 GroupChat with automatic tool execution

Key features

  • Zero setup: Uses Milvus Lite (in-process, no Docker needed)
  • OpenAI embeddings: text-embedding-3-small for text vectorization
  • Multi-agent RAG: Researcher retrieves, Analyst synthesizes
  • Tool registration: Milvus search exposed as AG2 tool via decorator pattern
  • Colab ready: Can run directly in Google Colab

Test plan

  • Tested locally with real OpenAI API calls — agents collaborate, retrieve from Milvus, and produce grounded answers
  • Black formatted
  • DCO signed

Add a Jupyter notebook demonstrating AG2 (formerly AutoGen) multi-agent
conversations using Milvus as the vector store for RAG.

- Milvus Lite for zero-setup vector search
- AG2 GroupChat with Research + Analyst agents
- Milvus search registered as AG2 tool via decorator pattern
- OpenAI embeddings for text vectorization
- Cleanup and Colab support

Signed-off-by: Faridun Mirzoev <faridun@ag2.ai>
@review-notebook-app
Copy link
Copy Markdown

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@sre-ci-robot
Copy link
Copy Markdown
Collaborator

[APPROVALNOTIFIER] This PR is NOT APPROVED

This pull-request has been approved by: faridun-ag2

The full list of commands accepted by this bot can be found here.

Details Needs approval from an approver in each of these files:

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

@sre-ci-robot
Copy link
Copy Markdown
Collaborator

Welcome @faridun-ag2! It looks like this is your first PR to milvus-io/bootcamp 🎉

@faridun-ag2
Copy link
Copy Markdown
Author

Hi @jaelgu !

Could you please review this PR when you get a chance? It adds an integration notebook for AG2 (formerly AutoGen) — a multi-agent conversation framework.

The notebook demonstrates AG2 agents collaborating to answer questions using Milvus as the vector store for RAG. It follows the same patterns as the existing LangGraph and LlamaIndex integration notebooks.

Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants