More graceful raise for status

search.py

@@ -32,15 +32,17 @@ class GoogleSearch:
 
         async with aiohttp.ClientSession() as session:
             async with session.get(url, params=params) as response:
-                response.raise_for_status()
-                data = await response.json()
-                results = "Web Search results:\n\n" + "\n\n".join(
-                    [
-                        f"Link:{result['link']}\nTitle:{result['title']}\nSnippet:{result['snippet']}"
-                        for result in data["items"][:num_results]
-                    ]
-                )
-                return results
+                if response.status == 200:
+                    data = await response.json()
+                    results = "Web Search results:\n\n" + "\n\n".join(
+                        [
+                            f"Link:{result['link']}\nTitle:{result['title']}\nSnippet:{result['snippet']}"
+                            for result in data["items"][:num_results]
+                        ]
+                    )
+                    return results
+                else:
+                    return f"Search failed with status {response.status}"
 
     async def google_image_search(self, query: str, num_results: int = 5) -> str:
         """
@@ -64,12 +66,14 @@ class GoogleSearch:
 
         async with aiohttp.ClientSession() as session:
             async with session.get(url, params=params) as response:
-                response.raise_for_status()
-                data = await response.json()
-                results = "Web Search results:\n\n" + "\n\n".join(
-                    [
-                        f"Link:{result['link']}\nTitle:{result['title']}"
-                        for result in data["items"][:num_results]
-                    ]
-                )
-                return results
+                if response.status == 200:
+                    data = await response.json()
+                    results = "Web Search results:\n\n" + "\n\n".join(
+                        [
+                            f"Link:{result['link']}\nTitle:{result['title']}"
+                            for result in data["items"][:num_results]
+                        ]
+                    )
+                    return results
+                else:
+                    return f"Search failed with status {response.status}"