Quickstart

from langchain.agents import create_agent
from langchain_openai import ChatOpenAI
from arcbeam_connector.langchain.connector import ArcbeamLangConnector

# Initialize the Arcbeam connector
connector = ArcbeamLangConnector(
    base_url="http://platform.arcbeam.ai",  # Or your self-hosted URL
    api_key="your-api-key-here",  # Your Arcbeam API key
)
connector.init()

# Create your model
model = ChatOpenAI(model="gpt-4o-mini")

# Define a simple tool
def get_weather(city: str) -> str:
    """Get weather for a given city."""
    return f"It's always sunny in {city}!"

# Create the agent with your tools
agent = create_agent(
    model=model,
    tools=[get_weather],
    system_prompt="You are a helpful assistant",
)

# Run the agent - this will be automatically traced!
result = agent.invoke(
    {"messages": [{"role": "user", "content": "what is the weather in sf"}]},
    config={"callbacks": []}
)
print(result["messages"][-1].content)

export OPENAI_API_KEY=your-openai-key
python your_app.py