File size: 1,596 Bytes
1e2f57f
 
 
 
 
1fc65b1
1e2f57f
1fc65b1
1e2f57f
 
 
 
1fc65b1
1e2f57f
 
 
 
 
 
 
 
4271ca6
1fc65b1
 
1e2f57f
 
 
 
 
 
1fc65b1
1e2f57f
 
 
 
 
 
5684fd5
1e2f57f
5684fd5
1e2f57f
 
 
5684fd5
1e2f57f
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import sys

from .typing import MetaModels, Union
from . import Provider


class Model(metaclass=MetaModels):

    class model:
        name: str
        base_provider: str
        best_site: str

    class gpt_35_turbo:
        name: str = 'gpt-3.5-turbo'
        base_provider: str = 'openai'
        best_site: Provider.Provider = Provider.Forefront

    class gpt_4:
        name: str = 'gpt-4'
        base_provider: str = 'openai'
        best_site: Provider.Provider = Provider.Forefront


class Utils:
    convert: dict = {
        'gpt-3.5-turbo': Model.gpt_35_turbo,
        'gpt-4': Model.gpt_4
    }


class ChatCompletion:
    @staticmethod
    def create(model: Model.model or str, messages: list, provider: Provider.Provider = None, stream: bool = False, **kwargs):
        try:
            if isinstance(model, str):
                model = Utils.convert[model]

            engine = model.best_site if not provider else provider
            if not engine.supports_stream and stream:
                print(
                    f"ValueError: {engine.__name__} does not support 'stream' argument", file=sys.stderr)
                sys.exit(1)

            return (engine._create_completion(model.name, messages, stream, **kwargs)
                    if stream else ''.join(engine._create_completion(model.name, messages, stream, **kwargs)))

        except TypeError as e:
            print(e)
            arg: str = str(e).split("'")[1]
            print(
                f"ValueError: {engine.__name__} does not support '{arg}' argument", file=sys.stderr)
            sys.exit(1)