Skip to content

Commit 50c60ff

Browse files
committed
graphrag_sdk terminology change from Schema to Ontology
1 parent 0d12839 commit 50c60ff

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/agentchat/contrib/graph_rag/test_falkor_graph_rag.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import pytest
44
from conftest import reason, skip_openai # noqa: E402
55
from graphrag_sdk import KnowledgeGraph, Source
6-
from graphrag_sdk.schema import Schema
6+
from graphrag_sdk.ontology import Ontology
77

88
try:
99
from autogen.agentchat.contrib.graph_rag.document import (
@@ -34,7 +34,7 @@ def test_falkor_db_query_engine():
3434
3. Query it with a question and verify the result contains the critical information.
3535
"""
3636
# Arrange
37-
test_schema = Schema()
37+
test_schema = Ontology()
3838
actor = test_schema.add_entity("Actor").add_attribute("name", str, unique=True)
3939
movie = test_schema.add_entity("Movie").add_attribute("title", str, unique=True)
4040
test_schema.add_relation("ACTED", actor, movie)

0 commit comments

Comments
 (0)