From d1356b20736be7012642efdc68fcc526761cb677 Mon Sep 17 00:00:00 2001
From: Andrew Charneski <acharneski@gmail.com>
Date: Sat, 28 Dec 2024 18:56:23 -0500
Subject: [PATCH] wip

---
 .../com/simiacryptus/skyenet/apps/general/OutlineApp.kt   | 8 ++++++--
 .../apps/plan/tools/online/SearchAndAnalyzeTask.kt        | 4 ++--
 .../com/simiacryptus/skyenet/util/TensorflowProjector.kt  | 5 ++---
 3 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/webui/src/main/kotlin/com/simiacryptus/skyenet/apps/general/OutlineApp.kt b/webui/src/main/kotlin/com/simiacryptus/skyenet/apps/general/OutlineApp.kt
index e5e9b82a..410fdda5 100644
--- a/webui/src/main/kotlin/com/simiacryptus/skyenet/apps/general/OutlineApp.kt
+++ b/webui/src/main/kotlin/com/simiacryptus/skyenet/apps/general/OutlineApp.kt
@@ -2,6 +2,7 @@ package com.simiacryptus.skyenet.apps.general
 
 import com.simiacryptus.jopenai.API
 import com.simiacryptus.jopenai.ChatClient
+import com.simiacryptus.jopenai.OpenAIClient
 import com.simiacryptus.jopenai.describe.JsonDescriber
 import com.simiacryptus.jopenai.models.ChatModel
 import com.simiacryptus.jopenai.models.OpenAIModels
@@ -30,6 +31,7 @@ open class OutlineApp(
   applicationName: String = "Outline Expansion Concept Map v1.1",
   val domainName: String,
   val settings: Settings? = null,
+  val api2: OpenAIClient,
 ) : ApplicationServer(
   applicationName = applicationName,
   path = "/idea_mapper",
@@ -80,6 +82,7 @@ open class OutlineApp(
     val settings = this.settings ?: getSettings(session, user)!!
     OutlineAgent(
       api = api,
+      api2 = api2,
       dataStorage = dataStorage,
       session = session,
       user = user,
@@ -99,6 +102,7 @@ open class OutlineApp(
 
 class OutlineAgent(
   val api: API,
+  val api2: OpenAIClient,
   dataStorage: StorageInterface,
   session: Session,
   user: User?,
@@ -167,7 +171,7 @@ class OutlineAgent(
     val finalOutline = finalOutline(outlineManager, sessionDir)
 
     if (showProjector) {
-      showProjector(api, outlineManager, finalOutline)
+      showProjector(api2, outlineManager, finalOutline)
     }
 
     if (writeFinalEssay) {
@@ -193,7 +197,7 @@ class OutlineAgent(
   }
 
   private fun showProjector(
-    api: ChatClient,
+    api: OpenAIClient,
     outlineManager: OutlineManager,
     finalOutline: List<OutlineManager.Node>
   ) {
diff --git a/webui/src/main/kotlin/com/simiacryptus/skyenet/apps/plan/tools/online/SearchAndAnalyzeTask.kt b/webui/src/main/kotlin/com/simiacryptus/skyenet/apps/plan/tools/online/SearchAndAnalyzeTask.kt
index ed978be8..c56fceaa 100644
--- a/webui/src/main/kotlin/com/simiacryptus/skyenet/apps/plan/tools/online/SearchAndAnalyzeTask.kt
+++ b/webui/src/main/kotlin/com/simiacryptus/skyenet/apps/plan/tools/online/SearchAndAnalyzeTask.kt
@@ -25,7 +25,7 @@ class SearchAndAnalyzeTask(
   class SearchAndAnalyzeTaskConfigData(
     @Description("The search query to use for Google search")
     val search_query: String = "",
-    @Description("The number of search results to analyze (max 5)")
+    @Description("The number of search results to analyze (max 20)")
     val num_results: Int = 3,
     @Description("The analysis goal or focus for the content")
     val analysis_goal: String = "",
@@ -42,7 +42,7 @@ class SearchAndAnalyzeTask(
   override fun promptSegment() = """
     SearchAndAnalyze - Search Google, fetch top results, and analyze content
     ** Specify the search query
-    ** Specify number of results to analyze (max 5)
+    ** Specify number of results to analyze (max 20)
     ** Specify the analysis goal or focus
   """.trimIndent()
 
diff --git a/webui/src/main/kotlin/com/simiacryptus/skyenet/util/TensorflowProjector.kt b/webui/src/main/kotlin/com/simiacryptus/skyenet/util/TensorflowProjector.kt
index 8b053f3a..ec44d874 100644
--- a/webui/src/main/kotlin/com/simiacryptus/skyenet/util/TensorflowProjector.kt
+++ b/webui/src/main/kotlin/com/simiacryptus/skyenet/util/TensorflowProjector.kt
@@ -1,6 +1,5 @@
 package com.simiacryptus.skyenet.util
 
-import com.simiacryptus.jopenai.API
 import com.simiacryptus.jopenai.OpenAIClient
 import com.simiacryptus.jopenai.models.ApiModel.EmbeddingRequest
 import com.simiacryptus.jopenai.models.EmbeddingModels
@@ -15,7 +14,7 @@ import java.io.IOException
 import java.util.*
 
 class TensorflowProjector(
-  val api: API,
+  val api: OpenAIClient,
   val dataStorage: StorageInterface,
   val sessionID: Session,
   val session: ApplicationInterface,
@@ -33,7 +32,7 @@ class TensorflowProjector(
   @Throws(IOException::class)
   private fun toVectorMap(vararg words: String): Map<String, DoubleArray> {
     val vectors = words.map { word ->
-      word to (api as OpenAIClient).createEmbedding(
+      word to api.createEmbedding(
         EmbeddingRequest(
           model = EmbeddingModels.AdaEmbedding.modelName,
           input = word.trim(),