Add Couchbase support [VectorDB] #2179
Conversation
| raise ImportError("`couchbase` not installed. Please install using `pip install couchbase`") | ||
|
|
||
|
|
||
| class CouchbaseFTS(VectorDb): |
There was a problem hiding this comment.
| class CouchbaseFTS(VectorDb): | |
| class Couchbase(VectorDb): |
What do you think about this change?
There was a problem hiding this comment.
While the Couchbase name is more concise, I believe retaining CouchbaseSearch (changed from CouchbaseFTS in the latest commit) is preferable as we're planning to support vector search through multiple Couchbase services (both FTS and the indexing service). The more specific naming clearly indicates which implementation is being used and will help avoid confusion when we add additional Couchbase vector search capabilities in the future.
| def insert(self, documents: List[Document], filters: Optional[Dict[str, Any]] = None) -> None: | ||
| """ | ||
| Insert documents into the Couchbase bucket. Fails if any document already exists. | ||
|
|
||
| Args: | ||
| documents: List of documents to insert | ||
| filters: Optional filters (not used) | ||
| """ | ||
| logger.info(f"Inserting {len(documents)} documents") | ||
|
|
||
| docs_to_insert: Dict[str, Any] = {} | ||
| for document in documents: | ||
| try: | ||
| doc_data = self.prepare_doc(document) | ||
| docs_to_insert[doc_data["_id"]] = doc_data | ||
| del doc_data["_id"] | ||
| except Exception as e: | ||
| logger.error(f"Error preparing document '{document.name}': {e}") | ||
|
|
||
| if docs_to_insert: | ||
| try: | ||
| result = self._collection.insert_multi(docs_to_insert) | ||
| if result.all_ok: | ||
| logger.info(f"Inserted {len(docs_to_insert)} documents successfully") | ||
| else: | ||
| logger.warning(f"Bulk write error while inserting documents: {result.exceptions}") | ||
| except Exception as e: | ||
| logger.error(f"Error during bulk insert: {e}") |
There was a problem hiding this comment.
same here. Why not use filters and insert it with documents?
|
Thanks for adding the tests! There are conflicts with the main branch that need resolution. |
# Conflicts: # libs/agno/pyproject.toml
@manthanguptaa Resolved the conflicts. |
Description
This PR integrates the Agno framework with Couchbase for vector search
Type of change
Please check the options that are relevant:
Checklist
./scripts/format.shand./scripts/validate.shto ensure code is formatted and linted.Additional Notes
You need to have a running Couchbase cluster with a created bucket.