google-adk 0.0.3__py3-none-any.whl → 0.0.4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (184) hide show
  1. google/adk/agents/run_config.py +4 -0
  2. google/adk/auth/auth_preprocessor.py +19 -16
  3. google/adk/cli/browser/index.html +1 -1
  4. google/adk/cli/browser/{main-SY2WYYGV.js → main-CU22TRPI.js} +30 -30
  5. google/adk/cli/cli.py +8 -8
  6. google/adk/cli/cli_deploy.py +2 -4
  7. google/adk/cli/cli_tools_click.py +6 -6
  8. google/adk/flows/llm_flows/contents.py +21 -1
  9. google/adk/flows/llm_flows/functions.py +3 -1
  10. google/adk/models/google_llm.py +0 -1
  11. google/adk/runners.py +13 -2
  12. google/adk/version.py +1 -1
  13. {google_adk-0.0.3.dist-info → google_adk-0.0.4.dist-info}/METADATA +4 -2
  14. google_adk-0.0.4.dist-info/RECORD +175 -0
  15. {google_adk-0.0.3.dist-info → google_adk-0.0.4.dist-info}/WHEEL +1 -1
  16. google_adk-0.0.4.dist-info/licenses/LICENSE +202 -0
  17. google/adk/._version.py +0 -0
  18. google/adk/docs/Makefile +0 -20
  19. google/adk/docs/build/doctrees/google-adk.doctree +0 -0
  20. google/adk/docs/build/html/_sources/google-adk.rst.txt +0 -98
  21. google/adk/docs/build/html/_sources/index.rst.txt +0 -7
  22. google/adk/docs/build/html/_static/autodoc_pydantic.css +0 -27
  23. google/adk/docs/build/html/_static/basic.css +0 -925
  24. google/adk/docs/build/html/_static/debug.css +0 -85
  25. google/adk/docs/build/html/_static/doctools.js +0 -156
  26. google/adk/docs/build/html/_static/documentation_options.js +0 -29
  27. google/adk/docs/build/html/_static/file.png +0 -0
  28. google/adk/docs/build/html/_static/language_data.js +0 -199
  29. google/adk/docs/build/html/_static/minus.png +0 -0
  30. google/adk/docs/build/html/_static/plus.png +0 -0
  31. google/adk/docs/build/html/_static/pygments.css +0 -274
  32. google/adk/docs/build/html/_static/scripts/furo-extensions.js +0 -16
  33. google/adk/docs/build/html/_static/scripts/furo.js +0 -19
  34. google/adk/docs/build/html/_static/scripts/furo.js.LICENSE.txt +0 -7
  35. google/adk/docs/build/html/_static/scripts/furo.js.map +0 -1
  36. google/adk/docs/build/html/_static/searchtools.js +0 -620
  37. google/adk/docs/build/html/_static/skeleton.css +0 -312
  38. google/adk/docs/build/html/_static/sphinx_highlight.js +0 -170
  39. google/adk/docs/build/html/_static/styles/furo-extensions.css +0 -18
  40. google/adk/docs/build/html/_static/styles/furo-extensions.css.map +0 -1
  41. google/adk/docs/build/html/_static/styles/furo.css +0 -18
  42. google/adk/docs/build/html/_static/styles/furo.css.map +0 -1
  43. google/adk/docs/build/html/genindex.html +0 -861
  44. google/adk/docs/build/html/google-adk.html +0 -5461
  45. google/adk/docs/build/html/index.html +0 -567
  46. google/adk/docs/build/html/objects.inv +0 -0
  47. google/adk/docs/build/html/py-modindex.html +0 -373
  48. google/adk/docs/build/html/search.html +0 -333
  49. google/adk/docs/build/html/searchindex.js +0 -17
  50. google/adk/docs/source/conf.py +0 -133
  51. google/adk/docs/source/google-adk.rst +0 -98
  52. google/adk/docs/source/index.rst +0 -7
  53. google/adk/tests/__init__.py +0 -14
  54. google/adk/tests/integration/.env.example +0 -10
  55. google/adk/tests/integration/__init__.py +0 -18
  56. google/adk/tests/integration/conftest.py +0 -119
  57. google/adk/tests/integration/fixture/__init__.py +0 -14
  58. google/adk/tests/integration/fixture/agent_with_config/__init__.py +0 -15
  59. google/adk/tests/integration/fixture/agent_with_config/agent.py +0 -88
  60. google/adk/tests/integration/fixture/callback_agent/__init__.py +0 -15
  61. google/adk/tests/integration/fixture/callback_agent/agent.py +0 -105
  62. google/adk/tests/integration/fixture/context_update_test/OWNERS +0 -1
  63. google/adk/tests/integration/fixture/context_update_test/__init__.py +0 -15
  64. google/adk/tests/integration/fixture/context_update_test/agent.py +0 -43
  65. google/adk/tests/integration/fixture/context_update_test/successful_test.session.json +0 -582
  66. google/adk/tests/integration/fixture/context_variable_agent/__init__.py +0 -15
  67. google/adk/tests/integration/fixture/context_variable_agent/agent.py +0 -115
  68. google/adk/tests/integration/fixture/customer_support_ma/__init__.py +0 -15
  69. google/adk/tests/integration/fixture/customer_support_ma/agent.py +0 -172
  70. google/adk/tests/integration/fixture/ecommerce_customer_service_agent/__init__.py +0 -15
  71. google/adk/tests/integration/fixture/ecommerce_customer_service_agent/agent.py +0 -338
  72. google/adk/tests/integration/fixture/ecommerce_customer_service_agent/order_query.test.json +0 -69
  73. google/adk/tests/integration/fixture/ecommerce_customer_service_agent/test_config.json +0 -6
  74. google/adk/tests/integration/fixture/flow_complex_spark/__init__.py +0 -15
  75. google/adk/tests/integration/fixture/flow_complex_spark/agent.py +0 -182
  76. google/adk/tests/integration/fixture/flow_complex_spark/sample.session.json +0 -190
  77. google/adk/tests/integration/fixture/hello_world_agent/__init__.py +0 -15
  78. google/adk/tests/integration/fixture/hello_world_agent/agent.py +0 -95
  79. google/adk/tests/integration/fixture/hello_world_agent/roll_die.test.json +0 -24
  80. google/adk/tests/integration/fixture/hello_world_agent/test_config.json +0 -6
  81. google/adk/tests/integration/fixture/home_automation_agent/__init__.py +0 -15
  82. google/adk/tests/integration/fixture/home_automation_agent/agent.py +0 -304
  83. google/adk/tests/integration/fixture/home_automation_agent/simple_test.test.json +0 -5
  84. google/adk/tests/integration/fixture/home_automation_agent/simple_test2.test.json +0 -5
  85. google/adk/tests/integration/fixture/home_automation_agent/test_config.json +0 -5
  86. google/adk/tests/integration/fixture/home_automation_agent/test_files/dependent_tool_calls.test.json +0 -18
  87. google/adk/tests/integration/fixture/home_automation_agent/test_files/memorizing_past_events/eval_data.test.json +0 -17
  88. google/adk/tests/integration/fixture/home_automation_agent/test_files/memorizing_past_events/test_config.json +0 -6
  89. google/adk/tests/integration/fixture/home_automation_agent/test_files/simple_multi_turn_conversation.test.json +0 -18
  90. google/adk/tests/integration/fixture/home_automation_agent/test_files/simple_test.test.json +0 -17
  91. google/adk/tests/integration/fixture/home_automation_agent/test_files/simple_test2.test.json +0 -5
  92. google/adk/tests/integration/fixture/home_automation_agent/test_files/test_config.json +0 -5
  93. google/adk/tests/integration/fixture/tool_agent/__init__.py +0 -15
  94. google/adk/tests/integration/fixture/tool_agent/agent.py +0 -218
  95. google/adk/tests/integration/fixture/tool_agent/files/Agent_test_plan.pdf +0 -0
  96. google/adk/tests/integration/fixture/trip_planner_agent/__init__.py +0 -15
  97. google/adk/tests/integration/fixture/trip_planner_agent/agent.py +0 -110
  98. google/adk/tests/integration/fixture/trip_planner_agent/initial.session.json +0 -13
  99. google/adk/tests/integration/fixture/trip_planner_agent/test_config.json +0 -5
  100. google/adk/tests/integration/fixture/trip_planner_agent/test_files/initial.session.json +0 -13
  101. google/adk/tests/integration/fixture/trip_planner_agent/test_files/test_config.json +0 -5
  102. google/adk/tests/integration/fixture/trip_planner_agent/test_files/trip_inquiry_sub_agent.test.json +0 -7
  103. google/adk/tests/integration/fixture/trip_planner_agent/trip_inquiry.test.json +0 -19
  104. google/adk/tests/integration/models/__init__.py +0 -14
  105. google/adk/tests/integration/models/test_google_llm.py +0 -65
  106. google/adk/tests/integration/test_callback.py +0 -70
  107. google/adk/tests/integration/test_context_variable.py +0 -67
  108. google/adk/tests/integration/test_evalute_agent_in_fixture.py +0 -76
  109. google/adk/tests/integration/test_multi_agent.py +0 -28
  110. google/adk/tests/integration/test_multi_turn.py +0 -42
  111. google/adk/tests/integration/test_single_agent.py +0 -23
  112. google/adk/tests/integration/test_sub_agent.py +0 -26
  113. google/adk/tests/integration/test_system_instruction.py +0 -177
  114. google/adk/tests/integration/test_tools.py +0 -287
  115. google/adk/tests/integration/test_with_test_file.py +0 -34
  116. google/adk/tests/integration/tools/__init__.py +0 -14
  117. google/adk/tests/integration/utils/__init__.py +0 -16
  118. google/adk/tests/integration/utils/asserts.py +0 -75
  119. google/adk/tests/integration/utils/test_runner.py +0 -97
  120. google/adk/tests/unittests/__init__.py +0 -14
  121. google/adk/tests/unittests/agents/__init__.py +0 -14
  122. google/adk/tests/unittests/agents/test_base_agent.py +0 -407
  123. google/adk/tests/unittests/agents/test_langgraph_agent.py +0 -191
  124. google/adk/tests/unittests/agents/test_llm_agent_callbacks.py +0 -138
  125. google/adk/tests/unittests/agents/test_llm_agent_fields.py +0 -231
  126. google/adk/tests/unittests/agents/test_loop_agent.py +0 -136
  127. google/adk/tests/unittests/agents/test_parallel_agent.py +0 -92
  128. google/adk/tests/unittests/agents/test_sequential_agent.py +0 -114
  129. google/adk/tests/unittests/artifacts/__init__.py +0 -14
  130. google/adk/tests/unittests/artifacts/test_artifact_service.py +0 -276
  131. google/adk/tests/unittests/auth/test_auth_handler.py +0 -575
  132. google/adk/tests/unittests/conftest.py +0 -73
  133. google/adk/tests/unittests/fast_api/__init__.py +0 -14
  134. google/adk/tests/unittests/fast_api/test_fast_api.py +0 -269
  135. google/adk/tests/unittests/flows/__init__.py +0 -14
  136. google/adk/tests/unittests/flows/llm_flows/__init__.py +0 -14
  137. google/adk/tests/unittests/flows/llm_flows/_test_examples.py +0 -142
  138. google/adk/tests/unittests/flows/llm_flows/test_agent_transfer.py +0 -311
  139. google/adk/tests/unittests/flows/llm_flows/test_functions_long_running.py +0 -244
  140. google/adk/tests/unittests/flows/llm_flows/test_functions_request_euc.py +0 -346
  141. google/adk/tests/unittests/flows/llm_flows/test_functions_sequential.py +0 -93
  142. google/adk/tests/unittests/flows/llm_flows/test_functions_simple.py +0 -258
  143. google/adk/tests/unittests/flows/llm_flows/test_identity.py +0 -66
  144. google/adk/tests/unittests/flows/llm_flows/test_instructions.py +0 -164
  145. google/adk/tests/unittests/flows/llm_flows/test_model_callbacks.py +0 -142
  146. google/adk/tests/unittests/flows/llm_flows/test_other_configs.py +0 -46
  147. google/adk/tests/unittests/flows/llm_flows/test_tool_callbacks.py +0 -269
  148. google/adk/tests/unittests/models/__init__.py +0 -14
  149. google/adk/tests/unittests/models/test_google_llm.py +0 -224
  150. google/adk/tests/unittests/models/test_litellm.py +0 -804
  151. google/adk/tests/unittests/models/test_models.py +0 -60
  152. google/adk/tests/unittests/sessions/__init__.py +0 -14
  153. google/adk/tests/unittests/sessions/test_session_service.py +0 -227
  154. google/adk/tests/unittests/sessions/test_vertex_ai_session_service.py +0 -246
  155. google/adk/tests/unittests/streaming/__init__.py +0 -14
  156. google/adk/tests/unittests/streaming/test_streaming.py +0 -50
  157. google/adk/tests/unittests/tools/__init__.py +0 -14
  158. google/adk/tests/unittests/tools/apihub_tool/clients/test_apihub_client.py +0 -499
  159. google/adk/tests/unittests/tools/apihub_tool/test_apihub_toolset.py +0 -204
  160. google/adk/tests/unittests/tools/application_integration_tool/clients/test_connections_client.py +0 -600
  161. google/adk/tests/unittests/tools/application_integration_tool/clients/test_integration_client.py +0 -630
  162. google/adk/tests/unittests/tools/application_integration_tool/test_application_integration_toolset.py +0 -345
  163. google/adk/tests/unittests/tools/google_api_tool/__init__.py +0 -13
  164. google/adk/tests/unittests/tools/google_api_tool/test_googleapi_to_openapi_converter.py +0 -657
  165. google/adk/tests/unittests/tools/openapi_tool/auth/credential_exchangers/test_auto_auth_credential_exchanger.py +0 -145
  166. google/adk/tests/unittests/tools/openapi_tool/auth/credential_exchangers/test_base_auth_credential_exchanger.py +0 -68
  167. google/adk/tests/unittests/tools/openapi_tool/auth/credential_exchangers/test_oauth2_exchanger.py +0 -153
  168. google/adk/tests/unittests/tools/openapi_tool/auth/credential_exchangers/test_service_account_exchanger.py +0 -196
  169. google/adk/tests/unittests/tools/openapi_tool/auth/test_auth_helper.py +0 -573
  170. google/adk/tests/unittests/tools/openapi_tool/common/test_common.py +0 -436
  171. google/adk/tests/unittests/tools/openapi_tool/openapi_spec_parser/test.yaml +0 -1367
  172. google/adk/tests/unittests/tools/openapi_tool/openapi_spec_parser/test_openapi_spec_parser.py +0 -628
  173. google/adk/tests/unittests/tools/openapi_tool/openapi_spec_parser/test_openapi_toolset.py +0 -139
  174. google/adk/tests/unittests/tools/openapi_tool/openapi_spec_parser/test_operation_parser.py +0 -406
  175. google/adk/tests/unittests/tools/openapi_tool/openapi_spec_parser/test_rest_api_tool.py +0 -966
  176. google/adk/tests/unittests/tools/openapi_tool/openapi_spec_parser/test_tool_auth_handler.py +0 -201
  177. google/adk/tests/unittests/tools/retrieval/__init__.py +0 -14
  178. google/adk/tests/unittests/tools/retrieval/test_vertex_ai_rag_retrieval.py +0 -147
  179. google/adk/tests/unittests/tools/test_agent_tool.py +0 -167
  180. google/adk/tests/unittests/tools/test_base_tool.py +0 -141
  181. google/adk/tests/unittests/tools/test_build_function_declaration.py +0 -277
  182. google/adk/tests/unittests/utils.py +0 -304
  183. google_adk-0.0.3.dist-info/RECORD +0 -340
  184. {google_adk-0.0.3.dist-info → google_adk-0.0.4.dist-info}/entry_points.txt +0 -0
google/adk/tests/integration/fixture/context_update_test/successful_test.session.json DELETED
@@ -1,582 +0,0 @@
1
- {
2
- "id": "ead43200-b575-4241-9248-233b4be4f29a",
3
- "context": {
4
- "_time": "2024-12-01 09:02:43.531503",
5
- "data_one": "RRRR",
6
- "data_two": "3.141529",
7
- "data_three": [
8
- "apple",
9
- "banana"
10
- ],
11
- "data_four": [
12
- "1",
13
- "hello",
14
- "3.14"
15
- ]
16
- },
17
- "events": [
18
- {
19
- "invocation_id": "6BGrtKJu",
20
- "author": "user",
21
- "content": {
22
- "parts": [
23
- {
24
- "text": "hi"
25
- }
26
- ],
27
- "role": "user"
28
- },
29
- "options": {},
30
- "id": "ltzQTqR4",
31
- "timestamp": 1733043686.8428597
32
- },
33
- {
34
- "invocation_id": "6BGrtKJu",
35
- "author": "root_agent",
36
- "content": {
37
- "parts": [
38
- {
39
- "text": "Hello! 👋 How can I help you today? \n"
40
- }
41
- ],
42
- "role": "model"
43
- },
44
- "options": {
45
- "partial": false
46
- },
47
- "id": "ClSROx8b",
48
- "timestamp": 1733043688.1030986
49
- },
50
- {
51
- "invocation_id": "M3dUcVa8",
52
- "author": "user",
53
- "content": {
54
- "parts": [
55
- {
56
- "text": "update data_one to be RRRR, data_two to be 3.141529, data_three to be apple and banana, data_four to be 1, hello, and 3.14"
57
- }
58
- ],
59
- "role": "user"
60
- },
61
- "options": {},
62
- "id": "yxigGwIZ",
63
- "timestamp": 1733043745.9900541
64
- },
65
- {
66
- "invocation_id": "M3dUcVa8",
67
- "author": "root_agent",
68
- "content": {
69
- "parts": [
70
- {
71
- "function_call": {
72
- "args": {
73
- "data_four": [
74
- "1",
75
- "hello",
76
- "3.14"
77
- ],
78
- "data_two": "3.141529",
79
- "data_three": [
80
- "apple",
81
- "banana"
82
- ],
83
- "data_one": "RRRR"
84
- },
85
- "name": "update_fc"
86
- }
87
- }
88
- ],
89
- "role": "model"
90
- },
91
- "options": {
92
- "partial": false
93
- },
94
- "id": "8V6de8th",
95
- "timestamp": 1733043747.4545543
96
- },
97
- {
98
- "invocation_id": "M3dUcVa8",
99
- "author": "root_agent",
100
- "content": {
101
- "parts": [
102
- {
103
- "function_response": {
104
- "name": "update_fc",
105
- "response": {}
106
- }
107
- }
108
- ],
109
- "role": "user"
110
- },
111
- "options": {
112
- "update_context": {
113
- "data_one": "RRRR",
114
- "data_two": "3.141529",
115
- "data_three": [
116
- "apple",
117
- "banana"
118
- ],
119
- "data_four": [
120
- "1",
121
- "hello",
122
- "3.14"
123
- ]
124
- },
125
- "function_call_event_id": "8V6de8th"
126
- },
127
- "id": "dkTj5v8B",
128
- "timestamp": 1733043747.457031
129
- },
130
- {
131
- "invocation_id": "M3dUcVa8",
132
- "author": "root_agent",
133
- "content": {
134
- "parts": [
135
- {
136
- "text": "OK. I've updated the data. Anything else? \n"
137
- }
138
- ],
139
- "role": "model"
140
- },
141
- "options": {
142
- "partial": false
143
- },
144
- "id": "OZ77XR41",
145
- "timestamp": 1733043748.7901294
146
- }
147
- ],
148
- "past_events": [],
149
- "pending_events": {},
150
- "artifacts": {},
151
- "event_logs": [
152
- {
153
- "invocation_id": "6BGrtKJu",
154
- "event_id": "ClSROx8b",
155
- "model_request": {
156
- "model": "gemini-1.5-flash",
157
- "contents": [
158
- {
159
- "parts": [
160
- {
161
- "text": "hi"
162
- }
163
- ],
164
- "role": "user"
165
- }
166
- ],
167
- "config": {
168
- "system_instruction": "You are an agent. Your name is root_agent.\nCall tools",
169
- "tools": [
170
- {
171
- "function_declarations": [
172
- {
173
- "description": "Hello",
174
- "name": "update_fc",
175
- "parameters": {
176
- "type": "OBJECT",
177
- "properties": {
178
- "data_one": {
179
- "type": "STRING"
180
- },
181
- "data_two": {
182
- "type": "STRING"
183
- },
184
- "data_three": {
185
- "type": "ARRAY",
186
- "items": {
187
- "type": "STRING"
188
- }
189
- },
190
- "data_four": {
191
- "type": "ARRAY",
192
- "items": {
193
- "any_of": [
194
- {
195
- "type": "INTEGER"
196
- },
197
- {
198
- "type": "NUMBER"
199
- },
200
- {
201
- "type": "STRING"
202
- }
203
- ],
204
- "type": "STRING"
205
- }
206
- }
207
- }
208
- }
209
- }
210
- ]
211
- }
212
- ]
213
- }
214
- },
215
- "model_response": {
216
- "candidates": [
217
- {
218
- "content": {
219
- "parts": [
220
- {
221
- "text": "Hello! 👋 How can I help you today? \n"
222
- }
223
- ],
224
- "role": "model"
225
- },
226
- "avg_logprobs": -0.15831730915949896,
227
- "finish_reason": "STOP",
228
- "safety_ratings": [
229
- {
230
- "category": "HARM_CATEGORY_HATE_SPEECH",
231
- "probability": "NEGLIGIBLE",
232
- "probability_score": 0.071777344,
233
- "severity": "HARM_SEVERITY_NEGLIGIBLE",
234
- "severity_score": 0.07080078
235
- },
236
- {
237
- "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
238
- "probability": "NEGLIGIBLE",
239
- "probability_score": 0.16308594,
240
- "severity": "HARM_SEVERITY_NEGLIGIBLE",
241
- "severity_score": 0.14160156
242
- },
243
- {
244
- "category": "HARM_CATEGORY_HARASSMENT",
245
- "probability": "NEGLIGIBLE",
246
- "probability_score": 0.09423828,
247
- "severity": "HARM_SEVERITY_NEGLIGIBLE",
248
- "severity_score": 0.037841797
249
- },
250
- {
251
- "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
252
- "probability": "NEGLIGIBLE",
253
- "probability_score": 0.059326172,
254
- "severity": "HARM_SEVERITY_NEGLIGIBLE",
255
- "severity_score": 0.02368164
256
- }
257
- ]
258
- }
259
- ],
260
- "model_version": "gemini-1.5-flash-001",
261
- "usage_metadata": {
262
- "candidates_token_count": 13,
263
- "prompt_token_count": 32,
264
- "total_token_count": 45
265
- }
266
- }
267
- },
268
- {
269
- "invocation_id": "M3dUcVa8",
270
- "event_id": "8V6de8th",
271
- "model_request": {
272
- "model": "gemini-1.5-flash",
273
- "contents": [
274
- {
275
- "parts": [
276
- {
277
- "text": "hi"
278
- }
279
- ],
280
- "role": "user"
281
- },
282
- {
283
- "parts": [
284
- {
285
- "text": "Hello! 👋 How can I help you today? \n"
286
- }
287
- ],
288
- "role": "model"
289
- },
290
- {
291
- "parts": [
292
- {
293
- "text": "update data_one to be RRRR, data_two to be 3.141529, data_three to be apple and banana, data_four to be 1, hello, and 3.14"
294
- }
295
- ],
296
- "role": "user"
297
- }
298
- ],
299
- "config": {
300
- "system_instruction": "You are an agent. Your name is root_agent.\nCall tools",
301
- "tools": [
302
- {
303
- "function_declarations": [
304
- {
305
- "description": "Hello",
306
- "name": "update_fc",
307
- "parameters": {
308
- "type": "OBJECT",
309
- "properties": {
310
- "data_one": {
311
- "type": "STRING"
312
- },
313
- "data_two": {
314
- "type": "STRING"
315
- },
316
- "data_three": {
317
- "type": "ARRAY",
318
- "items": {
319
- "type": "STRING"
320
- }
321
- },
322
- "data_four": {
323
- "type": "ARRAY",
324
- "items": {
325
- "any_of": [
326
- {
327
- "type": "INTEGER"
328
- },
329
- {
330
- "type": "NUMBER"
331
- },
332
- {
333
- "type": "STRING"
334
- }
335
- ],
336
- "type": "STRING"
337
- }
338
- }
339
- }
340
- }
341
- }
342
- ]
343
- }
344
- ]
345
- }
346
- },
347
- "model_response": {
348
- "candidates": [
349
- {
350
- "content": {
351
- "parts": [
352
- {
353
- "function_call": {
354
- "args": {
355
- "data_four": [
356
- "1",
357
- "hello",
358
- "3.14"
359
- ],
360
- "data_two": "3.141529",
361
- "data_three": [
362
- "apple",
363
- "banana"
364
- ],
365
- "data_one": "RRRR"
366
- },
367
- "name": "update_fc"
368
- }
369
- }
370
- ],
371
- "role": "model"
372
- },
373
- "avg_logprobs": -2.100960955431219e-6,
374
- "finish_reason": "STOP",
375
- "safety_ratings": [
376
- {
377
- "category": "HARM_CATEGORY_HATE_SPEECH",
378
- "probability": "NEGLIGIBLE",
379
- "probability_score": 0.12158203,
380
- "severity": "HARM_SEVERITY_NEGLIGIBLE",
381
- "severity_score": 0.13671875
382
- },
383
- {
384
- "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
385
- "probability": "NEGLIGIBLE",
386
- "probability_score": 0.421875,
387
- "severity": "HARM_SEVERITY_LOW",
388
- "severity_score": 0.24511719
389
- },
390
- {
391
- "category": "HARM_CATEGORY_HARASSMENT",
392
- "probability": "NEGLIGIBLE",
393
- "probability_score": 0.15722656,
394
- "severity": "HARM_SEVERITY_NEGLIGIBLE",
395
- "severity_score": 0.072753906
396
- },
397
- {
398
- "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
399
- "probability": "NEGLIGIBLE",
400
- "probability_score": 0.083984375,
401
- "severity": "HARM_SEVERITY_NEGLIGIBLE",
402
- "severity_score": 0.03564453
403
- }
404
- ]
405
- }
406
- ],
407
- "model_version": "gemini-1.5-flash-001",
408
- "usage_metadata": {
409
- "candidates_token_count": 32,
410
- "prompt_token_count": 94,
411
- "total_token_count": 126
412
- }
413
- }
414
- },
415
- {
416
- "invocation_id": "M3dUcVa8",
417
- "event_id": "OZ77XR41",
418
- "model_request": {
419
- "model": "gemini-1.5-flash",
420
- "contents": [
421
- {
422
- "parts": [
423
- {
424
- "text": "hi"
425
- }
426
- ],
427
- "role": "user"
428
- },
429
- {
430
- "parts": [
431
- {
432
- "text": "Hello! 👋 How can I help you today? \n"
433
- }
434
- ],
435
- "role": "model"
436
- },
437
- {
438
- "parts": [
439
- {
440
- "text": "update data_one to be RRRR, data_two to be 3.141529, data_three to be apple and banana, data_four to be 1, hello, and 3.14"
441
- }
442
- ],
443
- "role": "user"
444
- },
445
- {
446
- "parts": [
447
- {
448
- "function_call": {
449
- "args": {
450
- "data_four": [
451
- "1",
452
- "hello",
453
- "3.14"
454
- ],
455
- "data_two": "3.141529",
456
- "data_three": [
457
- "apple",
458
- "banana"
459
- ],
460
- "data_one": "RRRR"
461
- },
462
- "name": "update_fc"
463
- }
464
- }
465
- ],
466
- "role": "model"
467
- },
468
- {
469
- "parts": [
470
- {
471
- "function_response": {
472
- "name": "update_fc",
473
- "response": {}
474
- }
475
- }
476
- ],
477
- "role": "user"
478
- }
479
- ],
480
- "config": {
481
- "system_instruction": "You are an agent. Your name is root_agent.\nCall tools",
482
- "tools": [
483
- {
484
- "function_declarations": [
485
- {
486
- "description": "Hello",
487
- "name": "update_fc",
488
- "parameters": {
489
- "type": "OBJECT",
490
- "properties": {
491
- "data_one": {
492
- "type": "STRING"
493
- },
494
- "data_two": {
495
- "type": "STRING"
496
- },
497
- "data_three": {
498
- "type": "ARRAY",
499
- "items": {
500
- "type": "STRING"
501
- }
502
- },
503
- "data_four": {
504
- "type": "ARRAY",
505
- "items": {
506
- "any_of": [
507
- {
508
- "type": "INTEGER"
509
- },
510
- {
511
- "type": "NUMBER"
512
- },
513
- {
514
- "type": "STRING"
515
- }
516
- ],
517
- "type": "STRING"
518
- }
519
- }
520
- }
521
- }
522
- }
523
- ]
524
- }
525
- ]
526
- }
527
- },
528
- "model_response": {
529
- "candidates": [
530
- {
531
- "content": {
532
- "parts": [
533
- {
534
- "text": "OK. I've updated the data. Anything else? \n"
535
- }
536
- ],
537
- "role": "model"
538
- },
539
- "avg_logprobs": -0.22089435373033797,
540
- "finish_reason": "STOP",
541
- "safety_ratings": [
542
- {
543
- "category": "HARM_CATEGORY_HATE_SPEECH",
544
- "probability": "NEGLIGIBLE",
545
- "probability_score": 0.04663086,
546
- "severity": "HARM_SEVERITY_NEGLIGIBLE",
547
- "severity_score": 0.09423828
548
- },
549
- {
550
- "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
551
- "probability": "NEGLIGIBLE",
552
- "probability_score": 0.18554688,
553
- "severity": "HARM_SEVERITY_NEGLIGIBLE",
554
- "severity_score": 0.111328125
555
- },
556
- {
557
- "category": "HARM_CATEGORY_HARASSMENT",
558
- "probability": "NEGLIGIBLE",
559
- "probability_score": 0.071777344,
560
- "severity": "HARM_SEVERITY_NEGLIGIBLE",
561
- "severity_score": 0.03112793
562
- },
563
- {
564
- "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
565
- "probability": "NEGLIGIBLE",
566
- "probability_score": 0.043945313,
567
- "severity": "HARM_SEVERITY_NEGLIGIBLE",
568
- "severity_score": 0.057373047
569
- }
570
- ]
571
- }
572
- ],
573
- "model_version": "gemini-1.5-flash-001",
574
- "usage_metadata": {
575
- "candidates_token_count": 14,
576
- "prompt_token_count": 129,
577
- "total_token_count": 143
578
- }
579
- }
580
- }
581
- ]
582
- }
google/adk/tests/integration/fixture/context_variable_agent/__init__.py DELETED
@@ -1,15 +0,0 @@
1
- # Copyright 2025 Google LLC
2
- #
3
- # Licensed under the Apache License, Version 2.0 (the "License");
4
- # you may not use this file except in compliance with the License.
5
- # You may obtain a copy of the License at
6
- #
7
- # http://www.apache.org/licenses/LICENSE-2.0
8
- #
9
- # Unless required by applicable law or agreed to in writing, software
10
- # distributed under the License is distributed on an "AS IS" BASIS,
11
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
- # See the License for the specific language governing permissions and
13
- # limitations under the License.
14
-
15
- from . import agent