Skip to main content
Trubrics 是一个 LLM 用户分析平台，让你能够收集、分析和管理用户对 AI 模型的提示词与反馈。 请查看 Trubrics 仓库 以获取更多关于 Trubrics 的信息。
本指南将介绍如何设置 TrubricsCallbackHandler。

​
安装与设置

pip install -qU  trubrics langchain langchain-community

​
获取 Trubrics 凭据

如果你没有 Trubrics 账号，请在此处创建一个。在本教程中，我们将使用账号创建时自动生成的 default 项目。 现在将你的凭据设置为环境变量： 
import os

os.environ["TRUBRICS_EMAIL"] = "***@***"
os.environ["TRUBRICS_PASSWORD"] = "***"

from langchain_community.callbacks.trubrics_callback import TrubricsCallbackHandler

​
使用方法

TrubricsCallbackHandler 可以接收多种可选参数。传递给 Trubrics 提示词的 kwargs 参数请参见此处。 
class TrubricsCallbackHandler(BaseCallbackHandler):

    """
    Callback handler for Trubrics.

    Args:
        project: a trubrics project, default project is "default"
        email: a trubrics account email, can equally be set in env variables
        password: a trubrics account password, can equally be set in env variables
        **kwargs: all other kwargs are parsed and set to trubrics prompt variables, or added to the `metadata` dict
    """

​
示例

以下是两个使用 TrubricsCallbackHandler 与 LangChain LLM 或聊天模型配合的示例。我们将使用 OpenAI 模型，请在此设置你的 OPENAI_API_KEY 密钥： 
os.environ["OPENAI_API_KEY"] = "sk-***"

​
1. 与 LLM 配合使用

from langchain_openai import OpenAI

llm = OpenAI(callbacks=[TrubricsCallbackHandler()])

2023-09-26 11:30:02.149 | INFO     | trubrics.platform.auth:get_trubrics_auth_token:61 - User jeff.kayne@trubrics.com has been authenticated.

res = llm.generate(["Tell me a joke", "Write me a poem"])

2023-09-26 11:30:07.760 | INFO     | trubrics.platform:log_prompt:102 - User prompt saved to Trubrics.
2023-09-26 11:30:08.042 | INFO     | trubrics.platform:log_prompt:102 - User prompt saved to Trubrics.

print("--> GPT's joke: ", res.generations[0][0].text)
print()
print("--> GPT's poem: ", res.generations[1][0].text)

--> GPT's joke:

Q: What did the fish say when it hit the wall?
A: Dam!

--> GPT's poem:

A Poem of Reflection

I stand here in the night,
The stars above me filling my sight.
I feel such a deep connection,
To the world and all its perfection.

A moment of clarity,
The calmness in the air so serene.
My mind is filled with peace,
And I am released.

The past and the present,
My thoughts create a pleasant sentiment.
My heart is full of joy,
My soul soars like a toy.

I reflect on my life,
And the choices I have made.
My struggles and my strife,
The lessons I have paid.

The future is a mystery,
But I am ready to take the leap.
I am ready to take the lead,
And to create my own destiny.

​
2. 与聊天模型配合使用

from langchain.messages import HumanMessage, SystemMessage
from langchain_openai import ChatOpenAI

chat_llm = ChatOpenAI(
    callbacks=[
        TrubricsCallbackHandler(
            project="default",
            tags=["chat model"],
            user_id="user-id-1234",
            some_metadata={"hello": [1, 2]},
        )
    ]
)

chat_res = chat_llm.invoke(
    [
        SystemMessage(content="Every answer of yours must be about OpenAI."),
        HumanMessage(content="Tell me a joke"),
    ]
)

2023-09-26 11:30:10.550 | INFO     | trubrics.platform:log_prompt:102 - User prompt saved to Trubrics.

print(chat_res.content)

Why did the OpenAI computer go to the party?

Because it wanted to meet its AI friends and have a byte of fun!
在 GitHub 上编辑此页 或 提交问题。
将这些文档连接 到 Claude、VSCode 等，通过 MCP 获取实时解答。