17 lines
521 B
Python
17 lines
521 B
Python
"""Defines the ChatCompletionResponse class."""
|
|
|
|
from typing import Optional
|
|
|
|
from aisuite.framework.choice import Choice
|
|
from aisuite.framework.message import CompletionUsage
|
|
|
|
|
|
# pylint: disable=too-few-public-methods
|
|
class ChatCompletionResponse:
|
|
"""Used to conform to the response model of OpenAI."""
|
|
|
|
def __init__(self):
|
|
"""Initializes the ChatCompletionResponse."""
|
|
self.choices = [Choice()] # Adjust the range as needed for more choices
|
|
self.usage: Optional[CompletionUsage] = None
|