feat: working tests

Author: Valeriy-Burlaka

genai/text_generation/test_text_generation.py

@@ -12,13 +12,41 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
+
+from google import genai
+
+import pytest
+
 import textgen_chat_with_txt
 import textgen_chat_with_txt_stream
 import textgen_with_txt
 import textgen_with_txt_img
 import textgen_with_txt_stream
 
 
+PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")
+
+
+@pytest.fixture(autouse=True)
+def setup_client():
+    original_Client = genai.Client
+
+    class AutoInitClient(original_Client):
+        def __new__(cls, *args, **kwargs):
+            return original_Client(
+                vertexai=True,
+                project=PROJECT_ID,
+                location="us-central1"
+            )
+
+    genai.Client = AutoInitClient
+
+    yield
+
+    genai.Client = original_Client
+
+
 def test_textgen_with_txt() -> None:
     response = textgen_with_txt.generate_content()
     assert response