fix(pyright): suppress warnings

Author: dineshpinto

pyproject.toml

@@ -14,6 +14,7 @@ dependencies = [
     "requests>=2.32.3",
     "requests-unixsocket>=0.3.0",
     "structlog>=25.1.0",
+    "google-generativeai>=0.8.3",
 ]
 
 [dependency-groups]
@@ -27,9 +28,6 @@ validate = [
     "pyjwt>=2.10.1",
     "pyopenssl>=25.0.0",
 ]
-ai = [
-    "google-generativeai>=0.8.3",
-]
 onchain = [
     "web3>=7.7.0",
 ]
@@ -42,7 +40,7 @@ ui = [
 start-backend = "flare_ai_core.main:start"
 
 [tool.uv]
-default-groups = ["dev", "validate", "ai", "onchain", "ui"]
+default-groups = ["dev", "validate", "onchain", "ui"]
 
 [build-system]
 requires = ["hatchling"]

src/flare_ai_core/ai/gemini.py

@@ -73,9 +73,9 @@ def __init__(self, api_key: str, model: str, **kwargs: str) -> None:
             **kwargs (str): Additional configuration parameters including:
                 - system_instruction: Custom system prompt for the AI personality
         """
-        genai.configure(api_key=api_key)
-        self.chat: genai.ChatSession | None = None
-        self.model = genai.GenerativeModel(
+        genai.configure(api_key=api_key)  # pyright: ignore [reportPrivateImportUsage]
+        self.chat: genai.ChatSession | None = None  # pyright: ignore [reportPrivateImportUsage]
+        self.model = genai.GenerativeModel(  # pyright: ignore [reportPrivateImportUsage]
             model_name=model,
             system_instruction=kwargs.get("system_instruction", SYSTEM_INSTRUCTION),
         )
@@ -122,7 +122,7 @@ def generate(
         """
         response = self.model.generate_content(
             prompt,
-            generation_config=genai.GenerationConfig(
+            generation_config=genai.GenerationConfig(  # pyright: ignore [reportPrivateImportUsage]
                 response_mime_type=response_mime_type, response_schema=response_schema
             ),
         )