使用 Opik 记录追踪日志¶

本指南将指导您下载 Paul Graham 的文章作为数据源，然后通过 LlamaIndex 查询这些文章，并将追踪日志记录到 Opik 平台。

在 Comet.com 创建账户¶

Comet 提供 Opik 平台的托管版本，只需注册账户 并获取您的 API 密钥即可开始使用。

您也可以在本地运行 Opik 平台，详见 安装指南 获取更多信息。

In [ ]:

import os
import getpass

if "OPIK_API_KEY" not in os.environ:
    os.environ["OPIK_API_KEY"] = getpass.getpass("Opik API Key: ")
if "OPIK_WORKSPACE" not in os.environ:
    os.environ["OPIK_WORKSPACE"] = input(
        "Comet workspace (often the same as your username): "
    )

import os import getpass if "OPIK_API_KEY" not in os.environ: os.environ["OPIK_API_KEY"] = getpass.getpass("Opik API Key: ") if "OPIK_WORKSPACE" not in os.environ: os.environ["OPIK_WORKSPACE"] = input( "Comet workspace (often the same as your username): " )

如果您在本地运行 Opik 平台，只需设置：

In [ ]:

# import os
# os.environ["OPIK_URL_OVERRIDE"] = "http://localhost:5173/api"

# import os # os.environ["OPIK_URL_OVERRIDE"] = "http://localhost:5173/api"

环境准备¶

首先，我们将安装必要的库，下载 Chinook 数据库并配置不同的 API 密钥。

In [ ]:

%pip install opik llama-index llama-index-agent-openai llama-index-llms-openai --upgrade --quiet

%pip install opik llama-index llama-index-agent-openai llama-index-llms-openai --upgrade --quiet

并配置所需的环境变量：

In [ ]:

import os
import getpass

if "OPENAI_API_KEY" not in os.environ:
    os.environ["OPENAI_API_KEY"] = getpass.getpass(
        "Enter your OpenAI API key: "
    )

import os import getpass if "OPENAI_API_KEY" not in os.environ: os.environ["OPENAI_API_KEY"] = getpass.getpass( "Enter your OpenAI API key: " )

此外，我们将下载保罗·格雷厄姆的文集：

In [ ]:

import os
import requests

# Create directory if it doesn't exist
os.makedirs("./data/paul_graham/", exist_ok=True)

# Download the file using requests
url = "https://raw.githubusercontent.com/run-llama/llama_index/main/docs/docs/examples/data/paul_graham/paul_graham_essay.txt"
response = requests.get(url)
with open("./data/paul_graham/paul_graham_essay.txt", "wb") as f:
    f.write(response.content)

import os import requests # Create directory if it doesn't exist os.makedirs("./data/paul_graham/", exist_ok=True) # Download the file using requests url = "https://raw.githubusercontent.com/run-llama/llama_index/main/docs/docs/examples/data/paul_graham/paul_graham_essay.txt" response = requests.get(url) with open("./data/paul_graham/paul_graham_essay.txt", "wb") as f: f.write(response.content)

使用 LlamaIndex¶

配置 Opik 集成¶

您可以通过直接调用来使用 Opik 回调：

In [ ]:

from llama_index.core import set_global_handler

# You should provide your OPIK API key and Workspace using the following environment variables:
# OPIK_API_KEY, OPIK_WORKSPACE
set_global_handler(
    "opik",
)

from llama_index.core import set_global_handler # You should provide your OPIK API key and Workspace using the following environment variables: # OPIK_API_KEY, OPIK_WORKSPACE set_global_handler( "opik", )

回调处理器配置完成后，所有追踪记录将自动记录至 Opik 平台。

使用 LLamaIndex¶

第一步是将数据加载到 LlamaIndex 中。我们将使用 SimpleDirectoryReader 从 data/paul_graham 目录加载数据，同时创建向量存储库来索引所有已加载的文档。

In [ ]:

from llama_index.core import VectorStoreIndex, SimpleDirectoryReader

documents = SimpleDirectoryReader("./data/paul_graham").load_data()
index = VectorStoreIndex.from_documents(documents)
query_engine = index.as_query_engine()

from llama_index.core import VectorStoreIndex, SimpleDirectoryReader documents = SimpleDirectoryReader("./data/paul_graham").load_data() index = VectorStoreIndex.from_documents(documents) query_engine = index.as_query_engine()

现在我们可以通过 query_engine 对象来查询索引：

In [ ]:

response = query_engine.query("What did the author do growing up?")
print(response)

response = query_engine.query("What did the author do growing up?") print(response)

您现在可以前往 Opik 应用查看追踪记录：