LangGraph

import os

# Set OpenTelemetry environment variables BEFORE importing LangChain/LangGraph
os.environ['LANGSMITH_OTEL_ENABLED'] = 'true'
os.environ['LANGSMITH_TRACING'] = 'true'

# Now import the libraries
import gentrace
from gentrace import interaction
from langgraph.prebuilt import create_react_agent
from langchain_openai import ChatOpenAI

# Initialize Gentrace
gentrace.init(api_key=os.getenv("GENTRACE_API_KEY"))

import os

# Set OTEL environment variables first
os.environ['LANGSMITH_OTEL_ENABLED'] = 'true'
os.environ['LANGSMITH_TRACING'] = 'true'

import gentrace
from gentrace import interaction
from langgraph.prebuilt import create_react_agent
from langchain_openai import ChatOpenAI
from langchain_core.tools import tool

# Initialize Gentrace
gentrace.init(api_key=os.getenv("GENTRACE_API_KEY"))

# Define a tool for the agent
@tool
def add(a: int, b: int) -> int:
    """Add two numbers together."""
    return a + b

# Create the agent wrapped in a Gentrace interaction
@interaction(name="math-agent-interaction")
def run_math_agent():
    # Create a ReAct agent with the add tool
    math_agent = create_react_agent(
        'openai:gpt-4o',
        tools=[add],
        name='math_agent'
    )

    # Invoke the agent
    result = math_agent.invoke({
        'messages': [{
            'role': 'user',
            'content': "What's 123 + 456?"
        }]
    })

    return result

# Run the traced agent
if __name__ == "__main__":
    result = run_math_agent()
    print(result)

# Gentrace configuration
GENTRACE_API_KEY=your-gentrace-api-key

# OpenTelemetry configuration (required for LangGraph tracing)
LANGSMITH_OTEL_ENABLED=true
LANGSMITH_TRACING=true

# LLM provider credentials
OPENAI_API_KEY=your-openai-api-key

import os

# Set OTEL environment variables first
os.environ['LANGSMITH_OTEL_ENABLED'] = 'true'
os.environ['LANGSMITH_TRACING'] = 'true'

import gentrace
from gentrace import interaction
from langgraph.graph import StateGraph, END
from typing import TypedDict, Annotated
from operator import add

# Initialize Gentrace
gentrace.init(api_key=os.getenv("GENTRACE_API_KEY"))

class AgentState(TypedDict):
    messages: Annotated[list, add]
    next_action: str

def process_node(state):
    # Process the messages
    state["messages"].append("Processing completed")
    state["next_action"] = "analyze"
    return state

def analyze_node(state):
    # Analyze the processed data
    state["messages"].append("Analysis completed")
    return state

@interaction(name="custom-workflow-agent")
def run_custom_agent():
    # Build custom graph
    workflow = StateGraph(AgentState)

    # Add nodes and edges
    workflow.add_node("process", process_node)
    workflow.add_node("analyze", analyze_node)
    workflow.add_edge("process", "analyze")
    workflow.add_edge("analyze", END)

    # Set entry point
    workflow.set_entry_point("process")

    # Compile and run
    app = workflow.compile()
    result = app.invoke({"messages": [], "next_action": "start"})

    return result

if __name__ == "__main__":
    result = run_custom_agent()
    print("Final state:", result)

import os

# Set OTEL environment variables first
os.environ['LANGSMITH_OTEL_ENABLED'] = 'true'
os.environ['LANGSMITH_TRACING'] = 'true'

import gentrace
from gentrace import interaction
from langgraph.prebuilt import create_react_agent
from langchain_core.tools import tool

# Initialize Gentrace
gentrace.init(api_key=os.getenv("GENTRACE_API_KEY"))

# Define tools for the agents
@tool
def search_tool(query: str) -> str:
    """Search for information about a topic."""
    # Simulate search results
    return f"Search results for '{query}': Lorem ipsum dolor sit amet, consectetur adipiscing elit."

@tool
def write_tool(content: str) -> str:
    """Write formatted content based on input."""
    # Simulate writing process
    return f"Article: {content}\n\nFormatted and enhanced with additional context."

@interaction(name="multi-agent-system")
def coordinate_agents():
    # Create specialized agents
    research_agent = create_react_agent(
        'openai:gpt-4o',
        tools=[search_tool],
        name='researcher'
    )
    writer_agent = create_react_agent(
        'openai:gpt-4o',
        tools=[write_tool],
        name='writer'
    )

    # Research phase
    research_results = research_agent.invoke({
        'messages': [{
            'role': 'user',
            'content': 'Research the latest developments in artificial intelligence'
        }]
    })

    # Extract the content from research results
    research_content = research_results['messages'][-1].content

    # Writing phase using research
    final_output = writer_agent.invoke({
        'messages': [{
            'role': 'user',
            'content': f'Write an article based on this research: {research_content}'
        }]
    })

    return final_output

if __name__ == "__main__":
    result = coordinate_agents()
    print("Final output:", result['messages'][-1].content)