show cost

2023-03-31 04:50:49 +08:00
parent 11afa12b09
commit 05f57f29e5
4 changed files with 62 additions and 16 deletions
--- a/src/models.ts
+++ b/src/models.ts
@@ -0,0 +1,36 @@
+interface Model {
+  maxToken: number;
+  price: {
+    prompt: number;
+    completion: number;
+  };
+}
+
+const models: Record<string, Model> = {
+  "gpt-3.5-turbo": {
+    maxToken: 4096,
+    price: { prompt: 0.002 / 1000, completion: 0.002 / 1000 },
+  },
+  "gpt-3.5-turbo-0301": {
+    maxToken: 4096,
+    price: { prompt: 0.002 / 1000, completion: 0.002 / 1000 },
+  },
+  "gpt-4": {
+    maxToken: 8192,
+    price: { prompt: 0.03 / 1000, completion: 0.06 / 1000 },
+  },
+  "gpt-4-0314": {
+    maxToken: 8192,
+    price: { prompt: 0.03 / 1000, completion: 0.06 / 1000 },
+  },
+  "gpt-4-32k": {
+    maxToken: 8192,
+    price: { prompt: 0.03 / 1000, completion: 0.06 / 1000 },
+  },
+  "gpt-4-32k-0314": {
+    maxToken: 8192,
+    price: { prompt: 0.06 / 1000, completion: 0.12 / 1000 },
+  },
+};
+
+export default models;