mirror of
https://github.com/microsoft/autogen.git
synced 2025-08-04 14:52:10 +00:00
e579a4609d
* [Graph RAG] Init Commit with GraphRag interfaces * Add Document class for graph rag input document * Add Graph RAG Capability * Add unit test for graph rag interfaces --------- Co-authored-by: Li Jiang <bnujli@gmail.com> Co-authored-by: gagb <gagb@users.noreply.github.com>
18 lines
639 B
Python
18 lines
639 B
Python
from unittest.mock import Mock
|
|
|
|
from autogen.agentchat.contrib.graph_rag.graph_query_engine import GraphQueryEngine
|
|
from autogen.agentchat.contrib.graph_rag.graph_rag_capability import GraphRagCapability
|
|
from autogen.agentchat.conversable_agent import ConversableAgent
|
|
|
|
|
|
def test_dry_run():
|
|
"""Dry run for basic graph rag objects."""
|
|
mock_graph_query_engine = Mock(spec=GraphQueryEngine)
|
|
|
|
graph_rag_agent = ConversableAgent(
|
|
name="graph_rag_agent",
|
|
max_consecutive_auto_reply=3,
|
|
)
|
|
graph_rag_capability = GraphRagCapability(mock_graph_query_engine)
|
|
graph_rag_capability.add_to_agent(graph_rag_agent)
|