|
1 | 1 | import collections
|
2 | 2 | from collections.abc import Iterable
|
3 | 3 | import copy
|
| 4 | +import datetime |
4 | 5 | import pathlib
|
5 | 6 | from typing import Any
|
6 | 7 | import textwrap
|
@@ -112,7 +113,6 @@ def setUp(self):
|
112 | 113 | client_lib._client_manager.clients["generative"] = self.client
|
113 | 114 | client_lib._client_manager.clients["cache"] = self.client
|
114 | 115 |
|
115 |
| - |
116 | 116 | def test_hello(self):
|
117 | 117 | # Generate text from text prompt
|
118 | 118 | model = generative_models.GenerativeModel(model_name="gemini-pro")
|
@@ -338,7 +338,13 @@ def test_stream_prompt_feedback_not_blocked(self):
|
338 | 338 | dict(testcase_name="test_cached_content_as_id", cached_content="test-cached-content"),
|
339 | 339 | dict(
|
340 | 340 | testcase_name="test_cached_content_as_CachedContent_object",
|
341 |
| - cached_content=caching.CachedContent.get(name="cachedContents/test-cached-content"), |
| 341 | + cached_content=caching.CachedContent( |
| 342 | + name="cachedContents/test-cached-content", |
| 343 | + model="models/gemini-1.0-pro-001", |
| 344 | + create_time=datetime.datetime.now(), |
| 345 | + update_time=datetime.datetime.now(), |
| 346 | + expire_time=datetime.datetime.now(), |
| 347 | + ), |
342 | 348 | ),
|
343 | 349 | ],
|
344 | 350 | )
|
@@ -1289,7 +1295,7 @@ def test_repr_for_model_created_from_cahced_content(self):
|
1289 | 1295 | cached_content="test-cached-content"
|
1290 | 1296 | )
|
1291 | 1297 | result = repr(model)
|
1292 |
| - self.assertIn("cached_content=cachedContent/test-cached-content", result) |
| 1298 | + self.assertIn("cached_content=cachedContents/test-cached-content", result) |
1293 | 1299 | self.assertIn("model_name='models/gemini-1.0-pro-001'", result)
|
1294 | 1300 |
|
1295 | 1301 | def test_count_tokens_called_with_request_options(self):
|
|
0 commit comments