Turn off augmentation looping

microsoft · Mar 18, 2024 · 69c892a · 69c892a
1 parent 6505f18
commit 69c892a
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 10 deletions.
diff --git a/js/packages/teams-ai/src/models/LlamaModel.ts b/js/packages/teams-ai/src/models/LlamaModel.ts
@@ -48,21 +48,26 @@ export class LlamaModel implements PromptCompletionModel {
         if (last.role !== 'user') {
             last = undefined;
         }
-        // TODO: try catch
-        // TODO: make real interface
-        const res = await this._httpClient.post<{ output: string }>(this.options.endpoint, {
-            input_data: {
-                input_string: result.output,
-                parameters: template.config.completion
-            }
-        });
+        let res;
+
+        try {
+            res = await this._httpClient.post<{ output: string }>(this.options.endpoint, {
+                input_data: {
+                    input_string: result.output,
+                    parameters: template.config.completion
+                }
+            });
+        } catch (err) {
+            console.error(err);
+            throw err;
+        }
 
         return {
             status: 'success',
             input: last,
             message: {
                 role: 'assistant',
-                content: res.data.output
+                content: res!.data.output
             }
         };
     }

diff --git a/js/samples/04.ai.g.LLAMA/src/index.ts b/js/samples/04.ai.g.LLAMA/src/index.ts
@@ -99,7 +99,8 @@ const storage = new MemoryStorage();
 const app = new Application<ApplicationTurnState>({
     storage,
     ai: {
-        planner
+        planner,
+        allow_looping: false
     }
 });