Skip to content

Commit a7a7811

Browse files
authored
fix(2965): correct the number of arguments in BuiltinKnowledgeGraph.aload_document call (#2966)
1 parent 8c34893 commit a7a7811

File tree

1 file changed

+4
-1
lines changed
  • packages/dbgpt-core/src/dbgpt/storage

1 file changed

+4
-1
lines changed

packages/dbgpt-core/src/dbgpt/storage/base.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,11 +55,14 @@ def load_document(self, chunks: List[Chunk]) -> List[str]:
5555
"""
5656

5757
@abstractmethod
58-
async def aload_document(self, chunks: List[Chunk]) -> List[str]:
58+
async def aload_document(
59+
self, chunks: List[Chunk], file_id: Optional[str] = None
60+
) -> List[str]:
5961
"""Load document in index database.
6062
6163
Args:
6264
chunks(List[Chunk]): document chunks.
65+
file_id(Optional[str]): file id for document-level tracking.
6366
6467
Return:
6568
List[str]: chunk ids.

0 commit comments

Comments
 (0)