|
13 | 13 | # limitations under the License. |
14 | 14 |
|
15 | 15 | import backoff |
| 16 | +import pytest |
16 | 17 |
|
17 | 18 | from google.api_core.exceptions import ResourceExhausted |
18 | 19 | from vertexai.generative_models import GenerativeModel, Tool |
@@ -54,12 +55,14 @@ def test_function_calling_advanced_function_selection() -> None: |
54 | 55 | ) |
55 | 56 |
|
56 | 57 |
|
| 58 | +@pytest.mark.skip(reason="Blocked on b/... ") |
57 | 59 | @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=10) |
58 | 60 | def test_function_calling_basic() -> None: |
59 | 61 | response = chat_function_calling_basic.generate_text() |
60 | 62 | assert "get_current_weather" in response.choices[0].message.tool_calls[0].id |
61 | 63 |
|
62 | 64 |
|
| 65 | +@pytest.mark.skip(reason="Blocked on b/... ") |
63 | 66 | @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=10) |
64 | 67 | def test_function_calling_config() -> None: |
65 | 68 | response = chat_function_calling_config.generate_text() |
@@ -90,7 +93,7 @@ def test_parallel_function_calling() -> None: |
90 | 93 |
|
91 | 94 |
|
92 | 95 | def test_prototype() -> None: |
93 | | - func_declaration = declare_function_from_function.prototype() |
| 96 | + func_declaration = declare_function_from_function.function_declaration_as_func() |
94 | 97 | tools = Tool(function_declarations=[func_declaration]) |
95 | 98 | model = GenerativeModel(model_name="gemini-1.5-pro-002", tools=[tools]) |
96 | 99 | chat_session = model.start_chat() |
|
0 commit comments