refactor llm/embedding flow

app/gemini_client.py

@@ -13,12 +13,16 @@ class GeminiClient:
     def generate_text(self, prompt: str, **kwargs) -> str:
         try:
             response = self._model.generate_content(prompt, **kwargs)
-            logger.info(f"[GEMINI][RAW_RESPONSE] {response}")
+            if hasattr(response, 'usage_metadata'):
+                logger.info(f"[GEMINI][USAGE] Prompt Token Count: {response.usage_metadata.prompt_token_count} - Candidate Token Count: {response.usage_metadata.candidates_token_count} - Total Token Count: {response.usage_metadata.total_token_count}")
             # response có thể là GenerativeModelResponse, lấy text hoặc trả về str
             if hasattr(response, 'text'):
+                logger.info(f"[GEMINI][TEXT_RESPONSE] {response.text}")
                 return response.text
             elif hasattr(response, 'candidates') and response.candidates:
+                logger.info(f"[GEMINI][CANDIDATES_RESPONSE] {response.candidates[0].content.parts[0].text}")
                 return response.candidates[0].content.parts[0].text
+            logger.info(f"[GEMINI][RAW_RESPONSE] {response}")
             return str(response)
         except Exception as e:
             logger.error(f"[GEMINI] Error: {e}")
@@ -38,7 +42,7 @@ class GeminiClient:
                 content=text,
                 task_type="retrieval_query"
             )
-            logger.info(f"[GEMINI][EMBEDDING][RAW_RESPONSE] {response}")
+            logger.info(f"[GEMINI][EMBEDDING][RAW_RESPONSE] {response['embedding'][:10]} ..... {response['embedding'][-10:]}")
             return response['embedding']
         except Exception as e:
             logger.error(f"[GEMINI][EMBEDDING] Error: {e}")

app/sheets.py

@@ -146,7 +146,7 @@ class SheetsClient:
                 range=SHEET_RANGE
             ).execute()
             values = result.get('values', [])
-            logger.info(f"[DEBUG] Gsheet values {values}")
+            # logger.info(f"[DEBUG] Gsheet values {values}")
             ts = datetime.now().isoformat()
             # Đảm bảo timestamp luôn là list
             if timestamp is None: