From 5f31a2c5722daa188f75e489012a796d35aaf3b3 Mon Sep 17 00:00:00 2001
From: bel <bel>
Date: Fri, 12 Apr 2024 16:31:23 -0600
Subject: [PATCH] yay ai test fails

---
 .gitignore |  1 +
 ai_test.go | 44 +++++++++++++++++++++++++++++++++++++++-----
 2 files changed, 40 insertions(+), 5 deletions(-)

diff --git a/.gitignore b/.gitignore
index 7cde8b2..60c8781 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,3 @@
 /slack-bot-vr
+**/*.sw*
 /spoc-bot-vr
diff --git a/ai_test.go b/ai_test.go
index 7fece21..8da2b79 100644
--- a/ai_test.go
+++ b/ai_test.go
@@ -4,6 +4,7 @@ package main
 
 import (
 	"context"
+	"fmt"
 	"io"
 	"net/http"
 	"os"
@@ -54,9 +55,42 @@ func TestAILocal(t *testing.T) {
 		0.9,
 	)
 
-	if result, err := ai.Do(ctx, "hello world"); err != nil {
-		t.Fatal(err)
-	} else {
-		t.Logf("%s", result)
-	}
+	t.Run("mvp", func(t *testing.T) {
+		if result, err := ai.Do(ctx, "hello world"); err != nil {
+			t.Fatal(err)
+		} else if len(result) < 250 {
+			t.Error(result)
+		} else {
+			t.Logf("%s", result)
+		}
+	})
+
+	t.Run("simulation", func(t *testing.T) {
+		d := NewRAM()
+		FillWithTestdata(ctx, d)
+		s := NewStorage(d)
+
+		threads, err := s.Threads(ctx)
+		if err != nil || len(threads) < 1 {
+			t.Fatal(err)
+		}
+
+		thread, err := s.Thread(ctx, threads[0])
+		if err != nil || len(thread) < 1 {
+			t.Fatal(err)
+		}
+
+		input := fmt.Sprintf(`
+         Summarize the following forum converstion.
+         ---
+         %s
+      `, thread[0].Plaintext)
+		t.Logf("\n\t%s", input)
+		result, err := ai.Do(ctx, input)
+		if err != nil {
+			t.Fatal(err)
+		}
+		t.Logf("\n\t%s\n->\n\t%s", input, result)
+	})
+
 }