Do not issue beta or deprecation warnings on internal calls (#15641)

This commit is contained in:
Nuno Campos
2024-01-07 20:54:45 -08:00
committed by GitHub
parent ef22559f1f
commit 7ce4cd0709
12 changed files with 130 additions and 120 deletions

View File

@@ -4,7 +4,7 @@ import logging
from typing import TYPE_CHECKING, Dict, Iterable, Iterator, List, Optional, Union, cast
from langchain_core.chat_sessions import ChatSession
from langchain_core.load.load import _load_suppress_warning
from langchain_core.load.load import load
from langchain_community.chat_loaders.base import BaseChatLoader
@@ -66,10 +66,8 @@ class LangSmithRunChatLoader(BaseChatLoader):
raise ValueError(f"Run has no 'messages' inputs. Got {llm_run.inputs}")
if not llm_run.outputs:
raise ValueError("Cannot convert pending run")
messages = _load_suppress_warning(llm_run.inputs)["messages"]
message_chunk = _load_suppress_warning(llm_run.outputs)["generations"][0][
"message"
]
messages = load(llm_run.inputs)["messages"]
message_chunk = load(llm_run.outputs)["generations"][0]["message"]
return ChatSession(messages=messages + [message_chunk])
@staticmethod