From 1e80572bc3c44f6d1a32875642b1e3c0963a207f Mon Sep 17 00:00:00 2001 From: kausmeows Date: Fri, 9 May 2025 21:32:27 +0530 Subject: [PATCH 1/3] add workflow for cerebras --- .github/workflows/test_on_release.yml | 29 +++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) diff --git a/.github/workflows/test_on_release.yml b/.github/workflows/test_on_release.yml index e70bbc0076a..13ed97b489d 100644 --- a/.github/workflows/test_on_release.yml +++ b/.github/workflows/test_on_release.yml @@ -528,6 +528,35 @@ jobs: source .venv/bin/activate python -m pytest ./libs/agno/tests/integration/models/ibm/watsonx + # Run tests for Cerebras + test-google: + runs-on: ubuntu-latest + strategy: + matrix: + python-version: ["3.12"] + fail-fast: true # Stop all matrix jobs if one fails + env: + CEREBRAS_API_KEY: ${{ secrets.CEREBRAS_API_KEY }} + steps: + - uses: actions/checkout@v3 + - name: Set up Python ${{ matrix.python-version }} + uses: actions/setup-python@v4 + with: + python-version: ${{ matrix.python-version }} + - name: Install uv + run: | + curl -LsSf https://astral.sh/uv/install.sh | sh + echo "$HOME/.cargo/bin" >> $GITHUB_PATH + - name: Run dev setup + working-directory: . + run: | + ./scripts/dev_setup.sh + - name: Run Google tests + working-directory: . + run: | + source .venv/bin/activate + python -m pytest ./libs/agno/tests/integration/models/cerebras + test-deepinfra: runs-on: ubuntu-latest strategy: From 424f81c163973ffdf5fe70bda1cea67a88e3098f Mon Sep 17 00:00:00 2001 From: kausmeows Date: Fri, 9 May 2025 21:42:22 +0530 Subject: [PATCH 2/3] update --- .github/workflows/test_on_release.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test_on_release.yml b/.github/workflows/test_on_release.yml index 13ed97b489d..289858d96d9 100644 --- a/.github/workflows/test_on_release.yml +++ b/.github/workflows/test_on_release.yml @@ -529,7 +529,7 @@ jobs: python -m pytest ./libs/agno/tests/integration/models/ibm/watsonx # Run tests for Cerebras - test-google: + test-cerebras: runs-on: ubuntu-latest strategy: matrix: @@ -551,7 +551,7 @@ jobs: working-directory: . run: | ./scripts/dev_setup.sh - - name: Run Google tests + - name: Run Cerebras tests working-directory: . run: | source .venv/bin/activate From 51dbcbeee1342c5e4662b65153034b4e6a25e315 Mon Sep 17 00:00:00 2001 From: kausmeows Date: Fri, 9 May 2025 21:46:35 +0530 Subject: [PATCH 3/3] use duckduckgo --- .../integration/models/cerebras/test_tool_use.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/libs/agno/tests/integration/models/cerebras/test_tool_use.py b/libs/agno/tests/integration/models/cerebras/test_tool_use.py index 4afd796a8c8..e8f39bc3f9d 100644 --- a/libs/agno/tests/integration/models/cerebras/test_tool_use.py +++ b/libs/agno/tests/integration/models/cerebras/test_tool_use.py @@ -2,13 +2,13 @@ from agno.agent import Agent, RunResponse # noqa from agno.models.cerebras import Cerebras -from agno.tools.googlesearch import GoogleSearchTools +from agno.tools.duckduckgo import DuckDuckGoTools def test_tool_use(): agent = Agent( model=Cerebras(id="llama-4-scout-17b-16e-instruct"), - tools=[GoogleSearchTools(cache_results=True)], + tools=[DuckDuckGoTools(cache_results=True)], show_tool_calls=True, telemetry=False, monitoring=False, @@ -25,7 +25,7 @@ def test_tool_use(): def test_tool_use_stream(): agent = Agent( model=Cerebras(id="llama-4-scout-17b-16e-instruct"), - tools=[GoogleSearchTools(cache_results=True)], + tools=[DuckDuckGoTools(cache_results=True)], show_tool_calls=True, telemetry=False, monitoring=False, @@ -52,7 +52,7 @@ def test_tool_use_stream(): async def test_async_tool_use(): agent = Agent( model=Cerebras(id="llama-4-scout-17b-16e-instruct"), - tools=[GoogleSearchTools(cache_results=True)], + tools=[DuckDuckGoTools(cache_results=True)], show_tool_calls=True, telemetry=False, monitoring=False, @@ -70,7 +70,7 @@ async def test_async_tool_use(): async def test_async_tool_use_stream(): agent = Agent( model=Cerebras(id="llama-4-scout-17b-16e-instruct"), - tools=[GoogleSearchTools(cache_results=True)], + tools=[DuckDuckGoTools(cache_results=True)], show_tool_calls=True, telemetry=False, monitoring=False, @@ -96,7 +96,7 @@ async def test_async_tool_use_stream(): def test_tool_use_with_content(): agent = Agent( model=Cerebras(id="llama-4-scout-17b-16e-instruct"), - tools=[GoogleSearchTools(cache_results=True)], + tools=[DuckDuckGoTools(cache_results=True)], show_tool_calls=True, telemetry=False, monitoring=False,