With cometLLM, effortlessly track and visualize your LLM prompts and chains. Identify effective strategies, troubleshoot efficiently, and ensure reproducible workflows. Streamline your processes and unleash the full potential of your Large Language Models today.
Install comet_llm
Python library with pip:
pip install comet_llm
If you don't have already, create your free Comet account and grab your API Key from the account settings page.
Now you are all set to log your first prompt and response:
import comet_llm
comet_llm.log_prompt(
prompt="What is your name?",
output=" My name is Alex.",
api_key="<YOUR_COMET_API_KEY>",
)
- Log your prompts and responses, including prompt template, variables, timestamps and duration and any metadata that you need.
- Visualize your prompts and responses in the UI.
- Log your chain execution down to the level of granularity that you need.
- Visualize your chain execution in the UI.
- Diff your prompts and chain execution in the UI.
import comet_llm
comet_llm.log_prompt(
prompt="Answer the question and if the question can't be answered, say \"I don't know\"\n\n---\n\nQuestion: What is your name?\nAnswer:",
prompt_template="Answer the question and if the question can't be answered, say \"I don't know\"\n\n---\n\nQuestion: {{question}}?\nAnswer:",
prompt_template_variables={"question": "What is your name?"},
metadata= {
"usage.prompt_tokens": 7,
"usage.completion_tokens": 5,
"usage.total_tokens": 12,
},
output=" My name is Alex.",
duration=16.598,
)
You can configure your Comet credentials and where you are logging data to:
Name | Python parameter name | Environment variable name |
---|---|---|
Comet API KEY | api_key | COMET_API_KEY |
Comet Workspace name | workspace | COMET_WORKSPACE |
Comet Project name | project | COMET_PROJECT_NAME |
Copyright (c) Comet 2023-present. cometLLM
is free and open-source software licensed under the MIT License.