feat: add Google Gemini API support (#1418)

* feat: add Google Gemini API support

Co-Authored-By: Junyan Qin <Chin> <rockchinq@gmail.com>

* fix: remove unused imports

Co-Authored-By: Junyan Qin <Chin> <rockchinq@gmail.com>

* feat: add google-genai dependency

Co-Authored-By: Junyan Qin <Chin> <rockchinq@gmail.com>

* fix: update Gemini API implementation to use correct API methods

Co-Authored-By: Junyan Qin <Chin> <rockchinq@gmail.com>

* refactor: improve Gemini API implementation based on official documentation

Co-Authored-By: Junyan Qin <Chin> <rockchinq@gmail.com>

* fix: remove unsupported timeout parameter from Gemini API implementation

Co-Authored-By: Junyan Qin <Chin> <rockchinq@gmail.com>

* fix: correct Gemini API implementation based on official documentation

Co-Authored-By: Junyan Qin <Chin> <rockchinq@gmail.com>

* feat: update geminichatcmpl

* deps: add google-generativeai

---------

Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Co-authored-by: Junyan Qin <Chin> <rockchinq@gmail.com>

pkg/provider/modelmgr/requesters/gemini.svg

@@ -0,0 +1 @@
+<svg height="1em" style="flex:none;line-height:1" viewBox="0 0 24 24" width="1em" xmlns="http://www.w3.org/2000/svg"><title>Gemini</title><defs><linearGradient id="lobe-icons-gemini-fill" x1="0%" x2="68.73%" y1="100%" y2="30.395%"><stop offset="0%" stop-color="#1C7DFF"></stop><stop offset="52.021%" stop-color="#1C69FF"></stop><stop offset="100%" stop-color="#F0DCD6"></stop></linearGradient></defs><path d="M12 24A14.304 14.304 0 000 12 14.304 14.304 0 0012 0a14.305 14.305 0 0012 12 14.305 14.305 0 00-12 12" fill="url(#lobe-icons-gemini-fill)" fill-rule="nonzero"></path></svg>

pkg/provider/modelmgr/requesters/geminichatcmpl.py

@@ -0,0 +1,87 @@
+from __future__ import annotations
+
+import typing
+import google.genai
+from google.genai import types
+
+from .. import errors, requester
+from ....core import entities as core_entities
+from ... import entities as llm_entities
+from ...tools import entities as tools_entities
+
+
+class GeminiChatCompletions(requester.LLMAPIRequester):
+    """Google Gemini API 请求器"""
+
+    default_config: dict[str, typing.Any] = {
+        'base_url': 'https://generativelanguage.googleapis.com',
+        'timeout': 120,
+    }
+
+    async def initialize(self):
+        """初始化 Gemini API 客户端"""
+        pass
+
+    async def invoke_llm(
+        self,
+        query: core_entities.Query,
+        model: requester.RuntimeLLMModel,
+        messages: typing.List[llm_entities.Message],
+        funcs: typing.List[tools_entities.LLMFunction] = None,
+        extra_args: dict[str, typing.Any] = {},
+    ) -> llm_entities.Message:
+        """调用 Gemini API 生成回复"""
+        try:
+            self.client = google.genai.Client(
+                api_key=model.token_mgr.get_token(),
+                http_options=types.HttpOptions(api_version='v1alpha'),
+            )
+            contents = []
+
+            system_content = None
+
+            for message in messages:
+                role = message.role
+                parts = []
+
+                if isinstance(message.content, str):
+                    parts.append(types.Part.from_text(text=message.content))
+                elif isinstance(message.content, list):
+                    for content in message.content:
+                        if content.type == 'text':
+                            parts.append(types.Part.from_text(text=content.text))
+                        # elif content.type == 'image_url':
+                        #     parts.append(types.Part.from_image_url(url=content.image_url))
+
+                if role == 'system':
+                    system_content = parts
+                else:
+                    content = types.Content(role=role, parts=parts)
+                    contents.append(content)
+
+            response = self.client.models.generate_content(
+                model=model.model_entity.name,
+                contents=contents,
+                config=types.GenerateContentConfig(
+                    system_instruction=system_content,
+                    **extra_args,
+                ),
+            )
+
+            return llm_entities.Message(
+                role='assistant',
+                content=response.candidates[0].content.parts[0].text,
+            )
+
+        except Exception as e:
+            error_message = str(e).lower()
+            if 'invalid api key' in error_message:
+                raise errors.RequesterError(f'无效的 API 密钥: {str(e)}')
+            elif 'not found' in error_message:
+                raise errors.RequesterError(f'请求路径错误或模型无效: {str(e)}')
+            elif any(keyword in error_message for keyword in ['rate limit', 'quota', 'permission denied']):
+                raise errors.RequesterError(f'请求过于频繁或余额不足: {str(e)}')
+            elif 'timeout' in error_message:
+                raise errors.RequesterError(f'请求超时: {str(e)}')
+            else:
+                raise errors.RequesterError(f'Gemini API 请求错误: {str(e)}')

pkg/provider/modelmgr/requesters/geminichatcmpl.yaml

@@ -0,0 +1,28 @@
+apiVersion: v1
+kind: LLMAPIRequester
+metadata:
+  name: gemini-chat-completions
+  label:
+    en_US: Google Gemini
+    zh_CN: Google Gemini
+  icon: gemini.svg
+spec:
+  config:
+    - name: base_url
+      label:
+        en_US: Base URL
+        zh_CN: 基础 URL
+      type: string
+      required: true
+      default: "https://generativelanguage.googleapis.com"
+    - name: timeout
+      label:
+        en_US: Timeout
+        zh_CN: 超时时间
+      type: integer
+      required: true
+      default: 120
+execution:
+  python:
+    path: ./geminichatcmpl.py
+    attr: GeminiChatCompletions

requirements.txt

@@ -37,9 +37,11 @@ mcp
 sqlmodel
 slack_sdk
 telegramify-markdown
+google-genai
+google-generativeai
 
 # indirect
 taskgroup==0.0.0a4
 ruff
 pre-commit
-python-socks
+python-socks