ranjangoel/GPT-PDF
5
1U
2 �Qd� � @ s8 d dl mZ d dlZG dd� de�ZG dd� de�ZdS )� )�ListNc @ s" e Zd Zdd�dd�Zdd� ZdS )�ModelInterfaceN��returnc C s d S �N� )�selfr r �@/Users/alpaca/ChatGPT-Paper-Reader/gpt_reader/model_interface.py�__init__ s zModelInterface.__init__c G s d S r r )r �argsr r r �send_msg3 s zModelInterface.send_msg)�__name__�4__module__� __qualname__r5 r r r r r r s r c @ s0 e Zd Zd dd�dd�Zd ee d�d d6�ZdS )
� OpenAIModel�
gpt-3.5-turbo皙�����?Nr c C s |t _|| _|| _d S r )�openai�api_key�model� temperature)r r r r r r r r7 s zOpenAIModel.__init__T)�msgc C s6 t jj| j|| jd�}|r.|d d d d S |S d S )N)r �messagesr �choicesr �message�content)r �ChatCompletion�creater r )r r Zreturn_raw_text�responser r r r s �zOpenAIModel.send_msg)r r )T)r
r r r8 r �dictr r r r r r s r )�typingr r �objectr r r r r r �<module> s