backend.im/taskmanagerapi-ttkjqk
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

initial commit

Browse Source
This commit is contained in:
Agent@BackendIM 2025-05-17 08:45:19 +00:00
parent 7a291b7bba
commit d57ad1b56b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/services/llm_service.py
View File

@ -97,7 +97,7 @@ class GeminiService(LLMService):
"""Initialize the Gemini service.""" """Initialize the Gemini service."""
try: try:
import google.generativeai as genai import google.generativeai as genai
genai.configure(api_key=settings.gemini_api_key) genai.configure(api_key=settings.GEMINI_API_KEY)
self.model = genai.GenerativeModel(settings.gemini_model) self.model = genai.GenerativeModel(settings.gemini_model)
except (ImportError, AttributeError) as e: except (ImportError, AttributeError) as e:
logger.error(f"Failed to initialize Gemini service: {e}") logger.error(f"Failed to initialize Gemini service: {e}")
@ -198,9 +198,9 @@ def get_llm_service() -> LLMService:
""" """
llm_provider = settings.LLM_PROVIDER.lower() llm_provider = settings.LLM_PROVIDER.lower()
if llm_provider == "openai" and settings.openai_api_key: if llm_provider == "openai" and settings.OPEANAI_API_KEY:
return OpenAIService() return OpenAIService()
elif llm_provider == "gemini" and settings.gemini_api_key: elif llm_provider == "gemini" and settings.GEMINI_API_KEY:
return GeminiService() return GeminiService()
elif llm_provider == "mock" or settings.environment == "test": elif llm_provider == "mock" or settings.environment == "test":
# Use mock service for testing or when configured # Use mock service for testing or when configured