File tree Expand file tree Collapse file tree 1 file changed +1
-27
lines changed Expand file tree Collapse file tree 1 file changed +1
-27
lines changed Original file line number Diff line number Diff line change @@ -240,33 +240,7 @@ def unpatch_google_adk():
240240
241241# -------------------------------- Google Gemini --------------------------------- #
242242def trace_gemini (client ):
243- """Trace Google Gemini chat completions.
244-
245- This function patches a Google GenerativeModel client to automatically trace
246- all generate_content calls, including both streaming and non-streaming requests.
247-
248- Parameters
249- ----------
250- client : google.generativeai.GenerativeModel
251- The Gemini GenerativeModel to patch.
252-
253- Returns
254- -------
255- google.generativeai.GenerativeModel
256- The patched Gemini client.
257-
258- Example
259- -------
260- >>> import google.generativeai as genai
261- >>> from openlayer.lib import trace_gemini
262- >>> # Configure Gemini
263- >>> genai.configure(api_key="your-api-key")
264- >>> model = genai.GenerativeModel("gemini-pro")
265- >>> # Enable tracing
266- >>> traced_model = trace_gemini(model)
267- >>> # Use normally - tracing happens automatically
268- >>> response = traced_model.generate_content("Hello!")
269- """
243+ """Trace Google Gemini chat completions."""
270244 # pylint: disable=import-outside-toplevel
271245 try :
272246 import google .generativeai as genai
You can’t perform that action at this time.
0 commit comments