mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-09 06:53:59 +00:00
community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463)
Moved the following modules to new package langchain-community in a backwards compatible fashion: ``` mv langchain/langchain/adapters community/langchain_community mv langchain/langchain/callbacks community/langchain_community/callbacks mv langchain/langchain/chat_loaders community/langchain_community mv langchain/langchain/chat_models community/langchain_community mv langchain/langchain/document_loaders community/langchain_community mv langchain/langchain/docstore community/langchain_community mv langchain/langchain/document_transformers community/langchain_community mv langchain/langchain/embeddings community/langchain_community mv langchain/langchain/graphs community/langchain_community mv langchain/langchain/llms community/langchain_community mv langchain/langchain/memory/chat_message_histories community/langchain_community mv langchain/langchain/retrievers community/langchain_community mv langchain/langchain/storage community/langchain_community mv langchain/langchain/tools community/langchain_community mv langchain/langchain/utilities community/langchain_community mv langchain/langchain/vectorstores community/langchain_community mv langchain/langchain/agents/agent_toolkits community/langchain_community mv langchain/langchain/cache.py community/langchain_community mv langchain/langchain/adapters community/langchain_community mv langchain/langchain/callbacks community/langchain_community/callbacks mv langchain/langchain/chat_loaders community/langchain_community mv langchain/langchain/chat_models community/langchain_community mv langchain/langchain/document_loaders community/langchain_community mv langchain/langchain/docstore community/langchain_community mv langchain/langchain/document_transformers community/langchain_community mv langchain/langchain/embeddings community/langchain_community mv langchain/langchain/graphs community/langchain_community mv langchain/langchain/llms community/langchain_community mv langchain/langchain/memory/chat_message_histories community/langchain_community mv langchain/langchain/retrievers community/langchain_community mv langchain/langchain/storage community/langchain_community mv langchain/langchain/tools community/langchain_community mv langchain/langchain/utilities community/langchain_community mv langchain/langchain/vectorstores community/langchain_community mv langchain/langchain/agents/agent_toolkits community/langchain_community mv langchain/langchain/cache.py community/langchain_community ``` Moved the following to core ``` mv langchain/langchain/utils/json_schema.py core/langchain_core/utils mv langchain/langchain/utils/html.py core/langchain_core/utils mv langchain/langchain/utils/strings.py core/langchain_core/utils cat langchain/langchain/utils/env.py >> core/langchain_core/utils/env.py rm langchain/langchain/utils/env.py ``` See .scripts/community_split/script_integrations.sh for all changes
This commit is contained in:
143
libs/community/langchain_community/document_loaders/reddit.py
Normal file
143
libs/community/langchain_community/document_loaders/reddit.py
Normal file
@@ -0,0 +1,143 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import TYPE_CHECKING, Iterable, List, Optional, Sequence
|
||||
|
||||
from langchain_core.documents import Document
|
||||
|
||||
from langchain_community.document_loaders.base import BaseLoader
|
||||
|
||||
if TYPE_CHECKING:
|
||||
import praw
|
||||
|
||||
|
||||
def _dependable_praw_import() -> praw:
|
||||
try:
|
||||
import praw
|
||||
except ImportError:
|
||||
raise ImportError(
|
||||
"praw package not found, please install it with `pip install praw`"
|
||||
)
|
||||
return praw
|
||||
|
||||
|
||||
class RedditPostsLoader(BaseLoader):
|
||||
"""Load `Reddit` posts.
|
||||
|
||||
Read posts on a subreddit.
|
||||
First, you need to go to
|
||||
https://www.reddit.com/prefs/apps/
|
||||
and create your application
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
client_id: str,
|
||||
client_secret: str,
|
||||
user_agent: str,
|
||||
search_queries: Sequence[str],
|
||||
mode: str,
|
||||
categories: Sequence[str] = ["new"],
|
||||
number_posts: Optional[int] = 10,
|
||||
):
|
||||
"""
|
||||
Initialize with client_id, client_secret, user_agent, search_queries, mode,
|
||||
categories, number_posts.
|
||||
Example: https://www.reddit.com/r/learnpython/
|
||||
|
||||
Args:
|
||||
client_id: Reddit client id.
|
||||
client_secret: Reddit client secret.
|
||||
user_agent: Reddit user agent.
|
||||
search_queries: The search queries.
|
||||
mode: The mode.
|
||||
categories: The categories. Default: ["new"]
|
||||
number_posts: The number of posts. Default: 10
|
||||
"""
|
||||
self.client_id = client_id
|
||||
self.client_secret = client_secret
|
||||
self.user_agent = user_agent
|
||||
self.search_queries = search_queries
|
||||
self.mode = mode
|
||||
self.categories = categories
|
||||
self.number_posts = number_posts
|
||||
|
||||
def load(self) -> List[Document]:
|
||||
"""Load reddits."""
|
||||
praw = _dependable_praw_import()
|
||||
|
||||
reddit = praw.Reddit(
|
||||
client_id=self.client_id,
|
||||
client_secret=self.client_secret,
|
||||
user_agent=self.user_agent,
|
||||
)
|
||||
|
||||
results: List[Document] = []
|
||||
|
||||
if self.mode == "subreddit":
|
||||
for search_query in self.search_queries:
|
||||
for category in self.categories:
|
||||
docs = self._subreddit_posts_loader(
|
||||
search_query=search_query, category=category, reddit=reddit
|
||||
)
|
||||
results.extend(docs)
|
||||
|
||||
elif self.mode == "username":
|
||||
for search_query in self.search_queries:
|
||||
for category in self.categories:
|
||||
docs = self._user_posts_loader(
|
||||
search_query=search_query, category=category, reddit=reddit
|
||||
)
|
||||
results.extend(docs)
|
||||
|
||||
else:
|
||||
raise ValueError(
|
||||
"mode not correct, please enter 'username' or 'subreddit' as mode"
|
||||
)
|
||||
|
||||
return results
|
||||
|
||||
def _subreddit_posts_loader(
|
||||
self, search_query: str, category: str, reddit: praw.reddit.Reddit
|
||||
) -> Iterable[Document]:
|
||||
subreddit = reddit.subreddit(search_query)
|
||||
method = getattr(subreddit, category)
|
||||
cat_posts = method(limit=self.number_posts)
|
||||
|
||||
"""Format reddit posts into a string."""
|
||||
for post in cat_posts:
|
||||
metadata = {
|
||||
"post_subreddit": post.subreddit_name_prefixed,
|
||||
"post_category": category,
|
||||
"post_title": post.title,
|
||||
"post_score": post.score,
|
||||
"post_id": post.id,
|
||||
"post_url": post.url,
|
||||
"post_author": post.author,
|
||||
}
|
||||
yield Document(
|
||||
page_content=post.selftext,
|
||||
metadata=metadata,
|
||||
)
|
||||
|
||||
def _user_posts_loader(
|
||||
self, search_query: str, category: str, reddit: praw.reddit.Reddit
|
||||
) -> Iterable[Document]:
|
||||
user = reddit.redditor(search_query)
|
||||
method = getattr(user.submissions, category)
|
||||
cat_posts = method(limit=self.number_posts)
|
||||
|
||||
"""Format reddit posts into a string."""
|
||||
for post in cat_posts:
|
||||
metadata = {
|
||||
"post_subreddit": post.subreddit_name_prefixed,
|
||||
"post_category": category,
|
||||
"post_title": post.title,
|
||||
"post_score": post.score,
|
||||
"post_id": post.id,
|
||||
"post_url": post.url,
|
||||
"post_author": post.author,
|
||||
}
|
||||
yield Document(
|
||||
page_content=post.selftext,
|
||||
metadata=metadata,
|
||||
)
|
Reference in New Issue
Block a user