Format: application/x-python-code
1 o
2
3
4 �h� � @ s> d dl mZmZmZ d dlmZ d dlmZ G dd� d�ZdS )� )�Dict�Any�List)� query_ollama)� execute_toolc @ s� e Zd Zdd� Zdeddfdd�Zdefdd �Zd
5 edeeef fd d �Z d
6 ede
7 fdd�Z dededdfdd�Z deeef ddfdd�Z
8 dS )�Plannerc C s g | _ i | _d S �N)� chat_history�context)�self� r �:C:\Users\shubh\Videos\agent\agent\modules\agent\planner.py�__init__ s
9 zPlanner.__init__�
10 user_input�returnNc C s | j d|d� d S )N�user��role�content)�_update_history)r r r r r
11 �
12 receive_input
13 s zPlanner.receive_inputc � sh �d}t |�D ]*}t| j| jdd�I d H }d|v r'| �|�I d H }| �|� q| jd|d� | S dS )N� �planner)�promptr
14 �
15 agent_type�
16 [ToolCall:� assistantr z!Maximum processing cycles reached)�ranger r r
17 �_handle_tool_call�_update_contextr )r Z
18 max_cycles�_Z llm_responseZ tool_resultr r r
19 �process
20 s � � zPlanner.process�responsec � sR �|� d�d � d�d �� }| �|�\}}t||�I d H }| jdt|�d� |S )Nr � �]r �toolr )�split�strip�_parse_tool_callr r �str)r r" � tool_callZ tool_name�args�resultr r r
21 r s � zPlanner._handle_tool_callr* c C s* |� dd�d t|� dd�d �d��fS )N�(r# r �))r& �eval�rstrip)r r* r r r
22 r( % s *zPlanner._parse_tool_callr r c C s | j �||d�� d S )Nr )r �append)r r r r r r
23 r ( s zPlanner._update_history� tool_datac C s | j �|� d S r )r
24 �update)r r2 r r r
25 r + s zPlanner._update_context)�__name__�
26 __module__� __qualname__r r) r r! r r r �tupler( r r r r r r
27 r s r N) �typingr r r �modules.llm.ollama_clientr Zmodules.tools.tool_executorr r r r r r
28 �<module> s