Filter out cohere's pydantic warning (#1081)

griptape-ai · Aug 19, 2024 · ade80a5 · ade80a5
1 parent 49495dd
commit ade80a5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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