Comparing version 0.1.16 to 0.1.17
export type GenerationOptions = { | ||
provider?: "openai" | "cohere"; | ||
memoryId?: string; | ||
chatId?: string; | ||
}; | ||
@@ -5,0 +6,0 @@ declare function generateWithTokenUsage(task: string, options?: GenerationOptions): Promise<{ |
@@ -59,4 +59,9 @@ "use strict"; | ||
(0, ensurePolyfactToken_1.ensurePolyfactToken)(); | ||
const requestBody = { task, provider: options?.provider || "openai", memory_id: options?.memoryId || "" }; | ||
const resJSON = await (0, isomorphic_fetch_1.default)(`${POLYFACT_ENDPOINT}/generate`, { | ||
const requestBody = { | ||
task, | ||
provider: options?.provider || "openai", | ||
memory_id: options?.memoryId || "", | ||
chat_id: options?.chatId || "", | ||
}; | ||
const res = await (0, isomorphic_fetch_1.default)(`${POLYFACT_ENDPOINT}/generate`, { | ||
method: "POST", | ||
@@ -68,5 +73,3 @@ headers: { | ||
body: JSON.stringify(requestBody), | ||
}).then((res) => res.text()); | ||
console.log(resJSON); | ||
const res = JSON.parse(resJSON); | ||
}).then((res) => res.json()); | ||
if (!ResultType.is(res)) { | ||
@@ -73,0 +76,0 @@ throw new GenerationError(); |
@@ -7,2 +7,3 @@ import { generate, generateWithTokenUsage, GenerationOptions } from "./generate"; | ||
import { createMemory, updateMemory, getAllMemories } from "./memory"; | ||
export { generate, generateWithTokenUsage, generateWithType, generateWithTypeWithTokenUsage, splitString, t, GenerationOptions, transcribe, createMemory, updateMemory, getAllMemories, }; | ||
import { Chat } from "./chats"; | ||
export { generate, generateWithTokenUsage, generateWithType, generateWithTypeWithTokenUsage, splitString, t, GenerationOptions, transcribe, createMemory, updateMemory, getAllMemories, Chat, }; |
"use strict"; | ||
Object.defineProperty(exports, "__esModule", { value: true }); | ||
exports.getAllMemories = exports.updateMemory = exports.createMemory = exports.transcribe = exports.t = exports.splitString = exports.generateWithTypeWithTokenUsage = exports.generateWithType = exports.generateWithTokenUsage = exports.generate = void 0; | ||
exports.Chat = exports.getAllMemories = exports.updateMemory = exports.createMemory = exports.transcribe = exports.t = exports.splitString = exports.generateWithTypeWithTokenUsage = exports.generateWithType = exports.generateWithTokenUsage = exports.generate = void 0; | ||
const generate_1 = require("./generate"); | ||
@@ -20,1 +20,3 @@ Object.defineProperty(exports, "generate", { enumerable: true, get: function () { return generate_1.generate; } }); | ||
Object.defineProperty(exports, "getAllMemories", { enumerable: true, get: function () { return memory_1.getAllMemories; } }); | ||
const chats_1 = require("./chats"); | ||
Object.defineProperty(exports, "Chat", { enumerable: true, get: function () { return chats_1.Chat; } }); |
@@ -1,9 +0,31 @@ | ||
import { generate } from "../lib/index"; | ||
import { Chat } from "../lib/index"; | ||
(async () => { | ||
console.log( | ||
await generate("Who is the first man on the moon?", { | ||
provider: "cohere", | ||
}), | ||
); | ||
})(); | ||
async function main() { | ||
const chat = new Chat({ provider: "cohere" }); | ||
console.log(await chat.chatId); | ||
console.log(await chat.getMessages()); | ||
let prompt = "Who is the first person to have walked on the moon?"; | ||
console.log(prompt); | ||
let result = await chat.sendMessage(prompt); | ||
console.log(result); | ||
prompt = "Who is the second ?"; | ||
console.log(prompt); | ||
result = await chat.sendMessage(prompt); | ||
console.log(result); | ||
prompt = "Who was the president of the United States at the time?"; | ||
console.log(prompt); | ||
result = await chat.sendMessage(prompt); | ||
console.log(result); | ||
console.log(await chat.getMessages()); | ||
} | ||
main(); |
@@ -37,2 +37,3 @@ import fetch from "isomorphic-fetch"; | ||
memoryId?: string; | ||
chatId?: string; | ||
}; | ||
@@ -49,6 +50,13 @@ | ||
memory_id?: string; | ||
// eslint-disable-next-line camelcase | ||
chat_id?: string; | ||
provider: GenerationOptions["provider"]; | ||
} = { task, provider: options?.provider || "openai", memory_id: options?.memoryId || "" }; | ||
} = { | ||
task, | ||
provider: options?.provider || "openai", | ||
memory_id: options?.memoryId || "", | ||
chat_id: options?.chatId || "", | ||
}; | ||
const resJSON = await fetch(`${POLYFACT_ENDPOINT}/generate`, { | ||
const res = await fetch(`${POLYFACT_ENDPOINT}/generate`, { | ||
method: "POST", | ||
@@ -60,8 +68,4 @@ headers: { | ||
body: JSON.stringify(requestBody), | ||
}).then((res: any) => res.text()); | ||
}).then((res: any) => res.json()); | ||
console.log(resJSON); | ||
const res = JSON.parse(resJSON); | ||
if (!ResultType.is(res)) { | ||
@@ -68,0 +72,0 @@ throw new GenerationError(); |
@@ -10,2 +10,3 @@ import { generate, generateWithTokenUsage, GenerationOptions } from "./generate"; | ||
import { createMemory, updateMemory, getAllMemories } from "./memory"; | ||
import { Chat } from "./chats"; | ||
@@ -24,2 +25,3 @@ export { | ||
getAllMemories, | ||
Chat, | ||
}; |
{ | ||
"name": "polyfact", | ||
"version": "0.1.16", | ||
"version": "0.1.17", | ||
"main": "dist/index.js", | ||
@@ -5,0 +5,0 @@ "types": "dist/index.d.ts", |
Sorry, the diff of this file is not supported yet
878445
50
1761
6