github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add creds check on client

Browse Source
This commit is contained in:
Zamil Majdy
2024-07-01 08:40:15 +07:00
committed by Aarushi
parent f99c1c8c57
commit c1fc01036a
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rnd/autogpt_server/autogpt_server/blocks/reddit.py
View File

@@ -24,13 +24,18 @@ class RedditPost(BaseModel):
def get_praw(creds: RedditCredentials) -> praw.Reddit:
return praw.Reddit(
client = praw.Reddit(
client_id=creds.client_id,
client_secret=creds.client_secret,
user_agent=creds.user_agent,
username=creds.username,
password=creds.password,
)
me = client.user.me()
if not me:
raise ValueError("Invalid Reddit credentials.")
print(f"Logged in as {me.name}")
return client
class RedditGetPostsBlock(Block):
@@ -53,7 +58,6 @@ class RedditGetPostsBlock(Block):
def run(self, input_data: Input) -> BlockOutput:
client = get_praw(input_data.creds)
subreddit = client.subreddit(input_data.subreddit)
for post in subreddit.new(limit=None):
if input_data.last_post and post.created_utc < datetime.now() - \
timedelta(minutes=input_data.last_minutes):