veadk-python 0.2.2__tar.gz → 0.2.19__tar.gz

veadk_python-0.2.19/PKG-INFO

@@ -0,0 +1,370 @@
+Metadata-Version: 2.4
+Name: veadk-python
+Version: 0.2.19
+Summary: Volcengine agent development kit, integrations with Volcengine cloud services.
+Author-email: Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>, Zhi Han <sliverydayday@gmail.com>, Meng Wang <mengwangwm@gmail.com>
+License:                                  Apache License
+                                   Version 2.0, January 2004
+                                http://www.apache.org/licenses/
+        
+           TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+        
+           1. Definitions.
+        
+              "License" shall mean the terms and conditions for use, reproduction,
+              and distribution as defined by Sections 1 through 9 of this document.
+        
+              "Licensor" shall mean the copyright owner or entity authorized by
+              the copyright owner that is granting the License.
+        
+              "Legal Entity" shall mean the union of the acting entity and all
+              other entities that control, are controlled by, or are under common
+              control with that entity. For the purposes of this definition,
+              "control" means (i) the power, direct or indirect, to cause the
+              direction or management of such entity, whether by contract or
+              otherwise, or (ii) ownership of fifty percent (50%) or more of the
+              outstanding shares, or (iii) beneficial ownership of such entity.
+        
+              "You" (or "Your") shall mean an individual or Legal Entity
+              exercising permissions granted by this License.
+        
+              "Source" form shall mean the preferred form for making modifications,
+              including but not limited to software source code, documentation
+              source, and configuration files.
+        
+              "Object" form shall mean any form resulting from mechanical
+              transformation or translation of a Source form, including but
+              not limited to compiled object code, generated documentation,
+              and conversions to other media types.
+        
+              "Work" shall mean the work of authorship, whether in Source or
+              Object form, made available under the License, as indicated by a
+              copyright notice that is included in or attached to the work
+              (an example is provided in the Appendix below).
+        
+              "Derivative Works" shall mean any work, whether in Source or Object
+              form, that is based on (or derived from) the Work and for which the
+              editorial revisions, annotations, elaborations, or other modifications
+              represent, as a whole, an original work of authorship. For the purposes
+              of this License, Derivative Works shall not include works that remain
+              separable from, or merely link (or bind by name) to the interfaces of,
+              the Work and Derivative Works thereof.
+        
+              "Contribution" shall mean any work of authorship, including
+              the original version of the Work and any modifications or additions
+              to that Work or Derivative Works thereof, that is intentionally
+              submitted to Licensor for inclusion in the Work by the copyright owner
+              or by an individual or Legal Entity authorized to submit on behalf of
+              the copyright owner. For the purposes of this definition, "submitted"
+              means any form of electronic, verbal, or written communication sent
+              to the Licensor or its representatives, including but not limited to
+              communication on electronic mailing lists, source code control systems,
+              and issue tracking systems that are managed by, or on behalf of, the
+              Licensor for the purpose of discussing and improving the Work, but
+              excluding communication that is conspicuously marked or otherwise
+              designated in writing by the copyright owner as "Not a Contribution."
+        
+              "Contributor" shall mean Licensor and any individual or Legal Entity
+              on behalf of whom a Contribution has been received by Licensor and
+              subsequently incorporated within the Work.
+        
+           2. Grant of Copyright License. Subject to the terms and conditions of
+              this License, each Contributor hereby grants to You a perpetual,
+              worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+              copyright license to reproduce, prepare Derivative Works of,
+              publicly display, publicly perform, sublicense, and distribute the
+              Work and such Derivative Works in Source or Object form.
+        
+           3. Grant of Patent License. Subject to the terms and conditions of
+              this License, each Contributor hereby grants to You a perpetual,
+              worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+              (except as stated in this section) patent license to make, have made,
+              use, offer to sell, sell, import, and otherwise transfer the Work,
+              where such license applies only to those patent claims licensable
+              by such Contributor that are necessarily infringed by their
+              Contribution(s) alone or by combination of their Contribution(s)
+              with the Work to which such Contribution(s) was submitted. If You
+              institute patent litigation against any entity (including a
+              cross-claim or counterclaim in a lawsuit) alleging that the Work
+              or a Contribution incorporated within the Work constitutes direct
+              or contributory patent infringement, then any patent licenses
+              granted to You under this License for that Work shall terminate
+              as of the date such litigation is filed.
+        
+           4. Redistribution. You may reproduce and distribute copies of the
+              Work or Derivative Works thereof in any medium, with or without
+              modifications, and in Source or Object form, provided that You
+              meet the following conditions:
+        
+              (a) You must give any other recipients of the Work or
+                  Derivative Works a copy of this License; and
+        
+              (b) You must cause any modified files to carry prominent notices
+                  stating that You changed the files; and
+        
+              (c) You must retain, in the Source form of any Derivative Works
+                  that You distribute, all copyright, patent, trademark, and
+                  attribution notices from the Source form of the Work,
+                  excluding those notices that do not pertain to any part of
+                  the Derivative Works; and
+        
+              (d) If the Work includes a "NOTICE" text file as part of its
+                  distribution, then any Derivative Works that You distribute must
+                  include a readable copy of the attribution notices contained
+                  within such NOTICE file, excluding those notices that do not
+                  pertain to any part of the Derivative Works, in at least one
+                  of the following places: within a NOTICE text file distributed
+                  as part of the Derivative Works; within the Source form or
+                  documentation, if provided along with the Derivative Works; or,
+                  within a display generated by the Derivative Works, if and
+                  wherever such third-party notices normally appear. The contents
+                  of the NOTICE file are for informational purposes only and
+                  do not modify the License. You may add Your own attribution
+                  notices within Derivative Works that You distribute, alongside
+                  or as an addendum to the NOTICE text from the Work, provided
+                  that such additional attribution notices cannot be construed
+                  as modifying the License.
+        
+              You may add Your own copyright statement to Your modifications and
+              may provide additional or different license terms and conditions
+              for use, reproduction, or distribution of Your modifications, or
+              for any such Derivative Works as a whole, provided Your use,
+              reproduction, and distribution of the Work otherwise complies with
+              the conditions stated in this License.
+        
+           5. Submission of Contributions. Unless You explicitly state otherwise,
+              any Contribution intentionally submitted for inclusion in the Work
+              by You to the Licensor shall be under the terms and conditions of
+              this License, without any additional terms or conditions.
+              Notwithstanding the above, nothing herein shall supersede or modify
+              the terms of any separate license agreement you may have executed
+              with Licensor regarding such Contributions.
+        
+           6. Trademarks. This License does not grant permission to use the trade
+              names, trademarks, service marks, or product names of the Licensor,
+              except as required for reasonable and customary use in describing the
+              origin of the Work and reproducing the content of the NOTICE file.
+        
+           7. Disclaimer of Warranty. Unless required by applicable law or
+              agreed to in writing, Licensor provides the Work (and each
+              Contributor provides its Contributions) on an "AS IS" BASIS,
+              WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+              implied, including, without limitation, any warranties or conditions
+              of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+              PARTICULAR PURPOSE. You are solely responsible for determining the
+              appropriateness of using or redistributing the Work and assume any
+              risks associated with Your exercise of permissions under this License.
+        
+           8. Limitation of Liability. In no event and under no legal theory,
+              whether in tort (including negligence), contract, or otherwise,
+              unless required by applicable law (such as deliberate and grossly
+              negligent acts) or agreed to in writing, shall any Contributor be
+              liable to You for damages, including any direct, indirect, special,
+              incidental, or consequential damages of any character arising as a
+              result of this License or out of the use or inability to use the
+              Work (including but not limited to damages for loss of goodwill,
+              work stoppage, computer failure or malfunction, or any and all
+              other commercial damages or losses), even if such Contributor
+              has been advised of the possibility of such damages.
+        
+           9. Accepting Warranty or Additional Liability. While redistributing
+              the Work or Derivative Works thereof, You may choose to offer,
+              and charge a fee for, acceptance of support, warranty, indemnity,
+              or other liability obligations and/or rights consistent with this
+              License. However, in accepting such obligations, You may act only
+              on Your own behalf and on Your sole responsibility, not on behalf
+              of any other Contributor, and only if You agree to indemnify,
+              defend, and hold each Contributor harmless for any liability
+              incurred by, or claims asserted against, such Contributor by reason
+              of your accepting any such warranty or additional liability.
+        
+           END OF TERMS AND CONDITIONS
+        
+           APPENDIX: How to apply the Apache License to your work.
+        
+              To apply the Apache License to your work, attach the following
+              boilerplate notice, with the fields enclosed by brackets "[]"
+              replaced with your own identifying information. (Don't include
+              the brackets!)  The text should be enclosed in the appropriate
+              comment syntax for the file format. We also recommend that a
+              file or class name and description of purpose be included on the
+              same "printed page" as the copyright notice for easier
+              identification within third-party archives.
+        
+           Copyright [yyyy] [name of copyright owner]
+        
+           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.
+        
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pydantic-settings>=2.10.1
+Requires-Dist: a2a-sdk>=0.3.0
+Requires-Dist: deprecated>=1.2.18
+Requires-Dist: google-adk>=1.10.0
+Requires-Dist: litellm>=1.74.3
+Requires-Dist: loguru>=0.7.3
+Requires-Dist: opentelemetry-exporter-otlp>=1.35.0
+Requires-Dist: opentelemetry-instrumentation-logging>=0.56b0
+Requires-Dist: wrapt>=1.17.2
+Requires-Dist: openai<1.100
+Requires-Dist: volcengine-python-sdk>=4.0.18
+Requires-Dist: volcengine>=1.0.193
+Requires-Dist: agent-pilot-sdk>=0.1.2
+Requires-Dist: fastmcp>=2.11.3
+Requires-Dist: cookiecutter>=2.6.0
+Requires-Dist: omegaconf>=2.3.0
+Requires-Dist: llama-index>=0.14.0
+Requires-Dist: llama-index-embeddings-openai-like>=0.2.2
+Requires-Dist: llama-index-llms-openai-like>=0.5.1
+Requires-Dist: llama-index-vector-stores-opensearch>=0.6.1
+Requires-Dist: psycopg2-binary>=2.9.10
+Requires-Dist: pymysql>=1.1.1
+Requires-Dist: opensearch-py==2.8.0
+Requires-Dist: filetype>=1.2.0
+Provides-Extra: extensions
+Requires-Dist: redis>=5.0; extra == "extensions"
+Requires-Dist: tos>=2.8.4; extra == "extensions"
+Requires-Dist: llama-index-vector-stores-redis>=0.6.1; extra == "extensions"
+Provides-Extra: database
+Requires-Dist: redis>=5.0; extra == "database"
+Requires-Dist: pymysql>=1.1.1; extra == "database"
+Requires-Dist: volcengine>=1.0.193; extra == "database"
+Requires-Dist: tos>=2.8.4; extra == "database"
+Requires-Dist: mem0ai==0.1.118; extra == "database"
+Provides-Extra: eval
+Requires-Dist: prometheus-client>=0.22.1; extra == "eval"
+Requires-Dist: deepeval>=3.2.6; extra == "eval"
+Requires-Dist: google-adk[eval]; extra == "eval"
+Provides-Extra: cli
+Provides-Extra: dev
+Requires-Dist: pre-commit>=4.2.0; extra == "dev"
+Requires-Dist: pytest>=8.4.1; extra == "dev"
+Requires-Dist: pytest-asyncio>=1.0.0; extra == "dev"
+Requires-Dist: pytest-xdist>=3.8.0; extra == "dev"
+Dynamic: license-file
+
+<p align="center">
+    <img src="assets/images/logo.png" alt="Volcengine Agent Development Kit Logo" width="50%">
+</p>
+
+# Volcengine Agent Development Kit
+
+[![License](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](LICENSE)
+[![Deepwiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/volcengine/veadk-python)
+
+An open-source kit for agent development, integrated the powerful capabilities of Volcengine.
+
+For more details, see our [documents](https://volcengine.github.io/veadk-python/).
+
+A [tutorial](https://github.com/volcengine/veadk-python/blob/main/veadk_tutorial.ipynb) is available by Jupyter Notebook, or open it in [Google Colab](https://colab.research.google.com/github/volcengine/veadk-python/blob/main/veadk_tutorial.ipynb) directly.
+
+## Installation
+
+### From PyPI
+
+```python
+pip install veadk-python
+
+# install extensions
+pip install veadk-python[extensions]
+```
+
+### Build from source
+
+We use `uv` to build this project ([how-to-install-uv](https://docs.astral.sh/uv/getting-started/installation/)).
+
+```bash
+git clone ... # clone repo first
+
+cd veadk-python
+
+uv venv --python 3.10
+
+# only install necessary requirements
+uv sync
+
+# or, install extra requirements
+# uv sync --extra database
+# uv sync --extra eval
+# uv sync --extra cli
+
+# or, directly install all requirements
+# uv sync --all-extras
+
+# install veadk-python with editable mode
+uv pip install -e .
+```
+
+## Configuration
+
+We recommand you to create a `config.yaml` file in the root directory of your own project, `VeADK` is able to read it automatically. For running a minimal agent, you just need to set the following configs in your `config.yaml` file:
+
+```yaml
+model:
+  agent:
+    provider: openai
+    name: doubao-seed-1-6-250615
+    api_base: https://ark.cn-beijing.volces.com/api/v3/
+    api_key: # <-- set your Volcengine ARK api key here
+```
+
+You can refer to the [config instructions](https://volcengine.github.io/veadk-python/installation.html#%E9%85%8D%E7%BD%AE) for more details.
+
+## Have a try
+
+Enjoy a minimal agent from VeADK:
+
+```python
+from veadk import Agent
+import asyncio
+
+agent = Agent()
+
+res = asyncio.run(agent.run("hello!"))
+print(res)
+```
+
+## Command line tools
+
+VeADK provides several useful command line tools for faster deployment and optimization, such as:
+
+- `veadk deploy`: deploy your project to [Volcengine VeFaaS platform](https://www.volcengine.com/product/vefaas) (you can use `veadk init` to init a demo project first)
+- `veadk prompt`: otpimize the system prompt of your agent by [PromptPilot](https://promptpilot.volcengine.com)
+
+## Contribution
+
+Before making your contribution to our repository, please install and config the `pre-commit` linter first.
+
+```bash
+pip install pre-commit
+pre-commit install
+```
+
+Before commit or push your changes, please make sure the unittests are passed ,otherwise your PR will be rejected by CI/CD workflow. Running the unittests by:
+
+```bash
+pytest -n 16
+```
+
+## Contact with us
+
+Join our discussion group by scanning the QR code below:
+
+<p align="center">
+    <img src="assets/images/veadk_group_qrcode.jpg" alt="Volcengine Agent Development Kit Logo" width="40%">
+</p>
+
+## License
+
+This project is licensed under the [Apache 2.0 License](./LICENSE).

{veadk_python-0.2.2 → veadk_python-0.2.19}/README.md

@@ -15,6 +15,17 @@ A [tutorial](https://github.com/volcengine/veadk-python/blob/main/veadk_tutorial
 
 ## Installation
 
+### From PyPI
+
+```python
+pip install veadk-python
+
+# install extensions
+pip install veadk-python[extensions]
+```
+
+### Build from source
+
 We use `uv` to build this project ([how-to-install-uv](https://docs.astral.sh/uv/getting-started/installation/)).
 
 ```bash

{veadk_python-0.2.2 → veadk_python-0.2.19}/pyproject.toml

@@ -1,9 +1,10 @@
 [project]
 name = "veadk-python"
-version = "0.2.2"
+version = "0.2.19"
 description = "Volcengine agent development kit, integrations with Volcengine cloud services."
 readme = "README.md"
 requires-python = ">=3.10"
+license = { file = "LICENSE" }
 authors = [
     {name = "Yaozheng Fang", email = "fangyozheng@gmail.com"},
     {name = "Guodong Li", email = "cu.eric.lee@gmail.com"},
@@ -17,35 +18,48 @@ dependencies = [
     "google-adk>=1.10.0", # For basic agent architecture
     "litellm>=1.74.3", # For model inference
     "loguru>=0.7.3", # For better logging
-    "openinference-instrumentation-google-adk>=0.1.1", # For OpenInference instrumentation
     "opentelemetry-exporter-otlp>=1.35.0",
     "opentelemetry-instrumentation-logging>=0.56b0",
     "wrapt>=1.17.2", # For patching built-in functions
-    "openai==1.99.9" # For fix https://github.com/BerriAI/litellm/issues/13710
+    "openai<1.100", # For fix https://github.com/BerriAI/litellm/issues/13710
+    "volcengine-python-sdk>=4.0.18", # For Volcengine API
+    "volcengine>=1.0.193", # For Volcengine sign
+    "agent-pilot-sdk>=0.1.2", # Prompt optimization by Volcengine AgentPilot/PromptPilot toolkits
+    "fastmcp>=2.11.3", # For running MCP
+    "cookiecutter>=2.6.0", # For cloud deploy
+    "omegaconf>=2.3.0", # For agent builder
+    "llama-index>=0.14.0",
+    "llama-index-embeddings-openai-like>=0.2.2",
+    "llama-index-llms-openai-like>=0.5.1",
+    "llama-index-vector-stores-opensearch>=0.6.1",
+    "psycopg2-binary>=2.9.10", # For PostgreSQL database (short term memory)
+    "pymysql>=1.1.1", # For MySQL database (short term memory)
+    "opensearch-py==2.8.0",
+    "filetype>=1.2.0",
 ]
 
 [project.scripts]
 veadk = "veadk.cli.cli:veadk"
 
 [project.optional-dependencies]
+extensions = [
+    "redis>=5.0",                 # For Redis database
+    "tos>=2.8.4",                   # For TOS storage and Viking DB
+    "llama-index-vector-stores-redis>=0.6.1",
+]
 database = [
-    "opensearch-py==2.8.0",         # For OpenSearch database
-    "redis>=6.2.0",                 # For Redis database
+    "redis>=5.0",                 # For Redis database
     "pymysql>=1.1.1",               # For MySQL database
     "volcengine>=1.0.193",          # For Viking DB
     "tos>=2.8.4",                   # For TOS storage and Viking DB
+    "mem0ai==0.1.118",              # For mem0
 ]
 eval = [
     "prometheus-client>=0.22.1",    # For exporting data to Prometheus pushgateway
     "deepeval>=3.2.6",              # For DeepEval-based evaluation
     "google-adk[eval]",             # For Google ADK-based evaluation
 ]
-cli = [
-    "volcengine-python-sdk==4.0.3", # For Volcengine API
-    "agent-pilot-sdk>=0.0.9",       # Prompt optimization by Volcengine AgentPilot/PromptPilot toolkits
-    "fastmcp>=2.11.3",              # For running MCP
-    "cookiecutter>=2.6.0", # For cloud deploy
-]
+cli = []
 dev = [
     "pre-commit>=4.2.0",            # Format checking
     "pytest>=8.4.1",
@@ -75,5 +89,6 @@ include-package-data = true
 
 [tool.ruff]
 exclude = [
-    "veadk/integrations/ve_faas/template/*"
-]
+    "veadk/integrations/ve_faas/template/*",
+    "veadk/integrations/ve_faas/web_template/*"
+]

veadk_python-0.2.19/tests/test_agent.py

@@ -0,0 +1,209 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# 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 os
+from unittest.mock import Mock, patch
+
+from google.adk.agents.llm_agent import LlmAgent
+from google.adk.models.lite_llm import LiteLlm
+from google.adk.tools import load_memory
+
+from veadk import Agent
+from veadk.consts import (
+    DEFAULT_AGENT_NAME,
+    DEFAULT_MODEL_AGENT_API_BASE,
+    DEFAULT_MODEL_AGENT_NAME,
+    DEFAULT_MODEL_AGENT_PROVIDER,
+    DEFAULT_MODEL_EXTRA_CONFIG,
+)
+from veadk.knowledgebase import KnowledgeBase
+from veadk.memory.long_term_memory import LongTermMemory
+from veadk.tools import load_knowledgebase_tool
+from veadk.tracing.telemetry.opentelemetry_tracer import OpentelemetryTracer
+
+
+def test_agent():
+    os.environ["MODEL_EMBEDDING_API_KEY"] = "mocked_api_key"
+
+    knowledgebase = KnowledgeBase(index="test_index", backend="local")
+
+    long_term_memory = LongTermMemory(backend="local")
+    tracer = OpentelemetryTracer()
+
+    extra_config = {
+        "extra_headers": {"thinking": "test"},
+        "extra_body": {"content": "test"},
+    }
+
+    agent = Agent(
+        model_name="test_model_name",
+        model_provider="test_model_provider",
+        model_api_key="test_model_api_key",
+        model_api_base="test_model_api_base",
+        model_extra_config=extra_config,
+        tools=[],
+        sub_agents=[],
+        knowledgebase=knowledgebase,
+        long_term_memory=long_term_memory,
+        tracers=[tracer],
+    )
+
+    assert agent.model.model == f"{agent.model_provider}/{agent.model_name}"  # type: ignore
+
+    expected_config = DEFAULT_MODEL_EXTRA_CONFIG.copy()
+    expected_config["extra_headers"] |= extra_config["extra_headers"]
+    expected_config["extra_body"] |= extra_config["extra_body"]
+
+    assert agent.model_extra_config == expected_config
+
+    assert agent.knowledgebase == knowledgebase
+    assert agent.knowledgebase.backend == "local"  # type: ignore
+
+    assert agent.long_term_memory.backend == "local"  # type: ignore
+    assert load_memory in agent.tools
+
+
+@patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})
+def test_agent_default_values():
+    agent = Agent()
+
+    assert agent.name == DEFAULT_AGENT_NAME
+
+    assert agent.model_name == DEFAULT_MODEL_AGENT_NAME
+    assert agent.model_provider == DEFAULT_MODEL_AGENT_PROVIDER
+    assert agent.model_api_base == DEFAULT_MODEL_AGENT_API_BASE
+
+    assert agent.tools == []
+    assert agent.sub_agents == []
+    assert agent.knowledgebase is None
+    assert agent.long_term_memory is None
+    # assert agent.tracers == []
+
+
+@patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})
+def test_agent_without_knowledgebase():
+    agent = Agent()
+
+    assert agent.knowledgebase is None
+    assert load_knowledgebase_tool.load_knowledgebase_tool not in agent.tools
+
+
+@patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})
+def test_agent_without_long_term_memory():
+    agent = Agent()
+
+    assert agent.long_term_memory is None
+    assert load_memory not in agent.tools
+
+
+@patch("veadk.agent.LiteLlm")
+def test_agent_model_creation(mock_lite_llm):
+    mock_model = Mock()
+    mock_lite_llm.return_value = mock_model
+
+    agent = Agent(
+        model_name="test_model",
+        model_provider="test_provider",
+        model_api_key="test_key",
+        model_api_base="test_base",
+    )
+
+    mock_lite_llm.assert_called_once()
+    assert agent.model == mock_model
+
+
+@patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})
+def test_agent_with_existing_model():
+    existing_model = LiteLlm(model="test_model")
+    agent = Agent(model=existing_model)
+
+    assert agent.model == existing_model
+
+
+@patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})
+def test_agent_model_extra_config_merge():
+    user_config = {
+        "extra_headers": {"custom": "header"},
+        "extra_body": {"custom": "body"},
+        "other_param": "value",
+    }
+
+    agent = Agent(model_extra_config=user_config)
+
+    expected_headers = DEFAULT_MODEL_EXTRA_CONFIG["extra_headers"].copy()
+    expected_headers["custom"] = "header"
+
+    expected_body = DEFAULT_MODEL_EXTRA_CONFIG["extra_body"].copy()
+    expected_body["custom"] = "body"
+
+    assert agent.model_extra_config["extra_headers"] == expected_headers
+    assert agent.model_extra_config["extra_body"] == expected_body
+    assert agent.model_extra_config["other_param"] == "value"
+
+
+@patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})
+def test_agent_empty_model_extra_config():
+    agent = Agent(model_extra_config={})
+
+    assert (
+        agent.model_extra_config["extra_headers"]
+        == DEFAULT_MODEL_EXTRA_CONFIG["extra_headers"]
+    )
+    assert (
+        agent.model_extra_config["extra_body"]
+        == DEFAULT_MODEL_EXTRA_CONFIG["extra_body"]
+    )
+
+
+@patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})
+def test_agent_with_tools():
+    mock_tool = Mock()
+    agent = Agent(tools=[mock_tool])
+
+    assert mock_tool in agent.tools
+
+
+@patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})
+def test_agent_with_sub_agents():
+    adk_agent = LlmAgent(name="agent")
+    veadk_agent = Agent(name="agent")
+    agent = Agent(sub_agents=[adk_agent, veadk_agent])
+
+    assert adk_agent in agent.sub_agents
+    assert veadk_agent in agent.sub_agents
+    assert adk_agent.parent_agent == agent
+    assert veadk_agent.parent_agent == agent
+
+
+@patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})
+def test_agent_with_tracers():
+    tracer1 = OpentelemetryTracer()
+    tracer2 = OpentelemetryTracer()
+
+    agent = Agent(tracers=[tracer1, tracer2])
+
+    assert len(agent.tracers) == 2
+    assert tracer1 in agent.tracers
+    assert tracer2 in agent.tracers
+
+
+@patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})
+def test_agent_custom_name_and_description():
+    custom_name = "CustomAgent"
+    custom_description = "A custom agent for testing"
+
+    agent = Agent(name=custom_name, description=custom_description)
+
+    assert agent.name == custom_name
+    assert agent.description == custom_description