Can't create a Langgraph supervisor agent with cohere models AttributeError: 'CohereTool' object has no attribute 'get' #116

[neeleshkshukla]

Follwoing the readme of https://github.com/langchain-ai/langgraph-supervisor-py, I am trying to create Supervisor agent with Cohere Command R Refresh model. Fails with AttributeError: 'CohereTool' object has no attribute 'get'

https://docs.oracle.com/en-us/iaas/tools/python/2.149.2/api/generative_ai_inference/models/oci.generative_ai_inference.models.CohereTool.html

Code:

from langgraph_supervisor import create_supervisor
from langgraph.prebuilt import create_react_agent


# Create supervisor workflow
workflow = create_supervisor(
    [research_agent, math_agent],
    model=model,
    prompt=(
        "You are a team supervisor managing a research expert and a math expert. "
        "For current events, use research_agent. "
        "For math problems, use math_agent."
    )
)

Error Stacktrace:

AttributeError                            Traceback (most recent call last)
Cell In[19], line 2
      1 # Create supervisor workflow
----> 2 workflow = create_supervisor(
      3     [research_agent, math_agent],
      4     model=model,
      5     prompt=(
      6         "You are a team supervisor managing a research expert and a math expert. "
      7         "For current events, use research_agent. "
      8         "For math problems, use math_agent."
      9     )
     10 )

File ~/lib/python3.13/site-packages/langgraph_supervisor/supervisor.py:211, in create_supervisor(agents, model, tools, prompt, response_format, parallel_tool_calls, state_schema, config_schema, output_mode, add_handoff_back_messages, supervisor_name, include_agent_name)
    208 if include_agent_name:
    209     model = with_agent_name(model, include_agent_name)
--> 211 supervisor_agent = create_react_agent(
    212     name=supervisor_name,
    213     model=model,
    214     tools=all_tools,
    215     prompt=prompt,
    216     state_schema=state_schema,
    217     response_format=response_format,
    218 )
    220 builder = StateGraph(state_schema, config_schema=config_schema)
    221 builder.add_node(supervisor_agent, destinations=tuple(agent_names) + (END,))

File ~/lib/python3.13/site-packages/langgraph/prebuilt/chat_agent_executor.py:154, in _convert_modifier_to_prompt.<locals>.wrapper(*args, **kwargs)
    151     prompt = state_modifier
    153 kwargs["prompt"] = prompt
--> 154 return func(*args, **kwargs)

File ~/lib/python3.13/site-packages/langgraph/prebuilt/chat_agent_executor.py:653, in create_react_agent(model, tools, prompt, response_format, state_schema, config_schema, checkpointer, store, interrupt_before, interrupt_after, debug, version, name)
    649     model = cast(BaseChatModel, init_chat_model(model))
    651 tool_calling_enabled = len(tool_classes) > 0
--> 653 if _should_bind_tools(model, tool_classes) and tool_calling_enabled:
    654     model = cast(BaseChatModel, model).bind_tools(tool_classes)
    656 model_runnable = _get_prompt_runnable(prompt) | model

File ~/lib/python3.13/site-packages/langgraph/prebuilt/chat_agent_executor.py:186, in _should_bind_tools(model, tools)
    183 bound_tool_names = set()
    184 for bound_tool in bound_tools:
    185     # OpenAI-style tool
--> 186     if bound_tool.get("type") == "function":
    187         bound_tool_name = bound_tool["function"]["name"]
    188     # Anthropic-style tool

AttributeError: 'CohereTool' object has no attribute 'get'

[adizohar]

You created issue at OCI python sdk but your code is not using the OCI SDK please close the ticket and open in the right repository

[neeleshkshukla]

I am using Using from langchain_community.chat_models import ChatOCIGenAI to create the model.

https://docs.oracle.com/en-us/iaas/tools/python/2.149.2/api/generative_ai_inference/models/oci.generative_ai_inference.models.CohereTool.html

I do see its using oci python sdk AttributeError: 'CohereTool' object has no attribute 'get'

[adizohar]

I don’t see any error that coming from the OCI SDK
Please open ticket at langgraph repository they are not using the OCI sdk properly

[adizohar]

I checked the langraph repo and could not find any reference to OCI can you please help me where did you see reference to OCI SDK ?
Please share your full code, you probably didn't use the bound_tool properly, you cannot assign OCI Object to the bound tool

[voiddrum]

I too got this error.
I am trying out this example https://github.com/langchain-ai/langgraph-supervisor-py
With ChatOCIGenAI as the model

python supervisor.py
Traceback (most recent call last):
  File supervisor.py", line 85, in <module>
    workflow = create_supervisor(
  File "~/.pyenv/versions/langgraph-supervisor/lib/python3.10/site-packages/langgraph_supervisor/supervisor.py", line 404, in create_supervisor
    supervisor_agent = create_react_agent(
  File "~/.pyenv/versions/langgraph-supervisor/lib/python3.10/site-packages/langgraph/prebuilt/chat_agent_executor.py", line 447, in create_react_agent
    _should_bind_tools(model, tool_classes)
  File "~/.pyenv/versions/langgraph-supervisor/lib/python3.10/site-packages/langgraph/prebuilt/chat_agent_executor.py", line 170, in _should_bind_tools
    if bound_tool.get("type") == "function":
AttributeError: 'CohereTool' object has no attribute 'get'

the model object is created like this

COMPARTMENT_ID = "xxx"
    AUTH_TYPE = "SECURITY_TOKEN" # The authentication type to use, e.g., API_KEY (default), SECURITY_TOKEN, INSTANCE_PRINCIPAL, RESOURCE_PRINCIPAL.
    endpoint = "https://inference.generativeai.uk-london-1.oci.oraclecloud.com"
    CONFIG_PROFILE = "xxx"

    return ChatOCIGenAI(
        model_id="ocid1.generativeaimodel.oc1.uk-london-1.xxxx",
        service_endpoint=endpoint,
        compartment_id=COMPARTMENT_ID,
        provider="cohere",
        model_kwargs={
            "temperature": 1,
            "max_tokens": 600,
            "frequency_penalty": 0,
            "presence_penalty": 0,
            "top_k": 0,
            "top_p": 0.75
        },
        auth_type=AUTH_TYPE,
        auth_profile=CONFIG_PROFILE,
        auth_file_location="xxxx")