github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-11-02 09:14:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(document_loaders/telegram): fix pandas calls + add tests (#4806)

Browse Source 
# 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>
This commit is contained in:
Raduan Al-Shedivat
2023-05-16 23:35:25 +02:00
committed by GitHub
parent 206c87d525
commit 00c6ec8a2d
30 changed files with 137 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/unit_tests/document_loaders/test_web_base.py Normal file
View File

@@ -0,0 +1,10 @@
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