wip

gemini/sample-apps/conversational-genai-app-template/app/server.py

@@ -26,7 +26,7 @@
 from app.utils.tracing import CloudTraceLoggingSpanExporter
 from fastapi import FastAPI
 from fastapi.responses import RedirectResponse, StreamingResponse
-from google.cloud import logging as gcp_logging
+from google.cloud import logging as google_cloud_logging
 from traceloop.sdk import Instruments, Traceloop
 
 # Or choose one of the following pattern chains to test by uncommenting it:
@@ -48,7 +48,7 @@
 
 # Initialize FastAPI app and logging
 app = FastAPI()
-logging_client = gcp_logging.Client()
+logging_client = google_cloud_logging.Client()
 logger = logging_client.logger(__name__)
 
 # Initialize Traceloop

gemini/sample-apps/conversational-genai-app-template/app/utils/tracing.py

@@ -16,7 +16,7 @@
 import logging
 from typing import Any, Optional, Sequence
 
-from google.cloud import logging as gcp_logging
+from google.cloud import logging as google_cloud_logging
 from google.cloud import storage
 from opentelemetry.exporter.cloud_trace import CloudTraceSpanExporter
 from opentelemetry.sdk.trace import ReadableSpan
@@ -34,7 +34,7 @@ class CloudTraceLoggingSpanExporter(CloudTraceSpanExporter):
 
     def __init__(
         self,
-        logging_client: Optional[gcp_logging.Client] = None,
+        logging_client: Optional[google_cloud_logging.Client] = None,
         storage_client: Optional[storage.Client] = None,
         bucket_name: Optional[str] = None,
         debug: bool = False,
@@ -51,7 +51,7 @@ def __init__(
         """
         super().__init__(**kwargs)
         self.debug = debug
-        self.logging_client = logging_client or gcp_logging.Client(
+        self.logging_client = logging_client or google_cloud_logging.Client(
             project=self.project_id
         )
         self.logger = self.logging_client.logger(__name__)

gemini/sample-apps/conversational-genai-app-template/tests/unit/test_utils/test_tracing_exporter.py

@@ -16,7 +16,7 @@
 from unittest.mock import Mock, patch
 
 import pytest
-from google.cloud import logging as gcp_logging
+from google.cloud import logging as google_cloud_logging
 from google.cloud import storage
 from opentelemetry.sdk.trace import ReadableSpan
 
@@ -25,7 +25,7 @@
 
 @pytest.fixture
 def mock_logging_client() -> Mock:
-    return Mock(spec=gcp_logging.Client)
+    return Mock(spec=google_cloud_logging.Client)
 
 
 @pytest.fixture