+10
-9
firehose_classifier.py
+10
-9
firehose_classifier.py
···
90
90
91
91
try:
92
92
post_view = await self._get_post(evt.did, op.rkey)
93
+
94
+
if post_view.labels is not None:
95
+
for label in post_view.labels:
96
+
if label.val in ["sexual", "nudity", "porn"]:
97
+
logger.info("skipping bsky labeled post")
98
+
return
93
99
except Exception as e:
94
100
logger.error(f"Failed to get post {e}")
95
-
raise
96
-
97
-
if post_view.labels is not None:
98
-
for label in post_view.labels:
99
-
if label.val in ["sexual", "nudity", "porn"]:
100
-
logger.info("skipping bsky labeled post")
101
-
return
102
101
103
102
final_label = ""
104
103
final_pred = 0.0
···
149
148
self, did: str, rkey: str
150
149
) -> atproto.models.AppBskyFeedDefs.PostView:
151
150
at_uri = f"at://{did}/app.bsky.feed.post/{rkey}"
152
-
url = f"https://public.api.bsky.app/xrpc/app.bsky.feed.getPosts?uris={at_uri}"
151
+
url = f"{CONFIG.ozone_pds_url}/xrpc/app.bsky.feed.getPosts?uris={at_uri}"
152
+
153
+
headers = await self.ozone_client.get_auth_headers()
153
154
154
155
try:
155
-
resp = await self.http_client.get(url)
156
+
resp = await self.http_client.get(url, headers=headers)
156
157
resp.raise_for_status()
157
158
except Exception as e:
158
159
logger.error(f"Failed to fetch post from API: {e}")