mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 23:58:34 +00:00
29 lines
926 B
Python
29 lines
926 B
Python
import unittest
|
|
from unittest.mock import MagicMock
|
|
|
|
from crewai import Task
|
|
from crewai.utilities.planning_handler import CrewPlanner
|
|
|
|
|
|
def test_planning_llm_inherits_auth_params():
|
|
"""Test that planning LLM inherits authentication parameters from agent LLM."""
|
|
mock_llm = MagicMock()
|
|
mock_llm.base_url = "https://api.custom-provider.com/v1"
|
|
mock_llm.api_version = "2023-05-15"
|
|
|
|
task = Task(
|
|
description="Test Task",
|
|
expected_output="Test Output"
|
|
)
|
|
|
|
planner = CrewPlanner(
|
|
tasks=[task],
|
|
planning_agent_llm=None, # This should trigger the inheritance logic
|
|
agent_llm=mock_llm
|
|
)
|
|
|
|
assert hasattr(planner, 'planning_agent_llm')
|
|
assert hasattr(planner.planning_agent_llm, 'base_url')
|
|
assert planner.planning_agent_llm.base_url == "https://api.custom-provider.com/v1"
|
|
assert planner.planning_agent_llm.api_version == "2023-05-15"
|