impedance_agent.agent package

Subpackages

impedance_agent.agent.tools package
- Submodules
- impedance_agent.agent.tools.fitter_tools module
  - FitterTools

Submodules

impedance_agent.agent.analysis module

class ImpedanceAnalysisAgent(provider: str = 'deepseek')[source]

Bases: object

Agent for analyzing impedance data using LLM providers

__init__(provider: str = 'deepseek')[source]: Initialize with specified provider

async execute_tool_calls_concurrently(tool_calls, data: ImpedanceData) → Dict[str, Any][source]: Execute all tool calls concurrently

analyze(data: ImpedanceData, model_config: Dict | None = None) → AnalysisResult[source]: Analyze impedance data

impedance_agent.agent.base module

class BaseAgent[source]

Bases: ABC

Abstract base class for provider-specific agents

__init__()[source]: Initialize base agent components

abstract property system_prompt: str: Get provider-specific system prompt

abstract get_user_prompt(data: ImpedanceData, model_config: Dict | None) → str[source]

Get provider-specific user prompt

Parameters:

data – Impedance data to analyze
model_config – Optional ECM configuration

Returns:

Formatted prompt string

abstract create_chat_completion(messages: List[Dict], tools: List[Dict] | None = None, tool_choice: str = 'auto') → Any[source]

Make API call to provider

Parameters:

messages – List of conversation messages
tools – List of available tools
tool_choice – How tools should be selected

Returns:

Provider-specific response object

impedance_agent.agent.deepseek_agent module

class DeepSeekAgent[source]

Bases: BaseAgent

__init__()[source]: Initialize DeepSeek agent

property system_prompt: str: Get provider-specific system prompt

get_user_prompt(data: ImpedanceData, model_config: Dict | None) → str[source]

Get provider-specific user prompt

Parameters:

data – Impedance data to analyze
model_config – Optional ECM configuration

Returns:

Formatted prompt string

create_chat_completion(messages: List[Dict], tools: List[Dict] | None = None, tool_choice: str = 'auto') → Any[source]: Make API call to DeepSeek

impedance_agent.agent.openai_agent module

class OpenAIAgent[source]

Bases: BaseAgent

__init__()[source]: Initialize OpenAI agent

property system_prompt: str: Get provider-specific system prompt

get_user_prompt(data: ImpedanceData, model_config: Dict | None) → str[source]

Get provider-specific user prompt

Parameters:

data – Impedance data to analyze
model_config – Optional ECM configuration

Returns:

Formatted prompt string

create_chat_completion(messages: List[Dict], tools: List[Dict] | None = None, tool_choice: str = 'auto') → Any[source]: Make API call to OpenAI