feat: add fake completion

kirklin · Feb 25, 2024 · e1e6ced · e1e6ced
1 parent 7ee48f9
commit e1e6ced
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 55 deletions.
diff --git a/packages/ai/core/package.json b/packages/ai/core/package.json
@@ -15,6 +15,7 @@
   },
   "dependencies": {
     "@azure/openai": "^1.0.0-beta.11",
+    "@faker-js/faker": "^8.4.1",
     "ai": "^2.2.36",
     "gpt-tokenizer": "^2.1.2",
     "openai": "^4.28.0",

diff --git a/packages/ai/core/src/languageModelAgent/celeris/mock.ts b/packages/ai/core/src/languageModelAgent/celeris/mock.ts
@@ -0,0 +1,22 @@
+import { faker } from "@faker-js/faker";
+
+// Generate a fake completion using faker
+export const fakeCompletion = {
+  id: faker.string.uuid(),
+  object: "text_completion",
+  created: faker.date.soon,
+  model: "celeris-mock",
+  choices: [
+    {
+      text: faker.lorem.sentence({ min: 3, max: 300 }),
+      logprobs: null,
+      finish_reason: "length",
+      index: 0,
+    },
+  ],
+  usage: {
+    prompt_tokens: 10,
+    completion_tokens: 20,
+    total_tokens: 30,
+  },
+};
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml