-
Notifications
You must be signed in to change notification settings - Fork 5.8k
Add braintrust docs #3628
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Merged
Merged
Add braintrust docs #3628
Changes from all commits
Commits
Show all changes
5 commits
Select commit
Hold shift + click to select a range
File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,237 @@ | ||
| --- | ||
| title: Braintrust | ||
| description: Braintrust integration for CrewAI with OpenTelemetry tracing and evaluation | ||
| icon: magnifying-glass-chart | ||
| mode: "wide" | ||
| --- | ||
|
|
||
| # Braintrust Integration | ||
|
|
||
| This guide demonstrates how to integrate **Braintrust** with **CrewAI** using OpenTelemetry for comprehensive tracing and evaluation. By the end of this guide, you will be able to trace your CrewAI agents, monitor their performance, and evaluate their outputs using Braintrust's powerful observability platform. | ||
|
|
||
| > **What is Braintrust?** [Braintrust](https://www.braintrust.dev) is an AI evaluation and observability platform that provides comprehensive tracing, evaluation, and monitoring for AI applications with built-in experiment tracking and performance analytics. | ||
|
|
||
| ## Get Started | ||
|
|
||
| We'll walk through a simple example of using CrewAI and integrating it with Braintrust via OpenTelemetry for comprehensive observability and evaluation. | ||
|
|
||
| ### Step 1: Install Dependencies | ||
|
|
||
| ```bash | ||
| uv add braintrust[otel] crewai crewai-tools opentelemetry-instrumentation-openai opentelemetry-instrumentation-crewai python-dotenv | ||
| ``` | ||
|
|
||
| ### Step 2: Set Up Environment Variables | ||
|
|
||
| Setup Braintrust API keys and configure OpenTelemetry to send traces to Braintrust. You'll need a Braintrust API key and your OpenAI API key. | ||
|
|
||
| ```python | ||
| import os | ||
| from getpass import getpass | ||
|
|
||
| # Get your Braintrust credentials | ||
| BRAINTRUST_API_KEY = getpass("🔑 Enter your Braintrust API Key: ") | ||
|
|
||
| # Get API keys for services | ||
| OPENAI_API_KEY = getpass("🔑 Enter your OpenAI API key: ") | ||
|
|
||
| # Set environment variables | ||
| os.environ["BRAINTRUST_API_KEY"] = BRAINTRUST_API_KEY | ||
| os.environ["BRAINTRUST_PARENT"] = "project_name:crewai-demo" | ||
| os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY | ||
| ``` | ||
|
|
||
| ### Step 3: Initialize OpenTelemetry with Braintrust | ||
|
|
||
| Initialize the Braintrust OpenTelemetry instrumentation to start capturing traces and send them to Braintrust. | ||
|
|
||
| ```python | ||
| import os | ||
| from typing import Any, Dict | ||
|
|
||
| from braintrust.otel import BraintrustSpanProcessor | ||
| from crewai import Agent, Crew, Task | ||
| from crewai.llm import LLM | ||
| from opentelemetry import trace | ||
| from opentelemetry.instrumentation.crewai import CrewAIInstrumentor | ||
| from opentelemetry.instrumentation.openai import OpenAIInstrumentor | ||
| from opentelemetry.sdk.trace import TracerProvider | ||
|
|
||
| def setup_tracing() -> None: | ||
| """Setup OpenTelemetry tracing with Braintrust.""" | ||
| current_provider = trace.get_tracer_provider() | ||
| if isinstance(current_provider, TracerProvider): | ||
| provider = current_provider | ||
| else: | ||
| provider = TracerProvider() | ||
| trace.set_tracer_provider(provider) | ||
|
|
||
| provider.add_span_processor(BraintrustSpanProcessor()) | ||
| CrewAIInstrumentor().instrument(tracer_provider=provider) | ||
| OpenAIInstrumentor().instrument(tracer_provider=provider) | ||
|
|
||
|
|
||
| setup_tracing() | ||
| ``` | ||
|
|
||
| ### Step 4: Create a CrewAI Application | ||
|
|
||
| We'll create a CrewAI application where two agents collaborate to research and write a blog post about AI advancements, with comprehensive tracing enabled. | ||
|
|
||
| ```python | ||
| from crewai import Agent, Crew, Process, Task | ||
| from crewai_tools import SerperDevTool | ||
|
|
||
| def create_crew() -> Crew: | ||
| """Create a crew with multiple agents for comprehensive tracing.""" | ||
| llm = LLM(model="gpt-4o-mini") | ||
| search_tool = SerperDevTool() | ||
|
|
||
| # Define agents with specific roles | ||
| researcher = Agent( | ||
| role="Senior Research Analyst", | ||
| goal="Uncover cutting-edge developments in AI and data science", | ||
| backstory="""You work at a leading tech think tank. | ||
| Your expertise lies in identifying emerging trends. | ||
| You have a knack for dissecting complex data and presenting actionable insights.""", | ||
| verbose=True, | ||
| allow_delegation=False, | ||
| llm=llm, | ||
| tools=[search_tool], | ||
| ) | ||
|
|
||
| writer = Agent( | ||
| role="Tech Content Strategist", | ||
| goal="Craft compelling content on tech advancements", | ||
| backstory="""You are a renowned Content Strategist, known for your insightful and engaging articles. | ||
| You transform complex concepts into compelling narratives.""", | ||
| verbose=True, | ||
| allow_delegation=True, | ||
| llm=llm, | ||
| ) | ||
|
|
||
| # Create tasks for your agents | ||
| research_task = Task( | ||
| description="""Conduct a comprehensive analysis of the latest advancements in {topic}. | ||
| Identify key trends, breakthrough technologies, and potential industry impacts.""", | ||
| expected_output="Full analysis report in bullet points", | ||
| agent=researcher, | ||
| ) | ||
|
|
||
| writing_task = Task( | ||
| description="""Using the insights provided, develop an engaging blog | ||
| post that highlights the most significant {topic} advancements. | ||
| Your post should be informative yet accessible, catering to a tech-savvy audience. | ||
| Make it sound cool, avoid complex words so it doesn't sound like AI.""", | ||
| expected_output="Full blog post of at least 4 paragraphs", | ||
| agent=writer, | ||
| context=[research_task], | ||
| ) | ||
|
|
||
| # Instantiate your crew with a sequential process | ||
| crew = Crew( | ||
| agents=[researcher, writer], | ||
| tasks=[research_task, writing_task], | ||
| verbose=True, | ||
| process=Process.sequential | ||
| ) | ||
|
|
||
| return crew | ||
|
|
||
| def run_crew(): | ||
| """Run the crew and return results.""" | ||
| crew = create_crew() | ||
| result = crew.kickoff(inputs={"topic": "AI developments"}) | ||
| return result | ||
|
|
||
| # Run your crew | ||
| if __name__ == "__main__": | ||
| # Instrumentation is already initialized above in this module | ||
| result = run_crew() | ||
| print(result) | ||
| ``` | ||
|
|
||
| ### Step 5: View Traces in Braintrust | ||
|
|
||
| After running your crew, you can view comprehensive traces in Braintrust through different perspectives: | ||
|
|
||
| <Tabs> | ||
| <Tab title="Trace"> | ||
| <Frame> | ||
| <img src="/images/braintrust-trace-view.png" alt="Braintrust Trace View"/> | ||
| </Frame> | ||
| </Tab> | ||
|
|
||
| <Tab title="Timeline"> | ||
| <Frame> | ||
| <img src="/images/braintrust-timeline-view.png" alt="Braintrust Timeline View"/> | ||
| </Frame> | ||
| </Tab> | ||
|
|
||
| <Tab title="Thread"> | ||
| <Frame> | ||
| <img src="/images/braintrust-thread-view.png" alt="Braintrust Thread View"/> | ||
| </Frame> | ||
| </Tab> | ||
| </Tabs> | ||
|
|
||
| ### Step 6: Evaluate via SDK (Experiments) | ||
|
|
||
| You can also run evaluations using Braintrust's Eval SDK. This is useful for comparing versions or scoring outputs offline. Below is a Python example using the `Eval` class with the crew we created above: | ||
|
|
||
| ```python | ||
| # eval_crew.py | ||
| from braintrust import Eval | ||
| from autoevals import Levenshtein | ||
|
|
||
| def evaluate_crew_task(input_data): | ||
| """Task function that wraps our crew for evaluation.""" | ||
| crew = create_crew() | ||
| result = crew.kickoff(inputs={"topic": input_data["topic"]}) | ||
| return str(result) | ||
|
|
||
| Eval( | ||
| "AI Research Crew", # Project name | ||
| { | ||
| "data": lambda: [ | ||
| {"topic": "artificial intelligence trends 2024"}, | ||
| {"topic": "machine learning breakthroughs"}, | ||
| {"topic": "AI ethics and governance"}, | ||
| ], | ||
| "task": evaluate_crew_task, | ||
| "scores": [Levenshtein], | ||
| }, | ||
| ) | ||
| ``` | ||
|
|
||
| Setup your API key and run: | ||
|
|
||
| ```bash | ||
| export BRAINTRUST_API_KEY="YOUR_API_KEY" | ||
| braintrust eval eval_crew.py | ||
| ``` | ||
|
|
||
| See the [Braintrust Eval SDK guide](https://www.braintrust.dev/docs/start/eval-sdk) for more details. | ||
|
|
||
| ### Key Features of Braintrust Integration | ||
|
|
||
| - **Comprehensive Tracing**: Track all agent interactions, tool usage, and LLM calls | ||
| - **Performance Monitoring**: Monitor execution times, token usage, and success rates | ||
| - **Experiment Tracking**: Compare different crew configurations and models | ||
| - **Automated Evaluation**: Set up custom evaluation metrics for crew outputs | ||
| - **Error Tracking**: Monitor and debug failures across your crew executions | ||
| - **Cost Analysis**: Track token usage and associated costs | ||
|
|
||
| ### Version Compatibility Information | ||
| - Python 3.8+ | ||
| - CrewAI >= 0.86.0 | ||
| - Braintrust >= 0.1.0 | ||
| - OpenTelemetry SDK >= 1.31.0 | ||
|
|
||
| ### References | ||
| - [Braintrust Documentation](https://www.braintrust.dev/docs) - Overview of the Braintrust platform | ||
| - [Braintrust CrewAI Integration](https://www.braintrust.dev/docs/integrations/crew-ai) - Official CrewAI integration guide | ||
| - [Braintrust Eval SDK](https://www.braintrust.dev/docs/start/eval-sdk) - Run experiments via the SDK | ||
| - [CrewAI Documentation](https://docs.crewai.com/) - Overview of the CrewAI framework | ||
| - [OpenTelemetry Docs](https://opentelemetry.io/docs/) - OpenTelemetry guide | ||
| - [Braintrust GitHub](https://github.com/braintrustdata/braintrust) - Source code for Braintrust SDK | ||
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Oops, something went wrong.
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.