community[patch]: fix deprecation warnings on openai subclasses (#15621)

This commit is contained in:
Erick Friis
2024-01-05 18:02:17 -08:00
committed by GitHub
parent 4ac61670b2
commit d136925c49
7 changed files with 41 additions and 88 deletions

View File

@@ -2,6 +2,7 @@
import datetime
from typing import Any, Dict, List, Optional
from langchain_core._api.deprecation import suppress_langchain_deprecation_warning
from langchain_core.callbacks import (
AsyncCallbackManagerForLLMRun,
CallbackManagerForLLMRun,
@@ -39,6 +40,11 @@ class PromptLayerChatOpenAI(ChatOpenAI):
pl_tags: Optional[List[str]]
return_pl_id: Optional[bool] = False
def __init__(self, *kwargs: Any) -> None:
# bypass deprecation warning for ChatOpenAI
with suppress_langchain_deprecation_warning():
super().__init__(*kwargs)
@classmethod
def is_lc_serializable(cls) -> bool:
return False