Skip to content

Commit

Permalink
misc
Browse files Browse the repository at this point in the history
  • Loading branch information
acharneski committed Nov 24, 2023
1 parent 73ccea7 commit 55394db
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 9 deletions.
Original file line number Diff line number Diff line change
@@ -1,27 +1,24 @@
package com.simiacryptus.skyenet.core.actors

import com.simiacryptus.jopenai.API
import com.simiacryptus.jopenai.ApiModel
import com.simiacryptus.jopenai.ApiModel.*
import com.simiacryptus.jopenai.ApiModel.ChatMessage
import com.simiacryptus.jopenai.ApiModel.ImageGenerationRequest
import com.simiacryptus.jopenai.OpenAIClient
import com.simiacryptus.jopenai.models.ChatModels
import com.simiacryptus.jopenai.models.ImageModels
import com.simiacryptus.jopenai.models.OpenAITextModel
import com.simiacryptus.jopenai.proxy.ChatProxy
import java.awt.image.BufferedImage
import java.util.function.Function

open class ImageActor(
prompt: String = "Transform the user request into an image generation prompt that the user will like",
val action: String? = null,
name: String? = null,
textModel: ChatModels = ChatModels.GPT35Turbo,
val imageModel: ImageModels = ImageModels.DallE2,
temperature: Double = 0.3,
val width: Int = 1024,
val height: Int = 1024,
) : BaseActor<ImageResponse>(
prompt = prompt,
name = action,
name = name,
model = textModel,
temperature = temperature,
) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,15 +3,14 @@ package com.simiacryptus.skyenet.core.actors.record
import com.simiacryptus.jopenai.API
import com.simiacryptus.jopenai.models.OpenAIModel
import com.simiacryptus.skyenet.core.actors.ImageActor
import com.simiacryptus.skyenet.core.actors.ParsedResponse
import com.simiacryptus.skyenet.core.util.FunctionWrapper

class ImageActorInterceptor(
val inner: ImageActor,
private val functionInterceptor: FunctionWrapper,
) : ImageActor(
prompt = inner.prompt,
action = inner.action,
name = inner.name,
textModel = inner.model,
imageModel = inner.imageModel,
temperature = inner.temperature,
Expand Down

0 comments on commit 55394db

Please sign in to comment.