import lucidicai as lai
from langchain_openai import ChatOpenAI
from langchain.agents import create_react_agent, AgentExecutor
from langchain.tools import Tool
from langchain_core.prompts import PromptTemplate
lai.init(
session_name="langchain_agent",
providers=["langchain"]
)
# Define tools
tools = [
Tool(
name="Calculator",
func=lambda x: eval(x),
description="Useful for math calculations"
)
]
# Create agent
llm = ChatOpenAI(model="gpt-4")
prompt = PromptTemplate.from_template("You are a helpful assistant. {input}")
agent = create_react_agent(llm, tools, prompt)
executor = AgentExecutor(agent=agent, tools=tools)
lai.create_step(state="Ready", goal="Solve math problem")
# All LLM calls within the agent are automatically tracked!
result = executor.invoke({"input": "What is 25 * 4 + 10?"})
lai.end_step()