在Colab中安装OpenAI遇到问题

Question

你好，圣诞节假期后我在Google Colab上遇到了安装openai的问题。

我按照之前一直有效的方法进行操作（直到2023年12月中旬）：

!pip install --upgrade pip
!pip install --upgrade --quiet openai

import openai

但现在我收到以下错误信息：

ImportError                               Traceback (most recent call last)
<ipython-input-6-727253859e14> in <cell line: 4>
      2 get_ipython().system('pip install --upgrade --quiet openai')
      3 
---->4 import openai

5 frames
/usr/local/lib/python3.10/dist-packages/openai/_utils/_streams.py in 
      1 from typing import Any
----> 2 from typing_extensions import Iterator, AsyncIterator
      3 
      4 
      5 def consume_sync_iterator(iterator: Iterator[Any]) > None:

ImportError: cannot import name 'Iterator' from 'typing_extensions' (/usr/local/lib/python3.10/dist-packages/typing_extensions.py)

我还尝试了安装openai的旧版本，并对typing_extensions进行了升级和卸载重装等操作，但至今未能解决问题。

Mark B · Answer

我现在在Colab中执行以下操作，成功解决了问题：

!pip install cohere
!pip install tiktoken
!pip uninstall typing-extensions -y quiet
!pip install typing-extensions==4.5.0 --quiet
!pip install --upgrade tensorflow-probability
!pip install --upgrade --quiet openai

import openai

John La Rooy · Answer

你的问题出在笔记本内核上。有时，如果首先使用pip安装像langchain这样的包时可能会出现问题。你可以改变pip安装包的顺序，或者在pip安装后简单地重启内核即可。

!pip install openai

然后重启内核会话。

按下 Ctrl + M 键组合

之后，你就可以导入openai了。

import openai

Pascal Thivent · Answer

Matthew 的建议确实有效，但在每次想要运行Google Colab时都需要重启内核似乎较为繁琐。是否有办法通过代码直接解决这个问题呢？

我在OpenAI论坛上发现一个讨论帖，其中一位用户提出了如下建议：

引用内容：
“我在Google Colab上对试图从typing_extensions导入Iterator的文件做了如下修改：

%%writefile /usr/local/lib/python3.10/dist-packages/openai/_utils/_streams.py
from typing import Any
from typing_extensions import AsyncIterator
from typing import Iterator  # 从正确的库导入Iterator

def consume_sync_iterator(iterator: Iterator[Any]) -> None:
    for _ in iterator:
        ...

async def consume_async_iterator(iterator: AsyncIterator[Any]) -> None:
    async for _ in iterator:
        ...

对我而言，这样做是有效的。”

这个方法同样适用于我，但每次手动编辑文件也不够优雅。