Skip to content

Commit f064b95

Browse files
authored
Merge pull request #2027 from Matt23-star/main
Refactor: PostgreSQL
2 parents 9205818 + 24cb11f commit f064b95

File tree

2 files changed

+93
-117
lines changed

2 files changed

+93
-117
lines changed

lightrag/kg/deprecated/chroma_impl.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -164,9 +164,7 @@ async def upsert(self, data: dict[str, dict[str, Any]]) -> None:
164164
logger.error(f"Error during ChromaDB upsert: {str(e)}")
165165
raise
166166

167-
async def query(
168-
self, query: str, top_k: int, ids: list[str] | None = None
169-
) -> list[dict[str, Any]]:
167+
async def query(self, query: str, top_k: int) -> list[dict[str, Any]]:
170168
try:
171169
embedding = await self.embedding_func(
172170
[query], _priority=5

0 commit comments

Comments
 (0)