From ade80a500df7c90a8efa0ec89879fa62521eb2d9 Mon Sep 17 00:00:00 2001 From: Collin Dutter Date: Mon, 19 Aug 2024 15:29:14 -0700 Subject: [PATCH] Filter out cohere's pydantic warning (#1081) --- griptape/drivers/prompt/cohere_prompt_driver.py | 4 ++++ 1 file changed, 4 insertions(+) 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