firecoder/src/common/utils/configuration.ts at v0.0.33 · FireCoderAI/firecoder · GitHub

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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
import * as vscode from "vscode";
import { TypeModelsBase } from "../server";

const ConfigurationProperties = {
  "experimental.useGpu.windows.nvidia": {
    default: false,
  },
  "experimental.useGpu.linux.nvidia": {
    default: false,
  },
  "experimental.useGpu.osx.metal": {
    default: false,
  },
  "local.chat.use": {
    default: false,
  },
  "experimental.useopentabs": {
    default: true,
  },
  "completion.autoMode": {
    default: "base-small",
  },
  "completion.manuallyMode": {
    default: "base-small",
  },
  homedir: {
    default: "",
  },
  "cloud.endpoint": {
    default: "https://llm-api.firecoder.cc/v1",
  },
  "cloud.use": {
    default: false,
  },
  "server.usePreRelease": {
    default: false,
  },
  "cloud.autocomplete.use": {
    default: false,
  },
  "cloud.chat.use": {
    default: false,
  },
} as const;

interface ConfigurationPropertiesType
  extends Record<keyof typeof ConfigurationProperties, any> {
  "experimental.useGpu.windows.nvidia": {
    possibleValues: boolean;
  };
  "experimental.useGpu.linux.nvidia": {
    possibleValues: boolean;
  };
  "experimental.useGpu.osx.metal": {
    possibleValues: boolean;
  };
  "local.chat.use": {
    possibleValues: boolean;
  };
  "experimental.useopentabs": {
    possibleValues: boolean;
  };
  "completion.autoMode": {
    possibleValues: TypeModelsBase;
  };
  "completion.manuallyMode": {
    possibleValues: TypeModelsBase;
  };
  homedir: {
    possibleValues: string;
  };
  "server.usePreRelease": {
    possibleValues: boolean;
  };
  "cloud.endpoint": {
    possibleValues: string;
  };
  "cloud.use": {
    possibleValues: boolean;
  };

  "cloud.autocomplete.use": {
    possibleValues: boolean;
  };
  "cloud.chat.use": {
    possibleValues: boolean;
  };
}

class Configuration {
  public get<T extends keyof ConfigurationPropertiesType>(
    property: T
  ): ConfigurationPropertiesType[T]["possibleValues"] {
    const configuration = vscode.workspace.getConfiguration("firecoder");
    const value = configuration.get(property) as any;

    return value ?? ConfigurationProperties[property]["default"];
  }

  public async set<T extends keyof ConfigurationPropertiesType>(
    property: T,
    value: ConfigurationPropertiesType[T]["possibleValues"]
  ) {
    const configuration = vscode.workspace.getConfiguration("firecoder");
    await configuration.update(property, value, true);
  }
}

export const configuration = new Configuration();