vivexy-agents-core 0.2.0__tar.gz

vivexy_agents_core-0.2.0/.gitignore

@@ -0,0 +1 @@
+src/agents_core/file_to_process/

vivexy_agents_core-0.2.0/PKG-INFO

@@ -0,0 +1,70 @@
+Metadata-Version: 2.5
+Name: vivexy-agents-core
+Version: 0.2.0
+Summary: Core agent orchestration, tools, and business logic for Vivexy
+Project-URL: Repository, https://github.com/pranavubarhande/vivexy
+License-Expression: AGPL-3.0-or-later
+Requires-Python: >=3.11
+Requires-Dist: alembic>=1.13.0
+Requires-Dist: authlib>=1.3.2
+Requires-Dist: bcrypt>=4.1.2
+Requires-Dist: boto3>=1.42.43
+Requires-Dist: composio-langchain>=0.11.1
+Requires-Dist: crawl4ai>=0.4.0
+Requires-Dist: cryptography>=43.0.0
+Requires-Dist: daytona>=0.18.1
+Requires-Dist: deepagents>=0.3.3
+Requires-Dist: dnspython>=2.6.0
+Requires-Dist: exa-py>=0.1.8
+Requires-Dist: flasgger
+Requires-Dist: flask-cors>=4.0.0
+Requires-Dist: flask-sqlalchemy>=3.1.1
+Requires-Dist: flask>=3.0.0
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: langchain-anthropic>=1.1.0
+Requires-Dist: langchain-cohere>=0.3.0
+Requires-Dist: langchain-community==0.4.1
+Requires-Dist: langchain-core>=1.2.6
+Requires-Dist: langchain-daytona>=0.0.2
+Requires-Dist: langchain-fireworks>=0.2.0
+Requires-Dist: langchain-google-genai==4.1.2
+Requires-Dist: langchain-groq>=1.1.0
+Requires-Dist: langchain-huggingface==1.2.0
+Requires-Dist: langchain-mistralai>=0.2.0
+Requires-Dist: langchain-ollama>=0.3.0
+Requires-Dist: langchain-openai>=1.1.0
+Requires-Dist: langchain-together>=0.4.0
+Requires-Dist: langchain-xai>=0.2.0
+Requires-Dist: langchain==1.2.0
+Requires-Dist: langgraph-checkpoint-redis>=0.3.6
+Requires-Dist: langgraph==1.0.5
+Requires-Dist: markdown2>=2.5.4
+Requires-Dist: openpyxl==3.1.5
+Requires-Dist: pandas>=2.0.0
+Requires-Dist: pika>=1.3.0
+Requires-Dist: pinecone>=5.0.0
+Requires-Dist: playwright>=1.48.0
+Requires-Dist: psycopg2-binary>=2.9.0
+Requires-Dist: pydantic>=2.5.0
+Requires-Dist: pyjwt>=2.8.0
+Requires-Dist: pypandoc>=1.16.2
+Requires-Dist: pypdf==6.5.0
+Requires-Dist: python-docx==1.2.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: rank-bm25==0.2.2
+Requires-Dist: redis>=5.0.0
+Requires-Dist: reportlab>=4.0.0
+Requires-Dist: requests>=2.31.0
+Requires-Dist: resend>=2.5.1
+Requires-Dist: sentence-transformers==5.2.0
+Requires-Dist: sentry-sdk[flask]>=2.54.0
+Requires-Dist: sqlalchemy<2.1,>=2.0.0
+Requires-Dist: stripe>=11.0.0
+Requires-Dist: weasyprint>=68.1
+Requires-Dist: yfinance>=0.2.0
+Provides-Extra: dev
+Requires-Dist: black>=24.0.0; extra == 'dev'
+Requires-Dist: flake8>=7.0.0; extra == 'dev'
+Requires-Dist: isort>=5.0.0; extra == 'dev'
+Requires-Dist: pytest-flask>=1.3.0; extra == 'dev'
+Requires-Dist: pytest>=8.0.0; extra == 'dev'

vivexy_agents_core-0.2.0/constraints.txt

@@ -0,0 +1,278 @@
+aiodns==4.0.0
+aiofiles==24.1.0
+aiohappyeyeballs==2.6.1
+aiohttp==3.13.3
+aiohttp-retry==2.9.1
+aiosignal==1.4.0
+aiosqlite==0.22.1
+alembic==1.18.4
+alphashape==1.3.1
+annotated-types==0.7.0
+anthropic==0.86.0
+anyio==4.13.0
+async-timeout==5.0.1 ; python_full_version < '3.11.3'
+asyncstdlib-fw==3.13.2
+attrs==23.2.0
+authlib==1.8.0
+backports-zstd==1.3.0 ; python_full_version < '3.14' and platform_python_implementation == 'CPython'
+bcrypt==5.0.0
+beautifulsoup4==4.14.3
+betterproto-fw==2.0.3
+blinker==1.9.0
+boto3==1.42.74
+botocore==1.42.74
+bracex==2.6
+brotli==1.2.0
+brotlicffi==1.2.0.1 ; platform_python_implementation != 'CPython'
+certifi==2026.2.25
+cffi==2.0.0
+chardet==7.4.0.post1
+charset-normalizer==3.4.6
+click==8.3.1
+click-log==0.4.0
+cohere==5.21.1
+colorama==0.4.6 ; sys_platform == 'win32'
+composio==0.11.3
+composio-client==1.28.0
+composio-langchain==0.11.3
+crawl4ai==0.8.6
+cryptography==46.0.5
+cssselect==1.4.0
+cssselect2==0.9.0
+cuda-bindings==13.2.0 ; sys_platform == 'linux'
+cuda-pathfinder==1.4.4 ; sys_platform == 'linux'
+cuda-toolkit==13.0.2 ; sys_platform == 'linux'
+curl-cffi==0.13.0
+dataclasses-json==0.6.7
+daytona==0.155.0
+daytona-api-client==0.155.0
+daytona-api-client-async==0.155.0
+daytona-toolbox-api-client==0.155.0
+daytona-toolbox-api-client-async==0.155.0
+deepagents==0.3.3
+deprecated==1.3.1
+distro==1.9.0
+dnspython==2.9.0
+docstring-parser==0.17.0
+environs==14.6.0
+et-xmlfile==2.0.0
+exa-py==2.10.2
+fake-useragent==2.2.0
+fastavro==1.12.1
+fastuuid==0.14.0
+filelock==3.25.2
+filetype==1.2.0
+fireworks-ai==0.19.20
+flasgger==0.9.7.1
+flask==3.1.3
+flask-cors==6.0.2
+flask-sqlalchemy==3.1.1
+fonttools==4.62.1
+frozendict==2.4.7
+frozenlist==1.8.0
+fsspec==2026.2.0
+google-auth==2.49.1
+google-genai==1.68.0
+googleapis-common-protos==1.73.0
+greenlet==3.3.2
+groq==0.37.1
+grpcio==1.80.0
+grpclib==0.4.9
+h11==0.16.0
+h2==4.3.0
+hf-xet==1.4.2 ; platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64'
+hpack==4.1.0
+httpcore==1.0.9
+httpx==0.28.1
+httpx-sse==0.4.3
+httpx-ws==0.9.0
+huggingface-hub==0.36.2
+humanize==4.15.0
+hyperframe==6.1.0
+idna==3.11
+importlib-metadata==8.7.1
+itsdangerous==2.2.0
+jinja2==3.1.6
+jiter==0.13.0
+jmespath==1.1.0
+joblib==1.5.3
+joserfc==1.7.5
+json-schema-to-pydantic==0.4.11
+jsonpatch==1.33
+jsonpath-ng==1.8.0
+jsonpointer==3.1.1
+jsonschema==4.26.0
+jsonschema-specifications==2025.9.1
+langchain==1.2.0
+langchain-anthropic==1.3.1
+langchain-classic==1.0.1
+langchain-cohere==0.5.0
+langchain-community==0.4.1
+langchain-core==1.3.0
+langchain-daytona==0.0.2
+langchain-fireworks==1.1.0
+langchain-google-genai==4.1.2
+langchain-groq==1.1.1
+langchain-huggingface==1.2.0
+langchain-mistralai==1.1.1
+langchain-ollama==1.0.1
+langchain-openai==1.2.0
+langchain-text-splitters==1.1.0
+langchain-together==0.4.0
+langchain-xai==1.2.1
+langgraph==1.0.5
+langgraph-checkpoint==3.0.1
+langgraph-checkpoint-redis==0.3.6
+langgraph-prebuilt==1.0.8
+langgraph-sdk==0.3.12
+langsmith==0.7.22
+lark==1.3.1
+lxml==5.4.0
+mako==1.3.10
+markdown-it-py==4.0.0
+markdown2==2.5.5
+markupsafe==3.0.3
+marshmallow==3.26.2
+mdurl==0.1.2
+mistune==3.2.0
+ml-dtypes==0.6.0
+mmh3==5.2.1
+mpmath==1.3.0
+multidict==6.7.1
+multitasking==0.0.12
+mypy-extensions==1.1.0
+networkx==3.6.1
+nltk==3.9.4
+numpy==2.4.3
+nvidia-cublas==13.1.0.3 ; sys_platform == 'linux'
+nvidia-cuda-cupti==13.0.85 ; sys_platform == 'linux'
+nvidia-cuda-nvrtc==13.0.88 ; sys_platform == 'linux'
+nvidia-cuda-runtime==13.0.96 ; sys_platform == 'linux'
+nvidia-cudnn-cu13==9.19.0.56 ; sys_platform == 'linux'
+nvidia-cufft==12.0.0.61 ; sys_platform == 'linux'
+nvidia-cufile==1.15.1.6 ; sys_platform == 'linux'
+nvidia-curand==10.4.0.35 ; sys_platform == 'linux'
+nvidia-cusolver==12.0.4.66 ; sys_platform == 'linux'
+nvidia-cusparse==12.6.3.3 ; sys_platform == 'linux'
+nvidia-cusparselt-cu13==0.8.0 ; sys_platform == 'linux'
+nvidia-nccl-cu13==2.28.9 ; sys_platform == 'linux'
+nvidia-nvjitlink==13.0.88 ; sys_platform == 'linux'
+nvidia-nvshmem-cu13==3.4.5 ; sys_platform == 'linux'
+nvidia-nvtx==13.0.85 ; sys_platform == 'linux'
+obstore==0.8.2
+ollama==0.6.1
+openai==2.29.0
+openpyxl==3.1.5
+opentelemetry-api==1.40.0
+opentelemetry-exporter-otlp-proto-common==1.40.0
+opentelemetry-exporter-otlp-proto-http==1.40.0
+opentelemetry-instrumentation==0.61b0
+opentelemetry-instrumentation-aiohttp-client==0.61b0
+opentelemetry-proto==1.40.0
+opentelemetry-sdk==1.40.0
+opentelemetry-semantic-conventions==0.61b0
+opentelemetry-util-http==0.61b0
+orjson==3.11.7
+ormsgpack==1.12.2
+packaging==24.2
+pandas==3.0.1
+patchright==1.58.2
+peewee==4.0.2
+pika==1.3.2
+pillow==12.1.1
+pinecone==8.1.0
+pinecone-plugin-assistant==3.0.2
+pinecone-plugin-interface==0.0.7
+platformdirs==4.9.4
+playwright==1.58.0
+playwright-stealth==2.0.2
+propcache==0.4.1
+protobuf==5.29.3
+psutil==7.2.2
+psycopg2-binary==2.9.11
+pyasn1==0.6.3
+pyasn1-modules==0.4.2
+pycares==5.0.1
+pycparser==3.0 ; implementation_name != 'PyPy'
+pydantic==2.12.5
+pydantic-core==2.41.5
+pydantic-settings==2.13.1
+pydyf==0.12.1
+pyee==13.0.1
+pygments==2.19.2
+pyjwt==2.12.1
+pyopenssl==26.0.0
+pypandoc==1.17
+pypdf==6.5.0
+pyphen==0.17.2
+pysher==1.0.8
+python-dateutil==2.9.0.post0
+python-docx==1.2.0
+python-dotenv==1.2.2
+python-multipart==0.0.22
+python-ulid==4.0.1
+pytz==2026.1.post1
+pyyaml==6.0.3
+rank-bm25==0.2.2
+redis==7.4.0
+redisvl==0.25.1
+referencing==0.37.0
+regex==2026.2.28
+reportlab==4.4.10
+requests==2.32.5
+requests-toolbelt==1.0.0
+resend==2.49.1
+rich==14.3.3
+rpds-py==0.30.0
+rtree==1.4.1
+ruff==0.9.10
+s3transfer==0.16.0
+safetensors==0.7.0
+scikit-learn==1.8.0
+scipy==1.17.1
+sentence-transformers==5.2.0
+sentry-sdk==2.56.0
+setuptools==81.0.0
+shapely==2.1.2
+six==1.17.0
+sniffio==1.3.1
+snowballstemmer==2.2.0
+soupsieve==2.8.3
+sqlalchemy==2.0.48
+stripe==16.0.0
+sympy==1.14.0
+tenacity==9.1.4
+threadpoolctl==3.6.0
+tiktoken==0.12.0
+tinycss2==1.5.1
+tinyhtml5==2.1.0
+tokenizers==0.22.2
+toml==0.10.2
+torch==2.11.0
+tqdm==4.67.3
+transformers==4.57.6
+trimesh==4.11.5
+triton==3.6.0 ; sys_platform == 'linux'
+types-pyyaml==6.0.12.20260408
+types-requests==2.33.0.20260408
+typing-extensions==4.15.0
+typing-inspect==0.9.0
+typing-inspection==0.4.2
+tzdata==2025.3 ; sys_platform == 'emscripten' or sys_platform == 'win32'
+unclecode-litellm==1.81.13
+urllib3==2.6.3
+uuid-utils==0.14.1
+wcmatch==10.1
+weasyprint==68.1
+webencodings==0.5.1
+websocket-client==1.9.0
+websockets==15.0.1
+werkzeug==3.1.7
+wrapt==1.17.3
+wsproto==1.3.2
+xxhash==3.6.0
+yarl==1.23.0
+yfinance==1.2.0
+zipp==3.23.0
+zopfli==0.4.1
+zstandard==0.25.0

vivexy_agents_core-0.2.0/pyproject.toml

@@ -0,0 +1,96 @@
+[project]
+name = "vivexy-agents-core"
+version = "0.2.0"
+description = "Core agent orchestration, tools, and business logic for Vivexy"
+requires-python = ">=3.11"
+license = "AGPL-3.0-or-later"
+dependencies = [
+    "flask>=3.0.0",
+    "flask-cors>=4.0.0",
+    "flask-sqlalchemy>=3.1.1",
+    "sqlalchemy>=2.0.0,<2.1",
+    "alembic>=1.13.0",
+    "python-dotenv>=1.0.0",
+    "psycopg2-binary>=2.9.0",
+    "redis>=5.0.0",
+    "langchain==1.2.0",
+    "langchain-core>=1.2.6",
+    "langchain-google-genai==4.1.2",
+    "langchain-community==0.4.1",
+    "langchain-huggingface==1.2.0",
+    "sentence_transformers==5.2.0",
+    "rank-bm25==0.2.2",
+    "pypdf==6.5.0",
+    "openpyxl==3.1.5",
+    "python-docx==1.2.0",
+    "langgraph==1.0.5",
+    "pinecone>=5.0.0",
+    "pandas>=2.0.0",
+    "pydantic>=2.5.0",
+    "bcrypt>=4.1.2",
+    "pyjwt>=2.8.0",
+    "authlib>=1.3.2",
+    "resend>=2.5.1",
+    "yfinance>=0.2.0",
+    "requests>=2.31.0",
+    "boto3>=1.42.43",
+    "reportlab>=4.0.0",
+    "langchain-groq>=1.1.0",
+    "langchain-openai>=1.1.0",
+    "langchain-anthropic>=1.1.0",
+    "langchain-mistralai>=0.2.0",
+    "langchain-together>=0.4.0",
+    "langchain-fireworks>=0.2.0",
+    "langchain-cohere>=0.3.0",
+    "langchain-ollama>=0.3.0",
+    "langchain-xai>=0.2.0",
+    "cryptography>=43.0.0",
+    "composio-langchain>=0.11.1",
+    "pypandoc>=1.16.2",
+    "markdown2>=2.5.4",
+    "weasyprint>=68.1",
+    "pika>=1.3.0",
+    "sentry-sdk[flask]>=2.54.0",
+    "deepagents>=0.3.3",
+    "langchain-daytona>=0.0.2",
+    "daytona>=0.18.1",
+    "exa-py>=0.1.8",
+    "crawl4ai>=0.4.0",
+    "playwright>=1.48.0",
+    "flasgger",
+    "langgraph-checkpoint-redis>=0.3.6",
+    "httpx>=0.27.0",
+    "stripe>=11.0.0",
+    "dnspython>=2.6.0",
+]
+
+[project.urls]
+Repository = "https://github.com/pranavubarhande/vivexy"
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=8.0.0",
+    "black>=24.0.0",
+    "isort>=5.0.0",
+    "flake8>=7.0.0",
+    "pytest-flask>=1.3.0",
+]
+
+[tool.black]
+line-length = 88
+target-version = ["py311"]
+include = "\\.pyi?$"
+
+[tool.isort]
+profile = "black"
+multi_line_output = 3
+line_length = 88
+known_first_party = ["agents_core"]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/agents_core"]
+exclude = ["**/__tests__"]

vivexy_agents_core-0.2.0/pytest.ini

@@ -0,0 +1,5 @@
+[pytest]
+pythonpath = src
+testpaths = src
+python_files = *.test.py
+addopts = --import-mode=importlib

vivexy_agents_core-0.2.0/src/agents_core/__tests__/extensions.test.py

@@ -0,0 +1,151 @@
+from unittest.mock import MagicMock, patch
+
+import pytest
+from agents_core import extensions
+from agents_core.agents.config import (
+    UNIFIED_SYSTEM_PROMPT,
+    get_agent_metadata,
+    get_system_prompt,
+)
+from agents_core.agents.skill_loader import load_builtin_skills
+from agents_core.app_factory import create_app
+from agents_core.extensions import ToolContext, VivexyExtension
+from flask import Blueprint
+from langchain_core.tools import tool
+
+
+@tool
+def clause_lookup(clause: str) -> str:
+    """Look up a clause."""
+    return clause
+
+
+legal_bp = Blueprint("legal_extension", __name__, url_prefix="/legal")
+
+
+@legal_bp.route("/ping")
+def ping():
+    return {"ok": True}
+
+
+def _entry_point(name, value):
+    entry_point = MagicMock()
+    entry_point.name = name
+    entry_point.load.return_value = value
+    return entry_point
+
+
+@pytest.fixture
+def skills_dir(tmp_path):
+    skill_dir = tmp_path / "nda-review"
+    skill_dir.mkdir()
+    (skill_dir / "SKILL.md").write_text(
+        "---\nname: nda-review\ndescription: Review NDAs. More text.\n---\nBody"
+    )
+    return tmp_path
+
+
+@pytest.fixture
+def legal_extension(skills_dir):
+    return VivexyExtension(
+        name="legal",
+        tool_groups={"clauses": lambda context: [clause_lookup]},
+        agents={
+            "nda-reviewer": {
+                "name": "NDA Reviewer",
+                "description": "Reviews NDAs.",
+                "tools": ["base", "clauses"],
+                "system_prompt": "You review NDAs.",
+            }
+        },
+        blueprints=[legal_bp],
+        skill_dirs=[skills_dir],
+    )
+
+
+@pytest.fixture
+def installed(legal_extension):
+    def _install(*entry_point_values):
+        entry_points = [
+            _entry_point(f"ext{index}", value)
+            for index, value in enumerate(entry_point_values)
+        ]
+        patcher = patch.object(extensions, "entry_points", return_value=entry_points)
+        patcher.start()
+        extensions.load_extensions.cache_clear()
+        load_builtin_skills.cache_clear()
+        return patcher
+
+    patchers = []
+    yield lambda *values: patchers.append(_install(*values))
+    for patcher in patchers:
+        patcher.stop()
+    extensions.load_extensions.cache_clear()
+    load_builtin_skills.cache_clear()
+
+
+def test_no_extensions_keeps_core_agents(installed):
+    installed()
+    assert "contract-review" in get_agent_metadata()
+    assert extensions.get_extension_tool_groups() == {}
+
+
+def test_extension_agents_and_prompt(installed, legal_extension):
+    installed(legal_extension)
+    metadata = get_agent_metadata()["nda-reviewer"]
+    assert metadata["id"] == "nda-reviewer"
+    assert "system_prompt" not in metadata
+    assert get_system_prompt("nda-reviewer") == "You review NDAs."
+    assert get_system_prompt("unknown-agent") == UNIFIED_SYSTEM_PROMPT
+
+
+def test_extension_factory_and_tool_groups(installed, legal_extension):
+    installed(lambda: legal_extension)
+    loader = extensions.get_extension_tool_groups()["clauses"]
+    assert loader(ToolContext(chat_id="c1")) == [clause_lookup]
+
+
+def test_invalid_and_failing_extensions_are_skipped(installed, legal_extension):
+    broken = MagicMock()
+    broken.name = "broken"
+    broken.load.side_effect = ImportError("missing")
+    patcher = patch.object(
+        extensions,
+        "entry_points",
+        return_value=[
+            broken,
+            _entry_point("bad", object()),
+            _entry_point("ok", legal_extension),
+        ],
+    )
+    with patcher:
+        extensions.load_extensions.cache_clear()
+        assert [e.name for e in extensions.load_extensions()] == ["legal"]
+    extensions.load_extensions.cache_clear()
+
+
+def test_extension_skills_are_indexed(installed, legal_extension):
+    installed(legal_extension)
+    skill = load_builtin_skills()["nda-review"]
+    assert skill["description"] == "Review NDAs."
+
+
+def test_create_app_registers_extension_blueprints(installed, legal_extension):
+    installed(legal_extension)
+    config = type(
+        "TestConfig",
+        (),
+        {
+            "SQLALCHEMY_DATABASE_URI": "sqlite:///:memory:",
+            "TESTING": True,
+            "GOOGLE_CLIENT_ID": "",
+            "GOOGLE_CLIENT_SECRET": "",
+            "MICROSOFT_CLIENT_ID": "",
+            "MICROSOFT_CLIENT_SECRET": "",
+        },
+    )
+    extra_bp = Blueprint("extra", __name__, url_prefix="/extra")
+    app = create_app(config, blueprints=[extra_bp])
+    assert "legal_extension" in app.blueprints
+    assert "extra" in app.blueprints
+    assert app.test_client().get("/legal/ping").get_json() == {"ok": True}

vivexy_agents_core-0.2.0/src/agents_core/agents/__init__.py

@@ -0,0 +1,3 @@
+from agents_core.agents.agent_factory import create_agent, get_available_agents
+
+__all__ = ["create_agent", "get_available_agents"]

vivexy_agents_core-0.2.0/src/agents_core/agents/__tests__/config.test.py

@@ -0,0 +1,59 @@
+import os
+from unittest.mock import patch
+
+import pytest
+
+from agents_core.agents.config import AgentConfig, AgentPhase, AgentType, TaskItem, TaskStatus
+
+
+def test_task_item_to_dict():
+    task = TaskItem(
+        id="t1", title="Test Task", description="A description", status="RUNNING"
+    )
+    assert task.to_dict() == {
+        "id": "t1",
+        "title": "Test Task",
+        "description": "A description",
+        "status": "RUNNING",
+        "error": None,
+    }
+
+
+@pytest.mark.parametrize(
+    "provider, env_var, mocked_key",
+    [
+        ("google", "GOOGLE_AI_API_KEY", "google_key"),
+        ("openai", "OPENAI_API_KEY", "openai_key"),
+        ("anthropic", "ANTHROPIC_API_KEY", "anthropic_key"),
+        ("groq", "GROQ_API_KEY", "groq_key"),
+    ],
+)
+def test_agent_config_from_env(provider, env_var, mocked_key):
+    with patch.dict(os.environ, {env_var: mocked_key}):
+        # Mock other keys to ensure it picks the right one
+        for other_var in [
+            "GOOGLE_AI_API_KEY",
+            "OPENAI_API_KEY",
+            "ANTHROPIC_API_KEY",
+            "GROQ_API_KEY",
+        ]:
+            if other_var != env_var:
+                os.environ[other_var] = "wrong_key"
+
+        config = AgentConfig.from_env(provider=provider, model="test-model")
+
+        assert config.provider == provider
+        assert config.model == "test-model"
+        assert config.api_key == mocked_key
+
+
+def test_agent_config_fallback_google_key():
+    with patch.dict(os.environ, {}, clear=True):
+        os.environ["GOOGLE_API_KEY"] = "fallback_google_key"
+        config = AgentConfig.from_env(provider="google")
+        assert config.api_key == "fallback_google_key"
+
+
+def test_agent_config_unsupported_provider():
+    config = AgentConfig.from_env(provider="unsupported")
+    assert config.api_key == ""