Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions .semversioner/next-release/patch-20261012010114209801.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
{
"type": "patch",
"description": "Update dependencies."
}
2 changes: 1 addition & 1 deletion packages/graphrag-input/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ dependencies = [
"graphrag-common==3.3.0",
"graphrag-storage==3.3.0",
"pydantic~=2.13",
"markitdown[pdf]~=0.1.7",
"markitdown[pdf]~=0.1.8",
"pyarrow>=14.0.0"
]

Expand Down
4 changes: 2 additions & 2 deletions packages/graphrag-llm/graphrag_llm/completion/completion.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from graphrag_llm.threading.completion_thread_runner import completion_thread_runner

if TYPE_CHECKING:
from collections.abc import AsyncIterator, Iterator
from collections.abc import AsyncIterator, Generator, Iterator

from graphrag_cache import Cache, CacheKeyCreator

Expand Down Expand Up @@ -167,7 +167,7 @@ def completion_thread_pool(
response_handler: "ThreadedLLMCompletionResponseHandler",
concurrency: int,
queue_limit: int = 0,
) -> "Iterator[ThreadedLLMCompletionFunction]":
) -> "Generator[ThreadedLLMCompletionFunction, None, None]":
"""Run a completion thread pool.

Args
Expand Down
4 changes: 2 additions & 2 deletions packages/graphrag-llm/graphrag_llm/embedding/embedding.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from graphrag_llm.threading.embedding_thread_runner import embedding_thread_runner

if TYPE_CHECKING:
from collections.abc import Iterator
from collections.abc import Generator

from graphrag_cache import Cache, CacheKeyCreator

Expand Down Expand Up @@ -93,7 +93,7 @@ def embedding_thread_pool(
response_handler: "ThreadedLLMEmbeddingResponseHandler",
concurrency: int,
queue_limit: int = 0,
) -> "Iterator[ThreadedLLMEmbeddingFunction]":
) -> "Generator[ThreadedLLMEmbeddingFunction, None, None]":
"""Run an embedding thread pool.

Args
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import threading
import time
from collections import deque
from collections.abc import Iterator
from collections.abc import Generator
from contextlib import contextmanager
from typing import Any

Expand Down Expand Up @@ -62,7 +62,7 @@ def __init__(
self._stagger = self._period_in_seconds / self._rpp

@contextmanager
def acquire(self, token_count: int) -> Iterator[None]:
def acquire(self, token_count: int) -> Generator[None, None, None]:
"""
Acquire Rate Limiter.

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
import sys
import threading
import time
from collections.abc import Awaitable, Iterator
from collections.abc import Awaitable, Generator, Iterator
from contextlib import contextmanager
from queue import Empty, Queue
from typing import TYPE_CHECKING, Protocol, Unpack, runtime_checkable
Expand Down Expand Up @@ -147,7 +147,7 @@ def completion_thread_runner(
concurrency: int,
queue_limit: int = 0,
metrics_store: "MetricsStore | None" = None,
) -> Iterator[ThreadedLLMCompletionFunction]:
) -> Generator[ThreadedLLMCompletionFunction, None, None]:
"""Run a completion thread pool.

Args
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
import sys
import threading
import time
from collections.abc import Awaitable, Iterator
from collections.abc import Awaitable, Generator
from contextlib import contextmanager
from queue import Empty, Queue
from typing import TYPE_CHECKING, Protocol, Unpack, runtime_checkable
Expand Down Expand Up @@ -120,7 +120,7 @@ def embedding_thread_runner(
concurrency: int,
queue_limit: int = 0,
metrics_store: "MetricsStore | None" = None,
) -> Iterator[ThreadedLLMEmbeddingFunction]:
) -> Generator[ThreadedLLMEmbeddingFunction, None, None]:
"""Run an embedding thread pool.

Args
Expand Down
4 changes: 2 additions & 2 deletions packages/graphrag-llm/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -32,13 +32,13 @@ classifiers = [
"Programming Language :: Python :: 3.13",
]
dependencies = [
"azure-identity~=1.25",
"azure-identity~=1.26",
# fastapi is required for litellm tool calling. https://github.com/BerriAI/litellm/issues/32993
"fastapi>=0.136.3,<1.0",
"graphrag-cache==3.3.0",
"graphrag-common==3.3.0",
"jinja2~=3.1",
"litellm==1.100.1",
"litellm==1.102.1",
"nest-asyncio2~=1.7",
# orjson is required for litellm tool calling. https://github.com/BerriAI/litellm/issues/32993
"orjson>=3.11.6,<4.0",
Expand Down
6 changes: 3 additions & 3 deletions packages/graphrag-storage/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@ classifiers = [
]
dependencies = [
"aiofiles~=25.1",
"azure-cosmos~=4.16",
"azure-identity~=1.25",
"azure-storage-blob~=12.30",
"azure-cosmos~=4.17",
"azure-identity~=1.26",
"azure-storage-blob~=12.31",
"graphrag-common==3.3.0",
"pandas~=3.0",
"pydantic~=2.13",
Expand Down
4 changes: 2 additions & 2 deletions packages/graphrag-vectors/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -31,8 +31,8 @@ classifiers = [
]
dependencies = [
"azure-core~=1.41",
"azure-cosmos~=4.16",
"azure-identity~=1.25",
"azure-cosmos~=4.17",
"azure-identity~=1.26",
"azure-search-documents~=12.0",
"graphrag-common==3.3.0",
"lancedb~=0.38.0",
Expand Down
4 changes: 2 additions & 2 deletions packages/graphrag/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -32,9 +32,9 @@ classifiers = [
]

dependencies = [
"azure-identity~=1.25",
"azure-identity~=1.26",
"azure-search-documents~=12.0",
"azure-storage-blob~=12.30",
"azure-storage-blob~=12.31",
"devtools~=0.12",
"graphrag-cache==3.3.0",
"graphrag-chunking==3.3.0",
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ requires-python = ">=3.11,<3.14"
dev = [
"coverage~=7.16",
"deptry~=0.25",
"ipykernel~=7.3",
"ipykernel~=7.4",
"jupyter~=1.1",
"mkdocs-material~=9.7",
"mkdocs-jupyter~=0.26",
Expand Down
Loading
Loading