add logging

This commit is contained in:
Matteo Rosati
2026-01-29 14:20:01 +01:00
parent 20919298c8
commit ead140a2f4

16
lib.py
View File

@@ -4,6 +4,8 @@ This module provides functionality to generate content using Google's Gemini mod
with Vertex AI RAG (Retrieval-Augmented Generation) support. with Vertex AI RAG (Retrieval-Augmented Generation) support.
""" """
from llm_config import generate_content_config
import logging
import asyncio import asyncio
import json import json
import os import os
@@ -13,7 +15,8 @@ from google.genai import types
from google.oauth2 import service_account from google.oauth2 import service_account
from dotenv import load_dotenv from dotenv import load_dotenv
from llm_config import generate_content_config logger = logging.getLogger(__name__)
# Load environment variables from .env file # Load environment variables from .env file
load_dotenv() load_dotenv()
@@ -31,9 +34,14 @@ def get_credentials():
""" """
# Try to load credentials from JSON content directly # Try to load credentials from JSON content directly
credentials_json = os.getenv("GOOGLE_CREDENTIALS_JSON") credentials_json = os.getenv("GOOGLE_CREDENTIALS_JSON")
logger.info(f"creds JSON: {credentials_json}")
if credentials_json: if credentials_json:
try: try:
credentials_info = json.loads(credentials_json) credentials_info = json.loads(credentials_json)
logger.info(f"creds JSON parsed: {credentials_info}")
return service_account.Credentials.from_service_account_info( return service_account.Credentials.from_service_account_info(
credentials_info credentials_info
) )
@@ -42,6 +50,9 @@ def get_credentials():
# Fall back to file-based credentials (standard behavior) # Fall back to file-based credentials (standard behavior)
credentials_path = os.getenv("GOOGLE_APPLICATION_CREDENTIALS") credentials_path = os.getenv("GOOGLE_APPLICATION_CREDENTIALS")
logger.info(f"creds path: {credentials_path}")
if credentials_path and os.path.exists(credentials_path): if credentials_path and os.path.exists(credentials_path):
return service_account.Credentials.from_service_account_file( return service_account.Credentials.from_service_account_file(
credentials_path credentials_path
@@ -79,6 +90,9 @@ async def generate(prompt: str):
"""Run the synchronous streaming in a separate thread.""" """Run the synchronous streaming in a separate thread."""
try: try:
credentials = get_credentials() credentials = get_credentials()
logger.info(f"credentials: {credentials}")
client = genai.Client(vertexai=True, credentials=credentials) client = genai.Client(vertexai=True, credentials=credentials)
contents = [ contents = [