PyPI - google-adk - Versions diffs - 0.0.2__py3-none-any.whl → 0.0.4__py3-none-any.whl - Mend

google-adk 0.0.2py3-none-any.whl → 0.0.4py3-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 (153) hide show

google/adk/cli/media_streamer/index.html DELETED Viewed

@@ -1,228 +0,0 @@
-<!--
- Copyright 2025 Google LLC
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
-     http://www.apache.org/licenses/LICENSE-2.0
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
--->
-<html>
-  <head>
-    <style type="text/css">
-      body {
-        font-family: sans-serif;
-        padding: 10px;
-      }
-      button {
-        font-size: 16px;
-        padding: 10px 15px;
-        margin-right: 10px;
-      }
-      #status {
-        font-weight: bold;
-      }
-      /* Mirror the preview video */
-      #videoPreview {
-        border: 1px solid #ccc;
-        transform: scaleX(-1);
-      }
-    </style>
-  </head>
-  <body>
-    <h2>Audio &amp; Video Recorder Component</h2>
-    <!-- Video preview -->
-    <video id="videoPreview" autoplay muted playsinline width="320" height="240"></video>
-    <br/>
-    <!-- Unified start/stop buttons -->
-    <button id="startButton" onclick="startRecording()">Start Streaming</button>
-    <button id="stopButton" onclick="stopRecording()" disabled>Stop Streaming</button>
-    <span id="status">Idle</span>
-    <script type="text/javascript">
-      // --- Streamlit component lifecycle constants ---
-      const SET_COMPONENT_VALUE = "streamlit:setComponentValue";
-      const COMPONENT_READY = "streamlit:componentReady";
-      const SET_FRAME_HEIGHT = "streamlit:setFrameHeight";
-      const RENDER = "streamlit:render";
-      // Helper function to send messages to the Streamlit host.
-      function _sendMessage(type, data) {
-        const outboundData = Object.assign({ isStreamlitMessage: true, type: type }, data);
-        window.parent.postMessage(outboundData, "*");
-      }
-      function setFrameHeight(height) {
-        _sendMessage(SET_FRAME_HEIGHT, { height: height });
-      }
-      // This function sends the combined data.
-      function notifyHost(data) {
-        _sendMessage(SET_COMPONENT_VALUE, data);
-      }
-      function initialize() {
-        _sendMessage(COMPONENT_READY, { apiVersion: 1 });
-        window.addEventListener("load", function() {
-          setTimeout(() => setFrameHeight(document.documentElement.clientHeight), 0);
-        });
-      }
-      initialize();
-      // --- Global Variables ---
-      let isRecording = false;
-      let audioContext;
-      let scriptProcessor;
-      let audioAccumulatedChunks = [];
-      let audioFlushIntervalId = null;
-      let videoCaptureIntervalId = null;
-      const SAMPLE_RATE = 16000;
-      const BUFFER_SIZE = 4096;
-      let mediaStream = null; // for video (and optionally audio if needed)
-      // Global object to hold combined recording data.
-      let recordingData = {};
-      // Instead of calling notifyHost directly, update our combined data and send.
-      function sendCombinedUpdate(key, value) {
-        recordingData[key] = value;
-        notifyHost({
-          value: recordingData,
-          dataType: "json"
-        });
-      }
-      const videoElement = document.getElementById("videoPreview");
-      const statusSpan = document.getElementById("status");
-      const startButton = document.getElementById("startButton");
-      const stopButton = document.getElementById("stopButton");
-      // --- Start Recording: Capture both audio and video ---
-      async function startRecording() {
-        if (isRecording) return;
-        isRecording = true;
-        startButton.disabled = true;
-        stopButton.disabled = false;
-        statusSpan.innerText = "Streaming...";
-        // --- Setup audio recording ---
-        try {
-          audioContext = new (window.AudioContext || window.webkitAudioContext)({ sampleRate: SAMPLE_RATE });
-          let audioStream = await navigator.mediaDevices.getUserMedia({ audio: true });
-          let mediaStreamSource = audioContext.createMediaStreamSource(audioStream);
-          scriptProcessor = audioContext.createScriptProcessor(BUFFER_SIZE, 1, 1);
-          scriptProcessor.onaudioprocess = function(event) {
-            let inputBuffer = event.inputBuffer;
-            let inputData = inputBuffer.getChannelData(0);
-            let pcmData = new Int16Array(inputData.length);
-            for (let i = 0; i < inputData.length; i++) {
-              let s = Math.max(-1, Math.min(1, inputData[i]));
-              pcmData[i] = s < 0 ? s * 0x8000 : s * 0x7FFF;
-            }
-            audioAccumulatedChunks.push(pcmData);
-          };
-          mediaStreamSource.connect(scriptProcessor);
-          scriptProcessor.connect(audioContext.destination);
-        } catch (err) {
-          console.error("Error accessing microphone:", err);
-          statusSpan.innerText = "Audio Error: " + err;
-        }
-        // --- Setup video recording ---
-        try {
-          let videoStream = await navigator.mediaDevices.getUserMedia({ video: true });
-          videoElement.srcObject = videoStream;
-          // Save the video stream so we can stop it later.
-          mediaStream = videoStream;
-        } catch (err) {
-          console.error("Error accessing camera:", err);
-          statusSpan.innerText = "Video Error: " + err;
-        }
-        // --- Start interval timers for flushing audio and capturing video ---
-        videoCaptureIntervalId = setInterval(captureVideoFrame, 250); // every 250ms
-        audioFlushIntervalId = setInterval(flushAudioChunks, 250);  // every 250ms
-      }
-      // --- Stop Recording ---
-      function stopRecording() {
-        if (!isRecording) return;
-        isRecording = false;
-        startButton.disabled = false;
-        stopButton.disabled = true;
-        statusSpan.innerText = "Stopped.";
-        // Stop audio processing
-        if (scriptProcessor) {
-          scriptProcessor.disconnect();
-          scriptProcessor = null;
-        }
-        if (audioContext) {
-          audioContext.close();
-          audioContext = null;
-        }
-        if (audioFlushIntervalId) {
-          clearInterval(audioFlushIntervalId);
-          audioFlushIntervalId = null;
-        }
-        // Stop video stream
-        if (mediaStream) {
-          mediaStream.getTracks().forEach(track => track.stop());
-          mediaStream = null;
-        }
-        if (videoCaptureIntervalId) {
-          clearInterval(videoCaptureIntervalId);
-          videoCaptureIntervalId = null;
-        }
-        // Flush any remaining audio chunks
-        flushAudioChunks();
-        // Final notification that recording has stopped.
-        sendCombinedUpdate("message", "Recording stopped");
-        sendCombinedUpdate("isRecording", false);
-      }
-      // --- Flush accumulated audio chunks ---
-      function flushAudioChunks() {
-        if (audioAccumulatedChunks.length === 0) return;
-        // Combine all audio chunks into one array.
-        let totalLength = audioAccumulatedChunks.reduce((sum, chunk) => sum + chunk.length, 0);
-        let combined = new Int16Array(totalLength);
-        let offset = 0;
-        for (let chunk of audioAccumulatedChunks) {
-          combined.set(chunk, offset);
-          offset += chunk.length;
-        }
-        audioAccumulatedChunks = [];  // clear accumulator
-        // Convert combined PCM data to a Blob.
-        let blob = new Blob([combined.buffer], { type: 'application/octet-stream' });
-        let reader = new FileReader();
-        reader.onloadend = function() {
-          // Instead of sending only audio, update the combined data.
-          sendCombinedUpdate("audioChunk", reader.result);
-        };
-        reader.readAsDataURL(blob);
-      }
-      // --- Capture a single video frame ---
-      function captureVideoFrame() {
-        if (!videoElement.srcObject) return;
-        const canvas = document.createElement("canvas");
-        canvas.width = videoElement.videoWidth || 320;
-        canvas.height = videoElement.videoHeight || 240;
-        const ctx = canvas.getContext("2d");
-        ctx.drawImage(videoElement, 0, 0, canvas.width, canvas.height);
-        let dataUrl = canvas.toDataURL("image/jpeg");
-        // Instead of a separate notifyHost call, update the combined data.
-        sendCombinedUpdate("videoFrame", dataUrl);
-      }
-    </script>
-  </body>
-</html>

google/adk/tests/__init__.py DELETED Viewed

@@ -1,14 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.

google/adk/tests/integration/.env.example DELETED Viewed

@@ -1,10 +0,0 @@
-# Copy as .env file and fill your values below to run integration tests.
-# Choose Backend: GOOGLE_AI_ONLY | VERTEX_ONLY | BOTH (default)
-TEST_BACKEND=BOTH
-# ML Dev backend config
-GOOGLE_API_KEY=YOUR_VALUE_HERE
-# Vertex backend config
-GOOGLE_CLOUD_PROJECT=YOUR_VALUE_HERE
-GOOGLE_CLOUD_LOCATION=YOUR_VALUE_HERE

google/adk/tests/integration/__init__.py DELETED Viewed

@@ -1,18 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-import pytest
-# This allows pytest to show the values of the asserts.
-pytest.register_assert_rewrite('tests.integration.utils')

google/adk/tests/integration/conftest.py DELETED Viewed

@@ -1,119 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-import logging
-import os
-from typing import Literal
-import warnings
-from dotenv import load_dotenv
-from google.adk import Agent
-from pytest import fixture
-from pytest import FixtureRequest
-from pytest import hookimpl
-from pytest import Metafunc
-from .utils import TestRunner
-logger = logging.getLogger(__name__)
-def load_env_for_tests():
-  dotenv_path = os.path.join(os.path.dirname(__file__), '.env')
-  if not os.path.exists(dotenv_path):
-    warnings.warn(
-        f'Missing .env file at {dotenv_path}. See dotenv.sample for an example.'
-    )
-  else:
-    load_dotenv(dotenv_path, override=True, verbose=True)
-  if 'GOOGLE_API_KEY' not in os.environ:
-    warnings.warn(
-        'Missing GOOGLE_API_KEY in the environment variables. GOOGLE_AI backend'
-        ' integration tests will fail.'
-    )
-  for env_var in [
-      'GOOGLE_CLOUD_PROJECT',
-      'GOOGLE_CLOUD_LOCATION',
-  ]:
-    if env_var not in os.environ:
-      warnings.warn(
-          f'Missing {env_var} in the environment variables. Vertex backend'
-          ' integration tests will fail.'
-      )
-load_env_for_tests()
-BackendType = Literal['GOOGLE_AI', 'VERTEX']
-@fixture
-def agent_runner(request: FixtureRequest) -> TestRunner:
-  assert isinstance(request.param, dict)
-  if 'agent' in request.param:
-    assert isinstance(request.param['agent'], Agent)
-    return TestRunner(request.param['agent'])
-  elif 'agent_name' in request.param:
-    assert isinstance(request.param['agent_name'], str)
-    return TestRunner.from_agent_name(request.param['agent_name'])
-  raise NotImplementedError('Must provide agent or agent_name.')
-@fixture(autouse=True)
-def llm_backend(request: FixtureRequest):
-  # Set backend environment value.
-  original_val = os.environ.get('GOOGLE_GENAI_USE_VERTEXAI')
-  backend_type = request.param
-  if backend_type == 'GOOGLE_AI':
-    os.environ['GOOGLE_GENAI_USE_VERTEXAI'] = '0'
-  else:
-    os.environ['GOOGLE_GENAI_USE_VERTEXAI'] = '1'
-  yield  # Run the test
-  # Restore the environment
-  if original_val is None:
-    os.environ.pop('GOOGLE_GENAI_USE_VERTEXAI', None)
-  else:
-    os.environ['GOOGLE_GENAI_USE_VERTEXAI'] = original_val
-@hookimpl(tryfirst=True)
-def pytest_generate_tests(metafunc: Metafunc):
-  if llm_backend.__name__ in metafunc.fixturenames:
-    if not _is_explicitly_marked(llm_backend.__name__, metafunc):
-      test_backend = os.environ.get('TEST_BACKEND', 'BOTH')
-      if test_backend == 'GOOGLE_AI_ONLY':
-        metafunc.parametrize(llm_backend.__name__, ['GOOGLE_AI'], indirect=True)
-      elif test_backend == 'VERTEX_ONLY':
-        metafunc.parametrize(llm_backend.__name__, ['VERTEX'], indirect=True)
-      elif test_backend == 'BOTH':
-        metafunc.parametrize(
-            llm_backend.__name__, ['GOOGLE_AI', 'VERTEX'], indirect=True
-        )
-      else:
-        raise ValueError(
-            f'Invalid TEST_BACKEND value: {test_backend}, should be one of'
-            ' [GOOGLE_AI_ONLY, VERTEX_ONLY, BOTH]'
-        )
-def _is_explicitly_marked(mark_name: str, metafunc: Metafunc) -> bool:
-  if hasattr(metafunc.function, 'pytestmark'):
-    for mark in metafunc.function.pytestmark:
-      if mark.name == 'parametrize' and mark.args[0] == mark_name:
-        return True
-  return False

google/adk/tests/integration/fixture/__init__.py DELETED Viewed

@@ -1,14 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.

google/adk/tests/integration/fixture/agent_with_config/__init__.py DELETED Viewed

@@ -1,15 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-from . import agent

google/adk/tests/integration/fixture/agent_with_config/agent.py DELETED Viewed

@@ -1,88 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-from google.adk import Agent
-from google.genai import types
-new_message = types.Content(
-    role="user",
-    parts=[types.Part.from_text(text="Count a number")],
-)
-google_agent_1 = Agent(
-    model="gemini-1.5-flash",
-    name="agent_1",
-    description="The first agent in the team.",
-    instruction="Just say 1",
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.1,
-    ),
-)
-google_agent_2 = Agent(
-    model="gemini-1.5-flash",
-    name="agent_2",
-    description="The second agent in the team.",
-    instruction="Just say 2",
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.2,
-        safety_settings=[{
-            "category": "HARM_CATEGORY_HATE_SPEECH",
-            "threshold": "BLOCK_ONLY_HIGH",
-        }],
-    ),
-)
-google_agent_3 = Agent(
-    model="gemini-1.5-flash",
-    name="agent_3",
-    description="The third agent in the team.",
-    instruction="Just say 3",
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.5,
-        safety_settings=[{
-            "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-            "threshold": "BLOCK_NONE",
-        }],
-    ),
-)
-google_agent_with_instruction_in_config = Agent(
-    model="gemini-1.5-flash",
-    name="agent",
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.5, system_instruction="Count 1"
-    ),
-)
-def function():
-  pass
-google_agent_with_tools_in_config = Agent(
-    model="gemini-1.5-flash",
-    name="agent",
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.5, tools=[function]
-    ),
-)
-google_agent_with_response_schema_in_config = Agent(
-    model="gemini-1.5-flash",
-    name="agent",
-    generate_content_config=types.GenerateContentConfig(
-        temperature=0.5, response_schema={"key": "value"}
-    ),
-)

google/adk/tests/integration/fixture/callback_agent/__init__.py DELETED Viewed

@@ -1,15 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-from . import agent

google/adk/tests/integration/fixture/callback_agent/agent.py DELETED Viewed

@@ -1,105 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-from typing import Optional
-from google.adk import Agent
-from google.adk.agents.callback_context import CallbackContext
-from google.adk.agents.invocation_context import InvocationContext
-from google.adk.models import LlmRequest
-from google.adk.models import LlmResponse
-from google.genai import types
-def before_agent_call_end_invocation(
-    callback_context: CallbackContext,
-) -> types.Content:
-  return types.Content(
-      role='model',
-      parts=[types.Part(text='End invocation event before agent call.')],
-  )
-def before_agent_call(
-    invocation_context: InvocationContext,
-) -> types.Content:
-  return types.Content(
-      role='model',
-      parts=[types.Part.from_text(text='Plain text event before agent call.')],
-  )
-def before_model_call_end_invocation(
-    callback_context: CallbackContext, llm_request: LlmRequest
-) -> LlmResponse:
-  return LlmResponse(
-      content=types.Content(
-          role='model',
-          parts=[
-              types.Part.from_text(
-                  text='End invocation event before model call.'
-              )
-          ],
-      )
-  )
-def before_model_call(
-    invocation_context: InvocationContext, request: LlmRequest
-) -> LlmResponse:
-  request.config.system_instruction = 'Just return 999 as response.'
-  return LlmResponse(
-      content=types.Content(
-          role='model',
-          parts=[
-              types.Part.from_text(
-                  text='Update request event before model call.'
-              )
-          ],
-      )
-  )
-def after_model_call(
-    callback_context: CallbackContext,
-    llm_response: LlmResponse,
-) -> Optional[LlmResponse]:
-  content = llm_response.content
-  if not content or not content.parts or not content.parts[0].text:
-    return
-  content.parts[0].text += 'Update response event after model call.'
-  return llm_response
-before_agent_callback_agent = Agent(
-    model='gemini-1.5-flash',
-    name='before_agent_callback_agent',
-    instruction='echo 1',
-    before_agent_callback=before_agent_call_end_invocation,
-)
-before_model_callback_agent = Agent(
-    model='gemini-1.5-flash',
-    name='before_model_callback_agent',
-    instruction='echo 2',
-    before_model_callback=before_model_call_end_invocation,
-)
-after_model_callback_agent = Agent(
-    model='gemini-1.5-flash',
-    name='after_model_callback_agent',
-    instruction='Say hello',
-    after_model_callback=after_model_call,
-)

google/adk/tests/integration/fixture/context_update_test/OWNERS DELETED Viewed

	@@ -1 +0,0 @@
1	- gkcng

google/adk/tests/integration/fixture/context_update_test/__init__.py DELETED Viewed

@@ -1,15 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-from . import agent

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

google-adk 0.0.2py3-none-any.whl → 0.0.4py3-none-any.whl