forked from Archives/langchain
00c6ec8a2d
# Fix Telegram API loader + add tests. I was testing this integration and it was broken with next error: ```python message_threads = loader._get_message_threads(df) KeyError: False ``` Also, this particular loader didn't have any tests / related group in poetry, so I added those as well. @hwchase17 / @eyurtsev please take a look on this fix PR. --------- Co-authored-by: Dev 2049 <dev.dev2049@gmail.com>
11 lines
473 B
Python
11 lines
473 B
Python
from langchain.document_loaders.web_base import WebBaseLoader
|
|
|
|
|
|
class TestWebBaseLoader:
|
|
def test_respect_user_specified_user_agent(self) -> None:
|
|
user_specified_user_agent = "user_specified_user_agent"
|
|
header_template = {"User-Agent": user_specified_user_agent}
|
|
url = "https://www.example.com"
|
|
loader = WebBaseLoader(url, header_template=header_template)
|
|
assert loader.session.headers["User-Agent"] == user_specified_user_agent
|