레디스 및 구조 수정중
This commit is contained in:
@@ -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 ""
|
||||
|
||||
Reference in New Issue
Block a user