레디스 및 구조 수정중

This commit is contained in:
2026-02-25 16:56:34 +09:00
parent 89f54bae55
commit ec02f27200
10 changed files with 428 additions and 131 deletions

View File

@@ -13,8 +13,11 @@ class GeminiClient:
"""Gemini AI 클라이언트 (ClaudeClient와 동일한 인터페이스)"""
def __init__(self):
self.client = genai.Client(api_key=settings.GEMINI_API_KEY)
self.model_name = settings.GEMINI_MODEL
if not settings.GEMINI_API_KEY:
raise ValueError("GEMINI_API_KEY is not set")
genai.configure(api_key=settings.GEMINI_API_KEY)
self.model = genai.GenerativeModel(settings.GEMINI_MODEL)
self.generate_config = types.GenerateContentConfig(
max_output_tokens=settings.MAX_TOKENS,
temperature=settings.TEMPERATURE,
@@ -28,10 +31,13 @@ class GeminiClient:
) -> Dict[str, Any]:
"""Gemini를 통한 분석 실행"""
try:
response = await self.client.aio.models.generate_content(
model=self.model_name,
contents=f"{system_prompt}\n\n{user_message}",
config=self.generate_config,
model_with_system_prompt = genai.GenerativeModel(
model_name=self.model.model_name,
system_instruction=system_prompt
)
response = await model_with_system_prompt.generate_content_async(
user_message,
generation_config=self.generate_config,
)
text = response.text if response.text else ""