mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-15 17:30:13 +08:00
dcaaf6286a
- Add 100+ new test files covering provider sources, platform adapters, agent runners, star/plugin system, knowledge base, core utils, pipeline, computer tools, builtin commands, and dashboard routes - Fix Python type errors in sqlite.py (col() wrappers for SQLModel), astr_agent_tool_exec, core_lifecycle, star context/manager - Fix TypeScript strict mode errors across 30+ dashboard Vue files - Add import smoke tests, auth roundtrip, startup tests - Add compile-all check and CLI entry test for AUR compatibility - Restore BotMessageAccumulator and helpers lost in merge
63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
"""Smoke tests for GeminiEmbeddingProvider."""
|
|
from unittest.mock import patch
|
|
|
|
from astrbot.core.provider.sources.gemini_embedding_source import (
|
|
GeminiEmbeddingProvider,
|
|
)
|
|
|
|
|
|
def test_gemini_embedding_import():
|
|
assert GeminiEmbeddingProvider is not None
|
|
|
|
|
|
@patch("astrbot.core.provider.sources.gemini_embedding_source.genai")
|
|
def test_gemini_embedding_construction(mock_genai):
|
|
provider = GeminiEmbeddingProvider(
|
|
provider_config={
|
|
"embedding_api_key": "test-key",
|
|
"embedding_api_base": "https://generativelanguage.googleapis.com",
|
|
"embedding_model": "gemini-embedding-exp-03-07",
|
|
},
|
|
provider_settings={},
|
|
)
|
|
assert provider is not None
|
|
assert provider.model == "gemini-embedding-exp-03-07"
|
|
mock_genai.Client.assert_called_once()
|
|
|
|
|
|
@patch("astrbot.core.provider.sources.gemini_embedding_source.genai")
|
|
def test_gemini_embedding_default_model(mock_genai):
|
|
provider = GeminiEmbeddingProvider(
|
|
provider_config={
|
|
"embedding_api_key": "test-key",
|
|
"embedding_api_base": "https://generativelanguage.googleapis.com",
|
|
},
|
|
provider_settings={},
|
|
)
|
|
assert provider.model == "gemini-embedding-exp-03-07"
|
|
|
|
|
|
@patch("astrbot.core.provider.sources.gemini_embedding_source.genai")
|
|
def test_gemini_embedding_get_dim(mock_genai):
|
|
provider = GeminiEmbeddingProvider(
|
|
provider_config={
|
|
"embedding_api_key": "test-key",
|
|
"embedding_api_base": "https://generativelanguage.googleapis.com",
|
|
"embedding_dimensions": 512,
|
|
},
|
|
provider_settings={},
|
|
)
|
|
assert provider.get_dim() == 512
|
|
|
|
|
|
@patch("astrbot.core.provider.sources.gemini_embedding_source.genai")
|
|
def test_gemini_embedding_get_dim_default(mock_genai):
|
|
provider = GeminiEmbeddingProvider(
|
|
provider_config={
|
|
"embedding_api_key": "test-key",
|
|
"embedding_api_base": "https://generativelanguage.googleapis.com",
|
|
},
|
|
provider_settings={},
|
|
)
|
|
assert provider.get_dim() == 768
|