remove

Author: nina-kollman

packages/sample-app/sample_app/gemini.py

@@ -4,12 +4,8 @@
 from traceloop.sdk import Traceloop
 from traceloop.sdk.decorators import workflow
 
-# Initialize with console exporter for debugging
-Traceloop.init(
-    app_name="gemini_example",
-)
+Traceloop.init(app_name="gemini_example")
 
-# Configure client with API key
 client = genai.Client(api_key=os.environ.get("GENAI_API_KEY"))
 
 
@@ -40,19 +36,13 @@ async def async_predict_text() -> str:
 @workflow("chat")
 def chat() -> str:
     """Chat example with a Large Language Model"""
-
-    # For chat, we'll simulate with multiple generate_content calls
-    response1 = client.models.generate_content(
-        model="gemini-1.5-pro-002",
-        contents="Hello, how are you?",
-    )
 
-    response2 = client.models.generate_content(
+    response = client.models.generate_content(
         model="gemini-1.5-pro-002",
         contents="What is the capital of France?",
     )
 
-    return response2.text
+    return response.text
 
 
 if __name__ == "__main__":