agent.py

from __future__ import annotations
import os
from langchain_openai import ChatOpenAI
from langgraph.graph import StateGraph, START, END
from langchain.schema import HumanMessage, SystemMessage, AIMessage
from state import AgentState
from typing import Any, Dict, List, Optional
import json
from langgraph.prebuilt import create_react_agent

# ─────────────────────────── External tools ──────────────────────────────
from tools import (
    wikipedia_search_tool,
    arxiv_search_tool,
    audio_transcriber_tool,
    excel_tool,
    analyze_code_tool
)

# ─────────────────────────── Configuration ───────────────────────────────

MAX_TOOL_CALLS = 5

# ─────────────────────────── Helper utilities ────────────────────────────

# ─────────────────────────── Agent state ⬇ ───────────────────────────────

# ───────────────────────────── Nodes  ⬇ ───────────────────────────────────

# ------------- tool adapters -------------


# ─────────────────────────── Graph wiring ───────────────────────────────

def build_graph():
    graph = StateGraph(AgentState)

    llm = ChatOpenAI(model_name="gpt-4.1-mini", temperature=0.3)

    llm_tools = [
        wikipedia_search_tool,
        arxiv_search_tool,
        audio_transcriber_tool,
        excel_tool,
        analyze_code_tool,
    ]
    # llm = llm.bind_tools(llm_tools)
    agent = create_react_agent(llm, llm_tools)

    return agent