fix(sample-app): Update google genai package (#3358)

Author: nina-kollman

packages/sample-app/sample_app/gemini.py

@@ -1,21 +1,21 @@
 import os
 import asyncio
-import google.generativeai as genai
+import google.genai as genai
 from traceloop.sdk import Traceloop
 from traceloop.sdk.decorators import workflow
 
 Traceloop.init(app_name="gemini_example")
 
-genai.configure(api_key=os.environ.get("GENAI_API_KEY"))
+client = genai.Client(api_key=os.environ.get("GENAI_API_KEY"))
 
 
 @workflow("predict")
 def predict_text() -> str:
     """Ideation example with a Large Language Model"""
 
-    model = genai.GenerativeModel("gemini-1.5-pro-002")
-    response = model.generate_content(
-        "Give me ten interview questions for the role of program manager.",
+    response = client.models.generate_content(
+        model="gemini-1.5-pro-002",
+        contents="Give me ten interview questions for the role of program manager.",
     )
 
     return response.text
@@ -25,24 +25,37 @@ def predict_text() -> str:
 async def async_predict_text() -> str:
     """Async Ideation example with a Large Language Model"""
 
-    model = genai.GenerativeModel("gemini-1.5-pro-002")
-    response = await model.generate_content_async(
-        "Give me ten interview questions for the role of program manager.",
+    response = client.models.generate_content(
+        model="gemini-1.5-pro-002",
+        contents="Give me ten interview questions for the role of program manager.",
     )
 
     return response.text
 
 
 @workflow("chat")
 def chat() -> str:
-    """Chat example with a Large Language Model"""
+    """Real chat example with conversation context"""
 
-    model = genai.GenerativeModel("gemini-1.5-pro-002")
-    chat = model.start_chat()
-    response = chat.send_message("Hello, how are you?")
-    response = chat.send_message("What is the capital of France?")
+    # First message
+    response1 = client.models.generate_content(
+        model="gemini-1.5-pro-002",
+        contents="Hello, how are you?",
+    )
 
-    return response.text
+    # Second message with conversation history
+    conversation = [
+        {"role": "user", "parts": [{"text": "Hello, how are you?"}]},
+        {"role": "model", "parts": [{"text": response1.text}]},
+        {"role": "user", "parts": [{"text": "What is the capital of France?"}]},
+    ]
+
+    response2 = client.models.generate_content(
+        model="gemini-1.5-pro-002",
+        contents=conversation,
+    )
+
+    return response2.text
 
 
 if __name__ == "__main__":