| import openai |
|
|
|
|
| class MessageChatCompletion: |
| def __init__(self, |
| model: str = 'gpt-5-mini', |
| message: str = '', |
| api_key: str = '', |
| temperature: float = 0.10, |
| top_p: float = 0.95, |
| n: int = 1, |
| stream: bool = False, |
| stop: str = "\n", |
| max_tokens: int = 4096, |
| presence_penalty: float = 0.0, |
| frequency_penalty: float = 0.0, |
| logit_bias: int = None, |
| user: str = ''): |
|
|
| self.api_key = api_key |
| openai.api_key = self.api_key |
|
|
| if model in ["gpt-4o","gpt-4o-mini","gpt-5-mini","gpt-5"]: |
| self.endpoint = "https://api.openai.com/v1/chat/completions" |
| else: |
| self.endpoint = "https://api.openai.com/v1/completions" |
|
|
| self.headers = { |
| "Content-Type": "application/json", |
| "Authorization": f"Bearer {self.api_key}", |
| } |
|
|
| self.prompt = { |
| "model": model, |
| "messages": [], |
| "temperature": temperature, |
| "top_p": top_p, |
| "n": n, |
| "stream": stream, |
| "stop": stop, |
| "presence_penalty": presence_penalty, |
| "frequency_penalty": frequency_penalty |
| } |
|
|
| if max_tokens is not None: |
| self.prompt["max_tokens"] = max_tokens |
|
|
| if logit_bias is not None: |
| self.prompt["logit_bias"] = logit_bias |
|
|
| if user != '': |
| self.prompt["user"] = user |
|
|
| if message != '': |
| self.new_user_message(content=message) |
|
|
| self.response = '' |
|
|
| self.error = False |
|
|
| def new_message(self, role: str = 'user', content: str = '', name: str = ''): |
| new_message = {"role": role, "content": f"{content}"} |
| if name != '': |
| new_message['name'] = name |
|
|
| self.prompt['messages'].append(new_message) |
|
|
| def new_user_message(self, content: str = '', name: str = ''): |
| self.new_message(role='user', content=content, name=name) |
|
|
| def new_system_message(self, content: str = '', name: str = ''): |
| self.new_message(role='system', content=content, name=name) |
|
|
| def new_assistant_message(self, content: str = '', name: str = ''): |
| self.new_message(role='assistant', content=content, name=name) |
|
|
| def get_last_message(self): |
| return self.prompt['messages'][-1]['content'] |
|
|
| def send_message(self): |
|
|
| try: |
| self.error = False |
|
|
| response = openai.chat.completions.create( |
| model=self.prompt['model'], |
| messages=self.prompt['messages'], |
| frequency_penalty=self.prompt['frequency_penalty'], |
| temperature=self.prompt['temperature'], |
| max_tokens=self.prompt['max_tokens'], |
| top_p=self.prompt['top_p'], |
| presence_penalty=self.prompt['presence_penalty'], |
| stream=self.prompt['stream'] |
| ) |
|
|
| full_response = response.choices[0].message.content |
|
|
| |
| |
| |
| |
| |
| |
|
|
| self.new_system_message(content=full_response) |
|
|
| return self.response |
|
|
| except Exception as e: |
| self.error = True |
|
|
| if self.api_key == '' or self.api_key is None: |
| self.new_system_message(content="API key is missing") |
| else: |
| self.new_system_message(content=f"Unable to generate ChatCompletion response\nException: {e}") |
| return e |
|
|
