�
�n�hY= � � � d dl Z d dlmZmZmZ d dlmZ d dlZd dlZd dlZ e j e
� � Z G d� d� � ZdS )� N)�Dict�List�Any)�datetimec � � e Zd Zddefd�Z ddedeeef dee deeef fd�Zdedeeef fd �Z defd
�Z
dedeeef fd�Zdededee fd
�Zdeeef fd�Z
dee fd�Zd� ZdS )�AIChatAssistantN�openai_api_keyc �� � |pt j dd� � | _ t j dd� � | _ t j dd� � | _ g | _ i | _ d S )N�OPENAI_API_KEY�
c3_api_key�OPENAI_API_URLzhttps://api.comput3.ai/v1�OPENAI_MODELz
llama3:70b)�os�getenvr �openai_api_url�model�conversation_history�user_context)�selfr s �+/home/runner/workspace/ai_chat_assistant.py�__init__zAIChatAssistant.__init__ s] � �,�Y�� �:J�L�0Y�0Y��� �i�(8�:U�V�V����Y�~�|�<�<��
� %'��!������ �user_message�portfolio_data�transaction_history�returnc �
� |r
|| j d<