wip

Author: acharneski

core/src/main/kotlin/com/simiacryptus/skyenet/core/actors/ActorSystem.kt

@@ -457,16 +457,6 @@ Correct the code and try again.
       }
     }
 
-    fun String.upperSnakeCase(locale: Locale = Locale.getDefault()): String =
-      fromPascalCase().split(" ").map { it.trim() }.filter { it.isNotEmpty() }.joinToString("_") {
-        it.replaceFirstChar { c ->
-          when {
-            c.isLowerCase() -> c.titlecase(locale)
-            else -> c.toString()
-          }
-        }
-      }.uppercase(locale)
-
     fun String.imports(): List<String> {
       return this.split("\n").filter { it.trim().startsWith("import ") }.distinct().sorted()
     }

core/src/main/kotlin/com/simiacryptus/skyenet/core/actors/CodingActor.kt

@@ -4,12 +4,11 @@ abstract class ParsedResponse<T>(val clazz: Class<T>) {
   abstract val text: String
   abstract val obj: T
   override fun toString() = text
-  open fun <V> map(cls: Class<V>, fn: (T) -> V): ParsedResponse<V> = MappedResponse(cls, this.clazz, fn, this)
+  open fun <V> map(cls: Class<V>, fn: (T) -> V): ParsedResponse<V> = MappedResponse(cls, fn, this)
 }
 
 class MappedResponse<F, T>(
   clazz: Class<T>,
-  private val cls: Class<F>,
   private val fn: (F) -> T,
   private val inner: ParsedResponse<F>
 ) : ParsedResponse<T>(clazz) {
@@ -19,6 +18,6 @@ class MappedResponse<F, T>(
     get() = fn(inner.obj)
 
   override fun <V> map(cls: Class<V>, fn: (T) -> V): ParsedResponse<V> {
-    return MappedResponse(cls, this.clazz, fn, this)
+    return MappedResponse(cls, fn, this)
   }
 }

core/src/main/kotlin/com/simiacryptus/skyenet/core/actors/ParsedResponse.kt

@@ -1,6 +1,5 @@
 package com.simiacryptus.skyenet.core.platform.hsql
 
-import com.google.common.util.concurrent.AtomicDouble
 import com.simiacryptus.jopenai.models.ApiModel
 import com.simiacryptus.jopenai.models.ChatModel
 import com.simiacryptus.jopenai.models.OpenAIModel
@@ -12,7 +11,6 @@ import java.sql.Connection
 import java.sql.DriverManager
 import java.sql.ResultSet
 import java.sql.Timestamp
-import java.util.concurrent.atomic.AtomicLong
 
 class HSQLUsageManager(private val dbFile: File) : UsageInterface {
 
@@ -44,17 +42,6 @@ class HSQLUsageManager(private val dbFile: File) : UsageInterface {
   }
 
 
-  private fun updateSchema() {
-    log.info("Updating database schema if needed")
-    // Add schema update logic here if needed
-  }
-
-  private fun deleteSchema() {
-    log.info("Deleting database schema if exists")
-    connection.createStatement().executeUpdate("DROP TABLE IF EXISTS usage")
-    log.debug("Schema deleted")
-  }
-
   override fun incrementUsage(session: Session, apiKey: String?, model: OpenAIModel, tokens: ApiModel.Usage) {
     try {
       log.debug("Incrementing usage for session: ${session.sessionId}, apiKey: $apiKey, model: ${model.modelName}")
@@ -103,27 +90,6 @@ class HSQLUsageManager(private val dbFile: File) : UsageInterface {
     connection.createStatement().executeUpdate("DELETE FROM usage")
   }
 
-  private fun getUsageValues(usageKey: UsageInterface.UsageKey): UsageInterface.UsageValues {
-    log.debug("Getting usage values for session: ${usageKey.session.sessionId}, apiKey: ${usageKey.apiKey}, model: ${usageKey.model.modelName}")
-    val statement = connection.prepareStatement(
-      """
-         SELECT COALESCE(SUM(prompt_tokens), 0), COALESCE(SUM(completion_tokens), 0), COALESCE(SUM(cost), 0)
-         FROM usage
-         WHERE session_id = ? AND api_key = ? AND model = ?
-         """
-    )
-    statement.setString(1, usageKey.session.sessionId)
-    statement.setString(2, usageKey.apiKey ?: "")
-    statement.setString(3, usageKey.model.toString())
-    val resultSet = statement.executeQuery()
-    resultSet.next()
-    return UsageInterface.UsageValues(
-      AtomicLong(resultSet.getLong(1)),
-      AtomicLong(resultSet.getLong(2)),
-      AtomicDouble(resultSet.getDouble(3))
-    )
-  }
-
   private fun saveUsageValues(usageKey: UsageInterface.UsageKey, usageValues: UsageInterface.UsageValues) {
     log.debug("Saving usage values for session: ${usageKey.session.sessionId}, apiKey: ${usageKey.apiKey}, model: ${usageKey.model.modelName}")
     val statement = connection.prepareStatement(