import { CouchbaseDocumentLoader } from "@langchain/community/document_loaders/web/couchbase";
import { Cluster } from "couchbase";

const connectionString = "couchbase://localhost"; // 有效的 Couchbase 连接字符串
const dbUsername = "Administrator"; // 具有对正在查询的存储桶读取权限的有效数据库用户
const dbPassword = "Password"; // 数据库用户的密码

// query 是一个有效的 SQL++ 查询
const query = `
    SELECT h.* FROM \`travel-sample\`.inventory.hotel h
    WHERE h.country = 'United States'
    LIMIT 1
`;

const couchbaseClient = await Cluster.connect(connectionString, {
  username: dbUsername,
  password: dbPassword,
  configProfile: "wanDevelopment",
});

const loader = new CouchbaseDocumentLoader(
  couchbaseClient, // 已连接的 Couchbase 集群客户端
  query // 一个有效的 SQL++ 查询，将返回所需数据
);

// 使用 load 方法
docs = await loader.load();
console.log(docs);

// 使用 lazy_load
for await (const doc of this.lazyLoad()) {
  console.log(doc);
  break; // 根据所需条件中断
}

const loaderWithSelectedFields = new CouchbaseDocumentLoader(
  couchbaseClient,
  query,
  // pageContentFields
  [
    "address",
    "name",
    "city",
    "phone",
    "country",
    "geo",
    "description",
    "reviews",
  ],
  ["id"] // metadataFields
);

const filtered_docs = await loaderWithSelectedFields.load();
console.log(filtered_docs);