Company Brain: Company Q&A 是 cognee 仓库中的一个 cookbook 示例,位于 examples/cookbooks/company_brain/company_qa/。它的用途是演示如何把企业内部的多种数据来源,构建成一个统一的公司记忆,并基于该记忆跨来源回答自然语言问题。
用途与数据来源:该流程的目标数据源包括三类:一个 SQL 数据库(需提供数据库 URL,可选指定表)、一个或多个支持工单导出文件(JSON 或 CSV 格式)、以及一个文档文件夹。这些来源会在 cognee 的 company_qa 数据集中被记住,并通过同一个图模型进行抽取,使得同一个人、团队或项目在不同来源中归并为同一个节点,从而支持跨来源的问答。如果用户没有自己的数据源,或想先看效果,可以不传任何来源参数,此时运行会使用示例公司数据,流程会通过 [setup] 行说明这一点,并应告知用户答案来自示例数据。
主要能力与使用方式:运行所有命令都需要在仓库根目录执行。第一步是确定数据源,向用户询问他们想要使用的数据库 URL(及表)、一个或多个工单导出文件、以及文档文件夹。第二步执行检查命令:uv run python examples/cookbooks/company_brain/company_qa/company_qa.py --check,并带上数据源参数(--database URL [--tables a,b] --tickets FILE [FILE ...] --docs FOLDER),或不带参数以使用示例。若退出码为 0,则进入下一步;若退出码为 2,则每个 [setup] MISSING: 行会指明一项需要修复的内容,此时不应自行创建凭据,而应告知用户需要修复哪一行,并参考 README 的 "What it needs" 表格后停止。第三步运行同一命令但不带 --check,并加上用户的问题 --ask "..."。在示例数据上若不传 --ask,则会提出一个需要全部三个示例来源的问题。答案位于 [ask] A: 之后的所有内容,应交给用户;[ingest] 行会说明本次纳入了哪些来源。后续的问题无需再次提供数据源,可使用:uv run python examples/cookbooks/company_brain/company_qa/scripts/ask.py "..."。如果脚本失败,company_qa.py 会以退出码 1 退出,并打印一行说明出错原因,可单独从 examples/cookbooks/company_brain/company_qa/scripts/ 目录运行对应脚本以进一步排查。
适用场景与限制:适用于需要把分散在数据库、工单系统和文档中的企业信息整合为统一图谱记忆,并基于此进行跨来源问答的场景。限制与注意事项包括:数据来源属于用户公司数据,不应超出回答问题所需范围引用,也不要打印 .env 或数据库 URL 中的密码;每次运行都会消耗 LLM 额度,因此只有在用户明确要求时才进行 ingest;示例运行会在 ingest 前忘记 cookbook 的数据集 company_qa(即 [clear] 行),而在用户自己的数据上,只有在用户要求重新开始时才传入 --clear,因为它会忘记该数据集保存的全部内容;除非用户要求浏览图谱,否则不要传入 --ui,因为它会持续运行直到 Ctrl+C。清理命令为:uv run cognee-cli forget --dataset company_qa。
关于具体配置、脚本细节以及每一步的作用,资料指向 cookbook 的 README.md;其他未在资料中说明的内容暂未提供。