Skip to content

Commit a27b2e4

Browse files
fix: Use class-level storage for original method
Co-Authored-By: Alex Reibman <meta.alex.r@gmail.com>
1 parent 18143b5 commit a27b2e4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

agentops/llms/providers/gemini.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ class GeminiProvider(BaseProvider):
1616
is called and the google.generativeai package is imported. No manual
1717
initialization is required."""
1818

19-
_original_generate = None
19+
_original_generate = None # Store as class attribute
2020

2121
def __init__(self, client=None):
2222
"""Initialize the Gemini provider.

0 commit comments

Comments
 (0)