this repo has no description

use pds for fetching post

+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}")
+1 -1
ozone_client.py
··· 106 106 ): 107 107 await self._refresh_session() 108 108 109 - async def _get_auth_headers(self) -> Dict[str, str]: 109 + async def get_auth_headers(self) -> Dict[str, str]: 110 110 await self._ensure_valid_session() 111 111 112 112 return {