Skip to content

Commit 811379b

Browse files
committed
Fix typing
1 parent d4b07be commit 811379b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

fast_graphrag/_graphrag.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
"""This module implements a Graph-based Retrieval-Augmented Generation (GraphRAG) system."""
22

33
from dataclasses import dataclass, field
4-
from typing import Any, Dict, Generic, List, Optional, Union
4+
from typing import Any, Dict, Generic, List, Optional, Tuple, Union
55

66
from fast_graphrag._llm import BaseLLMService, format_and_send_prompt
77
from fast_graphrag._llm._base import BaseEmbeddingService
@@ -75,15 +75,15 @@ def insert(
7575
content: Union[str, List[str]],
7676
metadata: Union[List[Optional[Dict[str, Any]]], Optional[Dict[str, Any]]] = None,
7777
params: Optional[InsertParam] = None,
78-
) -> None:
78+
) -> Tuple[int, int, int]:
7979
return get_event_loop().run_until_complete(self.async_insert(content, metadata, params))
8080

8181
async def async_insert(
8282
self,
8383
content: Union[str, List[str]],
8484
metadata: Union[List[Optional[Dict[str, Any]]], Optional[Dict[str, Any]]] = None,
8585
params: Optional[InsertParam] = None,
86-
) -> None:
86+
) -> Tuple[int, int, int]:
8787
"""Insert a new memory or memories into the graph.
8888
8989
Args:

0 commit comments

Comments
 (0)