@graf-research/llm-runner 0.0.2 → 0.0.4

package/dist/multistep.d.ts

@@ -13,7 +13,9 @@ export declare namespace NativeStep {
     function askChoose(llm: LLMRunner.BaseLLM, q: string, options: string[], session_id?: string): Promise<string>;
     function streamAskChoose(llm: LLMRunner.BaseLLM, q: string, options: string[], session_id?: string): Promise<StreamResponseWithFinalAnswer<string>>;
     function askMultipleChoiceAnswer(llm: LLMRunner.BaseLLM, q: string, options: string[], session_id?: string): Promise<string[]>;
-    function streamAskMultipleChoiceAnswer(llm: LLMRunner.BaseLLM, q: string, options: string[], session_id?: string): Promise<StreamResponseWithFinalAnswer<string[]>>;
+    function streamMultipleChoiceAnswer(llm: LLMRunner.BaseLLM, q: string, options: string[], session_id?: string): Promise<StreamResponseWithFinalAnswer<string[]>>;
+    function askOpenListAnswer(llm: LLMRunner.BaseLLM, q: string, session_id?: string): Promise<string[]>;
+    function streamOpenListAnswer(llm: LLMRunner.BaseLLM, q: string, session_id?: string): Promise<StreamResponseWithFinalAnswer<string[]>>;
     function askPlan(llm: LLMRunner.BaseLLM, q: string, session_id?: string): Promise<string[]>;
     function streamPlan(llm: LLMRunner.BaseLLM, q: string, session_id?: string): Promise<StreamResponseWithFinalAnswer<string[]>>;
 }

package/dist/multistep.js

@@ -115,7 +115,7 @@ var NativeStep;
     // ================================= //
     function generateMultipleChoiceAnswerResolver(options) {
         return (response) => __awaiter(this, void 0, void 0, function* () {
-            const list_options_answer = response.trim().split('\n').map(s => s.trim());
+            const list_options_answer = response.trim().split(/[\n,]/).map(s => s.trim());
             return options.sort((a, b) => b.length - a.length).filter(a => list_options_answer.includes(a));
         });
     }
@@ -126,13 +126,33 @@ var NativeStep;
         });
     }
     NativeStep.askMultipleChoiceAnswer = askMultipleChoiceAnswer;
-    function streamAskMultipleChoiceAnswer(llm, q, options, session_id) {
+    function streamMultipleChoiceAnswer(llm, q, options, session_id) {
         return __awaiter(this, void 0, void 0, function* () {
             const prompt = `Respond this question: ${q}, choose on or more from this options as answer: ${options.map(a => `"${a}"`).join(',')}. Put all the answers into one line each. Don't make up an answer, only choose from available options`;
             return yield stream(llm, prompt, generateMultipleChoiceAnswerResolver(options), session_id);
         });
     }
-    NativeStep.streamAskMultipleChoiceAnswer = streamAskMultipleChoiceAnswer;
+    NativeStep.streamMultipleChoiceAnswer = streamMultipleChoiceAnswer;
+    // ================================= //
+    // Open List Answer
+    // ================================= //
+    const open_list_answer_resolver = (response) => __awaiter(this, void 0, void 0, function* () {
+        return response.trim().split(/\n+/).map(s => s.trim());
+    });
+    function askOpenListAnswer(llm, q, session_id) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const prompt = `Respond this question: ${q}, give answers as list line by line, remember: one line is one answer.`;
+            return yield ask(llm, prompt, open_list_answer_resolver, session_id);
+        });
+    }
+    NativeStep.askOpenListAnswer = askOpenListAnswer;
+    function streamOpenListAnswer(llm, q, session_id) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const prompt = `Respond this question: ${q}, give answers as list line by line, remember: one line is one answer.`;
+            return yield stream(llm, prompt, open_list_answer_resolver, session_id);
+        });
+    }
+    NativeStep.streamOpenListAnswer = streamOpenListAnswer;
     // ================================= //
     // Plan
     // ================================= //

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@graf-research/llm-runner",
-  "version": "0.0.2",
+  "version": "0.0.4",
   "main": "dist/index.js",
   "scripts": {
     "build": "rm -rf dist && tsc",