Update numpy to 2.1+ for Python 3.13 support

setup.py

@@ -30,7 +30,7 @@
     "termcolor",
     "flaml",
     # numpy is installed by flaml, but we want to pin the version to below 2.x (see https://github.com/microsoft/autogen/issues/1960)
-    "numpy>=1.17.0,<2",
+    "numpy>=2.1",  # Note, updated to 2.1+ because 2.1 is the lowest version usable with Python 3.13
     "python-dotenv",
     "tiktoken",
     # Disallowing 2.6.0 can be removed when this is fixed https://github.com/pydantic/pydantic/issues/8705

test/agentchat/contrib/graph_rag/test_falkor_graph_rag.py

@@ -3,7 +3,7 @@
 import pytest
 from conftest import reason, skip_openai  # noqa: E402
 from graphrag_sdk import KnowledgeGraph, Source
-from graphrag_sdk.schema import Schema
+from graphrag_sdk.ontology import Ontology
 
 try:
     from autogen.agentchat.contrib.graph_rag.document import (
@@ -34,7 +34,7 @@ def test_falkor_db_query_engine():
     3. Query it with a question and verify the result contains the critical information.
     """
     # Arrange
-    test_schema = Schema()
+    test_schema = Ontology()
     actor = test_schema.add_entity("Actor").add_attribute("name", str, unique=True)
     movie = test_schema.add_entity("Movie").add_attribute("title", str, unique=True)
     test_schema.add_relation("ACTED", actor, movie)