Using Opik with aisuite
Opik integrates with aisuite to provide a simple way to log traces for all aisuite LLM calls.
Creating an account on Comet.com
Comet provides a hosted version of the Opik platform, simply create an account and grab you API Key.
You can also run the Opik platform locally, see the installation guide for more information.
Preparing our environment
First, we will set up our OpenAI API keys.
Logging traces
In order to log traces to Opik, we need to wrap our OpenAI calls with the track_openai
function:
The prompt and response messages are automatically logged to Opik and can be viewed in the UI.
Using it with the track
decorator
If you have multiple steps in your LLM pipeline, you can use the track
decorator to log the traces for each step. If OpenAI is called within one of these steps, the LLM call with be associated with that corresponding step:
The trace can now be viewed in the UI: