We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0d12839 commit 50c60ffCopy full SHA for 50c60ff
test/agentchat/contrib/graph_rag/test_falkor_graph_rag.py
@@ -3,7 +3,7 @@
3
import pytest
4
from conftest import reason, skip_openai # noqa: E402
5
from graphrag_sdk import KnowledgeGraph, Source
6
-from graphrag_sdk.schema import Schema
+from graphrag_sdk.ontology import Ontology
7
8
try:
9
from autogen.agentchat.contrib.graph_rag.document import (
@@ -34,7 +34,7 @@ def test_falkor_db_query_engine():
34
3. Query it with a question and verify the result contains the critical information.
35
"""
36
# Arrange
37
- test_schema = Schema()
+ test_schema = Ontology()
38
actor = test_schema.add_entity("Actor").add_attribute("name", str, unique=True)
39
movie = test_schema.add_entity("Movie").add_attribute("title", str, unique=True)
40
test_schema.add_relation("ACTED", actor, movie)
0 commit comments