Vaibhav Prajapati
Vaibhav Prajapati

Reputation: 1

Failed to use model_dump to serialize <class 'langchain_community.tools.tavily_search.tool.TavilySearchResults'>

Failed to use model_dump to serialize <class 'langchain_community.tools.tavily_search.tool.TavilySearchResults'> to JSON: PydanticSerializationError(Unable to serialize unknown type: <class 'pydantic._internal._model_construction.ModelMetaclass'>)

I am working on integrating LangChain tools, specifically using ArxivQueryRun from the langchain_community.tools package. However, I am encountering the following errors when I try to bind tools to my LLM:

BadRequestError: Error code: 400 - {'error': {'message': "'tools.0.type' : property 'type' is missing", 'type': 'invalid_request_error'}}

Here is the relevant code snippet:

from langchain_community.utilities import ArxivAPIWrapper
from langchain_community.tools import ArxivQueryRun

# Initialize the Arxiv wrapper and tool
arxiv_wrapper = ArxivAPIWrapper(top_k_results=1, doc_content_chars_max=300)
arxiv_tool = ArxivQueryRun(api_wrapper=arxiv_wrapper)

tools = [tavily_search]  # Replace `tavily_search` with `arxiv_tool`
llm_with_tools = llm.bind(tools=tools)

Here is my graph code:

arcus_builder = StateGraph(ARCUSState)
arcus_builder.add_node("generate_co_info", generate_co_info)
arcus_builder.add_node("search_web", search_web)
arcus_builder.add_node("search_wiki", search_wiki)
arcus_builder.add_node("save_result", save_result)
arcus_builder.add_node("write_research", write_research)
arcus_builder.add_node("use_case_generator", use_case_generator)
arcus_builder.add_node("resource_collector", resource_collector)
tool_node = ToolNode(tools=tools)
arcus_builder.add_node("tools", tool_node)
arcus_builder.add_node("write_report", write_report)

arcus_builder.add_edge(START, "generate_co_info")
arcus_builder.add_edge("generate_co_info", "search_web")
arcus_builder.add_edge("generate_co_info", "search_wiki")
arcus_builder.add_edge("search_web", "save_result")
arcus_builder.add_edge("search_wiki", "save_result")
arcus_builder.add_edge("save_result", "write_research")
arcus_builder.add_edge("write_research", "use_case_generator")
arcus_builder.add_edge("use_case_generator", "resource_collector")
arcus_builder.add_conditional_edges(
    "resource_collector",
    tools_condition_custom,
    {
        "tools": "tools",
        "write_report": "write_report",
    }
)
arcus_builder.add_edge("tools", "resource_collector")
arcus_builder.add_edge("write_report", END)


memory = MemorySaver()
arcus_graph = arcus_builder.compile(checkpointer=memory)

display(Image(arcus_graph.get_graph(xray=1).draw_mermaid_png()))

Here is how I am calling:

company_name = "Tesla"
messages = [HumanMessage(f" Company Name: {company_name}")]
thread = {"configurable": {"thread_id": "1"}}
research = arcus_graph.invoke({"messages": messages}, thread)

Without tools its running fine, I am expecting tools to work.

Upvotes: 0

Views: 79

Answers (1)

Vaibhav Prajapati
Vaibhav Prajapati

Reputation: 1

It should be

llm_with_tools = llm.bind_tools(tools=tools)

Upvotes: 0

Related Questions