KaiserShultz's picture
Update src/config.py
d2fda9c verified
raw
history blame
1.2 kB
from langchain_openai import ChatOpenAI
from src.tools.tools import *
from src.tools.code_interpreter import safe_code_run
from langgraph.prebuilt import ToolNode
from src.schemas import PlannerPlan
from src.utils.utils import log_stage
from langchain_core.messages import HumanMessage, SystemMessage, AIMessage, ToolMessage
from dotenv import load_dotenv
load_dotenv()
config = {"configurable": {"thread_id": "1"}, "recursion_limit" : 50}
TOOLS = [download_file_from_url, web_search,
arxiv_search, wiki_search, add, subtract, multiply, divide,
power, analyze_excel_file, analyze_csv_file, analyze_docx_file,
analyze_pdf_file, analyze_txt_file,
vision_qa_gemma, safe_code_run]
TOOL_NODE = ToolNode(TOOLS)
DEBUGGING_TOOL_NODE = TOOL_NODE
llm = ChatOpenAI(model="gpt-4o-mini", temperature=0.7) #default 0.25
llm_deterministic = ChatOpenAI(model="gpt-5-mini", temperature=0.05)
planner_llm = ChatOpenAI(model="gpt-4o-mini", temperature=0.1).with_structured_output(PlannerPlan)
llm_criticist = ChatOpenAI(model="gpt-5-mini", temperature=0.3)
llm_with_tools = llm_deterministic.bind_tools(TOOLS)
llm_reasoning = ChatOpenAI(model="gpt-5-mini", temperature=0.3)