18 lines
335 B
Python
18 lines
335 B
Python
|
from dataclasses import dataclass
|
||
|
|
from typing import Literal
|
||
|
|
|
||
|
|
|
||
|
|
Service = Literal["openai", "bard"]
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class Model:
|
||
|
|
id: str
|
||
|
|
usage_name: str
|
||
|
|
service: Service
|
||
|
|
|
||
|
|
|
||
|
|
GPT_3 = Model(id="gpt-3.5-turbo-16k-0613", usage_name="GPT-3", service="openai")
|
||
|
|
|
||
|
|
GPT_4 = Model(id="gpt-4-16k-0613", usage_name="GPT-4", service="openai")
|