Add CardGeneratorError class

Author: kinto-b

karten/card.py

@@ -6,7 +6,7 @@
 
 
 class CardError(Exception):
-    """Card could not be created"""
+    """Card parsing failed"""
 
 
 class Card(BaseModel):

karten/card_generator.py

@@ -5,16 +5,20 @@
 from google import genai
 from google.genai import types
 
-from .card import Card, CardError
+from .card import Card
 from .prompt import CardPrompt
 
 
+class CardGeneratorError(Exception):
+    """Card generation failed"""
+
+
 class CardGenerator:
     """Generates flashcards using Google GenAI"""
 
     def __init__(self, api_key: str, model_name: str):
         if not api_key:
-            raise CardError("API key must be provided")
+            raise CardGeneratorError("API key must be provided")
         self.client = genai.Client(api_key=api_key)
         self.model_name = model_name
         self.config = types.GenerateContentConfig(
@@ -34,7 +38,7 @@ def generate_content(self, prompt: str) -> str:
             config=self.config,
         )
         if response.text is None:
-            raise CardError("No response text from model")
+            raise CardGeneratorError("No response text from model")
         return response.text
 
     def collect(self, word: str, lang: str) -> Card:

karten/cli.py

@@ -7,7 +7,7 @@
 
 from . import __version__
 from .card import CardError
-from .card_generator import CardGenerator
+from .card_generator import CardGenerator, CardGeneratorError
 from .cli_options import (
     option_date_from,
     option_file,
@@ -36,8 +36,8 @@ def card(word, lang, key, model):
         generator = CardGenerator(api_key=key, model_name=model)
         card = generator.collect(word, lang)  # pylint: disable=redefined-outer-name
         click.echo(card.model_dump_json(indent=2, exclude_none=True))
-    except CardError as e:
-        click.echo(e)
+    except (CardError, CardGeneratorError) as e:
+        click.echo(f"Error: {e}")
 
 
 @cli.command()
@@ -76,7 +76,7 @@ def _create_deck(
 
     try:
         generator = CardGenerator(api_key=key, model_name=model_name)
-    except CardError as e:
+    except CardGeneratorError as e:
         click.echo(f"Error: {e}")
         return