Merge pull request #1 from caseyclements/pull/3000

PYTHON-4506 Expanded tests and simplified vector search pipelines
Jibola · Jun 24, 2024 · 6245c30 · 6245c30
2 parents 334cc25 + f5e5fdf
commit 6245c30
Show file tree

Hide file tree

Showing 3 changed files with 550 additions and 250 deletions.
diff --git a/autogen/agentchat/contrib/vectordb/base.py b/autogen/agentchat/contrib/vectordb/base.py
@@ -171,7 +171,7 @@ def get_docs_by_ids(
             ids: List[ItemID] | A list of document ids. If None, will return all the documents. Default is None.
             collection_name: str | The name of the collection. Default is None.
             include: List[str] | The fields to include. Default is None.
-                If None, will include ["metadatas", "documents"], ids will always be included.
+                If None, will include ["metadata", "content"], ids will always be included.  # TODO - Confirm keys
             kwargs: dict | Additional keyword arguments.
 
         Returns: