GraphRAG-入门到实战：2-GraphRAG问答-本地搜索-解析知识图谱

一、GPT测试

import os
from dotenv import load_dotenv

load_dotenv()

True

# print(os.environ.get("OPENAI_API_KEY"))

from openai import OpenAI

# 实例化客户端
client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))

# 调用 GPT-4o-mini 模型
response = client.chat.completions.create(
    model="gpt-4o-mini",
    messages=[
        {
   
   "role": "user", "content": "你好，好久不见!"}
    ]
)

# 输出生成的响应内容
print(response.choices[0].message.content)

你好！很高兴见到你！你最近怎么样？有什么想聊的或者需要帮助的吗？

然后查看当前API-KEY可以调用的模型：

models_list = client.models.list()

models_list.data

[Model(id='gpt-4-0613', created=1686588896, object='model', owned_by='openai'),
 Model(id='gpt-4', created=1687882411, object='model', owned_by='openai'),
 Model(id='gpt-3.5-turbo', created=1677610602, object='model', owned_by='openai'),
 Model(id='gpt-5-nano', created=1754426384, object='model', owned_by='system'),
 Model(id='gpt-5', created=1754425777, object='model', owned_by='system'),
 Model(id='gpt-5-mini-2025-08-07', created=1754425867, object='model', owned_by='system'),
 Model(id='gpt-5-mini', created=1754425928, object='model', owned_by='system'),
 Model(id='gpt-5-nano-2025-08-07', created=1754426303, object='model', owned_by='system'),
 Model(id='davinci-002', created=1692634301, object='model', owned_by='system'),
 Model(id='babbage-002', created=1692634615, object='model', owned_by='system'),
 Model(id='gpt-3.5-turbo-instruct', created=1692901427, object='model', owned_by='system'),
 Model(id='gpt-3.5-turbo-instruct-0914', created=1694122472, object='model', owned_by='system'),
 Model(id='dall-e-3', created=1698785189, object='model', owned_by='system'),
 Model(id='dall-e-2', created=1698798177, object='model', owned_by='system'),
 Model(id='gpt-4-1106-preview', created=1698957206, object='model', owned_by='system'),
 Model(id='gpt-3.5-turbo-1106', created=1698959748, object='model', owned_by='system'),
 Model(id='tts-1-hd', created=1699046015, object='model', owned_by='system'),
 Model(id='tts-1-1106', created=1699053241, object='model', owned_by='system'),
 Model(id='tts-1-hd-1106', created=1699053533, object='model', owned_by='system'),
 Model(id='text-embedding-3-small', created=1705948997, object='model', owned_by='system'),
 Model(id='text-embedding-3-large', created=1705953180, object='model', owned_by='system'),
 Model(id='gpt-4-0125-preview', created=1706037612, object='model', owned_by='system'),
 Model(id='gpt-4-turbo-preview', created=1706037777, object='model', owned_by='system'),
 Model(id='gpt-3.5-turbo-0125', created=1706048358, object='model', owned_by='system'),
 Model(id='gpt-4-turbo', created=1712361441, object='model', owned_by='system'),
 Model(id='gpt-4-turbo-2024-04-09', created=1712601677, object='model', owned_by='system'),
 Model(id='gpt-4o', created=1715367049, object='model', owned_by='system'),
 Model(id='gpt-4o-2024-05-13', created=1715368132, object='model', owned_by='system'),
 Model(id='gpt-4o-mini-2024-07-18', created=1721172717, object='model', owned_by='system'),
 Model(id='gpt-4o-mini', created=1721172741, object='model', owned_by='system'),
 Model(id='gpt-4o-2024-08-06', created=1722814719, object='model', owned_by='system'),
 Model(id='chatgpt-4o-latest', created=1723515131, object='model', owned_by='system'),
 Model(id='o1-mini-2024-09-12', created=1725648979, object='model', owned_by='system'),
 Model(id='o1-mini', created=1725649008, object='model', owned_by='system'),
 Model(id='gpt-4o-realtime-preview-2024-10-01', created=1727131766, object='model', owned_by='system'),
 Model(id='gpt-4o-audio-preview-2024-10-01', created=1727389042, object='model', owned_by='system'),
 Model(id='gpt-4o-audio-preview', created=1727460443, object='model', owned_by='system'),
 Model(id='gpt-4o-realtime-preview', created=1727659998, object='model', owned_by='system'),
 Model(id='omni-moderation-latest', created=1731689265, object='model', owned_by='system'),
 Model(id='omni-moderation-2024-09-26', created=1732734466, object='model', owned_by='system'),
 Model(id='gpt-4o-realtime-preview-2024-12-17', created=1733945430, object='model', owned_by='system'),
 Model(id='gpt-4o-audio-preview-2024-12-17', created=1734034239, object='model', owned_by='system'),
 Model(id='gpt-4o-mini-realtime-preview-2024-12-17', created=1734112601, object='model', owned_by='system'),
 Model(id='gpt-4o-mini-audio-preview-2024-12-17', created=1734115920, object='model', owned_by='system'),
 Model(id='o1-2024-12-17', created=1734326976, object='model', owned_by='system'),
 Model(id='o1', created=1734375816, object='model', owned_by='system'),
 Model(id='gpt-4o-mini-realtime-preview', created=1734387380, object='model', owned_by='system'),
 Model(id='gpt-4o-mini-audio-preview', created=1734387424, object='model', owned_by='system'),
 Model(id='o3-mini', created=1737146383, object='model', owned_by='system'),
 Model(id='o3-mini-2025-01-31', created=1738010200, object='model', owned_by='system'),
 Model(id='gpt-4o-2024-11-20', created=1739331543, object='model', owned_by='system'),
 Model(id='gpt-4o-search-preview-2025-03-11', created=1741388170, object='model', owned_by='system'),
 Model(id='gpt-4o-search-preview', created=1741388720, object='model', owned_by='system'),
 Model(id='gpt-4o-mini-search-preview-2025-03-11', created=1741390858, object='model', owned_by='system'),
 Model(id='gpt-4o-mini-search-preview', created=1741391161, object='model', owned_by='system'),
 Model(id='gpt-4o-transcribe', created=1742068463, object='model', owned_by='system'),
 Model(id='gpt-4o-mini-transcribe', created=1742068596, object='model', owned_by='system'),
 Model(id='o1-pro-2025-03-19', created=1742251504, object='model', owned_by='system'),
 Model(id='o1-pro', created=1742251791, object='model', owned_by='system'),
 Model(id='gpt-4o-mini-tts', created=1742403959, object='model', owned_by='system'),
 Model(id='o3-2025-04-16', created=1744133301, object='model', owned_by='system'),
 Model(id='o4-mini-2025-04-16', created=1744133506, object='model', owned_by='system'),
 Model(id='o3', created=1744225308, object='model', owned_by='system'),
 Model(id='o4-mini', created=1744225351, object='model', owned_by='system'),
 Model(id='gpt-4.1-2025-04-14', created=1744315746, object='model', owned_by='system'),
 Model(id='gpt-4.1', created=1744316542, object='model', owned_by='system'),
 Model(id='gpt-4.1-mini-2025-04-14', created=1744317547, object='model', owned_by='system'),
 Model(id='gpt-4.1-mini', created=1744318173, object='model', owned_by='system'),
 Model(id='gpt-4.1-nano-2025-04-14', created=1744321025, object='model', owned_by='system'),
 Model(id='gpt-4.1-nano', created=1744321707, object='model', owned_by='system'),
 Model(id='gpt-image-1', created=1745517030, object='model', owned_by='system'),
 Model(id='codex-mini-latest', created=1746673257, object='model', owned_by='system'),
 Model(id='gpt-4o-realtime-preview-2025-06-03', created=1748907838, object='model', owned_by='system'),
 Model(id='gpt-4o-audio-preview-2025-06-03', created=1748908498, object='model', owned_by='system'),
 Model(id='o4-mini-deep-research', created=1749685485, object='model', owned_by='system'),
 Model(id='o4-mini-deep-research-2025-06-26', created=1750866121, object='model', owned_by='system'),
 Model(id='gpt-5-chat-latest', created=1754073306, object='model', owned_by='system'),
 Model(id='gpt-5-2025-08-07', created=1754075360, object='model', owned_by='system'),
 Model(id='gpt-3.5-turbo-16k', created=1683758102, object='model', owned_by='openai-internal'),
 Model(id='tts-1', created=1681940951, object='model', owned_by='openai-internal'),
 Model(id='whisper-1', created=1677532384, object='model', owned_by='openai-internal'),
 Model(id='text-embedding-ada-002', created=1671217299, object='model', owned_by='openai-internal')]

二、查看知识图谱相关表格

• 文件表（Documents Table）

import pandas as pd

documents_df = pd.read_parquet("/Users/dcs/study/ragtest/output/documents.parquet")
documents_df.head()

	id	human_readable_id	title	text	text_unit_ids	creation_date	metadata
0	4ad21bc3e51b59a81cf6154e067933779d25209edb4dd3...	1	百得利集团介绍-v1.1.txt	---\n### **百得利集团（BetterLife Group）详细介绍**\n\n--...	[fcfd223135ec68d9db3de6d96a9acf067bba754a821bd...	2025-05-16 11:09:52 +0800	None

• 文本单元表（TEXT UNIT Table）

text_unit_df = pd.read_parquet("/Users/dcs/study/ragtest/output/text_units.parquet")
text_unit_df.head()

	id	human_readable_id	text	n_tokens	document_ids	entity_ids	relationship_ids	covariate_ids