diff --git a/griptape/drivers/prompt/cohere_prompt_driver.py b/griptape/drivers/prompt/cohere_prompt_driver.py index ff1a8b482..05be5b7f2 100644 --- a/griptape/drivers/prompt/cohere_prompt_driver.py +++ b/griptape/drivers/prompt/cohere_prompt_driver.py @@ -1,5 +1,6 @@ from __future__ import annotations +import warnings from typing import TYPE_CHECKING, Any from attrs import Factory, define, field @@ -24,6 +25,9 @@ from griptape.tokenizers import BaseTokenizer, CohereTokenizer from griptape.utils import import_optional_dependency +# TODO Remove once https://github.com/cohere-ai/cohere-python/issues/559 is resolved +warnings.filterwarnings("ignore", module="pydantic") + if TYPE_CHECKING: from collections.abc import Iterator