KeyError: ~TContext while importing

[handrew]

This is a bit of a funky one.

On version 0.0.18 only (not 0.0.17 or before), I get the following trace when one of my FastAPI server files tries to import function_tool.

I cannot reproduce this on my local machine, so it is a bit odd

    from agents import function_tool
  File "/usr/local/lib/python3.11/site-packages/agents/__init__.py", line 8, in <module>
    from .agent import Agent, ToolsToFinalOutputFunction, ToolsToFinalOutputResult
  File "/usr/local/lib/python3.11/site-packages/agents/agent.py", line 18, in <module>
    from .mcp import MCPUtil
  File "/usr/local/lib/python3.11/site-packages/agents/mcp/__init__.py", line 14, in <module>
    from .util import MCPUtil
  File "/usr/local/lib/python3.11/site-packages/agents/mcp/util.py", line 11, in <module>
    from ..tool import FunctionTool, Tool
  File "/usr/local/lib/python3.11/site-packages/agents/tool.py", line 40, in <module>
    ToolFunctionWithToolContext[ToolParams],
    ~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^
  File "/usr/local/lib/python3.11/typing.py", line 360, in inner
    return cached(*args, **kwds)
           ^^^^^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.11/typing.py", line 1391, in __getitem__
    new_args = self._determine_new_args(args)
               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.11/typing.py", line 1440, in _determine_new_args
    new_arg = old_arg[tuple(subargs)]
              ~~~~~~~^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.11/typing.py", line 360, in inner
    return cached(*args, **kwds)
           ^^^^^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.11/typing.py", line 1391, in __getitem__
    new_args = self._determine_new_args(args)
               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.11/typing.py", line 1439, in _determine_new_args
    subargs.append(new_arg_by_param[x])
                   ~~~~~~~~~~~~~~~~^^^
KeyError: ~TContext

[handrew]

Tried python 3.12 on the server container. Worked.

[Navashub]

getting same error after trying to import Agent

---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
Cell In[3], [line 4](vscode-notebook-cell:?execution_count=3&line=4)
      1 # The imports
      3 from dotenv import load_dotenv
----> [4](vscode-notebook-cell:?execution_count=3&line=4) from agents import Agent, Runner

File d:\AI-Agents\deep_research_agent\researchenv\Lib\site-packages\agents\__init__.py:8
      5 from openai import AsyncOpenAI
      7 from . import _config
----> [8](file:///D:/AI-Agents/deep_research_agent/researchenv/Lib/site-packages/agents/__init__.py:8) from .agent import Agent, ToolsToFinalOutputFunction, ToolsToFinalOutputResult
      9 from .agent_output import AgentOutputSchema, AgentOutputSchemaBase
     10 from .computer import AsyncComputer, Button, Computer, Environment

File d:\AI-Agents\deep_research_agent\researchenv\Lib\site-packages\agents\agent.py:18
     16 from .items import ItemHelpers
     17 from .logger import logger
---> [18](file:///D:/AI-Agents/deep_research_agent/researchenv/Lib/site-packages/agents/agent.py:18) from .mcp import MCPUtil
     19 from .model_settings import ModelSettings
     20 from .models.interface import Model

File d:\AI-Agents\deep_research_agent\researchenv\Lib\site-packages\agents\mcp\__init__.py:14
     11 except ImportError:
     12     pass
---> [14](file:///D:/AI-Agents/deep_research_agent/researchenv/Lib/site-packages/agents/mcp/__init__.py:14) from .util import MCPUtil
...
   1444     #   C = Callable[[int], str]
   (...)   1451     # ...we need to be careful; `new_args` should end up as
   1452     # `(int, str, float)` rather than `([int, str], float)`.

KeyError: ~TContext

[daavoo]

I am also encountering the issue.
It was introduced by #766

[handrew]

I am also encountering the issue. It was introduced by #766

@daavoo do you have a sense of which line introduced the issue? Could make it easier for the devs to fix it.